rts: Add getPhysicalMemorySize
authorBen Gamari <bgamari.foss@gmail.com>
Fri, 13 Sep 2013 03:04:10 +0000 (23:04 -0400)
committerAustin Seipp <austin@well-typed.com>
Fri, 25 Oct 2013 14:14:29 +0000 (09:14 -0500)
rts/posix/OSMem.c
rts/sm/OSMem.h
rts/win32/OSMem.c

index 4328d9d..21d4e54 100644 (file)
@@ -263,6 +263,26 @@ W_ getPageSize (void)
     }
 }
 
+/* Returns 0 if physical memory size cannot be identified */
+StgWord64 getPhysicalMemorySize (void)
+{
+    static StgWord64 physMemSize = 0;
+    if (!physMemSize) {
+        long ret;
+        W_ pageSize = getPageSize();
+
+        ret = sysconf(_SC_PHYS_PAGES);
+        if (ret == -1) {
+#if defined(DEBUG)
+            errorBelch("warning: getPhysicsMemorySize: cannot get physical memory size");
+#endif
+            return 0;
+        }
+        physMemSize = ret * pageSize;
+    }
+    return physMemSize;
+}
+
 void setExecutable (void *p, W_ len, rtsBool exec)
 {
     StgWord pageSize = getPageSize();
index a0d615b..db704fc 100644 (file)
@@ -17,6 +17,7 @@ void osFreeMBlocks(char *addr, nat n);
 void osReleaseFreeMemory(void);
 void osFreeAllMBlocks(void);
 W_ getPageSize (void);
+StgWord64 getPhysicalMemorySize (void);
 void setExecutable (void *p, W_ len, rtsBool exec);
 
 #include "EndPrivate.h"
index 218b25d..f350076 100644 (file)
@@ -378,6 +378,24 @@ W_ getPageSize (void)
     }
 }
 
+/* Returns 0 if physical memory size cannot be identified */
+StgWord64 getPhysicalMemorySize (void)
+{
+    static StgWord64 physMemSize = 0;
+    if (!physMemSize) {
+        MEMORYSTATUSEX status;
+        status.dwLength = sizeof(status);
+        if (!GlobalMemoryStatusEx(&status)) {
+#if defined(DEBUG)
+            errorBelch("warning: getPhysicalMemorySize: cannot get physical memory size");
+#endif
+            return 0;
+        }
+        physMemSize = status.ullTotalPhys;
+    }
+    return physMemSize;
+}
+
 void setExecutable (void *p, W_ len, rtsBool exec)
 {
     DWORD dwOldProtect = 0;