wehub-resource-sync
79 lines
1.8 KiB
C++
Executable File
79 lines
1.8 KiB
C++
Executable File
#include "file_mem_mapper.h"
|
|
|
|
#include "logging.h"
|
|
|
|
#include <fcntl.h>
|
|
#include <sys/mman.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.h>
|
|
|
|
FileMemMapper::FileMemMapper(const std::string& path) {
|
|
// Get fd
|
|
mFd = open(path.c_str(), O_RDONLY);
|
|
if (mFd == -1) {
|
|
LOG(ERROR) << "Open file fail: " << path;
|
|
return;
|
|
}
|
|
|
|
// Get size
|
|
struct stat sb;
|
|
if (fstat(mFd, &sb) == -1) {
|
|
LOG(ERROR) << "fstat fail";
|
|
return;
|
|
}
|
|
mSize = sb.st_size;
|
|
|
|
// Map file data to memory
|
|
mBuffer = mmap(NULL, mSize, PROT_READ, MAP_SHARED, mFd, 0);
|
|
if (mBuffer == MAP_FAILED) {
|
|
LOG(ERROR) << "mmap fail";
|
|
return;
|
|
}
|
|
|
|
LOG(DEBUG) << "FileMemMapper: Mapped to "
|
|
<< "(fd=" << mFd << ", size=" << mSize << ", addr=" << mBuffer << "): " << path;
|
|
}
|
|
|
|
// Move ctor
|
|
FileMemMapper::FileMemMapper(FileMemMapper&& other)
|
|
: mFd(other.mFd), mBuffer(other.mBuffer), mSize(other.mSize) {
|
|
other.mFd = -1;
|
|
other.mBuffer = nullptr;
|
|
other.mSize = 0;
|
|
}
|
|
|
|
FileMemMapper::~FileMemMapper() {
|
|
if (!mBuffer && mFd == -1) {
|
|
return;
|
|
}
|
|
LOG(DEBUG) << "FileMemMapper: Unmapping "
|
|
<< "(fd=" << mFd << ", size=" << mSize << ", addr=" << mBuffer << ")";
|
|
if (mBuffer && munmap(mBuffer, mSize) == -1) {
|
|
LOG(ERROR) << "munmap fail";
|
|
return;
|
|
}
|
|
if (mFd != -1 && close(mFd) == -1) {
|
|
LOG(ERROR) << "close fail";
|
|
return;
|
|
}
|
|
}
|
|
|
|
FileMemMapper::operator bool() const {
|
|
return valid();
|
|
}
|
|
|
|
bool FileMemMapper::valid() const {
|
|
return (mBuffer != MAP_FAILED && mSize > 0);
|
|
}
|
|
|
|
std::pair<char*, size_t> FileMemMapper::get() const {
|
|
return {reinterpret_cast<char*>(mBuffer), mSize};
|
|
}
|
|
|
|
void* FileMemMapper::getAddr() const {
|
|
return mBuffer;
|
|
}
|
|
|
|
size_t FileMemMapper::getSize() const {
|
|
return mSize;
|
|
} |