Branch data Line data Source code
# 1 : : // Copyright (c) 2009-2010 Satoshi Nakamoto
# 2 : : // Copyright (c) 2009-2021 The Bitcoin Core developers
# 3 : : // Distributed under the MIT software license, see the accompanying
# 4 : : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
# 5 : :
# 6 : : #include <stdexcept>
# 7 : :
# 8 : : #include <flatfile.h>
# 9 : : #include <logging.h>
# 10 : : #include <tinyformat.h>
# 11 : : #include <util/system.h>
# 12 : :
# 13 : : FlatFileSeq::FlatFileSeq(fs::path dir, const char* prefix, size_t chunk_size) :
# 14 : : m_dir(std::move(dir)),
# 15 : : m_prefix(prefix),
# 16 : : m_chunk_size(chunk_size)
# 17 : 405890 : {
# 18 [ - + ]: 405890 : if (chunk_size == 0) {
# 19 : 0 : throw std::invalid_argument("chunk_size must be positive");
# 20 : 0 : }
# 21 : 405890 : }
# 22 : :
# 23 : : std::string FlatFilePos::ToString() const
# 24 : 109 : {
# 25 : 109 : return strprintf("FlatFilePos(nFile=%i, nPos=%i)", nFile, nPos);
# 26 : 109 : }
# 27 : :
# 28 : : fs::path FlatFileSeq::FileName(const FlatFilePos& pos) const
# 29 : 290404 : {
# 30 : 290404 : return m_dir / strprintf("%s%05u.dat", m_prefix, pos.nFile);
# 31 : 290404 : }
# 32 : :
# 33 : : FILE* FlatFileSeq::Open(const FlatFilePos& pos, bool read_only)
# 34 : 290460 : {
# 35 [ + + ]: 290460 : if (pos.IsNull()) {
# 36 : 109 : return nullptr;
# 37 : 109 : }
# 38 : 290351 : fs::path path = FileName(pos);
# 39 : 290351 : fs::create_directories(path.parent_path());
# 40 [ + + ]: 290351 : FILE* file = fsbridge::fopen(path, read_only ? "rb": "rb+");
# 41 [ + + ][ + + ]: 290351 : if (!file && !read_only)
# 42 : 835 : file = fsbridge::fopen(path, "wb+");
# 43 [ + + ]: 290351 : if (!file) {
# 44 : 6 : LogPrintf("Unable to open file %s\n", fs::PathToString(path));
# 45 : 6 : return nullptr;
# 46 : 6 : }
# 47 [ + + ][ - + ]: 290345 : if (pos.nPos && fseek(file, pos.nPos, SEEK_SET)) {
# 48 : 0 : LogPrintf("Unable to seek to position %u of %s\n", pos.nPos, fs::PathToString(path));
# 49 : 0 : fclose(file);
# 50 : 0 : return nullptr;
# 51 : 0 : }
# 52 : 290345 : return file;
# 53 : 290345 : }
# 54 : :
# 55 : : size_t FlatFileSeq::Allocate(const FlatFilePos& pos, size_t add_size, bool& out_of_space)
# 56 : 129686 : {
# 57 : 129686 : out_of_space = false;
# 58 : :
# 59 : 129686 : unsigned int n_old_chunks = (pos.nPos + m_chunk_size - 1) / m_chunk_size;
# 60 : 129686 : unsigned int n_new_chunks = (pos.nPos + add_size + m_chunk_size - 1) / m_chunk_size;
# 61 [ + + ]: 129686 : if (n_new_chunks > n_old_chunks) {
# 62 : 881 : size_t old_size = pos.nPos;
# 63 : 881 : size_t new_size = n_new_chunks * m_chunk_size;
# 64 : 881 : size_t inc_size = new_size - old_size;
# 65 : :
# 66 [ + - ]: 881 : if (CheckDiskSpace(m_dir, inc_size)) {
# 67 : 881 : FILE *file = Open(pos);
# 68 [ + - ]: 881 : if (file) {
# 69 [ + - ]: 881 : LogPrint(BCLog::VALIDATION, "Pre-allocating up to position 0x%x in %s%05u.dat\n", new_size, m_prefix, pos.nFile);
# 70 : 881 : AllocateFileRange(file, pos.nPos, inc_size);
# 71 : 881 : fclose(file);
# 72 : 881 : return inc_size;
# 73 : 881 : }
# 74 : 881 : } else {
# 75 : 0 : out_of_space = true;
# 76 : 0 : }
# 77 : 881 : }
# 78 : 128805 : return 0;
# 79 : 129686 : }
# 80 : :
# 81 : : bool FlatFileSeq::Flush(const FlatFilePos& pos, bool finalize)
# 82 : 3288 : {
# 83 : 3288 : FILE* file = Open(FlatFilePos(pos.nFile, 0)); // Avoid fseek to nPos
# 84 [ - + ]: 3288 : if (!file) {
# 85 : 0 : return error("%s: failed to open file %d", __func__, pos.nFile);
# 86 : 0 : }
# 87 [ + + ][ - + ]: 3288 : if (finalize && !TruncateFile(file, pos.nPos)) {
# 88 : 0 : fclose(file);
# 89 : 0 : return error("%s: failed to truncate file %d", __func__, pos.nFile);
# 90 : 0 : }
# 91 [ - + ]: 3288 : if (!FileCommit(file)) {
# 92 : 0 : fclose(file);
# 93 : 0 : return error("%s: failed to commit file %d", __func__, pos.nFile);
# 94 : 0 : }
# 95 : 3288 : DirectoryCommit(m_dir);
# 96 : :
# 97 : 3288 : fclose(file);
# 98 : 3288 : return true;
# 99 : 3288 : }
|