Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
make sure that we won't survive a parser error in the deployment file
[simgrid.git] / src / mc / mc_checkpoint.c
1 #include <libgen.h>
2 #include "private.h"
3
4 static mc_mem_region_t MC_region_new(void *start_addr, size_t size);
5 static void MC_region_restore(mc_mem_region_t reg);
6 static void MC_region_destroy(mc_mem_region_t reg);
7
8 static void MC_snapshot_add_region(mc_snapshot_t snapshot, void *start_addr, size_t size);
9
10 static mc_mem_region_t MC_region_new(void *start_addr, size_t size)
11 {
12   mc_mem_region_t new_reg = xbt_new0(s_mc_mem_region_t, 1);
13   new_reg->start_addr = start_addr;
14   new_reg->size = size;
15   new_reg->data = xbt_malloc0(size);
16   memcpy(new_reg->data, start_addr, size);
17   return new_reg;
18 }
19
20 static void MC_region_restore(mc_mem_region_t reg)
21 {
22   /*FIXME: check if start_addr is still mapped, if it is not, then map it
23     before copying the data */
24   memcpy(reg->start_addr, reg->data, reg->size);
25 }
26
27 static void MC_region_destroy(mc_mem_region_t reg)
28 {
29   xbt_free(reg->data);
30   xbt_free(reg);
31 }
32
33 static void MC_snapshot_add_region(mc_snapshot_t snapshot, void *start_addr, size_t size)
34 {
35   mc_mem_region_t new_reg = MC_region_new(start_addr, size);
36   snapshot->regions = xbt_realloc(snapshot->regions, (snapshot->num_reg + 1) * sizeof(mc_mem_region_t));
37   snapshot->regions[snapshot->num_reg] = new_reg;
38   snapshot->num_reg++;
39   return;
40
41
42 void MC_take_snapshot(mc_snapshot_t snapshot)
43 {
44   unsigned int i = 0;
45   s_map_region reg;
46   memory_map_t maps = get_memory_map();
47
48   /* Save the std heap and the writable mapped pages of libsimgrid */
49   while (i < maps->mapsize) {
50     reg = maps->regions[i];
51     if ((reg.prot & PROT_WRITE)){
52       if (maps->regions[i].pathname == NULL){
53         if (reg.start_addr == std_heap){
54           MC_snapshot_add_region(snapshot, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
55         }
56       } else {
57         if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
58           MC_snapshot_add_region(snapshot, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
59         }
60       }
61     }
62     i++;
63   }
64
65
66   /* FIXME: free the memory map */
67 }
68
69 void MC_restore_snapshot(mc_snapshot_t snapshot)
70 {
71   unsigned int i;
72   for(i=0; i < snapshot->num_reg; i++)
73     MC_region_restore(snapshot->regions[i]);
74 }
75
76 void MC_free_snapshot(mc_snapshot_t snapshot)
77 {
78   unsigned int i;
79   for(i=0; i < snapshot->num_reg; i++)
80     MC_region_destroy(snapshot->regions[i]);
81
82   xbt_free(snapshot);
83 }