model-checker : add snapshot of stack memory region for comparison of pairs reached...

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

diff --git a/src/mc/mc_checkpoint.c b/src/mc/mc_checkpoint.c
index 447a8db..aec3a71 100644 (file)
--- a/src/mc/mc_checkpoint.c
+++ b/src/mc/mc_checkpoint.c
@@ -18,7 +18,9 @@ static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size)
   new_reg->start_addr = start_addr;
   new_reg->size = size;
   new_reg->data = xbt_malloc0(size);
+  XBT_DEBUG("New reg data %p, start_addr %p", new_reg->data, start_addr);
   memcpy(new_reg->data, start_addr, size);
+  
   return new_reg;
 }
 
@@ -26,7 +28,37 @@ static void MC_region_restore(mc_mem_region_t reg)
 {
   /*FIXME: check if start_addr is still mapped, if it is not, then map it
     before copying the data */
-  memcpy(reg->start_addr, reg->data, reg->size);
+  if(reg->type == 3){
+    memory_map_t maps = get_memory_map();
+    MC_UNSET_RAW_MEM;
+    unsigned int i=0;
+    s_map_region r;
+    while(i < maps->mapsize){
+      r = maps->regions[i];
+      if (maps->regions[i].pathname != NULL){
+       if (!memcmp(maps->regions[i].pathname, "[stack]", 7)){
+         size_t diff = (char*)reg->start_addr - (char*)r.start_addr;
+         void *segment = malloc(reg->size + diff);
+         XBT_DEBUG("Size of segment : %lu", sizeof(segment));
+         memcpy((char *)segment + diff, reg->data, reg->size);
+         //memset(segment, '\0', diff);
+         //XBT_DEBUG("Memset ok");
+         //memcpy(segment+diff, reg->data, reg->size);
+         //XBT_DEBUG("Memcpy segment ok");
+         memcpy(r.start_addr, segment, sizeof(segment));
+         XBT_DEBUG("Memcpy region ok");
+         //memcpy(r.start_addr, reg->data, sizeof(reg->data));
+         break;
+       }
+      }
+      i++;
+    }
+  }else{
+    XBT_DEBUG("Memcpy : dest %p, src %p, size %Zu", reg->start_addr, reg->data, reg->size);
+    memcpy(reg->start_addr, reg->data, reg->size);
+  }
+  
+  //return;
 }
 
 static void MC_region_destroy(mc_mem_region_t reg)
@@ -37,6 +69,20 @@ static void MC_region_destroy(mc_mem_region_t reg)
 
 static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size)
 {
+  switch(type){
+  case 0 : 
+    XBT_DEBUG("New region heap");
+    break;
+  case 1 : 
+    XBT_DEBUG("New region libsimgrid");
+    break;
+  case 2 : 
+    XBT_DEBUG("New region program");
+    break;
+  case 3 : 
+    XBT_DEBUG("New region stack");
+    break;
+  }
   mc_mem_region_t new_reg = MC_region_new(type, start_addr, size);
   snapshot->regions = xbt_realloc(snapshot->regions, (snapshot->num_reg + 1) * sizeof(mc_mem_region_t));
   snapshot->regions[snapshot->num_reg] = new_reg;
@@ -77,6 +123,14 @@ void MC_take_snapshot_liveness(mc_snapshot_t snapshot, char *prgm)
   s_map_region reg;
   memory_map_t maps = get_memory_map();
 
+  for(i=0; i< snapshot->num_reg; i++){
+    MC_region_destroy(snapshot->regions[i]);
+  }
+
+  snapshot->num_reg = 0;
+
+  i = 0;
+
   /* Save the std heap and the writable mapped pages of libsimgrid */
   while (i < maps->mapsize) {
     reg = maps->regions[i];
@@ -90,8 +144,12 @@ void MC_take_snapshot_liveness(mc_snapshot_t snapshot, char *prgm)
           MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
         } else {
          if (!memcmp(basename(maps->regions[i].pathname), basename(prgm), strlen(basename(prgm)))){
-           MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
-         } 
+           MC_snapshot_add_region(snapshot, 2, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
+         } else {
+           if (!memcmp(maps->regions[i].pathname, "[stack]", 7)){
+             MC_snapshot_add_region(snapshot, 3, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
+           }
+         }
        }
       }
     }
@@ -104,8 +162,25 @@ void MC_take_snapshot_liveness(mc_snapshot_t snapshot, char *prgm)
 void MC_restore_snapshot(mc_snapshot_t snapshot)
 {
   unsigned int i;
-  for(i=0; i < snapshot->num_reg; i++)
+  for(i=0; i < snapshot->num_reg; i++){
     MC_region_restore(snapshot->regions[i]);
+    switch(snapshot->regions[i]->type){
+    case 0 : 
+      XBT_DEBUG("heap restored");
+      break;
+    case 1:
+      XBT_DEBUG("libsimgrid (data) restored");
+      break;
+    case 2:
+      XBT_DEBUG("program (data) restored");
+      break;
+    case 3:
+      XBT_DEBUG("stack restored");
+      break;
+    }
+
+  }
+
 }
 
 void MC_free_snapshot(mc_snapshot_t snapshot)
@@ -116,3 +191,5 @@ void MC_free_snapshot(mc_snapshot_t snapshot)
 
   xbt_free(snapshot);
 }
+
+