src/mc/mc_checkpoint.c

   1 #include <libgen.h>
   2 #include "private.h"
   3
   4
   5 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(mc_checkpoint, mc,
   6                                 "Logging specific to mc_checkpoint");
   7
   8 static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size);
   9 static void MC_region_restore(mc_mem_region_t reg);
  10 static void MC_region_destroy(mc_mem_region_t reg);
  11
  12 static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size);
  13
  14 static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size)
  15 {
  16   mc_mem_region_t new_reg = xbt_new0(s_mc_mem_region_t, 1);
  17   new_reg->type = type;
  18   new_reg->start_addr = start_addr;
  19   new_reg->size = size;
  20   new_reg->data = xbt_malloc0(size);
  21   XBT_DEBUG("New reg data %p, start_addr %p", new_reg->data, start_addr);
  22   memcpy(new_reg->data, start_addr, size);
  23
  24   return new_reg;
  25 }
  26
  27 static void MC_region_restore(mc_mem_region_t reg)
  28 {
  29   /*FIXME: check if start_addr is still mapped, if it is not, then map it
  30     before copying the data */
  31   if(reg->type == 3){
  32     memory_map_t maps = get_memory_map();
  33     MC_UNSET_RAW_MEM;
  34     unsigned int i=0;
  35     s_map_region r;
  36     while(i < maps->mapsize){
  37       r = maps->regions[i];
  38       if (maps->regions[i].pathname != NULL){
  39         if (!memcmp(maps->regions[i].pathname, "[stack]", 7)){
  40           size_t diff = (char*)reg->start_addr - (char*)r.start_addr;
  41           void *segment = malloc(reg->size + diff);
  42           XBT_DEBUG("Size of segment : %lu", sizeof(segment));
  43           memcpy((char *)segment + diff, reg->data, reg->size);
  44           //memset(segment, '\0', diff);
  45           //XBT_DEBUG("Memset ok");
  46           //memcpy(segment+diff, reg->data, reg->size);
  47           //XBT_DEBUG("Memcpy segment ok");
  48           memcpy(r.start_addr, segment, sizeof(segment));
  49           XBT_DEBUG("Memcpy region ok");
  50           //memcpy(r.start_addr, reg->data, sizeof(reg->data));
  51           break;
  52         }
  53       }
  54       i++;
  55     }
  56   }else{
  57     XBT_DEBUG("Memcpy : dest %p, src %p, size %Zu", reg->start_addr, reg->data, reg->size);
  58     memcpy(reg->start_addr, reg->data, reg->size);
  59   }
  60
  61   //return;
  62 }
  63
  64 static void MC_region_destroy(mc_mem_region_t reg)
  65 {
  66   xbt_free(reg->data);
  67   xbt_free(reg);
  68 }
  69
  70 static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size)
  71 {
  72   switch(type){
  73   case 0 :
  74     XBT_DEBUG("New region heap");
  75     break;
  76   case 1 :
  77     XBT_DEBUG("New region libsimgrid");
  78     break;
  79   case 2 :
  80     XBT_DEBUG("New region program");
  81     break;
  82   case 3 :
  83     XBT_DEBUG("New region stack");
  84     break;
  85   }
  86   mc_mem_region_t new_reg = MC_region_new(type, start_addr, size);
  87   snapshot->regions = xbt_realloc(snapshot->regions, (snapshot->num_reg + 1) * sizeof(mc_mem_region_t));
  88   snapshot->regions[snapshot->num_reg] = new_reg;
  89   snapshot->num_reg++;
  90   return;
  91 }
  92
  93 void MC_take_snapshot(mc_snapshot_t snapshot)
  94 {
  95   unsigned int i = 0;
  96   s_map_region reg;
  97   memory_map_t maps = get_memory_map();
  98
  99   /* Save the std heap and the writable mapped pages of libsimgrid */
 100   while (i < maps->mapsize) {
 101     reg = maps->regions[i];
 102     if ((reg.prot & PROT_WRITE)){
 103       if (maps->regions[i].pathname == NULL){
 104         if (reg.start_addr == std_heap){ // only save the std heap (and not the raw one)
 105           MC_snapshot_add_region(snapshot, 0, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 106         }
 107       } else {
 108         if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
 109           MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 110         }
 111       }
 112     }
 113     i++;
 114   }
 115
 116   /* FIXME: free the memory map */
 117 }
 118
 119
 120 void MC_take_snapshot_liveness(mc_snapshot_t snapshot, char *prgm)
 121 {
 122   unsigned int i = 0;
 123   s_map_region reg;
 124   memory_map_t maps = get_memory_map();
 125
 126   for(i=0; i< snapshot->num_reg; i++){
 127     MC_region_destroy(snapshot->regions[i]);
 128   }
 129
 130   snapshot->num_reg = 0;
 131
 132   i = 0;
 133
 134   /* Save the std heap and the writable mapped pages of libsimgrid */
 135   while (i < maps->mapsize) {
 136     reg = maps->regions[i];
 137     if ((reg.prot & PROT_WRITE)){
 138       if (maps->regions[i].pathname == NULL){
 139         if (reg.start_addr == std_heap){ // only save the std heap (and not the raw one)
 140           MC_snapshot_add_region(snapshot, 0, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 141         }
 142       } else {
 143         if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
 144           MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 145         } else {
 146           if (!memcmp(basename(maps->regions[i].pathname), basename(prgm), strlen(basename(prgm)))){
 147             MC_snapshot_add_region(snapshot, 2, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 148           } else {
 149             if (!memcmp(maps->regions[i].pathname, "[stack]", 7)){
 150               MC_snapshot_add_region(snapshot, 3, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 151             }
 152           }
 153         }
 154       }
 155     }
 156     i++;
 157   }
 158
 159   /* FIXME: free the memory map */
 160 }
 161
 162 void MC_restore_snapshot(mc_snapshot_t snapshot)
 163 {
 164   unsigned int i;
 165   for(i=0; i < snapshot->num_reg; i++){
 166     MC_region_restore(snapshot->regions[i]);
 167     switch(snapshot->regions[i]->type){
 168     case 0 :
 169       XBT_DEBUG("heap restored");
 170       break;
 171     case 1:
 172       XBT_DEBUG("libsimgrid (data) restored");
 173       break;
 174     case 2:
 175       XBT_DEBUG("program (data) restored");
 176       break;
 177     case 3:
 178       XBT_DEBUG("stack restored");
 179       break;
 180     }
 181
 182   }
 183
 184 }
 185
 186 void MC_free_snapshot(mc_snapshot_t snapshot)
 187 {
 188   unsigned int i;
 189   for(i=0; i < snapshot->num_reg; i++)
 190     MC_region_destroy(snapshot->regions[i]);
 191
 192   xbt_free(snapshot);
 193 }
 194
 195