X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/07c319ec54d6fc778ee3cc5e75a747242006723e..1dc68fca3f3bd110ddaaf88fb90a8bbb83a584b2:/src/mc/mc_checkpoint.c

diff --git a/src/mc/mc_checkpoint.c b/src/mc/mc_checkpoint.c
index d416f72a16..a43de263c4 100644
--- a/src/mc/mc_checkpoint.c
+++ b/src/mc/mc_checkpoint.c
@@ -1,3 +1,4 @@
+#include <libgen.h>
 #include "private.h"
 
 static mc_mem_region_t MC_region_new(void *start_addr, size_t size);
@@ -40,21 +41,35 @@ static void MC_snapshot_add_region(mc_snapshot_t snapshot, void *start_addr, siz
 
 void MC_take_snapshot(mc_snapshot_t snapshot)
 {
-  unsigned int i;
+  unsigned int i = 0;
+  char copy = 0;
   s_map_region reg;
   memory_map_t maps = get_memory_map();
 
-  /* Save all the writable mapped pages except the  and the stack */
-  for (i = 0; i < maps->mapsize; i++) {
-    reg = maps->regions[i];
-    if((reg.prot & PROT_WRITE)
-       && (reg.pathname == NULL
-           || (strncmp(reg.pathname, "/dev/zero", 9)
-               && strncmp(reg.pathname, "[stack]", 7)))){
-      MC_snapshot_add_region(snapshot, reg.start_addr,
-                             (char*)reg.end_addr - (char*)reg.start_addr);
+  /* Save the std heap and the writtable mapped pages of libsimgrid */
+    while(i < maps->mapsize
+          && (maps->regions[i].pathname == NULL
+              || memcmp(maps->regions[i].pathname, "/lib/ld", 7))){
+      reg = maps->regions[i];
+      if((reg.prot & PROT_WRITE)){
+        if(reg.start_addr == std_heap){
+          MC_snapshot_add_region(snapshot, reg.start_addr,
+                                 (char*)reg.end_addr - (char*)reg.start_addr);
+
+        }else if(copy || (reg.pathname != NULL
+                 && !memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10))){
+          MC_snapshot_add_region(snapshot, reg.start_addr,
+                                 (char*)reg.end_addr - (char*)reg.start_addr);
+          /* This will force the save of the regions in the next iterations,
+           * but we assume that ld will be found mapped and break the loop
+           * before saving a wrong region.(This is ugly I know). */
+          copy = TRUE;
+        }
+      }
+      i++;
     }
-  }
+
+  /* FIXME: free the memory map */
 }
 
 void MC_restore_snapshot(mc_snapshot_t snapshot)