src/mc/sosp/RegionSnapshot.cpp

   1 /* Copyright (c) 2007-2019. The 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 <cstdlib>
   7
   8 #include <sys/mman.h>
   9 #ifdef __FreeBSD__
  10 #define MAP_POPULATE MAP_PREFAULT_READ
  11 #endif
  12
  13 #include "src/mc/ModelChecker.hpp"
  14 #include "src/mc/mc_config.hpp"
  15 #include "src/mc/mc_forward.hpp"
  16
  17 #include "src/mc/mc_smx.hpp"
  18 #include "src/mc/sosp/RegionSnapshot.hpp"
  19
  20 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(mc_RegionSnaphot, mc, "Logging specific to region snapshots");
  21
  22 namespace simgrid {
  23 namespace mc {
  24
  25 RegionDense::RegionDense(RegionType region_type, void* start_addr, void* permanent_addr, size_t size)
  26     : RegionSnapshot(region_type, start_addr, permanent_addr, size)
  27 {
  28   flat_data_ = Buffer::malloc(size);
  29
  30   mc_model_checker->process().read_bytes(flat_data_.get(), size, remote(permanent_addr));
  31
  32   storage_type_ = StorageType::Flat;
  33   page_numbers_.clear();
  34
  35   XBT_DEBUG("New region : type : %s, data : %p (real addr %p), size : %zu",
  36             (region_type == RegionType::Heap ? "Heap" : (region_type == RegionType::Data ? "Data" : "?")),
  37             flat_data().get(), permanent_addr, size);
  38 }
  39
  40 /** @brief Take a snapshot of a given region
  41  *
  42  * @param type
  43  * @param start_addr   Address of the region in the simulated process
  44  * @param permanent_addr Permanent address of this data (for privatized variables, this is the virtual address of the
  45  * privatized mapping)
  46  * @param size         Size of the data*
  47  */
  48 RegionSnapshot* region(RegionType type, void* start_addr, void* permanent_addr, size_t size)
  49 {
  50   if (_sg_mc_sparse_checkpoint)
  51     return new RegionSparse(type, start_addr, permanent_addr, size);
  52   else
  53     return new RegionDense(type, start_addr, permanent_addr, size);
  54 }
  55
  56 RegionSparse::RegionSparse(RegionType region_type, void* start_addr, void* permanent_addr, size_t size)
  57     : RegionSnapshot(region_type, start_addr, permanent_addr, size)
  58 {
  59   simgrid::mc::RemoteClient* process = &mc_model_checker->process();
  60   assert(process != nullptr);
  61
  62   xbt_assert((((uintptr_t)start_addr) & (xbt_pagesize - 1)) == 0, "Start address not at the beginning of a page");
  63   xbt_assert((((uintptr_t)permanent_addr) & (xbt_pagesize - 1)) == 0,
  64              "Permanent address not at the beginning of a page");
  65
  66   storage_type_ = StorageType::Chunked;
  67   flat_data_.clear();
  68
  69   page_numbers_ =
  70       ChunkedData(mc_model_checker->page_store(), *process, RemotePtr<void>(permanent_addr), mmu::chunk_count(size));
  71 }
  72
  73 /** @brief Restore a region from a snapshot
  74  *
  75  *  @param region     Target region
  76  */
  77 void RegionSnapshot::restore()
  78 {
  79   switch (storage_type()) {
  80     case simgrid::mc::StorageType::Flat:
  81       mc_model_checker->process().write_bytes(flat_data().get(), size(), permanent_address());
  82       break;
  83
  84     case simgrid::mc::StorageType::Chunked:
  85       xbt_assert(((permanent_address().address()) & (xbt_pagesize - 1)) == 0, "Not at the beginning of a page");
  86       xbt_assert(simgrid::mc::mmu::chunk_count(size()) == page_data().page_count());
  87
  88       for (size_t i = 0; i != page_data().page_count(); ++i) {
  89         void* target_page = (void*)simgrid::mc::mmu::join(i, (std::uintptr_t)(void*)permanent_address().address());
  90         const void* source_page = page_data().page(i);
  91         mc_model_checker->process().write_bytes(source_page, xbt_pagesize, remote(target_page));
  92       }
  93
  94       break;
  95
  96     default: // includes StorageType::NoData
  97       xbt_die("Storage type not supported");
  98       break;
  99   }
 100 }
 101
 102 } // namespace mc
 103 } // namespace simgrid