Add FindNS3 module to cmake.

[simgrid.git] / src / mc / mc_checkpoint.c

diff --git a/src/mc/mc_checkpoint.c b/src/mc/mc_checkpoint.c
index c1e8ea9..a43de26 100644 (file)
--- 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,19 +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 raw_heap and the stack */
-  for (i = 0; i < maps->mapsize; i++) {
-    reg = maps->regions[i];
-    if((reg.prot & PROT_WRITE) && reg.start_addr != raw_heap
-       && (reg.pathname == NULL || 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)