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