Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
model-checker : remove stack from regions added in snapshot
[simgrid.git] / src / mc / mc_checkpoint.c
1 #include <libgen.h>
2 #include "private.h"
3
4
5 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(mc_checkpoint, mc,
6                                 "Logging specific to mc_checkpoint");
7
8 static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size);
9 static void MC_region_restore(mc_mem_region_t reg);
10 static void MC_region_destroy(mc_mem_region_t reg);
11
12 static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size);
13
14 static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size)
15 {
16   mc_mem_region_t new_reg = xbt_new0(s_mc_mem_region_t, 1);
17   new_reg->type = type;
18   new_reg->start_addr = start_addr;
19   new_reg->size = size;
20   new_reg->data = xbt_malloc0(size);
21   XBT_DEBUG("New reg data %p, start_addr %p", new_reg->data, start_addr);
22   memcpy(new_reg->data, start_addr, size);
23   
24   return new_reg;
25 }
26
27 static void MC_region_restore(mc_mem_region_t reg)
28 {
29   /*FIXME: check if start_addr is still mapped, if it is not, then map it
30     before copying the data */
31   if(reg->type == 3){
32     memory_map_t maps = get_memory_map();
33     MC_UNSET_RAW_MEM;
34     unsigned int i=0;
35     s_map_region r;
36     while(i < maps->mapsize){
37       r = maps->regions[i];
38       if (maps->regions[i].pathname != NULL){
39         if (!memcmp(maps->regions[i].pathname, "[stack]", 7)){
40           size_t diff = (char*)reg->start_addr - (char*)r.start_addr;
41           void *segment = malloc(reg->size + diff);
42           XBT_DEBUG("Size of segment : %lu", sizeof(segment));
43           memcpy((char *)segment + diff, reg->data, reg->size);
44           //memset(segment, '\0', diff);
45           //XBT_DEBUG("Memset ok");
46           //memcpy(segment+diff, reg->data, reg->size);
47           //XBT_DEBUG("Memcpy segment ok");
48           memcpy(r.start_addr, segment, sizeof(segment));
49           XBT_DEBUG("Memcpy region ok");
50           //memcpy(r.start_addr, reg->data, sizeof(reg->data));
51           break;
52         }
53       }
54       i++;
55     }
56   }else{
57     XBT_DEBUG("Memcpy : dest %p, src %p, size %Zu", reg->start_addr, reg->data, reg->size);
58     memcpy(reg->start_addr, reg->data, reg->size);
59   }
60   
61   //return;
62 }
63
64 static void MC_region_destroy(mc_mem_region_t reg)
65 {
66   xbt_free(reg->data);
67   xbt_free(reg);
68 }
69
70 static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size)
71 {
72   switch(type){
73   case 0 : 
74     XBT_DEBUG("New region heap");
75     break;
76   case 1 : 
77     XBT_DEBUG("New region libsimgrid");
78     break;
79   case 2 : 
80     XBT_DEBUG("New region program");
81     break;
82   case 3 : 
83     XBT_DEBUG("New region stack");
84     break;
85   }
86   mc_mem_region_t new_reg = MC_region_new(type, start_addr, size);
87   snapshot->regions = xbt_realloc(snapshot->regions, (snapshot->num_reg + 1) * sizeof(mc_mem_region_t));
88   snapshot->regions[snapshot->num_reg] = new_reg;
89   snapshot->num_reg++;
90   return;
91
92
93 void MC_take_snapshot(mc_snapshot_t snapshot)
94 {
95   unsigned int i = 0;
96   s_map_region reg;
97   memory_map_t maps = get_memory_map();
98
99   /* Save the std heap and the writable mapped pages of libsimgrid */
100   while (i < maps->mapsize) {
101     reg = maps->regions[i];
102     if ((reg.prot & PROT_WRITE)){
103       if (maps->regions[i].pathname == NULL){
104         if (reg.start_addr == std_heap){ // only save the std heap (and not the raw one)
105           MC_snapshot_add_region(snapshot, 0, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
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         } 
111       }
112     }
113     i++;
114   }
115
116   /* FIXME: free the memory map */
117 }
118
119
120 void MC_take_snapshot_liveness(mc_snapshot_t snapshot, char *prgm)
121 {
122   unsigned int i = 0;
123   s_map_region reg;
124   memory_map_t maps = get_memory_map();
125
126   for(i=0; i< snapshot->num_reg; i++){
127     MC_region_destroy(snapshot->regions[i]);
128   }
129
130   snapshot->num_reg = 0;
131
132   i = 0;
133
134   /* Save the std heap and the writable mapped pages of libsimgrid */
135   while (i < maps->mapsize) {
136     reg = maps->regions[i];
137     if ((reg.prot & PROT_WRITE)){
138       if (maps->regions[i].pathname == NULL){
139         if (reg.start_addr == std_heap){ // only save the std heap (and not the raw one)
140           MC_snapshot_add_region(snapshot, 0, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
141         }
142       } else {
143         if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
144           MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
145         } else {
146           if (!memcmp(basename(maps->regions[i].pathname), basename(prgm), strlen(basename(prgm)))){
147             MC_snapshot_add_region(snapshot, 2, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
148           } //else {
149             //if (!memcmp(maps->regions[i].pathname, "[stack]", 7)){
150           //MC_snapshot_add_region(snapshot, 3, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
151           //}
152           //}
153         }
154       }
155     }
156     i++;
157   }
158
159   /* FIXME: free the memory map */
160 }
161
162 void MC_restore_snapshot(mc_snapshot_t snapshot)
163 {
164   unsigned int i;
165   for(i=0; i < snapshot->num_reg; i++){
166     MC_region_restore(snapshot->regions[i]);
167     switch(snapshot->regions[i]->type){
168     case 0 : 
169       XBT_DEBUG("heap restored");
170       break;
171     case 1:
172       XBT_DEBUG("libsimgrid (data) restored");
173       break;
174     case 2:
175       XBT_DEBUG("program (data) restored");
176       break;
177     case 3:
178       XBT_DEBUG("stack restored");
179       break;
180     }
181
182   }
183
184 }
185
186 void MC_free_snapshot(mc_snapshot_t snapshot)
187 {
188   unsigned int i;
189   for(i=0; i < snapshot->num_reg; i++)
190     MC_region_destroy(snapshot->regions[i]);
191
192   xbt_free(snapshot);
193 }
194
195