Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
b2a5d0ddbab7f2de1f4443155359490a9d6cb62c
[simgrid.git] / 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