+#include <libgen.h>
#include "private.h"
+
+XBT_LOG_NEW_DEFAULT_SUBCATEGORY(mc_checkpoint, mc,
+ "Logging specific to mc_checkpoint");
+
+static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size);
+static void MC_region_restore(mc_mem_region_t reg);
+static void MC_region_destroy(mc_mem_region_t reg);
+static void MC_snapshot_destroy(mc_snapshot_t s);
+
+static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size);
+
+static mc_mem_region_t MC_region_new(int type, void *start_addr, size_t size)
+{
+ mc_mem_region_t new_reg = xbt_new0(s_mc_mem_region_t, 1);
+ new_reg->type = type;
+ new_reg->start_addr = start_addr;
+ new_reg->size = size;
+ new_reg->data = xbt_malloc0(size);
+ memcpy(new_reg->data, start_addr, size);
+ return new_reg;
+}
+
+static void MC_region_restore(mc_mem_region_t reg)
+{
+ /*FIXME: check if start_addr is still mapped, if it is not, then map it
+ before copying the data */
+ memcpy(reg->start_addr, reg->data, reg->size);
+}
+
+static void MC_region_destroy(mc_mem_region_t reg)
+{
+ xbt_free(reg->data);
+ xbt_free(reg);
+}
+
+static void MC_snapshot_add_region(mc_snapshot_t snapshot, int type, void *start_addr, size_t size)
+{
+ mc_mem_region_t new_reg = MC_region_new(type, start_addr, size);
+ snapshot->regions = xbt_realloc(snapshot->regions, (snapshot->num_reg + 1) * sizeof(mc_mem_region_t));
+ snapshot->regions[snapshot->num_reg] = new_reg;
+ snapshot->num_reg++;
+ return;
+}
+
void MC_take_snapshot(mc_snapshot_t snapshot)
-{
-/* Save the heap! */
- snapshot->heap_size = MC_save_heap(&(snapshot->heap));
+{
+ unsigned int i = 0;
+ s_map_region reg;
+ memory_map_t maps = get_memory_map();
+
+ XBT_DEBUG("Take snapshot");
+
+ /* Save the std heap and the writable mapped pages of libsimgrid */
+ while (i < maps->mapsize) {
+ reg = maps->regions[i];
+ if ((reg.prot & PROT_WRITE)){
+ if (maps->regions[i].pathname == NULL){
+ if (reg.start_addr == std_heap){ // only save the std heap (and not the raw one)
+ MC_snapshot_add_region(snapshot, 0, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
+ }
+ } else {
+ if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
+ MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
+ }
+ }
+ }
+ i++;
+ }
-/* Save data and bss that */
- snapshot->data_size = MC_save_dataseg(&(snapshot->data));
+
+ /* FIXME: free the memory map */
+}
+
+static void MC_snapshot_destroy(mc_snapshot_t s){
+
+ int i;
+
+ for(i=0; i< s->num_reg; i++){
+ MC_region_destroy(s->regions[i]);
+ }
+
+ s->num_reg=0;
+
+}
+
+void MC_take_snapshot_liveness(mc_snapshot_t snapshot, char *prgm)
+{
+ unsigned int i = 0;
+ s_map_region reg;
+ memory_map_t maps = get_memory_map();
+
+ MC_snapshot_destroy(snapshot);
+
+ XBT_DEBUG("Take snapshot");
+
+ /* Save the std heap and the writable mapped pages of libsimgrid */
+ while (i < maps->mapsize) {
+ reg = maps->regions[i];
+ if ((reg.prot & PROT_WRITE)){
+ if (maps->regions[i].pathname == NULL){
+ if (reg.start_addr == std_heap){ // only save the std heap (and not the raw one)
+ MC_snapshot_add_region(snapshot, 0, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
+ //XBT_DEBUG("Region heap");
+ }
+ } else {
+ if (!memcmp(basename(maps->regions[i].pathname), "libsimgrid", 10)){
+ MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
+ //XBT_DEBUG("Region simgrid");
+ } else {
+ if (!memcmp(basename(maps->regions[i].pathname), basename(prgm), strlen(basename(prgm)))){
+ MC_snapshot_add_region(snapshot, 1, reg.start_addr, (char*)reg.end_addr - (char*)reg.start_addr);
+ //XBT_DEBUG("Region prog");
+ }
+ }
+ }
+ }
+ i++;
+ }
+
+
+ /* FIXME: free the memory map */
}
void MC_restore_snapshot(mc_snapshot_t snapshot)
-{
- MC_restore_heap(snapshot->heap, snapshot->heap_size);
- MC_restore_dataseg(snapshot->data, snapshot->data_size);
+{
+ unsigned int i;
+ for(i=0; i < snapshot->num_reg; i++)
+ MC_region_restore(snapshot->regions[i]);
}
void MC_free_snapshot(mc_snapshot_t snapshot)
{
- xbt_free(snapshot->heap);
- xbt_free(snapshot->data);
- xbt_free(snapshot);
+ unsigned int i;
+ for(i=0; i < snapshot->num_reg; i++)
+ MC_region_destroy(snapshot->regions[i]);
+
+ xbt_free(snapshot);
}