src/mc/mc_checkpoint.c

   1 /* Copyright (c) 2008-2012 Da SimGrid Team. All rights reserved.            */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #include <libgen.h>
   7 #include "mc_private.h"
   8 #include "xbt/module.h"
   9
  10
  11 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(mc_checkpoint, mc,
  12                                 "Logging specific to mc_checkpoint");
  13
  14 static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size);
  15 static void MC_region_restore(mc_mem_region_t reg);
  16 static void MC_region_destroy(mc_mem_region_t reg);
  17
  18 static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size);
  19
  20 static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size)
  21 {
  22   mc_mem_region_t new_reg = xbt_new0(s_mc_mem_region_t, 1);
  23   new_reg->type = type;
  24   new_reg->start_addr = start_addr;
  25   new_reg->size = size;
  26   new_reg->data = xbt_malloc0(size);
  27   XBT_DEBUG("New reg data %p, start_addr %p, size %zu", new_reg->data, start_addr, size);
  28   memcpy(new_reg->data, start_addr, size);
  29
  30   return new_reg;
  31 }
  32
  33 static void MC_region_restore(mc_mem_region_t reg)
  34 {
  35   /*FIXME: check if start_addr is still mapped, if it is not, then map it
  36     before copying the data */
  37   if(reg->type == 3){
  38     memory_map_t maps = get_memory_map();
  39     MC_UNSET_RAW_MEM;
  40     unsigned int i=0;
  41     s_map_region_t r;
  42     while(i < maps->mapsize){
  43       r = maps->regions[i];
  44       if (maps->regions[i].pathname != NULL){
  45         if (!memcmp(maps->regions[i].pathname, "[stack]", 7)){
  46           size_t diff = (char*)reg->start_addr - (char*)r.start_addr;
  47           void *segment = malloc(reg->size + diff);
  48           XBT_DEBUG("Size of segment : %zu", sizeof(segment));
  49           memcpy((char *)segment + diff, reg->data, reg->size);
  50           memcpy(r.start_addr, segment, sizeof(segment));
  51           XBT_DEBUG("Memcpy region ok");
  52           break;
  53         }
  54       }
  55       i++;
  56     }
  57   }else{
  58     XBT_DEBUG("Memcpy : dest %p, src %p, size %zu", reg->start_addr, reg->data, reg->size);
  59     memcpy(reg->start_addr, reg->data, reg->size);
  60   }
  61
  62   return;
  63 }
  64
  65 static void MC_region_destroy(mc_mem_region_t reg)
  66 {
  67   xbt_free(reg->data);
  68   xbt_free(reg);
  69 }
  70
  71 static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size)
  72 {
  73   switch(type){
  74   case 0 :
  75     XBT_DEBUG("New region heap (%zu)", size);
  76     break;
  77   case 1 :
  78     XBT_DEBUG("New region libsimgrid (%zu)", size);
  79     break;
  80   case 2 :
  81     XBT_DEBUG("New region program data (%zu)", size);
  82     break;
  83   }
  84   mc_mem_region_t new_reg = MC_region_new(type, start_addr, size);
  85   snapshot->regions = xbt_realloc(snapshot->regions, (snapshot->num_reg + 1) * sizeof(mc_mem_region_t));
  86   snapshot->regions[snapshot->num_reg] = new_reg;
  87   snapshot->num_reg++;
  88   return;
  89 }
  90
  91 void MC_take_snapshot(mc_snapshot_t snapshot)
  92 {
  93   unsigned int i = 0;
  94   s_map_region_t reg;
  95   memory_map_t maps = get_memory_map();
  96
  97   /* Save the std heap and the writable mapped pages of libsimgrid */
  98   while (i < maps->mapsize) {
  99     reg = maps->regions[i];
 100     if ((reg.prot & PROT_WRITE)){
 101       if (maps->regions[i].pathname == NULL){
 102         if (reg.start_addr == std_heap){ // only save the std heap (and not the raw one)
 103           MC_snapshot_add_region(snapshot, 0, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 104         }
 105       } else {
 106         if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
 107           MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 108         }
 109       }
 110     }
 111     i++;
 112   }
 113
 114   /* FIXME: free the memory map */
 115 }
 116
 117 void MC_take_snapshot_liveness(mc_snapshot_t snapshot)
 118 {
 119   unsigned int i = 0;
 120   s_map_region_t reg;
 121   memory_map_t maps = get_memory_map();
 122
 123   for(i=0; i< snapshot->num_reg; i++){
 124     MC_region_destroy(snapshot->regions[i]);
 125   }
 126
 127   snapshot->num_reg = 0;
 128
 129   i = 0;
 130
 131   /* Save the std heap and the writable mapped pages of libsimgrid */
 132   while (i < maps->mapsize) {
 133     reg = maps->regions[i];
 134     if ((reg.prot & PROT_WRITE)){
 135       if (maps->regions[i].pathname == NULL){
 136         if (reg.start_addr == std_heap){ // only save the std heap (and not the raw one)
 137           MC_snapshot_add_region(snapshot, 0, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 138         }
 139       } else {
 140         if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
 141           MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 142         } else {
 143           if (!memcmp(basename(maps->regions[i].pathname), basename(xbt_binary_name), strlen(basename(xbt_binary_name)))){
 144             MC_snapshot_add_region(snapshot, 2, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 145           }
 146         }
 147       }
 148     }
 149     i++;
 150   }
 151
 152   /* FIXME: free the memory map */
 153 }
 154
 155 void MC_restore_snapshot(mc_snapshot_t snapshot)
 156 {
 157   unsigned int i;
 158   for(i=0; i < snapshot->num_reg; i++){
 159     MC_region_restore(snapshot->regions[i]);
 160     switch(snapshot->regions[i]->type){
 161     case 0 :
 162       XBT_DEBUG("heap restored");
 163       break;
 164     case 1:
 165       XBT_DEBUG("libsimgrid (data) restored");
 166       break;
 167     case 2:
 168       XBT_DEBUG("data program restored");
 169       break;
 170     }
 171
 172   }
 173
 174 }
 175
 176 void MC_free_snapshot(mc_snapshot_t snapshot)
 177 {
 178   unsigned int i;
 179   for(i=0; i < snapshot->num_reg; i++)
 180     MC_region_destroy(snapshot->regions[i]);
 181
 182   xbt_free(snapshot);
 183 }
 184
 185 static int data_program_region_compare(void *d1, void *d2, size_t size);
 186 static int data_libsimgrid_region_compare(void *d1, void *d2, size_t size);
 187
 188 static int data_program_region_compare(void *d1, void *d2, size_t size){
 189   int distance = 0;
 190   int i;
 191
 192   for(i=0; i<size; i++){
 193     if(memcmp(((char *)d1) + i, ((char *)d2) + i, 1) != 0){
 194       fprintf(stderr,"Different byte (offset=%d) (%p - %p) in data program region\n", i, (char *)d1 + i, (char *)d2 + i);
 195       distance++;
 196     }
 197   }
 198
 199   fprintf(stderr, "Hamming distance between data program regions : %d\n", distance);
 200
 201   return distance;
 202 }
 203
 204 int data_libsimgrid_region_compare(void *d1, void *d2, size_t size){
 205   int distance = 0;
 206   int i;
 207
 208   for(i=0; i<size; i++){
 209     if(memcmp(((char *)d1) + i, ((char *)d2) + i, 1) != 0){
 210       fprintf(stderr, "Different byte (offset=%d) (%p - %p) in data libsimgrid region\n", i, (char *)d1 + i, (char *)d2 + i);
 211       distance++;
 212     }
 213   }
 214
 215   fprintf(stderr, "Hamming distance between data libsimgrid regions : %d\n", distance);
 216
 217   return distance;
 218 }
 219
 220 int snapshot_compare(mc_snapshot_t s1, mc_snapshot_t s2){
 221
 222
 223   if(s1->num_reg != s2->num_reg){
 224     XBT_INFO("Different num_reg (s1 = %u, s2 = %u)", s1->num_reg, s2->num_reg);
 225     return 1;
 226   }
 227
 228   int i;
 229   int errors = 0;
 230
 231   for(i=0 ; i< s1->num_reg ; i++){
 232
 233     if(s1->regions[i]->type != s2->regions[i]->type){
 234       XBT_INFO("Different type of region");
 235       errors++;
 236     }
 237
 238     switch(s1->regions[i]->type){
 239     case 0:
 240       if(s1->regions[i]->size != s2->regions[i]->size){
 241         XBT_INFO("Different size of heap (s1 = %zu, s2 = %zu)", s1->regions[i]->size, s2->regions[i]->size);
 242         errors++;
 243       }
 244       if(s1->regions[i]->start_addr != s2->regions[i]->start_addr){
 245         XBT_INFO("Different start addr of heap (s1 = %p, s2 = %p)", s1->regions[i]->start_addr, s2->regions[i]->start_addr);
 246         errors++;
 247       }
 248       if(mmalloc_compare_heap(s1->regions[i]->data, s2->regions[i]->data)){
 249         XBT_INFO("Different heap (mmalloc_compare)");
 250         errors++;
 251       }
 252       break;
 253     case 1 :
 254       if(s1->regions[i]->size != s2->regions[i]->size){
 255         XBT_INFO("Different size of libsimgrid (s1 = %zu, s2 = %zu)", s1->regions[i]->size, s2->regions[i]->size);
 256         errors++;
 257       }
 258       if(s1->regions[i]->start_addr != s2->regions[i]->start_addr){
 259         XBT_INFO("Different start addr of libsimgrid (s1 = %p, s2 = %p)", s1->regions[i]->start_addr, s2->regions[i]->start_addr);
 260         errors++;
 261       }
 262       if(data_libsimgrid_region_compare(s1->regions[i]->data, s2->regions[i]->data, s1->regions[i]->size) != 0){
 263         XBT_INFO("Different memcmp for data in libsimgrid");
 264         errors++;
 265       }
 266       break;
 267     case 2 :
 268       if(s1->regions[i]->size != s2->regions[i]->size){
 269         XBT_INFO("Different size of data program (s1 = %zu, s2 = %zu)", s1->regions[i]->size, s2->regions[i]->size);
 270         errors++;
 271       }
 272       if(s1->regions[i]->start_addr != s2->regions[i]->start_addr){
 273         XBT_INFO("Different start addr of data program (s1 = %p, s2 = %p)", s1->regions[i]->start_addr, s2->regions[i]->start_addr);
 274         errors++;
 275       }
 276       if(data_program_region_compare(s1->regions[i]->data, s2->regions[i]->data, s1->regions[i]->size) != 0){
 277         XBT_INFO("Different memcmp for data in program");
 278         errors++;
 279       }
 280       break;
 281     default:
 282       break;
 283     }
 284   }
 285
 286   return (errors > 0);
 287
 288 }
 289