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 static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size);
  14 static void MC_region_restore(mc_mem_region_t reg);
  15 static void MC_region_destroy(mc_mem_region_t reg);
  16
  17 static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size);
  18
  19 static int data_program_region_compare(void *d1, void *d2, size_t size);
  20 static int data_libsimgrid_region_compare(void *d1, void *d2, size_t size);
  21
  22 static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size)
  23 {
  24   mc_mem_region_t new_reg = xbt_new0(s_mc_mem_region_t, 1);
  25   new_reg->type = type;
  26   new_reg->start_addr = start_addr;
  27   new_reg->size = size;
  28   new_reg->data = xbt_malloc0(size);
  29   memcpy(new_reg->data, start_addr, size);
  30
  31   XBT_DEBUG("New region : type : %d, data : %p, size : %zu", type, new_reg->data, size);
  32
  33   return new_reg;
  34 }
  35
  36 static void MC_region_restore(mc_mem_region_t reg)
  37 {
  38   /*FIXME: check if start_addr is still mapped, if it is not, then map it
  39     before copying the data */
  40
  41   memcpy(reg->start_addr, reg->data, reg->size);
  42
  43   return;
  44 }
  45
  46 static void MC_region_destroy(mc_mem_region_t reg)
  47 {
  48   xbt_free(reg->data);
  49   xbt_free(reg);
  50 }
  51
  52 static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size)
  53 {
  54   mc_mem_region_t new_reg = MC_region_new(type, start_addr, size);
  55   snapshot->regions = xbt_realloc(snapshot->regions, (snapshot->num_reg + 1) * sizeof(mc_mem_region_t));
  56   snapshot->regions[snapshot->num_reg] = new_reg;
  57   snapshot->num_reg++;
  58   return;
  59 }
  60
  61 void MC_take_snapshot(mc_snapshot_t snapshot)
  62 {
  63   unsigned int i = 0;
  64   s_map_region_t reg;
  65   memory_map_t maps = get_memory_map();
  66
  67   /* Save the std heap and the writable mapped pages of libsimgrid */
  68   while (i < maps->mapsize) {
  69     reg = maps->regions[i];
  70     if ((reg.prot & PROT_WRITE)){
  71       if (maps->regions[i].pathname == NULL){
  72         if (reg.start_addr == std_heap){ // only save the std heap (and not the raw one)
  73           MC_snapshot_add_region(snapshot, 0, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
  74         }
  75       } else {
  76         if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
  77           MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
  78         }
  79       }
  80     }
  81     i++;
  82   }
  83
  84   free_memory_map(maps);
  85 }
  86
  87 void MC_take_snapshot_liveness(mc_snapshot_t snapshot)
  88 {
  89   unsigned int i = 0;
  90   s_map_region_t reg;
  91   memory_map_t maps = get_memory_map();
  92   int nb_reg = 0;
  93
  94   /* Save the std heap and the writable mapped pages of libsimgrid */
  95   while (i < maps->mapsize && nb_reg < 3) {
  96     reg = maps->regions[i];
  97     if ((reg.prot & PROT_WRITE)){
  98       if (maps->regions[i].pathname == NULL){
  99         if (reg.start_addr == std_heap){ // only save the std heap (and not the raw one)
 100           MC_snapshot_add_region(snapshot, 0, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 101           nb_reg++;
 102         }
 103       } else {
 104         if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
 105           MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 106           nb_reg++;
 107         } else {
 108           if (!memcmp(basename(maps->regions[i].pathname), basename(xbt_binary_name), strlen(basename(xbt_binary_name)))){
 109             MC_snapshot_add_region(snapshot, 2, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
 110             nb_reg++;
 111           }
 112         }
 113       }
 114     }
 115     i++;
 116   }
 117
 118   free_memory_map(maps);
 119
 120 }
 121
 122 void MC_restore_snapshot(mc_snapshot_t snapshot)
 123 {
 124   unsigned int i;
 125   for(i=0; i < snapshot->num_reg; i++){
 126     MC_region_restore(snapshot->regions[i]);
 127   }
 128
 129 }
 130
 131 void MC_free_snapshot(mc_snapshot_t snapshot)
 132 {
 133   unsigned int i;
 134   for(i=0; i < snapshot->num_reg; i++)
 135     MC_region_destroy(snapshot->regions[i]);
 136
 137   xbt_free(snapshot);
 138 }
 139
 140 static int data_program_region_compare(void *d1, void *d2, size_t size){
 141   int distance = 0;
 142   size_t i = 0;
 143
 144   for(i=0; i<size; i++){
 145     if(memcmp(((char *)d1) + i, ((char *)d2) + i, 1) != 0){
 146       XBT_DEBUG("Different byte (offset=%zu) (%p - %p) in data program region", i, (char *)d1 + i, (char *)d2 + i);
 147       distance++;
 148     }
 149   }
 150
 151   XBT_DEBUG("Hamming distance between data program regions : %d", distance);
 152
 153   return distance;
 154 }
 155
 156 static int data_libsimgrid_region_compare(void *d1, void *d2, size_t size){
 157   int distance = 0;
 158   size_t i = 0;
 159   int pointer_align;
 160   void *addr_pointed1 = NULL, *addr_pointed2 = NULL;
 161
 162   for(i=0; i<size; i++){
 163     if(memcmp(((char *)d1) + i, ((char *)d2) + i, 1) != 0){
 164       XBT_DEBUG("Different byte (offset=%zu) (%p - %p) in data libsimgrid region", i, (char *)d1 + i, (char *)d2 + i);
 165       pointer_align = (i / sizeof(void*)) * sizeof(void*);
 166       addr_pointed1 = *((void **)((char *)d1 + pointer_align));
 167       addr_pointed2 = *((void **)((char *)d2 + pointer_align));
 168       XBT_DEBUG("Addresses pointed : %p - %p", addr_pointed1, addr_pointed2);
 169       distance++;
 170     }
 171   }
 172
 173   XBT_DEBUG("Hamming distance between data libsimgrid regions : %d", distance); fflush(NULL);
 174
 175   return distance;
 176 }
 177
 178 int snapshot_compare(mc_snapshot_t s1, mc_snapshot_t s2){
 179
 180   int errors = 0, i;
 181
 182   if(s1->num_reg != s2->num_reg){
 183     XBT_DEBUG("Different num_reg (s1 = %u, s2 = %u)", s1->num_reg, s2->num_reg);
 184     return 1;
 185   }
 186
 187   for(i=0 ; i< s1->num_reg ; i++){
 188
 189     if(s1->regions[i]->type != s2->regions[i]->type){
 190       XBT_INFO("Different type of region");
 191       errors++;
 192     }
 193
 194     switch(s1->regions[i]->type){
 195     case 0 :
 196       /* Compare heapregion */
 197       if(s1->regions[i]->size != s2->regions[i]->size){
 198         XBT_DEBUG("Different size of heap (s1 = %zu, s2 = %zu)", s1->regions[i]->size, s2->regions[i]->size);
 199         errors++;
 200       }
 201       if(s1->regions[i]->start_addr != s2->regions[i]->start_addr){
 202         XBT_DEBUG("Different start addr of heap (s1 = %p, s2 = %p)", s1->regions[i]->start_addr, s2->regions[i]->start_addr);
 203         errors++;
 204       }
 205       if(mmalloc_compare_heap((xbt_mheap_t)s1->regions[i]->data, (xbt_mheap_t)s2->regions[i]->data)){
 206         XBT_DEBUG("Different heap (mmalloc_compare)");
 207         errors++;
 208       }
 209       break;
 210     case 1 :
 211       /* Compare data libsimgrid region */
 212       if(s1->regions[i]->size != s2->regions[i]->size){
 213         XBT_DEBUG("Different size of libsimgrid (data) (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 libsimgrid (data) (s1 = %p, s2 = %p)", s1->regions[i]->start_addr, s2->regions[i]->start_addr);
 218         errors++;
 219       }
 220       if(data_libsimgrid_region_compare(s1->regions[i]->data, s2->regions[i]->data, s1->regions[i]->size) != 0){
 221         XBT_DEBUG("Different memcmp for data in libsimgrid");
 222         errors++;
 223       }
 224       break;
 225
 226     case 2 :
 227        /* Compare data program region */
 228       if(s1->regions[i]->size != s2->regions[i]->size){
 229         XBT_DEBUG("Different size of data program (s1 = %zu, s2 = %zu)", s1->regions[i]->size, s2->regions[i]->size);
 230         //errors++;
 231         return 1;
 232       }
 233       if(s1->regions[i]->start_addr != s2->regions[i]->start_addr){
 234         XBT_DEBUG("Different start addr of data program (s1 = %p, s2 = %p)", s1->regions[i]->start_addr, s2->regions[i]->start_addr);
 235         //errors++;
 236         return 1;
 237       }
 238       if(data_program_region_compare(s1->regions[i]->data, s2->regions[i]->data, s1->regions[i]->size) != 0){
 239         XBT_DEBUG("Different memcmp for data in program");
 240         //errors++;
 241         return 1;
 242       }
 243       break;
 244
 245     }
 246
 247   }
 248
 249
 250   return errors > 0;
 251
 252 }
 253