#include <cstddef>
#include <utility>
+#include <memory>
+#include <vector>
+
#include <xbt/base.h>
-#include "PageStore.hpp"
-#include "AddressSpace.hpp"
+#include "src/mc/RemotePtr.hpp"
+#include "src/mc/PageStore.hpp"
+#include "src/mc/AddressSpace.hpp"
+#include "src/mc/ChunkedData.hpp"
namespace simgrid {
namespace mc {
-class PerPageCopy {
- PageStore* store_;
- std::vector<std::size_t> pagenos_;
+enum class RegionType {
+ Unknown = 0,
+ Heap = 1,
+ Data = 2
+};
+
+enum class StorageType {
+ NoData = 0,
+ Flat = 1,
+ Chunked = 2,
+ Privatized = 3
+};
+
+class Buffer {
+private:
+ enum class Type {
+ Malloc,
+ Mmap
+ };
+ void* data_ = nullptr;
+ std::size_t size_;
+ Type type_ = Type::Malloc;
+private:
+ Buffer(std::size_t size, Type type = Type::Malloc);
+ Buffer(void* data, std::size_t size, Type type = Type::Malloc) :
+ data_(data), size_(size), type_(type) {}
public:
- PerPageCopy() : store_(nullptr) {}
- PerPageCopy(PerPageCopy const& that)
- {
- store_ = that.store_;
- pagenos_ = that.pagenos_;
- for (std::size_t pageno : pagenos_)
- store_->ref_page(pageno);
- }
- void clear()
- {
- for (std::size_t pageno : pagenos_)
- store_->unref_page(pageno);
- pagenos_.clear();
- }
- ~PerPageCopy() {
- clear();
- }
+ Buffer() {}
+ void clear() noexcept;
+ ~Buffer() noexcept { clear(); }
- PerPageCopy(PerPageCopy&& that)
- {
- store_ = that.store_;
- that.store_ = nullptr;
- pagenos_ = std::move(that.pagenos_);
- that.pagenos_.clear();
- }
- PerPageCopy& operator=(PerPageCopy const& that)
+ static Buffer malloc(std::size_t size)
{
- this->clear();
- store_ = that.store_;
- pagenos_ = that.pagenos_;
- for (std::size_t pageno : pagenos_)
- store_->ref_page(pageno);
- return *this;
+ return Buffer(size, Type::Malloc);
}
- PerPageCopy& operator=(PerPageCopy && that)
+ static Buffer mmap(std::size_t size)
{
- this->clear();
- store_ = that.store_;
- that.store_ = nullptr;
- pagenos_ = std::move(that.pagenos_);
- that.pagenos_.clear();
- return *this;
+ return Buffer(size, Type::Mmap);
}
- std::size_t page_count() const
- {
- return pagenos_.size();
- }
+ // No copy
+ Buffer(Buffer const& buffer) = delete;
+ Buffer& operator=(Buffer const& buffer) = delete;
- std::size_t pageno(std::size_t i) const
+ // Move
+ Buffer(Buffer&& that) noexcept
+ : data_(that.data_), size_(that.size_), type_(that.type_)
{
- return pagenos_[i];
+ that.data_ = nullptr;
+ that.size_ = 0;
+ that.type_ = Type::Malloc;
}
-
- const void* page(std::size_t i) const
+ Buffer& operator=(Buffer&& that) noexcept
{
- return store_->get_page(pagenos_[i]);
+ clear();
+ data_ = that.data_;
+ size_ = that.size_;
+ type_ = that.type_;
+ that.data_ = nullptr;
+ that.size_ = 0;
+ that.type_ = Type::Malloc;
+ return *this;
}
- PerPageCopy(PageStore& store, AddressSpace& as,
- remote_ptr<void> addr, std::size_t page_count);
-};
-
-enum class RegionType {
- Unknown = 0,
- Heap = 1,
- Data = 2
-};
-
-// TODO, use Boost.Variant instead of this
-enum class StorageType {
- NoData = 0,
- Flat = 1,
- Chunked = 2,
- Privatized = 3
+ void* get() { return data_; }
+ const void* get() const { return data_; }
+ std::size_t size() const { return size_; }
};
-/** @brief Copy/snapshot of a given memory region
+/** A copy/snapshot of a given memory region
*
* Different types of region snapshot storage types exist:
- * <ul>
- * <li>flat/dense snapshots are a simple copy of the region;</li>
- * <li>sparse/per-page snapshots are snaapshots which shared
- * identical pages.</li>
- * <li>privatized (SMPI global variable privatisation).
- * </ul>
*
- * This is handled with a variant based approch:
+ * * flat/dense snapshots are a simple copy of the region;
+ *
+ * * sparse/per-page snapshots are snaapshots which shared
+ * identical pages.
*
- * * `storage_type` identified the type of storage;
- * * an anonymous enum is used to distinguish the relevant types for
- * each type.
+ * * privatized (SMPI global variable privatisation).
+ *
+ * This is handled with a variant based approach:
+ *
+ * * `storage_type` identified the type of storage;
+ *
+ * * an anonymous enum is used to distinguish the relevant types for
+ * each type.
*/
class RegionSnapshot {
+public:
static const RegionType UnknownRegion = RegionType::Unknown;
static const RegionType HeapRegion = RegionType::Heap;
static const RegionType DataRegion = RegionType::Data;
- static const StorageType NoData = StorageType::NoData;
- static const StorageType FlatData = StorageType::Flat;
- static const StorageType ChunkedData = StorageType::Chunked;
- static const StorageType PrivatizedData = StorageType::Privatized;
private:
RegionType region_type_;
StorageType storage_type_;
void *start_addr_;
/** @brief Size of the data region in bytes */
- size_t size_;
+ std::size_t size_;
/** @brief Permanent virtual address of the region
*
* */
void *permanent_addr_;
- std::vector<char> flat_data_;
- PerPageCopy page_numbers_;
+ Buffer flat_data_;
+ ChunkedData page_numbers_;
std::vector<RegionSnapshot> privatized_regions_;
public:
RegionSnapshot() :
region_type_(UnknownRegion),
- storage_type_(NoData),
+ storage_type_(StorageType::NoData),
object_info_(nullptr),
start_addr_(nullptr),
size_(0),
{}
RegionSnapshot(RegionType type, void *start_addr, void* permanent_addr, size_t size) :
region_type_(type),
- storage_type_(NoData),
+ storage_type_(StorageType::NoData),
object_info_(nullptr),
start_addr_(start_addr),
size_(size),
void clear()
{
region_type_ = UnknownRegion;
- storage_type_ = NoData;
+ storage_type_ = StorageType::NoData;
privatized_regions_.clear();
page_numbers_.clear();
flat_data_.clear();
void clear_data()
{
- storage_type_ = NoData;
+ storage_type_ = StorageType::NoData;
flat_data_.clear();
page_numbers_.clear();
privatized_regions_.clear();
}
- void flat_data(std::vector<char> data)
+ void flat_data(Buffer data)
{
- storage_type_ = FlatData;
+ storage_type_ = StorageType::Flat;
flat_data_ = std::move(data);
page_numbers_.clear();
privatized_regions_.clear();
}
- std::vector<char> const& flat_data() const { return flat_data_; }
+ const Buffer& flat_data() const { return flat_data_; }
+ Buffer& flat_data() { return flat_data_; }
- void page_data(PerPageCopy page_data)
+ void page_data(ChunkedData page_data)
{
- storage_type_ = ChunkedData;
+ storage_type_ = StorageType::Chunked;
flat_data_.clear();
page_numbers_ = std::move(page_data);
privatized_regions_.clear();
}
- PerPageCopy const& page_data() const { return page_numbers_; }
+ ChunkedData const& page_data() const { return page_numbers_; }
void privatized_data(std::vector<RegionSnapshot> data)
{
- storage_type_ = PrivatizedData;
+ storage_type_ = StorageType::Privatized;
flat_data_.clear();
page_numbers_.clear();
privatized_regions_ = std::move(data);
// Other getters
- remote_ptr<void> start() const { return remote(start_addr_); }
- remote_ptr<void> end() const { return remote((char*)start_addr_ + size_); }
- remote_ptr<void> permanent_address() const { return remote(permanent_addr_); }
+ RemotePtr<void> start() const { return remote(start_addr_); }
+ RemotePtr<void> end() const { return remote((char*)start_addr_ + size_); }
+ RemotePtr<void> permanent_address() const { return remote(permanent_addr_); }
std::size_t size() const { return size_; }
StorageType storage_type() const { return storage_type_; }
RegionType region_type() const { return region_type_; }
- bool contain(remote_ptr<void> p) const
+ bool contain(RemotePtr<void> p) const
{
return p >= start() && p < end();
}
};
RegionSnapshot privatized_region(
- RegionType type, void *start_addr, void* data_addr, size_t size);
+ RegionType region_type, void *start_addr, void* permanent_addr,
+ std::size_t size);
RegionSnapshot dense_region(
- RegionType type, void *start_addr, void* data_addr, size_t size);
+ RegionType type, void *start_addr, void* data_addr, std::size_t size);
simgrid::mc::RegionSnapshot sparse_region(
- RegionType type, void *start_addr, void* data_addr, size_t size);
+ RegionType type, void *start_addr, void* data_addr, std::size_t size);
simgrid::mc::RegionSnapshot region(
- RegionType type, void *start_addr, void* data_addr, size_t size);
-
+ RegionType type, void *start_addr, void* data_addr, std::size_t size);
+
}
}