72 lines
1.5 KiB
C++
72 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include "./file.hpp"
|
|
|
|
#include <fstream>
|
|
#include <limits>
|
|
|
|
struct FileRWFile : public FileI {
|
|
std::fstream _file;
|
|
|
|
// dont truncate by default
|
|
FileRWFile(std::string_view file_path, uint64_t file_size, bool trunc = false)
|
|
: _file(
|
|
static_cast<std::string>(file_path),
|
|
std::ios::in |
|
|
std::ios::out |
|
|
(trunc ? std::ios::trunc | std::ios::binary : std::ios::binary) // hacky but type safe
|
|
) {
|
|
_file_size = file_size;
|
|
|
|
if (!_file.is_open()) {
|
|
return; // TODO: error
|
|
}
|
|
}
|
|
|
|
virtual ~FileRWFile(void) {}
|
|
|
|
bool isGood(void) override {
|
|
return _file.is_open() && _file.good();
|
|
}
|
|
|
|
std::vector<uint8_t> read(uint64_t pos, uint64_t size) override {
|
|
if (pos >= _file_size) {
|
|
return {};
|
|
}
|
|
|
|
// TODO: error check
|
|
_file.seekg(pos, std::ios::beg);
|
|
|
|
std::vector<uint8_t> chunk(size);
|
|
const auto nread = _file.read(reinterpret_cast<char*>(chunk.data()), chunk.size()).gcount();
|
|
if (nread != std::numeric_limits<std::streamsize>::max()) {
|
|
chunk.resize(nread); // usually a noop
|
|
} else {
|
|
chunk.clear();
|
|
}
|
|
|
|
_bytes_read += chunk.size();
|
|
|
|
return chunk;
|
|
}
|
|
|
|
bool write(uint64_t pos, const std::vector<uint8_t>& data) override {
|
|
if (pos >= _file_size) {
|
|
return false;
|
|
}
|
|
|
|
// if out-of-order, seek
|
|
if (_file.tellp() != int64_t(pos)) {
|
|
// TODO: error check
|
|
if (_file.seekp(pos, std::ios::beg).fail()) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
_file.write(reinterpret_cast<const char*>(data.data()), data.size());
|
|
|
|
return _file.good();
|
|
}
|
|
};
|
|
|