X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=rts%2Fposix%2FOSMem.c;h=bfe12964a23cea2f5f13dda3e4f7dad432e0f371;hb=b076da6047bf4c6b3d74a97c6c5fd59ed3bdd114;hp=bdae2739f1191d9856bdeb79aa9b68295094b7a6;hpb=eb90092dad2a0b614d0aba5ed56d7d4eaf14b2ea;p=ghc-hetmet.git

diff --git a/rts/posix/OSMem.c b/rts/posix/OSMem.c
index bdae273..bfe1296 100644
--- a/rts/posix/OSMem.c
+++ b/rts/posix/OSMem.c
@@ -6,12 +6,13 @@
  *
  * ---------------------------------------------------------------------------*/
 
-/* This is non-posix compliant. */
-/* #include "PosixSource.h" */
+// This is non-posix compliant.
+// #include "PosixSource.h"
 
 #include "Rts.h"
-#include "OSMem.h"
-#include "RtsFlags.h"
+
+#include "RtsUtils.h"
+#include "sm/OSMem.h"
 
 #ifdef HAVE_UNISTD_H
 #include <unistd.h>
@@ -121,8 +122,8 @@ my_mmap (void *addr, lnat size)
 	    (errno == EINVAL && sizeof(void*)==4 && size >= 0xc0000000)) {
 	    // If we request more than 3Gig, then we get EINVAL
 	    // instead of ENOMEM (at least on Linux).
-	    errorBelch("out of memory (requested %lu bytes)", size);
-	    stg_exit(EXIT_FAILURE);
+            errorBelch("out of memory (requested %lu bytes)", size);
+            stg_exit(EXIT_FAILURE);
 	} else {
 	    barf("getMBlock: mmap: %s", strerror(errno));
 	}
@@ -138,7 +139,7 @@ static void *
 gen_map_mblocks (lnat size)
 {
     int slop;
-    void *ret;
+    StgWord8 *ret;
 
     // Try to map a larger block, and take the aligned portion from
     // it (unmap the rest).
@@ -148,10 +149,10 @@ gen_map_mblocks (lnat size)
     // unmap the slop bits around the chunk we allocated
     slop = (W_)ret & MBLOCK_MASK;
     
-    if (munmap(ret, MBLOCK_SIZE - slop) == -1) {
+    if (munmap((void*)ret, MBLOCK_SIZE - slop) == -1) {
       barf("gen_map_mblocks: munmap failed");
     }
-    if (slop > 0 && munmap(ret+size-slop, slop) == -1) {
+    if (slop > 0 && munmap((void*)(ret+size-slop), slop) == -1) {
       barf("gen_map_mblocks: munmap failed");
     }
 
@@ -177,7 +178,7 @@ osGetMBlocks(nat n)
 {
   caddr_t ret;
   lnat size = MBLOCK_SIZE * (lnat)n;
- 
+
   if (next_request == 0) {
       // use gen_map_mblocks the first time.
       ret = gen_map_mblocks(size);
@@ -198,7 +199,6 @@ osGetMBlocks(nat n)
 	  ret = gen_map_mblocks(size);
       }
   }
-
   // Next time, we'll try to allocate right after the block we just got.
   // ToDo: check that we haven't already grabbed the memory at next_request
   next_request = ret + size;
@@ -206,9 +206,24 @@ osGetMBlocks(nat n)
   return ret;
 }
 
+void osFreeMBlocks(char *addr, nat n)
+{
+    munmap(addr, n * MBLOCK_SIZE);
+}
+
+void osReleaseFreeMemory(void) {
+    /* Nothing to do on POSIX */
+}
+
 void osFreeAllMBlocks(void)
 {
-    /* XXX Do something here (bug #711) */
+    void *mblock;
+
+    for (mblock = getFirstMBlock();
+         mblock != NULL;
+         mblock = getNextMBlock(mblock)) {
+        munmap(mblock, MBLOCK_SIZE);
+    }
 }
 
 lnat getPageSize (void)
@@ -237,6 +252,6 @@ void setExecutable (void *p, lnat len, rtsBool exec)
     StgWord size             = startOfLastPage - startOfFirstPage + pageSize;
     if (mprotect((void*)startOfFirstPage, (size_t)size, 
 		 (exec ? PROT_EXEC : 0) | PROT_READ | PROT_WRITE) != 0) {
-	barf("makeExecutable: failed to protect 0x%p\n", p);
+	barf("setExecutable: failed to protect 0x%p\n", p);
     }
 }