/root/bitcoin/src/flatfile.cpp
Line | Count | Source |
1 | | // Copyright (c) 2009-2010 Satoshi Nakamoto |
2 | | // Copyright (c) 2009-present 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 <flatfile.h> |
7 | | |
8 | | #include <tinyformat.h> |
9 | | #include <util/fs_helpers.h> |
10 | | #include <util/log.h> |
11 | | #include <util/overflow.h> |
12 | | |
13 | | #include <stdexcept> |
14 | | |
15 | | FlatFileSeq::FlatFileSeq(fs::path dir, const char* prefix, size_t chunk_size) : |
16 | 0 | m_dir(std::move(dir)), |
17 | 0 | m_prefix(prefix), |
18 | 0 | m_chunk_size(chunk_size) |
19 | 0 | { |
20 | 0 | if (chunk_size == 0) { |
21 | 0 | throw std::invalid_argument("chunk_size must be positive"); |
22 | 0 | } |
23 | 0 | } |
24 | | |
25 | | std::string FlatFilePos::ToString() const |
26 | 0 | { |
27 | 0 | return strprintf("FlatFilePos(nFile=%i, nPos=%i)", nFile, nPos);Line | Count | Source | 1172 | 0 | #define strprintf tfm::format |
|
28 | 0 | } |
29 | | |
30 | | fs::path FlatFileSeq::FileName(const FlatFilePos& pos) const |
31 | 0 | { |
32 | 0 | return m_dir / fs::u8path(strprintf("%s%05u.dat", m_prefix, pos.nFile));Line | Count | Source | 1172 | 0 | #define strprintf tfm::format |
|
33 | 0 | } |
34 | | |
35 | | FILE* FlatFileSeq::Open(const FlatFilePos& pos, bool read_only) const |
36 | 0 | { |
37 | 0 | if (pos.IsNull()) { |
38 | 0 | return nullptr; |
39 | 0 | } |
40 | 0 | fs::path path = FileName(pos); |
41 | 0 | fs::create_directories(path.parent_path()); |
42 | 0 | FILE* file = fsbridge::fopen(path, read_only ? "rb": "rb+"); |
43 | 0 | if (!file && !read_only) |
44 | 0 | file = fsbridge::fopen(path, "wb+"); |
45 | 0 | if (!file) { |
46 | 0 | LogError("Unable to open file %s", fs::PathToString(path));Line | Count | Source | 97 | 0 | #define LogError(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Error, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 89 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
|
|
47 | 0 | return nullptr; |
48 | 0 | } |
49 | 0 | if (pos.nPos && fseek(file, pos.nPos, SEEK_SET)) { |
50 | 0 | LogError("Unable to seek to position %u of %s", pos.nPos, fs::PathToString(path));Line | Count | Source | 97 | 0 | #define LogError(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Error, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 89 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
|
|
51 | 0 | if (fclose(file) != 0) { |
52 | 0 | LogError("Unable to close file %s", fs::PathToString(path));Line | Count | Source | 97 | 0 | #define LogError(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Error, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 89 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
|
|
53 | 0 | } |
54 | 0 | return nullptr; |
55 | 0 | } |
56 | 0 | return file; |
57 | 0 | } |
58 | | |
59 | | size_t FlatFileSeq::Allocate(const FlatFilePos& pos, size_t add_size, bool& out_of_space) const |
60 | 0 | { |
61 | 0 | out_of_space = false; |
62 | |
|
63 | 0 | unsigned int n_old_chunks = CeilDiv(pos.nPos, m_chunk_size); |
64 | 0 | unsigned int n_new_chunks = CeilDiv(pos.nPos + add_size, m_chunk_size); |
65 | 0 | if (n_new_chunks > n_old_chunks) { |
66 | 0 | size_t old_size = pos.nPos; |
67 | 0 | size_t new_size = n_new_chunks * m_chunk_size; |
68 | 0 | size_t inc_size = new_size - old_size; |
69 | |
|
70 | 0 | if (CheckDiskSpace(m_dir, inc_size)) { |
71 | 0 | FILE *file = Open(pos); |
72 | 0 | if (file) { |
73 | 0 | LogDebug(BCLog::VALIDATION, "Pre-allocating up to position 0x%x in %s%05u.dat\n", new_size, m_prefix, pos.nFile); Line | Count | Source | 115 | 0 | #define LogDebug(category, ...) detail_LogIfCategoryAndLevelEnabled(category, BCLog::Level::Debug, __VA_ARGS__) Line | Count | Source | 106 | 0 | do { \ | 107 | 0 | if (util::log::ShouldLog((category), (level))) { \ | 108 | 0 | bool rate_limit{level >= BCLog::Level::Info}; \ | 109 | 0 | Assume(!rate_limit); /*Only called with the levels below*/ \ Line | Count | Source | 125 | 0 | #define Assume(val) inline_assertion_check<false>(val, std::source_location::current(), #val) |
| 110 | 0 | LogPrintLevel_(category, level, rate_limit, __VA_ARGS__); \ Line | Count | Source | 89 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
| 111 | 0 | } \ | 112 | 0 | } while (0) |
|
|
74 | 0 | AllocateFileRange(file, pos.nPos, inc_size); |
75 | 0 | if (fclose(file) != 0) { |
76 | 0 | LogError("Cannot close file %s%05u.dat after extending it with %u bytes", m_prefix, pos.nFile, new_size);Line | Count | Source | 97 | 0 | #define LogError(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Error, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 89 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
|
|
77 | 0 | return 0; |
78 | 0 | } |
79 | 0 | return inc_size; |
80 | 0 | } |
81 | 0 | } else { |
82 | 0 | out_of_space = true; |
83 | 0 | } |
84 | 0 | } |
85 | 0 | return 0; |
86 | 0 | } |
87 | | |
88 | | bool FlatFileSeq::Flush(const FlatFilePos& pos, bool finalize) const |
89 | 0 | { |
90 | 0 | FILE* file = Open(FlatFilePos(pos.nFile, 0)); // Avoid fseek to nPos |
91 | 0 | if (!file) { |
92 | 0 | LogError("%s: failed to open file %d\n", __func__, pos.nFile);Line | Count | Source | 97 | 0 | #define LogError(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Error, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 89 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
|
|
93 | 0 | return false; |
94 | 0 | } |
95 | 0 | if (finalize && !TruncateFile(file, pos.nPos)) { |
96 | 0 | LogError("%s: failed to truncate file %d\n", __func__, pos.nFile);Line | Count | Source | 97 | 0 | #define LogError(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Error, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 89 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
|
|
97 | 0 | if (fclose(file) != 0) { |
98 | 0 | LogError("Failed to close file %d", pos.nFile);Line | Count | Source | 97 | 0 | #define LogError(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Error, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 89 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
|
|
99 | 0 | } |
100 | 0 | return false; |
101 | 0 | } |
102 | 0 | if (!FileCommit(file)) { |
103 | 0 | LogError("%s: failed to commit file %d\n", __func__, pos.nFile);Line | Count | Source | 97 | 0 | #define LogError(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Error, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 89 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
|
|
104 | 0 | if (fclose(file) != 0) { |
105 | 0 | LogError("Failed to close file %d", pos.nFile);Line | Count | Source | 97 | 0 | #define LogError(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Error, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 89 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
|
|
106 | 0 | } |
107 | 0 | return false; |
108 | 0 | } |
109 | 0 | DirectoryCommit(m_dir); |
110 | |
|
111 | 0 | if (fclose(file) != 0) { |
112 | 0 | LogError("Failed to close file %d after flush", pos.nFile);Line | Count | Source | 97 | 0 | #define LogError(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Error, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 89 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
|
|
113 | 0 | return false; |
114 | 0 | } |
115 | 0 | return true; |
116 | 0 | } |