38 lines
1.0 KiB
C++
38 lines
1.0 KiB
C++
|
#include "mmap_file.h"
|
||
|
|
#include "vtr_error.h"
|
||
|
|
#include "vtr_util.h"
|
||
|
|
|
||
|
|
#include <fcntl.h>
|
||
|
|
#include <sys/stat.h>
|
||
|
|
#include <sys/mman.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
|
||
|
|
#include "kj/filesystem.h"
|
||
|
|
|
||
|
|
MmapFile::MmapFile(const std::string& file)
|
||
|
|
: size_(0) {
|
||
|
|
try {
|
||
|
|
auto fs = kj::newDiskFilesystem();
|
||
|
|
auto path = fs->getCurrentPath().evalNative(file);
|
||
|
|
|
||
|
|
const auto& dir = fs->getRoot();
|
||
|
|
auto stat = dir.lstat(path);
|
||
|
|
auto f = dir.openFile(path);
|
||
|
|
size_ = stat.size;
|
||
|
|
data_ = f->mmap(0, stat.size);
|
||
|
|
} catch (kj::Exception& e) {
|
||
|
|
throw vtr::VtrError(e.getDescription().cStr(), e.getFile(), e.getLine());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const kj::ArrayPtr<const ::capnp::word> MmapFile::getData() const {
|
||
|
|
if ((size_ % sizeof(::capnp::word)) != 0) {
|
||
|
|
throw vtr::VtrError(
|
||
|
|
vtr::string_fmt("size_ %d is not a multiple of capnp::word", size_),
|
||
|
|
__FILE__, __LINE__);
|
||
|
|
}
|
||
|
|
|
||
|
|
return kj::arrayPtr(reinterpret_cast<const ::capnp::word*>(data_.begin()),
|
||
|
|
size_ / sizeof(::capnp::word));
|
||
|
|
}
|