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 #include "../simix/smx_private.h"
  11
  12 #include <libunwind.h>
  13
  14 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(mc_checkpoint, mc,
  15                                 "Logging specific to mc_checkpoint");
  16
  17 void *start_text_libsimgrid;
  18 void *start_plt, *end_plt;
  19 char *libsimgrid_path;
  20
  21 static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size);
  22 static void MC_region_restore(mc_mem_region_t reg);
  23 static void MC_region_destroy(mc_mem_region_t reg);
  24
  25 static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size);
  26
  27 static void add_value(xbt_dynar_t *list, const char *type, unsigned long int val);
  28 static xbt_dynar_t take_snapshot_stacks(void *heap);
  29 static void get_local_variables_values(xbt_dynar_t *all_variables, stack_region_t stack, void *heap);
  30 static void print_local_variables_values(xbt_dynar_t all_variables);
  31
  32 static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size)
  33 {
  34   mc_mem_region_t new_reg = xbt_new0(s_mc_mem_region_t, 1);
  35   new_reg->type = type;
  36   new_reg->start_addr = start_addr;
  37   new_reg->size = size;
  38   new_reg->data = xbt_malloc0(size);
  39   memcpy(new_reg->data, start_addr, size);
  40
  41   XBT_DEBUG("New region : type : %d, data : %p, size : %zu", type, new_reg->data, size);
  42
  43   return new_reg;
  44 }
  45
  46 static void MC_region_restore(mc_mem_region_t reg)
  47 {
  48   /*FIXME: check if start_addr is still mapped, if it is not, then map it
  49     before copying the data */
  50
  51   memcpy(reg->start_addr, reg->data, reg->size);
  52
  53   return;
  54 }
  55
  56 static void MC_region_destroy(mc_mem_region_t reg)
  57 {
  58   xbt_free(reg->data);
  59   xbt_free(reg);
  60 }
  61
  62 static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size)
  63 {
  64   mc_mem_region_t new_reg = MC_region_new(type, start_addr, size);
  65   snapshot->regions = xbt_realloc(snapshot->regions, (snapshot->num_reg + 1) * sizeof(mc_mem_region_t));
  66   snapshot->regions[snapshot->num_reg] = new_reg;
  67   snapshot->num_reg++;
  68   return;
  69 }
  70
  71 void MC_take_snapshot(mc_snapshot_t snapshot)
  72 {
  73   unsigned int i = 0;
  74   s_map_region_t reg;
  75   memory_map_t maps = get_memory_map();
  76
  77   /* Save the std heap and the writable mapped pages of libsimgrid */
  78   while (i < maps->mapsize) {
  79     reg = maps->regions[i];
  80     if ((reg.prot & PROT_WRITE)){
  81       if (maps->regions[i].pathname == NULL){
  82         if (reg.start_addr == std_heap){ // only save the std heap (and not the raw one)
  83           MC_snapshot_add_region(snapshot, 0, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
  84         }
  85       } else {
  86         if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
  87           MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
  88         }
  89       }
  90     }
  91     i++;
  92   }
  93
  94   free_memory_map(maps);
  95 }
  96
  97 void MC_take_snapshot_liveness(mc_snapshot_t snapshot)
  98 {
  99   unsigned int i = 0;
 100   s_map_region_t reg;
 101   memory_map_t maps = get_memory_map();
 102   int nb_reg = 0;
 103   void *heap = NULL;
 104
 105   /* Save the std heap and the writable mapped pages of libsimgrid */
 106   while (i < maps->mapsize && nb_reg < 3) {
 107     reg = maps->regions[i];
 108     if ((reg.prot & PROT_WRITE)){
 109       if (maps->regions[i].pathname == NULL){
 110         if (reg.start_addr == std_heap){ // only save the std heap (and not the raw one)
 111           MC_snapshot_add_region(snapshot, 0, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 112           heap = snapshot->regions[nb_reg]->data;
 113           nb_reg++;
 114         }
 115       } else {
 116         if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
 117           MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 118           nb_reg++;
 119         } else {
 120           if (!memcmp(basename(maps->regions[i].pathname), basename(xbt_binary_name), strlen(basename(xbt_binary_name)))){
 121             MC_snapshot_add_region(snapshot, 2, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 122             nb_reg++;
 123           }
 124         }
 125       }
 126     }else if ((reg.prot & PROT_READ)){
 127       if (maps->regions[i].pathname != NULL){
 128         if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
 129           start_text_libsimgrid = reg.start_addr;
 130           libsimgrid_path = strdup(maps->regions[i].pathname);
 131         }
 132       }
 133     }
 134     i++;
 135   }
 136
 137   snapshot->stacks = take_snapshot_stacks(heap);
 138
 139   free_memory_map(maps);
 140
 141 }
 142
 143 void MC_restore_snapshot(mc_snapshot_t snapshot)
 144 {
 145   unsigned int i;
 146   for(i=0; i < snapshot->num_reg; i++){
 147     MC_region_restore(snapshot->regions[i]);
 148   }
 149
 150 }
 151
 152 void MC_free_snapshot(mc_snapshot_t snapshot)
 153 {
 154   unsigned int i;
 155   for(i=0; i < snapshot->num_reg; i++)
 156     MC_region_destroy(snapshot->regions[i]);
 157
 158   xbt_free(snapshot);
 159 }
 160
 161 static int data_program_region_compare(void *d1, void *d2, size_t size){
 162   int distance = 0;
 163   size_t i = 0;
 164
 165   for(i=0; i<size; i++){
 166     if(memcmp(((char *)d1) + i, ((char *)d2) + i, 1) != 0){
 167       XBT_DEBUG("Different byte (offset=%zu) (%p - %p) in data program region", i, (char *)d1 + i, (char *)d2 + i);
 168       distance++;
 169     }
 170   }
 171
 172   XBT_DEBUG("Hamming distance between data program regions : %d", distance);
 173
 174   return distance;
 175 }
 176
 177 static int data_libsimgrid_region_compare(void *d1, void *d2, size_t size){
 178   int distance = 0;
 179   size_t i = 0;
 180   int pointer_align;
 181   void *addr_pointed1 = NULL, *addr_pointed2 = NULL;
 182
 183   for(i=0; i<size; i++){
 184     if(memcmp(((char *)d1) + i, ((char *)d2) + i, 1) != 0){
 185       pointer_align = (i / sizeof(void*)) * sizeof(void*);
 186       addr_pointed1 = *((void **)((char *)d1 + pointer_align));
 187       addr_pointed2 = *((void **)((char *)d2 + pointer_align));
 188       if((addr_pointed1 > start_plt && addr_pointed1 < end_plt) || (addr_pointed2 > start_plt && addr_pointed2 < end_plt)){
 189         continue;
 190       }else{
 191         XBT_DEBUG("Different byte (offset=%zu) (%p - %p) in data libsimgrid region", i, (char *)d1 + i, (char *)d2 + i);
 192         XBT_DEBUG("Addresses pointed : %p - %p\n", addr_pointed1, addr_pointed2);
 193         distance++;
 194       }
 195     }
 196   }
 197
 198   XBT_DEBUG("Hamming distance between data libsimgrid regions : %d", distance); fflush(NULL);
 199
 200   return distance;
 201 }
 202
 203 static int heap_region_compare(void *d1, void *d2, size_t size);
 204
 205 static int heap_region_compare(void *d1, void *d2, size_t size){
 206
 207   int distance = 0;
 208   size_t i = 0;
 209
 210   for(i=0; i<size; i++){
 211     if(memcmp(((char *)d1) + i, ((char *)d2) + i, 1) != 0){
 212       //XBT_DEBUG("Different byte (offset=%zu) (%p - %p) in heap region", i, (char *)d1 + i, (char *)d2 + i);
 213       distance++;
 214     }
 215   }
 216
 217   XBT_DEBUG("Hamming distance between heap regions : %d (total size : %zu)", distance, size);
 218
 219   return distance;
 220 }
 221
 222 int snapshot_compare(mc_snapshot_t s1, mc_snapshot_t s2){
 223
 224   int errors = 0, i;
 225   //int dist = 0;
 226
 227
 228   if(s1->num_reg != s2->num_reg){
 229     XBT_DEBUG("Different num_reg (s1 = %u, s2 = %u)", s1->num_reg, s2->num_reg);
 230     return 1;
 231   }
 232
 233   for(i=0 ; i< s1->num_reg ; i++){
 234
 235     if(s1->regions[i]->type != s2->regions[i]->type){
 236       XBT_INFO("Different type of region");
 237       errors++;
 238     }
 239
 240     switch(s1->regions[i]->type){
 241     case 0 :
 242       /* Compare heapregion */
 243       if(s1->regions[i]->size != s2->regions[i]->size){
 244         XBT_DEBUG("Different size of heap (s1 = %zu, s2 = %zu)", s1->regions[i]->size, s2->regions[i]->size);
 245         errors++;
 246       }
 247       if(s1->regions[i]->start_addr != s2->regions[i]->start_addr){
 248         XBT_DEBUG("Different start addr of heap (s1 = %p, s2 = %p)", s1->regions[i]->start_addr, s2->regions[i]->start_addr);
 249         errors++;
 250       }
 251       if(mmalloc_compare_heap((xbt_mheap_t)s1->regions[i]->data, (xbt_mheap_t)s2->regions[i]->data)){
 252         XBT_DEBUG("Different heap (mmalloc_compare)");
 253         errors++;
 254       }
 255       /*if(heap_region_compare(s1->regions[i]->data, s2->regions[i]->data, s1->regions[i]->size) != 0){
 256         XBT_DEBUG("Different memcmp for heap");
 257         errors++;
 258         }*/
 259       /*if((dist = mmalloc_linear_compare_heap((xbt_mheap_t)s1->regions[i]->data, (xbt_mheap_t)s2->regions[i]->data)) > 0){
 260         XBT_DEBUG("Different heap (mmalloc_linear_compare) : %d", dist);
 261         errors++;
 262         }*/
 263       break;
 264     case 1 :
 265       /* Compare data libsimgrid region */
 266       if(s1->regions[i]->size != s2->regions[i]->size){
 267         XBT_DEBUG("Different size of libsimgrid (data) (s1 = %zu, s2 = %zu)", s1->regions[i]->size, s2->regions[i]->size);
 268         errors++;
 269       }
 270       if(s1->regions[i]->start_addr != s2->regions[i]->start_addr){
 271         XBT_DEBUG("Different start addr of libsimgrid (data) (s1 = %p, s2 = %p)", s1->regions[i]->start_addr, s2->regions[i]->start_addr);
 272         errors++;
 273       }
 274       if(data_libsimgrid_region_compare(s1->regions[i]->data, s2->regions[i]->data, s1->regions[i]->size) != 0){
 275         XBT_DEBUG("Different memcmp for data in libsimgrid");
 276         errors++;
 277       }
 278       break;
 279
 280     case 2 :
 281        /* Compare data program region */
 282       if(s1->regions[i]->size != s2->regions[i]->size){
 283         XBT_DEBUG("Different size of data program (s1 = %zu, s2 = %zu)", s1->regions[i]->size, s2->regions[i]->size);
 284         //errors++;
 285         return 1;
 286       }
 287       if(s1->regions[i]->start_addr != s2->regions[i]->start_addr){
 288         XBT_DEBUG("Different start addr of data program (s1 = %p, s2 = %p)", s1->regions[i]->start_addr, s2->regions[i]->start_addr);
 289         //errors++;
 290         return 1;
 291       }
 292       if(data_program_region_compare(s1->regions[i]->data, s2->regions[i]->data, s1->regions[i]->size) != 0){
 293         XBT_DEBUG("Different memcmp for data in program");
 294         //errors++;
 295         return 1;
 296       }
 297       break;
 298
 299     }
 300
 301   }
 302
 303
 304   return errors > 0;
 305
 306 }
 307
 308 void get_plt_section(){
 309
 310   FILE *fp;
 311   char *line = NULL;            /* Temporal storage for each line that is readed */
 312   ssize_t read;                 /* Number of bytes readed */
 313   size_t n = 0;                 /* Amount of bytes to read by getline */
 314
 315   char *lfields[7];
 316   int i, plt_not_found = 1;
 317   unsigned long int size, offset;
 318
 319   char *command = bprintf( "objdump --section-headers %s", libsimgrid_path);
 320
 321   fp = popen(command, "r");
 322
 323   if(fp == NULL)
 324     perror("popen failed");
 325
 326   while ((read = getline(&line, &n, fp)) != -1 && plt_not_found == 1) {
 327
 328     if(n == 0)
 329       continue;
 330
 331      /* Wipeout the new line character */
 332     line[read - 1] = '\0';
 333
 334     lfields[0] = strtok(line, " ");
 335
 336     if(lfields[0] == NULL)
 337       continue;
 338
 339     if(strcmp(lfields[0], "Sections:") == 0 || strcmp(lfields[0], "Idx") == 0 || strcmp(lfields[0], "libsimgrid.so:") == 0)
 340       continue;
 341
 342     for (i = 1; i < 7 && lfields[i - 1] != NULL; i++) {
 343       lfields[i] = strtok(NULL, " ");
 344     }
 345
 346     if(i>=5){
 347       if(strcmp(lfields[1], ".plt") == 0){
 348         size = strtoul(lfields[2], NULL, 16);
 349         offset = strtoul(lfields[4], NULL, 16);
 350         start_plt = (char *)start_text_libsimgrid + offset;
 351         end_plt = (char *)start_plt + size;
 352         plt_not_found = 0;
 353       }
 354     }
 355
 356
 357   }
 358
 359   free(command);
 360   free(line);
 361   pclose(fp);
 362
 363 }
 364