better memory detection (from wonderful drakx)

1 files changed, 8 insertions, 30 deletions

diff --git a/mdk-stage1/tools.c b/mdk-stage1/tools.c
index 9acabdb5f..8b2931ad2 100644
--- a/mdk-stage1/tools.c
+++ b/mdk-stage1/tools.c
@@ -183,38 +183,16 @@ int charstar_to_int(char * s)
 
 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");
+	int value;
+	struct stat statr;
+	if (stat("/proc/kcore", &statr))
+		return 0;
 
-	memtotal = charstar_to_int(&(buf[i]));
-	
-	log_message("%s %d kB", memtotal_tag, memtotal);
+	/* drakx powered: use /proc/kcore and rounds every 4 Mbytes */
+	value = 4 * ((int)((float)statr.st_size / 1024 / 1024 / 4 + 0.5));
+	log_message("Total Memory: %d Mbytes", value);
 
-	return memtotal;
+	return value;
 }