Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
model-checker : previous algorithm for heap comparison, used to compare the results...
[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 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 static int heap_region_compare(void *d1, void *d2, size_t size);
194
195 static int heap_region_compare(void *d1, void *d2, size_t size){
196   
197   int distance = 0;
198   size_t i = 0;
199   
200   for(i=0; i<size; i++){
201     if(memcmp(((char *)d1) + i, ((char *)d2) + i, 1) != 0){
202       //XBT_DEBUG("Different byte (offset=%zu) (%p - %p) in heap region", i, (char *)d1 + i, (char *)d2 + i);
203       distance++;
204     }
205   }
206   
207   XBT_DEBUG("Hamming distance between heap regions : %d (total size : %zu)", distance, size);
208
209   return distance;
210 }
211
212 int snapshot_compare(mc_snapshot_t s1, mc_snapshot_t s2){
213
214   int errors = 0, i;
215   //int dist = 0;
216
217   
218   if(s1->num_reg != s2->num_reg){
219     XBT_DEBUG("Different num_reg (s1 = %u, s2 = %u)", s1->num_reg, s2->num_reg);
220     return 1;
221   }
222
223   for(i=0 ; i< s1->num_reg ; i++){
224     
225     if(s1->regions[i]->type != s2->regions[i]->type){
226       XBT_INFO("Different type of region");
227       errors++;
228     }
229     
230     switch(s1->regions[i]->type){
231     case 0 :
232       /* Compare heapregion */
233       if(s1->regions[i]->size != s2->regions[i]->size){
234         XBT_DEBUG("Different size of heap (s1 = %zu, s2 = %zu)", s1->regions[i]->size, s2->regions[i]->size);
235         errors++;
236       }
237       if(s1->regions[i]->start_addr != s2->regions[i]->start_addr){
238         XBT_DEBUG("Different start addr of heap (s1 = %p, s2 = %p)", s1->regions[i]->start_addr, s2->regions[i]->start_addr);
239         errors++;
240       }
241       if(mmalloc_compare_heap((xbt_mheap_t)s1->regions[i]->data, (xbt_mheap_t)s2->regions[i]->data)){
242         XBT_DEBUG("Different heap (mmalloc_compare)");
243         errors++; 
244       }
245       /*if(heap_region_compare(s1->regions[i]->data, s2->regions[i]->data, s1->regions[i]->size) != 0){
246         XBT_DEBUG("Different memcmp for heap");
247         errors++;
248         }*/
249       /*if((dist = mmalloc_linear_compare_heap((xbt_mheap_t)s1->regions[i]->data, (xbt_mheap_t)s2->regions[i]->data)) > 0){
250         XBT_DEBUG("Different heap (mmalloc_linear_compare) : %d", dist);
251         errors++; 
252         }*/
253       break;
254     case 1 :
255       /* Compare data libsimgrid region */
256       if(s1->regions[i]->size != s2->regions[i]->size){
257         XBT_DEBUG("Different size of libsimgrid (data) (s1 = %zu, s2 = %zu)", s1->regions[i]->size, s2->regions[i]->size);
258         errors++;
259       }
260       if(s1->regions[i]->start_addr != s2->regions[i]->start_addr){
261         XBT_DEBUG("Different start addr of libsimgrid (data) (s1 = %p, s2 = %p)", s1->regions[i]->start_addr, s2->regions[i]->start_addr);
262         errors++;
263       }
264       if(data_libsimgrid_region_compare(s1->regions[i]->data, s2->regions[i]->data, s1->regions[i]->size) != 0){
265         XBT_DEBUG("Different memcmp for data in libsimgrid");
266         errors++;
267       }
268       break;
269
270     case 2 :
271        /* Compare data program region */
272       if(s1->regions[i]->size != s2->regions[i]->size){
273         XBT_DEBUG("Different size of data program (s1 = %zu, s2 = %zu)", s1->regions[i]->size, s2->regions[i]->size);
274         //errors++;
275         return 1;
276       }
277       if(s1->regions[i]->start_addr != s2->regions[i]->start_addr){
278         XBT_DEBUG("Different start addr of data program (s1 = %p, s2 = %p)", s1->regions[i]->start_addr, s2->regions[i]->start_addr);
279         //errors++;
280         return 1;
281       }
282       if(data_program_region_compare(s1->regions[i]->data, s2->regions[i]->data, s1->regions[i]->size) != 0){
283         XBT_DEBUG("Different memcmp for data in program");
284         //errors++;
285         return 1;
286       }
287       break;
288  
289     }
290
291   }
292
293
294   return errors > 0;
295   
296 }
297
298 void get_plt_section(){
299
300   FILE *fp;
301   char *line = NULL;            /* Temporal storage for each line that is readed */
302   ssize_t read;                 /* Number of bytes readed */
303   size_t n = 0;                 /* Amount of bytes to read by getline */
304
305   char *lfields[7];
306   int i, plt_not_found = 1;
307   unsigned long int size, offset;
308
309   char *command = bprintf( "objdump --section-headers %s", libsimgrid_path);
310
311   fp = popen(command, "r");
312
313   if(fp == NULL)
314     perror("popen failed");
315
316   while ((read = getline(&line, &n, fp)) != -1 && plt_not_found == 1) {
317
318     if(n == 0)
319       continue;
320
321      /* Wipeout the new line character */
322     line[read - 1] = '\0';
323
324     lfields[0] = strtok(line, " ");
325
326     if(lfields[0] == NULL)
327       continue;
328
329     if(strcmp(lfields[0], "Sections:") == 0 || strcmp(lfields[0], "Idx") == 0 || strcmp(lfields[0], "libsimgrid.so:") == 0)
330       continue;
331
332     for (i = 1; i < 7 && lfields[i - 1] != NULL; i++) {
333       lfields[i] = strtok(NULL, " ");
334     }
335
336     if(i>=5){
337       if(strcmp(lfields[1], ".plt") == 0){
338         size = strtoul(lfields[2], NULL, 16);
339         offset = strtoul(lfields[4], NULL, 16);
340         start_plt = (char *)start_text_libsimgrid + offset;
341         end_plt = (char *)start_plt + size;
342         plt_not_found = 0;
343       }
344     }
345     
346     
347   }
348
349   free(command);
350   free(line);
351   pclose(fp);
352
353 }
354