#include <cstddef>
#include <utility>
+#include <memory>
+#include <vector>
+
#include <xbt/base.h>
-#include <mc/PageStore.hpp>
-#include <mc/AddressSpace.hpp>
+#include "src/mc/AddressSpace.hpp"
+#include "src/mc/ChunkedData.hpp"
+#include "src/mc/PageStore.hpp"
+#include "src/mc/remote/RemotePtr.hpp"
namespace simgrid {
namespace mc {
-class PerPageCopy {
- PageStore* store_;
- std::vector<std::size_t> pagenos_;
-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();
- }
-
- PerPageCopy(PerPageCopy&& that)
- {
- store_ = that.store_;
- that.store_ = nullptr;
- pagenos_ = std::move(that.pagenos_);
- that.pagenos_.clear();
- }
- PerPageCopy& operator=(PerPageCopy const& that)
- {
- this->clear();
- store_ = that.store_;
- pagenos_ = that.pagenos_;
- for (std::size_t pageno : pagenos_)
- store_->ref_page(pageno);
- return *this;
- }
- PerPageCopy& operator=(PerPageCopy && that)
- {
- this->clear();
- store_ = that.store_;
- that.store_ = nullptr;
- pagenos_ = std::move(that.pagenos_);
- that.pagenos_.clear();
- return *this;
- }
-
- std::size_t page_count() const
- {
- return pagenos_.size();
- }
-
- std::size_t pageno(std::size_t i) const
- {
- return pagenos_[i];
- }
-
- const std::size_t* pagenos() const { return pagenos_.data(); }
- std::size_t* pagenos() { return pagenos_.data(); }
-
- const void* page(std::size_t i) const
- {
- return store_->get_page(pagenos_[i]);
- }
-
- PerPageCopy(PageStore& store, AddressSpace& as,
- remote_ptr<void> addr, std::size_t page_count,
- const size_t* ref_page_numbers, const std::uint64_t* pagemap);
-};
-
enum class RegionType {
Unknown = 0,
Heap = 1,
Data = 2
};
-// TODO, use Boost.Variant instead of this
enum class StorageType {
NoData = 0,
Flat = 1,
Privatized = 3
};
-class data_deleter {
-public:
- enum Type {
- Free,
- Munmap
- };
+class Buffer {
private:
- Type type_;
+ 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:
- data_deleter() : type_(Free) {}
- data_deleter(Type type, std::size_t size) : type_(type), size_(size) {}
- void operator()(void* p) const;
-};
+ Buffer() {}
+ void clear() noexcept;
+ ~Buffer() noexcept { clear(); }
+
+ static Buffer malloc(std::size_t size)
+ {
+ return Buffer(size, Type::Malloc);
+ }
+ static Buffer mmap(std::size_t size)
+ {
+ return Buffer(size, Type::Mmap);
+ }
-typedef std::unique_ptr<char[], data_deleter> unique_data_ptr;
+ // No copy
+ Buffer(Buffer const& buffer) = delete;
+ Buffer& operator=(Buffer const& buffer) = delete;
+
+ // Move
+ Buffer(Buffer&& that) noexcept
+ : data_(that.data_), size_(that.size_), type_(that.type_)
+ {
+ that.data_ = nullptr;
+ that.size_ = 0;
+ that.type_ = Type::Malloc;
+ }
+ Buffer& operator=(Buffer&& that) noexcept
+ {
+ clear();
+ data_ = that.data_;
+ size_ = that.size_;
+ type_ = that.type_;
+ that.data_ = nullptr;
+ that.size_ = 0;
+ that.type_ = Type::Malloc;
+ return *this;
+ }
+
+ void* get() { return data_; }
+ const void* get() const { return data_; }
+ std::size_t size() const { return size_; }
+};
/** A copy/snapshot of a given memory region
*
*
* * privatized (SMPI global variable privatisation).
*
- * This is handled with a variant based approch:
+ * This is handled with a variant based approach:
*
* * `storage_type` identified the type of storage;
*
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;
-public:
- typedef unique_data_ptr flat_data_ptr;
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_;
- flat_data_ptr 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_.reset();
+ flat_data_.clear();
object_info_ = nullptr;
start_addr_ = nullptr;
size_ = 0;
void clear_data()
{
- storage_type_ = NoData;
- flat_data_.reset();
+ storage_type_ = StorageType::NoData;
+ flat_data_.clear();
page_numbers_.clear();
privatized_regions_.clear();
}
- void flat_data(flat_data_ptr data)
+ void flat_data(Buffer data)
{
- storage_type_ = FlatData;
+ storage_type_ = StorageType::Flat;
flat_data_ = std::move(data);
page_numbers_.clear();
privatized_regions_.clear();
}
- const char* flat_data() const { return flat_data_.get(); }
+ 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;
- flat_data_.reset();
+ 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;
- flat_data_.reset();
+ 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,
- RegionSnapshot const* ref_region);
+ 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,
- RegionSnapshot const* ref_region);
+ RegionType type, void *start_addr, void* data_addr, std::size_t size);
}
}