summaryrefslogtreecommitdiffstats
path: root/mdk-stage1/tools.c
blob: e07fcd34715090cd0ac5efa2229b6d97eeed82e8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
/*
 * Guillaume Cottenceau (gc@mandrakesoft.com)
 *
 * Copyright 2000 MandrakeSoft
 *
 * This software may be freely redistributed under the terms of the GNU
 * public license.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 */

/*
 * Portions from Erik Troan (ewt@redhat.com)
 *
 * Copyright 1996 Red Hat Software 
 *
 */

#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <string.h>
#include <ctype.h>
#include "stage1.h"
#include "log.h"

#include "tools.h"

int total_memory(void)
{
	int fd;
	int i;
	char buf[4096];
	char * memtotal_tag = "MemTotal:";
	int memtotal = 0;
    
	fd = open("/proc/meminfo", O_RDONLY);
	if (fd == -1)
		fatal_error("could not open /proc/meminfo");

	i = read(fd, buf, sizeof(buf));
	if (i < 0)
		fatal_error("could not read /proc/meminfo");
		
	close(fd);
	buf[i] = 0;

	i = 0;
	while (buf[i] != 0 && strncmp(&buf[i], memtotal_tag, strlen(memtotal_tag)))
		i++;

	while (buf[i] != 0 && buf[i] != '\n' && !isdigit(buf[i]))
		i++;

	if (buf[i] == 0 || buf[i] == '\n')
		fatal_error("could not read MemTotal");

	while (buf[i] != 0 && isdigit(buf[i])) {
	    memtotal = (memtotal * 10) + (buf[i] - '0');
	    i++;
	}
	
	log_message("%s %d kB", memtotal_tag, memtotal);

	return memtotal;
}