mc: Also remove the process_index

[simgrid.git] / src / mc / sosp / RegionSnapshot.cpp

diff --git a/src/mc/sosp/RegionSnapshot.cpp b/src/mc/sosp/RegionSnapshot.cpp
index a657ea0..56a2395 100644 (file)
--- a/src/mc/sosp/RegionSnapshot.cpp
+++ b/src/mc/sosp/RegionSnapshot.cpp
@@ -10,11 +10,11 @@
 #define MAP_POPULATE MAP_PREFAULT_READ
 #endif
 
-#include "mc/mc.h"
+#include "src/mc/ModelChecker.hpp"
 #include "src/mc/mc_config.hpp"
-#include "src/mc/sosp/mc_snapshot.hpp"
+#include "src/mc/mc_forward.hpp"
 
-#include "src/mc/sosp/ChunkedData.hpp"
+#include "src/mc/mc_smx.hpp"
 #include "src/mc/sosp/RegionSnapshot.hpp"
 
 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(mc_RegionSnaphot, mc, "Logging specific to region snapshots");
@@ -22,19 +22,19 @@ XBT_LOG_NEW_DEFAULT_SUBCATEGORY(mc_RegionSnaphot, mc, "Logging specific to regio
 namespace simgrid {
 namespace mc {
 
-RegionSnapshot* dense_region(RegionType region_type, void* start_addr, void* permanent_addr, size_t size)
+RegionDense::RegionDense(RegionType region_type, void* start_addr, void* permanent_addr, size_t size)
+    : RegionSnapshot(region_type, start_addr, permanent_addr, size)
 {
-  simgrid::mc::Buffer data = Buffer::malloc(size);
+  flat_data_ = Buffer::malloc(size);
 
-  mc_model_checker->process().read_bytes(data.get(), size, remote(permanent_addr), simgrid::mc::ProcessIndexDisabled);
+  mc_model_checker->process().read_bytes(flat_data_.get(), size, remote(permanent_addr));
 
-  simgrid::mc::RegionSnapshot* region = new RegionSnapshot(region_type, start_addr, permanent_addr, size);
-  region->flat_data(std::move(data));
+  storage_type_ = StorageType::Flat;
+  page_numbers_.clear();
 
   XBT_DEBUG("New region : type : %s, data : %p (real addr %p), size : %zu",
             (region_type == RegionType::Heap ? "Heap" : (region_type == RegionType::Data ? "Data" : "?")),
-            region->flat_data().get(), permanent_addr, size);
-  return region;
+            flat_data().get(), permanent_addr, size);
 }
 
 /** @brief Take a snapshot of a given region
@@ -48,12 +48,13 @@ RegionSnapshot* dense_region(RegionType region_type, void* start_addr, void* per
 RegionSnapshot* region(RegionType type, void* start_addr, void* permanent_addr, size_t size)
 {
   if (_sg_mc_sparse_checkpoint)
-    return sparse_region(type, start_addr, permanent_addr, size);
+    return new RegionSparse(type, start_addr, permanent_addr, size);
   else
-    return dense_region(type, start_addr, permanent_addr, size);
+    return new RegionDense(type, start_addr, permanent_addr, size);
 }
 
-RegionSnapshot* sparse_region(RegionType region_type, void* start_addr, void* permanent_addr, size_t size)
+RegionSparse::RegionSparse(RegionType region_type, void* start_addr, void* permanent_addr, size_t size)
+    : RegionSnapshot(region_type, start_addr, permanent_addr, size)
 {
   simgrid::mc::RemoteClient* process = &mc_model_checker->process();
   assert(process != nullptr);
@@ -61,14 +62,41 @@ RegionSnapshot* sparse_region(RegionType region_type, void* start_addr, void* pe
   xbt_assert((((uintptr_t)start_addr) & (xbt_pagesize - 1)) == 0, "Start address not at the beginning of a page");
   xbt_assert((((uintptr_t)permanent_addr) & (xbt_pagesize - 1)) == 0,
              "Permanent address not at the beginning of a page");
-  size_t page_count = simgrid::mc::mmu::chunk_count(size);
 
-  simgrid::mc::ChunkedData page_data(mc_model_checker->page_store(), *process, RemotePtr<void>(permanent_addr),
-                                     page_count);
+  storage_type_ = StorageType::Chunked;
+  flat_data_.clear();
 
-  simgrid::mc::RegionSnapshot* region = new RegionSnapshot(region_type, start_addr, permanent_addr, size);
-  region->page_data(std::move(page_data));
-  return region;
+  page_numbers_ =
+      ChunkedData(mc_model_checker->page_store(), *process, RemotePtr<void>(permanent_addr), mmu::chunk_count(size));
+}
+
+/** @brief Restore a region from a snapshot
+ *
+ *  @param region     Target region
+ */
+void RegionSnapshot::restore()
+{
+  switch (storage_type()) {
+    case simgrid::mc::StorageType::Flat:
+      mc_model_checker->process().write_bytes(flat_data().get(), size(), permanent_address());
+      break;
+
+    case simgrid::mc::StorageType::Chunked:
+      xbt_assert(((permanent_address().address()) & (xbt_pagesize - 1)) == 0, "Not at the beginning of a page");
+      xbt_assert(simgrid::mc::mmu::chunk_count(size()) == page_data().page_count());
+
+      for (size_t i = 0; i != page_data().page_count(); ++i) {
+        void* target_page = (void*)simgrid::mc::mmu::join(i, (std::uintptr_t)(void*)permanent_address().address());
+        const void* source_page = page_data().page(i);
+        mc_model_checker->process().write_bytes(source_page, xbt_pagesize, remote(target_page));
+      }
+
+      break;
+
+    default: // includes StorageType::NoData
+      xbt_die("Storage type not supported");
+      break;
+  }
 }
 
 } // namespace mc