Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Throw an exception if return value shows an error.
[simgrid.git] / src / xbt / mmalloc / mmap-sup.c
index f32e0aa..28f89bc 100644 (file)
  * under the terms of the license (GNU LGPL) which comes with this package. */
 
 #ifndef MAP_ANONYMOUS
-       #define MAP_ANONYMOUS MAP_ANON
+#define MAP_ANONYMOUS MAP_ANON
 #endif
 
 #ifdef HAVE_UNISTD_H
-#include <unistd.h>    /* Prototypes for lseek */
+#include <unistd.h>             /* Prototypes for lseek */
 #endif
 #include <stdio.h>
 #include <fcntl.h>
@@ -25,6 +25,7 @@
 #endif
 
 #include "mmprivate.h"
+#include "xbt/ex.h"
 
 /* Cache the pagesize for the current host machine.  Note that if the host
    does not readily provide a getpagesize() function, we need to emulate it
@@ -33,7 +34,7 @@
 
 static size_t pagesize;
 #if NEED_DECLARATION_GETPAGESIZE
-extern int getpagesize (void);
+extern int getpagesize(void);
 #endif
 
 #define PAGE_ALIGN(addr) (void*) (((long)(addr) + pagesize - 1) & \
@@ -61,108 +62,102 @@ extern int getpagesize (void);
     amount to either add to or subtract from the existing region.  Works
     like sbrk(), but using mmap(). */
 
-void*
-__mmalloc_mmap_morecore (struct mdesc *mdp, int size)
+void *__mmalloc_mmap_morecore(struct mdesc *mdp, int size)
 {
-  void* result = NULL;
-  off_t foffset;       /* File offset at which new mapping will start */
-  size_t mapbytes;     /* Number of bytes to map */
-  void* moveto;        /* Address where we wish to move "break value" to */
-  void* mapto; /* Address we actually mapped to */
-  char buf = 0;                /* Single byte to write to extend mapped file */
+  ssize_t test = 0;
+  void *result = NULL;
+  off_t foffset;                /* File offset at which new mapping will start */
+  size_t mapbytes;              /* Number of bytes to map */
+  void *moveto;                 /* Address where we wish to move "break value" to */
+  void *mapto;                  /* Address we actually mapped to */
+  char buf = 0;                 /* Single byte to write to extend mapped file */
 
   if (pagesize == 0)
     pagesize = getpagesize();
 
-  if (size == 0)
-  {
+  if (size == 0) {
     /* Just return the current "break" value. */
-    result = mdp -> breakval;
-  }
-  else if (size < 0)
-  {
+    result = mdp->breakval;
+
+  } else if (size < 0) {
     /* We are deallocating memory.  If the amount requested would cause
-            us to try to deallocate back past the base of the mmap'd region
-            then do nothing, and return NULL.  Otherwise, deallocate the
-            memory and return the old break value. */
-    if (((char*)mdp -> breakval) + size >= (char*)mdp -> base)
-         {
-           result = (void*) mdp -> breakval;
-           mdp -> breakval = (char*) mdp->breakval + size;
-           moveto = PAGE_ALIGN (mdp -> breakval);
-           munmap (moveto, (size_t) (((char*)mdp -> top) - ((char*)moveto)) - 1);
-           mdp -> top = moveto;
-         }
-  }
-  else
-  {
+       us to try to deallocate back past the base of the mmap'd region
+       then do nothing, and return NULL.  Otherwise, deallocate the
+       memory and return the old break value. */
+    if (((char *) mdp->breakval) + size >= (char *) mdp->base) {
+      result = (void *) mdp->breakval;
+      mdp->breakval = (char *) mdp->breakval + size;
+      moveto = PAGE_ALIGN(mdp->breakval);
+      munmap(moveto,
+             (size_t) (((char *) mdp->top) - ((char *) moveto)) - 1);
+      mdp->top = moveto;
+    }
+  } else {
     /* We are allocating memory. Make sure we have an open file
-            descriptor if not working with anonymous memory. */
-    if ( !(mdp->flags & MMALLOC_ANONYMOUS) && mdp -> fd < 0)
-         {
-           result = NULL;
-         }
-    else if ((char*)mdp -> breakval + size > (char*)mdp -> top)
-         {
-         /* The request would move us past the end of the currently
-            mapped memory, so map in enough more memory to satisfy
-            the request.  This means we also have to grow the mapped-to
-            file by an appropriate amount, since mmap cannot be used
-            to extend a file. */
-           moveto = PAGE_ALIGN ((char*)mdp -> breakval + size);
-           mapbytes = (char*)moveto - (char*)mdp -> top;
-           foffset = (char*)mdp -> top - (char*)mdp -> base;
-
-      if( mdp -> fd > 0){
-         /* FIXME:  Test results of lseek() and write() */
-        lseek (mdp -> fd, foffset + mapbytes - 1, SEEK_SET);
-             write (mdp -> fd, &buf, 1);
+       descriptor if not working with anonymous memory. */
+    if (!(mdp->flags & MMALLOC_ANONYMOUS) && mdp->fd < 0) {
+      THROWF(system_error,0,"mmap file descriptor <0 (%d), without MMALLOC_ANONYMOUS being in the flags",mdp->fd);
+      result = NULL;
+    } else if ((char *) mdp->breakval + size > (char *) mdp->top) {
+      /* The request would move us past the end of the currently
+         mapped memory, so map in enough more memory to satisfy
+         the request.  This means we also have to grow the mapped-to
+         file by an appropriate amount, since mmap cannot be used
+         to extend a file. */
+      moveto = PAGE_ALIGN((char *) mdp->breakval + size);
+      mapbytes = (char *) moveto - (char *) mdp->top;
+      foffset = (char *) mdp->top - (char *) mdp->base;
+
+      if (mdp->fd > 0) {
+        /* FIXME:  Test results of lseek() */
+        lseek(mdp->fd, foffset + mapbytes - 1, SEEK_SET);
+        test = write(mdp->fd, &buf, 1);
+        if (test == -1)
+          THROWF(system_error, 0, "write to mmap'ed fd failed! error: %s", strerror(errno));
       }
-           
-           /* Let's call mmap. Note that it is possible that mdp->top
-              is 0. In this case mmap will choose the address for us */
-      mapto = mmap (mdp->top, mapbytes, PROT_READ | PROT_WRITE,
-        MAP_PRIVATE_OR_SHARED(mdp) | MAP_IS_ANONYMOUS(mdp) | MAP_FIXED,
-        MAP_ANON_OR_FD(mdp), foffset);
-
-      if (mapto != (void*) -1){
-           
-        if(mdp -> top == 0)
-          mdp -> base = mdp -> breakval = mapto;
-        
-        mdp -> top = PAGE_ALIGN ((char*)mdp -> breakval + size);
-        result = (void*) mdp -> breakval;
-        mdp -> breakval = (char*)mdp->breakval + size;
+
+      /* Let's call mmap. Note that it is possible that mdp->top
+         is 0. In this case mmap will choose the address for us */
+      mapto = mmap(mdp->top, mapbytes, PROT_READ | PROT_WRITE,
+                   MAP_PRIVATE_OR_SHARED(mdp) | MAP_IS_ANONYMOUS(mdp) |
+                   MAP_FIXED, MAP_ANON_OR_FD(mdp), foffset);
+
+      if (mapto != (void *) -1/* That's MAP_FAILED */) {
+
+        if (mdp->top == 0)
+          mdp->base = mdp->breakval = mapto;
+
+        mdp->top = PAGE_ALIGN((char *) mdp->breakval + size);
+        result = (void *) mdp->breakval;
+        mdp->breakval = (char *) mdp->breakval + size;
+      } else {
+       THROWF(system_error,0,"mmap returned MAP_FAILED! error: %s",strerror(errno));
       }
-         }
-    else
-         {
-           result = (void*) mdp -> breakval;
-           mdp -> breakval = (char*)mdp->breakval + size;
-         }
+    } else {
+      result = (void *) mdp->breakval;
+      mdp->breakval = (char *) mdp->breakval + size;
+    }
   }
   return (result);
 }
 
-void*
-__mmalloc_remap_core (struct mdesc *mdp)
+void *__mmalloc_remap_core(struct mdesc *mdp)
 {
-  voidbase;
+  void *base;
 
   /* FIXME:  Quick hack, needs error checking and other attention. */
 
-  base = mmap (mdp -> base, (char*)mdp -> top - (char*)mdp -> base,
-              PROT_READ | PROT_WRITE | PROT_EXEC, MAP_PRIVATE_OR_SHARED (mdp) | MAP_FIXED,
-              mdp -> fd, 0);
-  return ((void*) base);
+  base = mmap(mdp->base, (char *) mdp->top - (char *) mdp->base,
+              PROT_READ | PROT_WRITE | PROT_EXEC,
+              MAP_PRIVATE_OR_SHARED(mdp) | MAP_FIXED, mdp->fd, 0);
+  return ((void *) base);
 }
 
-void*
-mmalloc_findbase (int size)
+void *mmalloc_findbase(int size)
 {
   int fd;
   int flags;
-  voidbase = NULL;
+  void *base = NULL;
 
 #ifdef MAP_ANONYMOUS
   flags = MAP_PRIVATE | MAP_ANONYMOUS;
@@ -173,32 +168,26 @@ mmalloc_findbase (int size)
 #else
   flags = MAP_PRIVATE;
 #endif
-  fd = open ("/dev/zero", O_RDWR);
-  if (fd != -1)
-    {
-      return ((void*) NULL);
-    }
+  fd = open("/dev/zero", O_RDWR);
+  if (fd != -1) {
+    return ((void *) NULL);
+  }
 #endif
-  base = mmap (0, size, PROT_READ | PROT_WRITE, flags, fd, 0);
-  if (base != (void*) -1)
-    {
-      munmap (base, (size_t) size);
-    }
-  if (fd != -1)
-    {
-      close (fd);
-    }
-  if (base == 0)
-    {
-      /* Don't allow mapping at address zero.  We use that value
-        to signal an error return, and besides, it is useful to
-        catch NULL pointers if it is unmapped.  Instead start
-        at the next page boundary. */
-      base = (void*) getpagesize ();
-    }
-  else if (base == (void*) -1)
-    {
-      base = NULL;
-    }
-  return ((void*) base);
+  base = mmap(0, size, PROT_READ | PROT_WRITE, flags, fd, 0);
+  if (base != (void *) -1) {
+    munmap(base, (size_t) size);
+  }
+  if (fd != -1) {
+    close(fd);
+  }
+  if (base == 0) {
+    /* Don't allow mapping at address zero.  We use that value
+       to signal an error return, and besides, it is useful to
+       catch NULL pointers if it is unmapped.  Instead start
+       at the next page boundary. */
+    base = (void *) (long) getpagesize();
+  } else if (base == (void *) -1) {
+    base = NULL;
+  }
+  return ((void *) base);
 }