Coverage Report

Created: 2026-07-23 20:35

next uncovered line (L), next uncovered region (R), next uncovered branch (B)
/tmp/bitcoin/src/node/blockstorage.cpp
Line
Count
Source
1
// Copyright (c) 2011-present The Bitcoin Core developers
2
// Distributed under the MIT software license, see the accompanying
3
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
4
5
#include <node/blockstorage.h>
6
7
#include <arith_uint256.h>
8
#include <chain.h>
9
#include <consensus/params.h>
10
#include <crypto/hex_base.h>
11
#include <dbwrapper.h>
12
#include <flatfile.h>
13
#include <hash.h>
14
#include <kernel/blockmanager_opts.h>
15
#include <kernel/chainparams.h>
16
#include <kernel/messagestartchars.h>
17
#include <kernel/notifications_interface.h>
18
#include <kernel/types.h>
19
#include <pow.h>
20
#include <primitives/block.h>
21
#include <primitives/transaction.h>
22
#include <random.h>
23
#include <serialize.h>
24
#include <signet.h>
25
#include <streams.h>
26
#include <sync.h>
27
#include <tinyformat.h>
28
#include <uint256.h>
29
#include <undo.h>
30
#include <util/check.h>
31
#include <util/expected.h>
32
#include <util/fs.h>
33
#include <util/log.h>
34
#include <util/obfuscation.h>
35
#include <util/overflow.h>
36
#include <util/result.h>
37
#include <util/signalinterrupt.h>
38
#include <util/strencodings.h>
39
#include <util/syserror.h>
40
#include <util/time.h>
41
#include <util/translation.h>
42
#include <validation.h>
43
44
#include <cerrno>
45
#include <compare>
46
#include <cstddef>
47
#include <cstdio>
48
#include <exception>
49
#include <map>
50
#include <optional>
51
#include <ostream>
52
#include <span>
53
#include <stdexcept>
54
#include <system_error>
55
#include <unordered_map>
56
57
namespace kernel {
58
static constexpr uint8_t DB_BLOCK_FILES{'f'};
59
static constexpr uint8_t DB_BLOCK_INDEX{'b'};
60
static constexpr uint8_t DB_FLAG{'F'};
61
static constexpr uint8_t DB_REINDEX_FLAG{'R'};
62
static constexpr uint8_t DB_LAST_BLOCK{'l'};
63
// Keys used in previous version that might still be found in the DB:
64
// BlockTreeDB::DB_TXINDEX_BLOCK{'T'};
65
// BlockTreeDB::DB_TXINDEX{'t'}
66
// BlockTreeDB::ReadFlag("txindex")
67
68
bool BlockTreeDB::ReadBlockFileInfo(int nFile, CBlockFileInfo& info)
69
2.42k
{
70
2.42k
    return Read(std::make_pair(DB_BLOCK_FILES, nFile), info);
71
2.42k
}
72
73
void BlockTreeDB::WriteReindexing(bool fReindexing)
74
29
{
75
29
    if (fReindexing) {
76
15
        Write(DB_REINDEX_FLAG, uint8_t{'1'});
77
15
    } else {
78
14
        Erase(DB_REINDEX_FLAG);
79
14
    }
80
29
}
81
82
void BlockTreeDB::ReadReindexing(bool& fReindexing)
83
1.20k
{
84
1.20k
    fReindexing = Exists(DB_REINDEX_FLAG);
85
1.20k
}
86
87
bool BlockTreeDB::ReadLastBlockFile(int& nFile)
88
1.20k
{
89
1.20k
    return Read(DB_LAST_BLOCK, nFile);
90
1.20k
}
91
92
void BlockTreeDB::WriteBatchSync(const std::vector<std::pair<int, const CBlockFileInfo*>>& fileInfo, int nLastFile, const std::vector<const CBlockIndex*>& blockinfo)
93
3.44k
{
94
3.44k
    CDBBatch batch(*this);
95
3.44k
    for (const auto& [file, info] : fileInfo) {
96
1.65k
        batch.Write(std::make_pair(DB_BLOCK_FILES, file), *info);
97
1.65k
    }
98
3.44k
    batch.Write(DB_LAST_BLOCK, nLastFile);
99
124k
    for (const CBlockIndex* bi : blockinfo) {
100
124k
        batch.Write(std::make_pair(DB_BLOCK_INDEX, bi->GetBlockHash()), CDiskBlockIndex{bi});
101
124k
    }
102
3.44k
    WriteBatch(batch, true);
103
3.44k
}
104
105
void BlockTreeDB::WriteFlag(const std::string& name, bool fValue)
106
8
{
107
8
    Write(std::make_pair(DB_FLAG, name), fValue ? uint8_t{'1'} : uint8_t{'0'});
108
8
}
109
110
bool BlockTreeDB::ReadFlag(const std::string& name, bool& fValue)
111
1.20k
{
112
1.20k
    uint8_t ch;
113
1.20k
    if (!Read(std::make_pair(DB_FLAG, name), ch)) {
114
1.19k
        return false;
115
1.19k
    }
116
2
    fValue = ch == uint8_t{'1'};
117
2
    return true;
118
1.20k
}
119
120
bool BlockTreeDB::LoadBlockIndexGuts(const Consensus::Params& consensusParams, std::function<CBlockIndex*(const uint256&)> insertBlockIndex, const util::SignalInterrupt& interrupt)
121
1.20k
{
122
1.20k
    AssertLockHeld(::cs_main);
123
1.20k
    std::unique_ptr<CDBIterator> pcursor(NewIterator());
124
1.20k
    pcursor->Seek(std::make_pair(DB_BLOCK_INDEX, uint256()));
125
126
    // Load m_block_index
127
137k
    while (pcursor->Valid()) {
128
136k
        if (interrupt) return false;
129
136k
        std::pair<uint8_t, uint256> key;
130
136k
        if (pcursor->GetKey(key) && key.first == DB_BLOCK_INDEX) {
131
135k
            CDiskBlockIndex diskindex;
132
135k
            if (pcursor->GetValue(diskindex)) {
133
                // Construct block index object
134
135k
                CBlockIndex* pindexNew = insertBlockIndex(diskindex.ConstructBlockHash());
135
135k
                pindexNew->pprev          = insertBlockIndex(diskindex.hashPrev);
136
135k
                pindexNew->nHeight        = diskindex.nHeight;
137
135k
                pindexNew->nFile          = diskindex.nFile;
138
135k
                pindexNew->nDataPos       = diskindex.nDataPos;
139
135k
                pindexNew->nUndoPos       = diskindex.nUndoPos;
140
135k
                pindexNew->nVersion       = diskindex.nVersion;
141
135k
                pindexNew->hashMerkleRoot = diskindex.hashMerkleRoot;
142
135k
                pindexNew->nTime          = diskindex.nTime;
143
135k
                pindexNew->nBits          = diskindex.nBits;
144
135k
                pindexNew->nNonce         = diskindex.nNonce;
145
135k
                pindexNew->nStatus        = diskindex.nStatus;
146
135k
                pindexNew->nTx            = diskindex.nTx;
147
148
135k
                if (!CheckProofOfWork(pindexNew->GetBlockHash(), pindexNew->nBits, consensusParams)) {
149
0
                    LogError("%s: CheckProofOfWork failed: %s\n", __func__, pindexNew->ToString());
150
0
                    return false;
151
0
                }
152
153
135k
                pcursor->Next();
154
135k
            } else {
155
0
                LogError("%s: failed to read value\n", __func__);
156
0
                return false;
157
0
            }
158
135k
        } else {
159
749
            break;
160
749
        }
161
136k
    }
162
163
1.20k
    return true;
164
1.20k
}
165
166
std::string CBlockFileInfo::ToString() const
167
1.23k
{
168
1.23k
    return strprintf("CBlockFileInfo(blocks=%u, size=%u, heights=%u...%u, time=%s...%s)", nBlocks, nSize, nHeightFirst, nHeightLast, FormatISO8601Date(nTimeFirst), FormatISO8601Date(nTimeLast));
169
1.23k
}
170
} // namespace kernel
171
172
namespace node {
173
174
bool CBlockIndexWorkComparator::operator()(const CBlockIndex* pa, const CBlockIndex* pb) const
175
461M
{
176
    // First sort by most total work, ...
177
461M
    if (pa->nChainWork > pb->nChainWork) return false;
178
294M
    if (pa->nChainWork < pb->nChainWork) return true;
179
180
    // ... then by earliest activatable time, ...
181
2.32M
    if (pa->nSequenceId < pb->nSequenceId) return false;
182
2.28M
    if (pa->nSequenceId > pb->nSequenceId) return true;
183
184
    // Use pointer address as tie breaker (should only happen with blocks
185
    // loaded from disk, as those share the same id: 0 for blocks on the
186
    // best chain, 1 for all others).
187
2.24M
    if (pa < pb) return false;
188
2.24M
    if (pa > pb) return true;
189
190
    // Identical blocks.
191
2.24M
    return false;
192
2.24M
}
193
194
bool CBlockIndexHeightOnlyComparator::operator()(const CBlockIndex* pa, const CBlockIndex* pb) const
195
2.51M
{
196
2.51M
    return pa->nHeight < pb->nHeight;
197
2.51M
}
198
199
std::vector<CBlockIndex*> BlockManager::GetAllBlockIndices()
200
3.63k
{
201
3.63k
    AssertLockHeld(cs_main);
202
3.63k
    std::vector<CBlockIndex*> rv;
203
3.63k
    rv.reserve(m_block_index.size());
204
412k
    for (auto& [_, block_index] : m_block_index) {
205
412k
        rv.push_back(&block_index);
206
412k
    }
207
3.63k
    return rv;
208
3.63k
}
209
210
CBlockIndex* BlockManager::LookupBlockIndex(const uint256& hash)
211
662k
{
212
662k
    AssertLockHeld(cs_main);
213
662k
    BlockMap::iterator it = m_block_index.find(hash);
214
662k
    return it == m_block_index.end() ? nullptr : &it->second;
215
662k
}
216
217
const CBlockIndex* BlockManager::LookupBlockIndex(const uint256& hash) const
218
6
{
219
6
    AssertLockHeld(cs_main);
220
6
    BlockMap::const_iterator it = m_block_index.find(hash);
221
6
    return it == m_block_index.end() ? nullptr : &it->second;
222
6
}
223
224
CBlockIndex* BlockManager::AddToBlockIndex(const CBlockHeader& block, CBlockIndex*& best_header)
225
123k
{
226
123k
    AssertLockHeld(cs_main);
227
228
123k
    auto [mi, inserted] = m_block_index.try_emplace(block.GetHash(), block);
229
123k
    if (!inserted) {
230
3
        return &mi->second;
231
3
    }
232
123k
    CBlockIndex* pindexNew = &(*mi).second;
233
234
    // We assign the sequence id to blocks only when the full data is available,
235
    // to avoid miners withholding blocks but broadcasting headers, to get a
236
    // competitive advantage.
237
123k
    pindexNew->nSequenceId = SEQ_ID_INIT_FROM_DISK;
238
239
123k
    pindexNew->phashBlock = &((*mi).first);
240
123k
    BlockMap::iterator miPrev = m_block_index.find(block.hashPrevBlock);
241
123k
    if (miPrev != m_block_index.end()) {
242
123k
        pindexNew->pprev = &(*miPrev).second;
243
123k
        pindexNew->nHeight = pindexNew->pprev->nHeight + 1;
244
123k
        pindexNew->BuildSkip();
245
123k
    }
246
123k
    pindexNew->nTimeMax = (pindexNew->pprev ? std::max(pindexNew->pprev->nTimeMax, pindexNew->nTime) : pindexNew->nTime);
247
123k
    pindexNew->nChainWork = (pindexNew->pprev ? pindexNew->pprev->nChainWork : 0) + GetBlockProof(*pindexNew);
248
123k
    pindexNew->RaiseValidity(BLOCK_VALID_TREE);
249
123k
    if (best_header == nullptr || best_header->nChainWork < pindexNew->nChainWork) {
250
104k
        best_header = pindexNew;
251
104k
    }
252
253
123k
    m_dirty_blockindex.insert(pindexNew);
254
255
123k
    return pindexNew;
256
123k
}
257
258
void BlockManager::AddUnlinkedBlock(CBlockIndex* block)
259
3.32k
{
260
3.32k
    AssertLockHeld(cs_main);
261
3.32k
    Assume(block != nullptr);
262
3.32k
    Assume(block->nStatus & BLOCK_HAVE_DATA);
263
3.32k
    auto range = m_blocks_unlinked.equal_range(block->pprev);
264
3.34k
    for (auto it = range.first; it != range.second; ++it) {
265
19
        if (it->second == block) return;  // don't insert duplicates
266
19
    }
267
3.32k
    m_blocks_unlinked.emplace(block->pprev, block);
268
3.32k
}
269
270
void BlockManager::PruneOneBlockFile(const int fileNumber)
271
15
{
272
15
    AssertLockHeld(cs_main);
273
274
9.28k
    for (auto& entry : m_block_index) {
275
9.28k
        CBlockIndex* pindex = &entry.second;
276
9.28k
        if (pindex->nFile == fileNumber) {
277
2.93k
            pindex->nStatus &= ~BLOCK_HAVE_DATA;
278
2.93k
            pindex->nStatus &= ~BLOCK_HAVE_UNDO;
279
2.93k
            pindex->nFile = 0;
280
2.93k
            pindex->nDataPos = 0;
281
2.93k
            pindex->nUndoPos = 0;
282
2.93k
            m_dirty_blockindex.insert(pindex);
283
284
            // Prune from m_blocks_unlinked -- any block we prune would have
285
            // to be downloaded again in order to consider its chain, at which
286
            // point it would be considered as a candidate for
287
            // m_blocks_unlinked or setBlockIndexCandidates.
288
2.93k
            auto range = m_blocks_unlinked.equal_range(pindex->pprev);
289
2.93k
            while (range.first != range.second) {
290
1
                std::multimap<CBlockIndex*, CBlockIndex*>::iterator _it = range.first;
291
1
                range.first++;
292
1
                if (_it->second == pindex) {
293
1
                    m_blocks_unlinked.erase(_it);
294
1
                }
295
1
            }
296
2.93k
        }
297
9.28k
    }
298
299
15
    m_blockfile_info.at(fileNumber) = CBlockFileInfo{};
300
15
    m_dirty_fileinfo.insert(fileNumber);
301
15
}
302
303
void BlockManager::FindFilesToPruneManual(
304
    std::set<int>& setFilesToPrune,
305
    int nManualPruneHeight,
306
    const Chainstate& chain)
307
11
{
308
11
    assert(IsPruneMode() && nManualPruneHeight > 0);
309
310
11
    LOCK(::cs_main);
311
11
    if (chain.m_chain.Height() < 0) {
312
0
        return;
313
0
    }
314
315
11
    const auto [min_block_to_prune, last_block_can_prune] = chain.GetPruneRange(nManualPruneHeight);
316
317
11
    int count = 0;
318
36
    for (int fileNumber = 0; fileNumber < this->MaxBlockfileNum(); fileNumber++) {
319
25
        const auto& fileinfo = m_blockfile_info[fileNumber];
320
25
        if (fileinfo.nSize == 0 || fileinfo.nHeightLast > (unsigned)last_block_can_prune || fileinfo.nHeightFirst < (unsigned)min_block_to_prune) {
321
13
            continue;
322
13
        }
323
324
12
        PruneOneBlockFile(fileNumber);
325
12
        setFilesToPrune.insert(fileNumber);
326
12
        count++;
327
12
    }
328
11
    LogInfo("[%s] Prune (Manual): prune_height=%d removed %d blk/rev pairs",
329
11
        chain.GetRole(), last_block_can_prune, count);
330
11
}
331
332
void BlockManager::FindFilesToPrune(
333
    std::set<int>& setFilesToPrune,
334
    int last_prune,
335
    const Chainstate& chain,
336
    ChainstateManager& chainman)
337
142
{
338
142
    LOCK(::cs_main);
339
    // Compute `target` value with maximum size (in bytes) of blocks below the
340
    // `last_prune` height which should be preserved and not pruned. The
341
    // `target` value will be derived from the -prune preference provided by the
342
    // user. If there is a historical chainstate being used to populate indexes
343
    // and validate the snapshot, the target is divided by two so half of the
344
    // block storage will be reserved for the historical chainstate, and the
345
    // other half will be reserved for the most-work chainstate.
346
142
    const int num_chainstates{chainman.HistoricalChainstate() ? 2 : 1};
347
142
    const auto target = std::max(
348
142
        MIN_DISK_SPACE_FOR_BLOCK_FILES, GetPruneTarget() / num_chainstates);
349
142
    const uint64_t target_sync_height = chainman.m_best_header->nHeight;
350
351
142
    if (chain.m_chain.Height() < 0 || target == 0) {
352
14
        return;
353
14
    }
354
128
    if (static_cast<uint64_t>(chain.m_chain.Height()) <= chainman.GetParams().PruneAfterHeight()) {
355
20
        return;
356
20
    }
357
358
108
    const auto [min_block_to_prune, last_block_can_prune] = chain.GetPruneRange(last_prune);
359
360
108
    uint64_t nCurrentUsage = CalculateCurrentUsage();
361
    // We don't check to prune until after we've allocated new space for files
362
    // So we should leave a buffer under our target to account for another allocation
363
    // before the next pruning.
364
108
    uint64_t nBuffer = BLOCKFILE_CHUNK_SIZE + UNDOFILE_CHUNK_SIZE;
365
108
    uint64_t nBytesToPrune;
366
108
    int count = 0;
367
368
108
    if (nCurrentUsage + nBuffer >= target) {
369
        // On a prune event, the chainstate DB is flushed.
370
        // To avoid excessive prune events negating the benefit of high dbcache
371
        // values, we should not prune too rapidly.
372
        // So when pruning in IBD, increase the buffer to avoid a re-prune too soon.
373
0
        const auto chain_tip_height = chain.m_chain.Height();
374
0
        if (chainman.IsInitialBlockDownload() && target_sync_height > (uint64_t)chain_tip_height) {
375
            // Since this is only relevant during IBD, we assume blocks are at least 1 MB on average
376
0
            static constexpr uint64_t average_block_size = 1000000;  /* 1 MB */
377
0
            const uint64_t remaining_blocks = target_sync_height - chain_tip_height;
378
0
            nBuffer += average_block_size * remaining_blocks;
379
0
        }
380
381
0
        for (int fileNumber = 0; fileNumber < this->MaxBlockfileNum(); fileNumber++) {
382
0
            const auto& fileinfo = m_blockfile_info[fileNumber];
383
0
            nBytesToPrune = fileinfo.nSize + fileinfo.nUndoSize;
384
385
0
            if (fileinfo.nSize == 0) {
386
0
                continue;
387
0
            }
388
389
0
            if (nCurrentUsage + nBuffer < target) { // are we below our target?
390
0
                break;
391
0
            }
392
393
            // don't prune files that could have a block that's not within the allowable
394
            // prune range for the chain being pruned.
395
0
            if (fileinfo.nHeightLast > (unsigned)last_block_can_prune || fileinfo.nHeightFirst < (unsigned)min_block_to_prune) {
396
0
                continue;
397
0
            }
398
399
0
            PruneOneBlockFile(fileNumber);
400
            // Queue up the files for removal
401
0
            setFilesToPrune.insert(fileNumber);
402
0
            nCurrentUsage -= nBytesToPrune;
403
0
            count++;
404
0
        }
405
0
    }
406
407
108
    LogDebug(BCLog::PRUNE, "[%s] target=%dMiB actual=%dMiB diff=%dMiB min_height=%d max_prune_height=%d removed %d blk/rev pairs\n",
408
108
             chain.GetRole(), target / 1_MiB, nCurrentUsage / 1_MiB,
409
108
             (int64_t(target) - int64_t(nCurrentUsage)) / int64_t(1_MiB),
410
108
             min_block_to_prune, last_block_can_prune, count);
411
108
}
412
413
7.21k
void BlockManager::UpdatePruneLock(const std::string& name, const PruneLockInfo& lock_info) {
414
7.21k
    AssertLockHeld(::cs_main);
415
7.21k
    m_prune_locks[name] = lock_info;
416
7.21k
}
417
418
bool BlockManager::DeletePruneLock(const std::string& name)
419
3
{
420
3
    AssertLockHeld(::cs_main);
421
3
    return m_prune_locks.erase(name) > 0;
422
3
}
423
424
CBlockIndex* BlockManager::InsertBlockIndex(const uint256& hash)
425
271k
{
426
271k
    AssertLockHeld(cs_main);
427
428
271k
    if (hash.IsNull()) {
429
748
        return nullptr;
430
748
    }
431
432
271k
    const auto [mi, inserted]{m_block_index.try_emplace(hash)};
433
271k
    CBlockIndex* pindex = &(*mi).second;
434
271k
    if (inserted) {
435
135k
        pindex->phashBlock = &((*mi).first);
436
135k
    }
437
271k
    return pindex;
438
271k
}
439
440
bool BlockManager::LoadBlockIndex(const std::optional<uint256>& snapshot_blockhash)
441
1.20k
{
442
1.20k
    if (!m_block_tree_db->LoadBlockIndexGuts(
443
271k
            GetConsensus(), [this](const uint256& hash) EXCLUSIVE_LOCKS_REQUIRED(cs_main) { return this->InsertBlockIndex(hash); }, m_interrupt)) {
444
2
        return false;
445
2
    }
446
447
1.20k
    if (snapshot_blockhash) {
448
7
        const std::optional<AssumeutxoData> maybe_au_data = GetParams().AssumeutxoForBlockhash(*snapshot_blockhash);
449
7
        if (!maybe_au_data) {
450
1
            m_opts.notifications.fatalError(strprintf(_("Assumeutxo data not found for the given blockhash '%s'."), snapshot_blockhash->ToString()));
451
1
            return false;
452
1
        }
453
6
        const AssumeutxoData& au_data = *Assert(maybe_au_data);
454
6
        m_snapshot_height = au_data.height;
455
6
        CBlockIndex* base{LookupBlockIndex(*snapshot_blockhash)};
456
457
        // Since m_chain_tx_count (responsible for estimated progress) isn't persisted
458
        // to disk, we must bootstrap the value for assumedvalid chainstates
459
        // from the hardcoded assumeutxo chainparams.
460
6
        base->m_chain_tx_count = au_data.m_chain_tx_count;
461
6
        LogInfo("[snapshot] set m_chain_tx_count=%d for %s", au_data.m_chain_tx_count, snapshot_blockhash->ToString());
462
1.19k
    } else {
463
        // If this isn't called with a snapshot blockhash, make sure the cached snapshot height
464
        // is null. This is relevant during snapshot completion, when the blockman may be loaded
465
        // with a height that then needs to be cleared after the snapshot is fully validated.
466
1.19k
        m_snapshot_height.reset();
467
1.19k
    }
468
469
1.20k
    Assert(m_snapshot_height.has_value() == snapshot_blockhash.has_value());
470
471
    // Calculate nChainWork
472
1.20k
    std::vector<CBlockIndex*> vSortedByHeight{GetAllBlockIndices()};
473
1.20k
    std::sort(vSortedByHeight.begin(), vSortedByHeight.end(),
474
1.20k
              CBlockIndexHeightOnlyComparator());
475
476
1.20k
    CBlockIndex* previous_index{nullptr};
477
135k
    for (CBlockIndex* pindex : vSortedByHeight) {
478
135k
        if (m_interrupt) return false;
479
135k
        if (previous_index && pindex->nHeight > previous_index->nHeight + 1) {
480
1
            LogError("%s: block index is non-contiguous, index of height %d missing\n", __func__, previous_index->nHeight + 1);
481
1
            return false;
482
1
        }
483
135k
        previous_index = pindex;
484
135k
        pindex->nChainWork = (pindex->pprev ? pindex->pprev->nChainWork : 0) + GetBlockProof(*pindex);
485
135k
        pindex->nTimeMax = (pindex->pprev ? std::max(pindex->pprev->nTimeMax, pindex->nTime) : pindex->nTime);
486
487
        // We can link the chain of blocks for which we've received transactions at some point, or
488
        // blocks that are assumed-valid on the basis of snapshot load (see
489
        // PopulateAndValidateSnapshot()).
490
        // Pruned nodes may have deleted the block.
491
135k
        if (pindex->nTx > 0) {
492
134k
            if (pindex->pprev) {
493
134k
                if (m_snapshot_height && pindex->nHeight == *m_snapshot_height &&
494
134k
                        pindex->GetBlockHash() == *snapshot_blockhash) {
495
                    // Should have been set above; don't disturb it with code below.
496
4
                    Assert(pindex->m_chain_tx_count > 0);
497
134k
                } else if (pindex->pprev->m_chain_tx_count > 0) {
498
134k
                    pindex->m_chain_tx_count = pindex->pprev->m_chain_tx_count + pindex->nTx;
499
134k
                } else {
500
11
                    pindex->m_chain_tx_count = 0;
501
11
                    if (pindex->nStatus & BLOCK_HAVE_DATA) {
502
10
                        AddUnlinkedBlock(pindex);
503
10
                    }
504
11
                }
505
134k
            } else {
506
749
                pindex->m_chain_tx_count = pindex->nTx;
507
749
            }
508
134k
        }
509
510
135k
        if (pindex->nStatus & BLOCK_FAILED_CHILD) {
511
            // BLOCK_FAILED_CHILD is deprecated, but may still exist on disk. Replace it with BLOCK_FAILED_VALID.
512
1
            pindex->nStatus = (pindex->nStatus & ~BLOCK_FAILED_CHILD) | BLOCK_FAILED_VALID;
513
1
            m_dirty_blockindex.insert(pindex);
514
1
        }
515
135k
        if (!(pindex->nStatus & BLOCK_FAILED_VALID) && pindex->pprev && (pindex->pprev->nStatus & BLOCK_FAILED_VALID)) {
516
            // All descendants of invalid blocks are invalid too.
517
1
            pindex->nStatus |= BLOCK_FAILED_VALID;
518
1
            m_dirty_blockindex.insert(pindex);
519
1
        }
520
521
135k
        if (pindex->pprev) {
522
134k
            pindex->BuildSkip();
523
134k
        }
524
135k
    }
525
526
1.20k
    return true;
527
1.20k
}
528
529
void BlockManager::WriteBlockIndexDB()
530
3.44k
{
531
3.44k
    AssertLockHeld(::cs_main);
532
3.44k
    std::vector<std::pair<int, const CBlockFileInfo*>> vFiles;
533
3.44k
    vFiles.reserve(m_dirty_fileinfo.size());
534
5.09k
    for (std::set<int>::iterator it = m_dirty_fileinfo.begin(); it != m_dirty_fileinfo.end();) {
535
1.65k
        vFiles.emplace_back(*it, &m_blockfile_info[*it]);
536
1.65k
        m_dirty_fileinfo.erase(it++);
537
1.65k
    }
538
3.44k
    std::vector<const CBlockIndex*> vBlocks;
539
3.44k
    vBlocks.reserve(m_dirty_blockindex.size());
540
127k
    for (std::set<CBlockIndex*>::iterator it = m_dirty_blockindex.begin(); it != m_dirty_blockindex.end();) {
541
124k
        vBlocks.push_back(*it);
542
124k
        m_dirty_blockindex.erase(it++);
543
124k
    }
544
3.44k
    int max_blockfile{this->MaxBlockfileNum()};
545
3.44k
    m_block_tree_db->WriteBatchSync(vFiles, max_blockfile, vBlocks);
546
3.44k
}
547
548
bool BlockManager::LoadBlockIndexDB(const std::optional<uint256>& snapshot_blockhash)
549
1.20k
{
550
1.20k
    AssertLockHeld(::cs_main);
551
1.20k
    if (!LoadBlockIndex(snapshot_blockhash)) {
552
4
        return false;
553
4
    }
554
1.20k
    int max_blockfile_num{0};
555
556
    // Load block file info
557
1.20k
    m_block_tree_db->ReadLastBlockFile(max_blockfile_num);
558
1.20k
    m_blockfile_info.resize(max_blockfile_num + 1);
559
1.20k
    LogInfo("Loading block index db: last block file = %i", max_blockfile_num);
560
2.42k
    for (int nFile = 0; nFile <= max_blockfile_num; nFile++) {
561
1.22k
        m_block_tree_db->ReadBlockFileInfo(nFile, m_blockfile_info[nFile]);
562
1.22k
    }
563
1.20k
    LogInfo("Loading block index db: last block file info: %s", m_blockfile_info[max_blockfile_num].ToString());
564
1.20k
    for (int nFile = max_blockfile_num + 1; true; nFile++) {
565
1.20k
        CBlockFileInfo info;
566
1.20k
        if (m_block_tree_db->ReadBlockFileInfo(nFile, info)) {
567
0
            m_blockfile_info.push_back(info);
568
1.20k
        } else {
569
1.20k
            break;
570
1.20k
        }
571
1.20k
    }
572
573
    // Check presence of blk files
574
1.20k
    LogInfo("Checking all blk files are present...");
575
1.20k
    std::set<int> setBlkDataFiles;
576
135k
    for (const auto& [_, block_index] : m_block_index) {
577
135k
        if (block_index.nStatus & BLOCK_HAVE_DATA) {
578
134k
            setBlkDataFiles.insert(block_index.nFile);
579
134k
        }
580
135k
    }
581
1.96k
    for (std::set<int>::iterator it = setBlkDataFiles.begin(); it != setBlkDataFiles.end(); it++) {
582
764
        FlatFilePos pos(*it, 0);
583
764
        if (OpenBlockFile(pos, /*fReadOnly=*/true).IsNull()) {
584
1
            return false;
585
1
        }
586
764
    }
587
588
1.20k
    {
589
        // Initialize the blockfile cursors.
590
2.42k
        for (size_t i = 0; i < m_blockfile_info.size(); ++i) {
591
1.21k
            const auto last_height_in_file = m_blockfile_info[i].nHeightLast;
592
1.21k
            m_blockfile_cursors[BlockfileTypeForHeight(last_height_in_file)] = {static_cast<int>(i), 0};
593
1.21k
        }
594
1.20k
    }
595
596
    // Check whether we have ever pruned block & undo files
597
1.20k
    m_block_tree_db->ReadFlag("prunedblockfiles", m_have_pruned);
598
1.20k
    if (m_have_pruned) {
599
2
        LogInfo("Loading block index db: Block files have previously been pruned");
600
2
    }
601
602
    // Check whether we need to continue reindexing
603
1.20k
    bool fReindexing = false;
604
1.20k
    m_block_tree_db->ReadReindexing(fReindexing);
605
1.20k
    if (fReindexing) m_blockfiles_indexed = false;
606
607
1.20k
    return true;
608
1.20k
}
609
610
void BlockManager::ScanAndUnlinkAlreadyPrunedFiles()
611
1.20k
{
612
1.20k
    AssertLockHeld(::cs_main);
613
1.20k
    int max_blockfile{this->MaxBlockfileNum()};
614
1.20k
    if (!m_have_pruned) {
615
1.20k
        return;
616
1.20k
    }
617
618
4
    std::set<int> block_files_to_prune;
619
11
    for (int file_number = 0; file_number < max_blockfile; file_number++) {
620
7
        if (m_blockfile_info[file_number].nSize == 0) {
621
5
            block_files_to_prune.insert(file_number);
622
5
        }
623
7
    }
624
625
4
    UnlinkPrunedFiles(block_files_to_prune);
626
4
}
627
628
bool BlockManager::IsBlockPruned(const CBlockIndex& block) const
629
447
{
630
447
    AssertLockHeld(::cs_main);
631
447
    return m_have_pruned && !(block.nStatus & BLOCK_HAVE_DATA) && (block.nTx > 0);
632
447
}
633
634
const CBlockIndex& BlockManager::GetFirstBlock(const CBlockIndex& upper_block, uint32_t status_mask, const CBlockIndex* lower_block) const
635
71
{
636
71
    AssertLockHeld(::cs_main);
637
71
    const CBlockIndex* last_block = &upper_block;
638
71
    assert((last_block->nStatus & status_mask) == status_mask); // 'upper_block' must satisfy the status mask
639
13.8k
    while (last_block->pprev && ((last_block->pprev->nStatus & status_mask) == status_mask)) {
640
13.7k
        if (lower_block) {
641
            // Return if we reached the lower_block
642
13.6k
            if (last_block == lower_block) return *lower_block;
643
            // if range was surpassed, means that 'lower_block' is not part of the 'upper_block' chain
644
            // and so far this is not allowed.
645
13.6k
            assert(last_block->nHeight >= lower_block->nHeight);
646
13.6k
        }
647
13.7k
        last_block = last_block->pprev;
648
13.7k
    }
649
71
    assert(last_block != nullptr);
650
58
    return *last_block;
651
58
}
652
653
bool BlockManager::CheckBlockDataAvailability(const CBlockIndex& upper_block, const CBlockIndex& lower_block, BlockStatus block_status)
654
37
{
655
37
    if (!(upper_block.nStatus & block_status)) return false;
656
37
    const auto& first_block = GetFirstBlock(upper_block, block_status, &lower_block);
657
    // Special case: the genesis block has no undo data
658
37
    if (block_status & BLOCK_HAVE_UNDO && lower_block.nHeight == 0 && first_block.nHeight == 1) {
659
        // This might indicate missing data, or it could simply reflect the expected absence of undo data for the genesis block.
660
        // To distinguish between the two, check if all required block data *except* undo is available up to the genesis block.
661
13
        BlockStatus flags{block_status & ~BLOCK_HAVE_UNDO};
662
13
        return first_block.pprev && first_block.pprev->nStatus & flags;
663
13
    }
664
24
    return &first_block == &lower_block;
665
37
}
666
667
// If we're using -prune with -reindex, then delete block files that will be ignored by the
668
// reindex.  Since reindexing works by starting at block file 0 and looping until a blockfile
669
// is missing, do the same here to delete any later block files after a gap.  Also delete all
670
// rev files since they'll be rewritten by the reindex anyway.  This ensures that m_blockfile_info
671
// is in sync with what's actually on disk by the time we start downloading, so that pruning
672
// works correctly.
673
void BlockManager::CleanupBlockRevFiles() const
674
2
{
675
2
    std::map<std::string, fs::path> mapBlockFiles;
676
677
    // Glob all blk?????.dat and rev?????.dat files from the blocks directory.
678
    // Remove the rev files immediately and insert the blk file paths into an
679
    // ordered map keyed by block file index.
680
2
    LogInfo("Removing unusable blk?????.dat and rev?????.dat files for -reindex with -prune");
681
14
    for (fs::directory_iterator it(m_opts.blocks_dir); it != fs::directory_iterator(); it++) {
682
12
        const std::string path = fs::PathToString(it->path().filename());
683
12
        if (fs::is_regular_file(*it) &&
684
12
            path.length() == 12 &&
685
12
            path.ends_with(".dat"))
686
6
        {
687
6
            if (path.starts_with("blk")) {
688
3
                mapBlockFiles[path.substr(3, 5)] = it->path();
689
3
            } else if (path.starts_with("rev")) {
690
3
                remove(it->path());
691
3
            }
692
6
        }
693
12
    }
694
695
    // Remove all block files that aren't part of a contiguous set starting at
696
    // zero by walking the ordered map (keys are block file indices) by
697
    // keeping a separate counter.  Once we hit a gap (or if 0 doesn't exist)
698
    // start removing block files.
699
2
    int nContigCounter = 0;
700
3
    for (const std::pair<const std::string, fs::path>& item : mapBlockFiles) {
701
3
        if (LocaleIndependentAtoi<int>(item.first) == nContigCounter) {
702
1
            nContigCounter++;
703
1
            continue;
704
1
        }
705
2
        remove(item.second);
706
2
    }
707
2
}
708
709
CBlockFileInfo* BlockManager::GetBlockFileInfo(size_t n)
710
3
{
711
3
    AssertLockHeld(::cs_main);
712
3
    return &m_blockfile_info.at(n);
713
3
}
714
715
bool BlockManager::ReadBlockUndo(CBlockUndo& blockundo, const CBlockIndex& index) const
716
36.5k
{
717
36.5k
    const FlatFilePos pos{WITH_LOCK(::cs_main, return index.GetUndoPos())};
718
719
    // Open history file to read
720
36.5k
    AutoFile file{OpenUndoFile(pos, true)};
721
36.5k
    if (file.IsNull()) {
722
5
        LogError("OpenUndoFile failed for %s while reading block undo", pos.ToString());
723
5
        return false;
724
5
    }
725
36.5k
    BufferedReader filein{std::move(file)};
726
727
36.5k
    try {
728
        // Read block
729
36.5k
        HashVerifier verifier{filein}; // Use HashVerifier, as reserializing may lose data, c.f. commit d3424243
730
731
36.5k
        verifier << index.pprev->GetBlockHash();
732
36.5k
        verifier >> blockundo;
733
734
36.5k
        uint256 hashChecksum;
735
36.5k
        filein >> hashChecksum;
736
737
        // Verify checksum
738
36.5k
        if (hashChecksum != verifier.GetHash()) {
739
0
            LogError("Checksum mismatch at %s while reading block undo", pos.ToString());
740
0
            return false;
741
0
        }
742
36.5k
    } catch (const std::exception& e) {
743
1
        LogError("Deserialize or I/O error - %s at %s while reading block undo", e.what(), pos.ToString());
744
1
        return false;
745
1
    }
746
747
36.5k
    return true;
748
36.5k
}
749
750
bool BlockManager::FlushUndoFile(int block_file, bool finalize)
751
3.45k
{
752
3.45k
    FlatFilePos undo_pos_old(block_file, m_blockfile_info[block_file].nUndoSize);
753
3.45k
    if (!m_undo_file_seq.Flush(undo_pos_old, finalize)) {
754
0
        m_opts.notifications.flushError(_("Flushing undo file to disk failed. This is likely the result of an I/O error."));
755
0
        return false;
756
0
    }
757
3.45k
    return true;
758
3.45k
}
759
760
bool BlockManager::FlushBlockFile(int blockfile_num, bool fFinalize, bool finalize_undo)
761
3.45k
{
762
3.45k
    AssertLockHeld(::cs_main);
763
3.45k
    bool success = true;
764
765
3.45k
    if (m_blockfile_info.size() < 1) {
766
        // Return if we haven't loaded any blockfiles yet. This happens during
767
        // chainstate init, when we call ChainstateManager::MaybeRebalanceCaches() (which
768
        // then calls FlushStateToDisk()), resulting in a call to this function before we
769
        // have populated `m_blockfile_info` via LoadBlockIndexDB().
770
2
        return true;
771
2
    }
772
3.45k
    assert(static_cast<int>(m_blockfile_info.size()) > blockfile_num);
773
774
3.45k
    FlatFilePos block_pos_old(blockfile_num, m_blockfile_info[blockfile_num].nSize);
775
3.45k
    if (!m_block_file_seq.Flush(block_pos_old, fFinalize)) {
776
0
        m_opts.notifications.flushError(_("Flushing block file to disk failed. This is likely the result of an I/O error."));
777
0
        success = false;
778
0
    }
779
    // we do not always flush the undo file, as the chain tip may be lagging behind the incoming blocks,
780
    // e.g. during IBD or a sync after a node going offline
781
3.45k
    if (!fFinalize || finalize_undo) {
782
3.45k
        if (!FlushUndoFile(blockfile_num, finalize_undo)) {
783
0
            success = false;
784
0
        }
785
3.45k
    }
786
3.45k
    return success;
787
3.45k
}
788
789
BlockfileType BlockManager::BlockfileTypeForHeight(int height)
790
227k
{
791
227k
    if (!m_snapshot_height) {
792
221k
        return BlockfileType::NORMAL;
793
221k
    }
794
5.52k
    return (height >= *m_snapshot_height) ? BlockfileType::ASSUMED : BlockfileType::NORMAL;
795
227k
}
796
797
bool BlockManager::FlushChainstateBlockFile(int tip_height)
798
3.44k
{
799
3.44k
    AssertLockHeld(::cs_main);
800
3.44k
    auto& cursor = m_blockfile_cursors[BlockfileTypeForHeight(tip_height)];
801
    // If the cursor does not exist, it means an assumeutxo snapshot is loaded,
802
    // but no blocks past the snapshot height have been written yet, so there
803
    // is no data associated with the chainstate, and it is safe not to flush.
804
3.44k
    if (cursor) {
805
3.41k
        return FlushBlockFile(cursor->file_num, /*fFinalize=*/false, /*finalize_undo=*/false);
806
3.41k
    }
807
    // No need to log warnings in this case.
808
22
    return true;
809
3.44k
}
810
811
uint64_t BlockManager::CalculateCurrentUsage()
812
15.3k
{
813
15.3k
    AssertLockHeld(::cs_main);
814
15.3k
    uint64_t retval = 0;
815
15.5k
    for (const CBlockFileInfo& file : m_blockfile_info) {
816
15.5k
        retval += file.nSize + file.nUndoSize;
817
15.5k
    }
818
15.3k
    return retval;
819
15.3k
}
820
821
void BlockManager::UnlinkPrunedFiles(const std::set<int>& setFilesToPrune) const
822
16
{
823
16
    std::error_code ec;
824
35
    for (std::set<int>::iterator it = setFilesToPrune.begin(); it != setFilesToPrune.end(); ++it) {
825
19
        FlatFilePos pos(*it, 0);
826
19
        const bool removed_blockfile{fs::remove(m_block_file_seq.FileName(pos), ec)};
827
19
        const bool removed_undofile{fs::remove(m_undo_file_seq.FileName(pos), ec)};
828
19
        if (removed_blockfile || removed_undofile) {
829
15
            LogDebug(BCLog::BLOCKSTORAGE, "Prune: %s deleted blk/rev (%05u)\n", __func__, *it);
830
15
        }
831
19
    }
832
16
}
833
834
AutoFile BlockManager::OpenBlockFile(const FlatFilePos& pos, bool fReadOnly) const
835
274k
{
836
274k
    return AutoFile{m_block_file_seq.Open(pos, fReadOnly), m_obfuscation};
837
274k
}
838
839
/** Open an undo file (rev?????.dat) */
840
AutoFile BlockManager::OpenUndoFile(const FlatFilePos& pos, bool fReadOnly) const
841
143k
{
842
143k
    return AutoFile{m_undo_file_seq.Open(pos, fReadOnly), m_obfuscation};
843
143k
}
844
845
fs::path BlockManager::GetBlockPosFilename(const FlatFilePos& pos) const
846
34
{
847
34
    return m_block_file_seq.FileName(pos);
848
34
}
849
850
FlatFilePos BlockManager::FindNextBlockPos(unsigned int nAddSize, unsigned int nHeight, uint64_t nTime)
851
109k
{
852
109k
    AssertLockHeld(::cs_main);
853
109k
    const BlockfileType chain_type = BlockfileTypeForHeight(nHeight);
854
855
109k
    if (!m_blockfile_cursors[chain_type]) {
856
        // If a snapshot is loaded during runtime, we may not have initialized this cursor yet.
857
12
        assert(chain_type == BlockfileType::ASSUMED);
858
12
        const auto new_cursor = BlockfileCursor{this->MaxBlockfileNum() + 1};
859
12
        m_blockfile_cursors[chain_type] = new_cursor;
860
12
        LogDebug(BCLog::BLOCKSTORAGE, "[%s] initializing blockfile cursor to %s\n", chain_type, new_cursor);
861
12
    }
862
109k
    const int last_blockfile = m_blockfile_cursors[chain_type]->file_num;
863
864
109k
    int nFile = last_blockfile;
865
109k
    if (static_cast<int>(m_blockfile_info.size()) <= nFile) {
866
16
        m_blockfile_info.resize(nFile + 1);
867
16
    }
868
869
109k
    bool finalize_undo = false;
870
109k
    unsigned int max_blockfile_size{MAX_BLOCKFILE_SIZE};
871
    // Use smaller blockfiles in test-only -fastprune mode - but avoid
872
    // the possibility of having a block not fit into the block file.
873
109k
    if (m_opts.fast_prune) {
874
4.74k
        max_blockfile_size = 0x10000; // 64kiB
875
4.74k
        if (nAddSize >= max_blockfile_size) {
876
            // dynamically adjust the blockfile size to be larger than the added size
877
1
            max_blockfile_size = nAddSize + 1;
878
1
        }
879
4.74k
    }
880
109k
    assert(nAddSize < max_blockfile_size);
881
882
109k
    while (m_blockfile_info[nFile].nSize + nAddSize >= max_blockfile_size) {
883
        // when the undo file is keeping up with the block file, we want to flush it explicitly
884
        // when it is lagging behind (more blocks arrive than are being connected), we let the
885
        // undo block write case handle it
886
33
        finalize_undo = (static_cast<int>(m_blockfile_info[nFile].nHeightLast) ==
887
33
                         Assert(m_blockfile_cursors[chain_type])->undo_height);
888
889
        // Try the next unclaimed blockfile number
890
33
        nFile = this->MaxBlockfileNum() + 1;
891
        // Set to increment MaxBlockfileNum() for next iteration
892
33
        m_blockfile_cursors[chain_type] = BlockfileCursor{nFile};
893
894
33
        if (static_cast<int>(m_blockfile_info.size()) <= nFile) {
895
33
            m_blockfile_info.resize(nFile + 1);
896
33
        }
897
33
    }
898
109k
    FlatFilePos pos;
899
109k
    pos.nFile = nFile;
900
109k
    pos.nPos = m_blockfile_info[nFile].nSize;
901
902
109k
    if (nFile != last_blockfile) {
903
33
        LogDebug(BCLog::BLOCKSTORAGE, "Leaving block file %i: %s (onto %i) (height %i)\n",
904
33
                 last_blockfile, m_blockfile_info[last_blockfile].ToString(), nFile, nHeight);
905
906
        // Do not propagate the return code. The flush concerns a previous block
907
        // and undo file that has already been written to. If a flush fails
908
        // here, and we crash, there is no expected additional block data
909
        // inconsistency arising from the flush failure here. However, the undo
910
        // data may be inconsistent after a crash if the flush is called during
911
        // a reindex. A flush error might also leave some of the data files
912
        // untrimmed.
913
33
        if (!FlushBlockFile(last_blockfile, /*fFinalize=*/true, finalize_undo)) {
914
0
            LogWarning(
915
0
                          "Failed to flush previous block file %05i (finalize=1, finalize_undo=%i) before opening new block file %05i\n",
916
0
                          last_blockfile, finalize_undo, nFile);
917
0
        }
918
        // No undo data yet in the new file, so reset our undo-height tracking.
919
33
        m_blockfile_cursors[chain_type] = BlockfileCursor{nFile};
920
33
    }
921
922
109k
    m_blockfile_info[nFile].AddBlock(nHeight, nTime);
923
109k
    m_blockfile_info[nFile].nSize += nAddSize;
924
925
109k
    bool out_of_space;
926
109k
    size_t bytes_allocated = m_block_file_seq.Allocate(pos, nAddSize, out_of_space);
927
109k
    if (out_of_space) {
928
0
        m_opts.notifications.fatalError(_("Disk space is too low!"));
929
0
        return {};
930
0
    }
931
109k
    if (bytes_allocated != 0 && IsPruneMode()) {
932
90
        m_check_for_pruning = true;
933
90
    }
934
935
109k
    m_dirty_fileinfo.insert(nFile);
936
109k
    return pos;
937
109k
}
938
939
void BlockManager::UpdateBlockInfo(const CBlock& block, unsigned int nHeight, const FlatFilePos& pos)
940
2.10k
{
941
2.10k
    AssertLockHeld(::cs_main);
942
    // Update the cursor so it points to the last file.
943
2.10k
    const BlockfileType chain_type{BlockfileTypeForHeight(nHeight)};
944
2.10k
    auto& cursor{m_blockfile_cursors[chain_type]};
945
2.10k
    if (!cursor || cursor->file_num < pos.nFile) {
946
1
        m_blockfile_cursors[chain_type] = BlockfileCursor{pos.nFile};
947
1
    }
948
949
    // Update the file information with the current block.
950
2.10k
    const unsigned int added_size = ::GetSerializeSize(TX_WITH_WITNESS(block));
951
2.10k
    const int nFile = pos.nFile;
952
2.10k
    if (static_cast<int>(m_blockfile_info.size()) <= nFile) {
953
14
        m_blockfile_info.resize(nFile + 1);
954
14
    }
955
2.10k
    m_blockfile_info[nFile].AddBlock(nHeight, block.GetBlockTime());
956
2.10k
    m_blockfile_info[nFile].nSize = std::max(pos.nPos + added_size, m_blockfile_info[nFile].nSize);
957
2.10k
    m_dirty_fileinfo.insert(nFile);
958
2.10k
}
959
960
bool BlockManager::FindUndoPos(BlockValidationState& state, int nFile, FlatFilePos& pos, unsigned int nAddSize)
961
106k
{
962
106k
    AssertLockHeld(::cs_main);
963
106k
    pos.nFile = nFile;
964
965
106k
    pos.nPos = m_blockfile_info[nFile].nUndoSize;
966
106k
    m_blockfile_info[nFile].nUndoSize += nAddSize;
967
106k
    m_dirty_fileinfo.insert(nFile);
968
969
106k
    bool out_of_space;
970
106k
    size_t bytes_allocated = m_undo_file_seq.Allocate(pos, nAddSize, out_of_space);
971
106k
    if (out_of_space) {
972
0
        return FatalError(m_opts.notifications, state, _("Disk space is too low!"));
973
0
    }
974
106k
    if (bytes_allocated != 0 && IsPruneMode()) {
975
28
        m_check_for_pruning = true;
976
28
    }
977
978
106k
    return true;
979
106k
}
980
981
bool BlockManager::WriteBlockUndo(const CBlockUndo& blockundo, BlockValidationState& state, CBlockIndex& block)
982
111k
{
983
111k
    AssertLockHeld(::cs_main);
984
111k
    const BlockfileType type = BlockfileTypeForHeight(block.nHeight);
985
111k
    auto& cursor = *Assert(m_blockfile_cursors[type]);
986
987
    // Write undo information to disk
988
111k
    if (block.GetUndoPos().IsNull()) {
989
106k
        FlatFilePos pos;
990
106k
        const auto blockundo_size{static_cast<uint32_t>(GetSerializeSize(blockundo))};
991
106k
        if (!FindUndoPos(state, block.nFile, pos, blockundo_size + UNDO_DATA_DISK_OVERHEAD)) {
992
0
            LogError("FindUndoPos failed for %s while writing block undo", pos.ToString());
993
0
            return false;
994
0
        }
995
996
        // Open history file to append
997
106k
        AutoFile file{OpenUndoFile(pos)};
998
106k
        if (file.IsNull()) {
999
0
            LogError("OpenUndoFile failed for %s while writing block undo", pos.ToString());
1000
0
            return FatalError(m_opts.notifications, state, _("Failed to write undo data."));
1001
0
        }
1002
106k
        {
1003
106k
            BufferedWriter fileout{file};
1004
1005
            // Write index header
1006
106k
            fileout << GetParams().MessageStart() << blockundo_size;
1007
106k
            pos.nPos += STORAGE_HEADER_BYTES;
1008
106k
            {
1009
                // Calculate checksum
1010
106k
                HashWriter hasher{};
1011
106k
                hasher << block.pprev->GetBlockHash() << blockundo;
1012
                // Write undo data & checksum
1013
106k
                fileout << blockundo << hasher.GetHash();
1014
106k
            }
1015
            // BufferedWriter will flush pending data to file when fileout goes out of scope.
1016
106k
        }
1017
1018
        // Make sure that the file is closed before we call `FlushUndoFile`.
1019
106k
        if (file.fclose() != 0) {
1020
0
            LogError("Failed to close block undo file %s: %s", pos.ToString(), SysErrorString(errno));
1021
0
            return FatalError(m_opts.notifications, state, _("Failed to close block undo file."));
1022
0
        }
1023
1024
        // rev files are written in block height order, whereas blk files are written as blocks come in (often out of order)
1025
        // we want to flush the rev (undo) file once we've written the last block, which is indicated by the last height
1026
        // in the block file info as below; note that this does not catch the case where the undo writes are keeping up
1027
        // with the block writes (usually when a synced up node is getting newly mined blocks) -- this case is caught in
1028
        // the FindNextBlockPos function
1029
106k
        if (pos.nFile < cursor.file_num && static_cast<uint32_t>(block.nHeight) == m_blockfile_info[pos.nFile].nHeightLast) {
1030
            // Do not propagate the return code, a failed flush here should not
1031
            // be an indication for a failed write. If it were propagated here,
1032
            // the caller would assume the undo data not to be written, when in
1033
            // fact it is. Note though, that a failed flush might leave the data
1034
            // file untrimmed.
1035
1
            if (!FlushUndoFile(pos.nFile, true)) {
1036
0
                LogWarning("Failed to flush undo file %05i\n", pos.nFile);
1037
0
            }
1038
106k
        } else if (pos.nFile == cursor.file_num && block.nHeight > cursor.undo_height) {
1039
96.6k
            cursor.undo_height = block.nHeight;
1040
96.6k
        }
1041
        // update nUndoPos in block index
1042
106k
        block.nUndoPos = pos.nPos;
1043
106k
        block.nStatus |= BLOCK_HAVE_UNDO;
1044
106k
        m_dirty_blockindex.insert(&block);
1045
106k
    }
1046
1047
111k
    return true;
1048
111k
}
1049
1050
bool BlockManager::ReadBlock(CBlock& block, const FlatFilePos& pos, const std::optional<uint256>& expected_hash) const
1051
133k
{
1052
133k
    block.SetNull();
1053
1054
    // Open history file to read
1055
133k
    const auto block_data{ReadRawBlock(pos)};
1056
133k
    if (!block_data) {
1057
206
        return false;
1058
206
    }
1059
1060
133k
    try {
1061
        // Read block
1062
133k
        SpanReader{*block_data} >> TX_WITH_WITNESS(block);
1063
133k
    } catch (const std::exception& e) {
1064
0
        LogError("Deserialize or I/O error - %s at %s while reading block", e.what(), pos.ToString());
1065
0
        return false;
1066
0
    }
1067
1068
133k
    const auto block_hash{block.GetHash()};
1069
1070
    // Check the header
1071
133k
    if (!CheckProofOfWork(block_hash, block.nBits, GetConsensus())) {
1072
3
        LogError("Errors in block header at %s while reading block", pos.ToString());
1073
3
        return false;
1074
3
    }
1075
1076
    // Signet only: check block solution
1077
133k
    if (GetConsensus().signet_blocks && !CheckSignetBlockSolution(block, GetConsensus())) {
1078
0
        LogError("Errors in block solution at %s while reading block", pos.ToString());
1079
0
        return false;
1080
0
    }
1081
1082
133k
    if (expected_hash && block_hash != *expected_hash) {
1083
1
        LogError("GetHash() doesn't match index at %s while reading block (%s != %s)",
1084
1
                 pos.ToString(), block_hash.ToString(), expected_hash->ToString());
1085
1
        return false;
1086
1
    }
1087
1088
133k
    return true;
1089
133k
}
1090
1091
bool BlockManager::ReadBlock(CBlock& block, const CBlockIndex& index) const
1092
127k
{
1093
127k
    const FlatFilePos block_pos{WITH_LOCK(cs_main, return index.GetBlockPos())};
1094
127k
    return ReadBlock(block, block_pos, index.GetBlockHash());
1095
127k
}
1096
1097
BlockManager::ReadRawBlockResult BlockManager::ReadRawBlock(const FlatFilePos& pos, std::optional<std::pair<size_t, size_t>> block_part) const
1098
164k
{
1099
164k
    if (pos.nPos < STORAGE_HEADER_BYTES) {
1100
        // If nPos is less than STORAGE_HEADER_BYTES, we can't read the header that precedes the block data
1101
        // This would cause an unsigned integer underflow when trying to position the file cursor
1102
        // This can happen after pruning or default constructed positions
1103
203
        LogError("Failed for %s while reading raw block storage header", pos.ToString());
1104
203
        return util::Unexpected{ReadRawError::IO};
1105
203
    }
1106
164k
    AutoFile filein{OpenBlockFile({pos.nFile, pos.nPos - STORAGE_HEADER_BYTES}, /*fReadOnly=*/true)};
1107
164k
    if (filein.IsNull()) {
1108
6
        LogError("OpenBlockFile failed for %s while reading raw block", pos.ToString());
1109
6
        return util::Unexpected{ReadRawError::IO};
1110
6
    }
1111
1112
164k
    try {
1113
164k
        MessageStartChars blk_start;
1114
164k
        unsigned int blk_size;
1115
1116
164k
        filein >> blk_start >> blk_size;
1117
1118
164k
        if (blk_start != GetParams().MessageStart()) {
1119
1
            LogError("Block magic mismatch for %s: %s versus expected %s while reading raw block",
1120
1
                pos.ToString(), HexStr(blk_start), HexStr(GetParams().MessageStart()));
1121
1
            return util::Unexpected{ReadRawError::IO};
1122
1
        }
1123
1124
164k
        if (blk_size > MAX_SIZE) {
1125
0
            LogError("Block data is larger than maximum deserialization size for %s: %s versus %s while reading raw block",
1126
0
                pos.ToString(), blk_size, MAX_SIZE);
1127
0
            return util::Unexpected{ReadRawError::IO};
1128
0
        }
1129
1130
164k
        if (block_part) {
1131
39
            const auto [offset, size]{*block_part};
1132
39
            if (size == 0 || SaturatingAdd(offset, size) > blk_size) {
1133
24
                return util::Unexpected{ReadRawError::BadPartRange}; // Avoid logging - offset/size come from untrusted REST input
1134
24
            }
1135
15
            filein.seek(offset, SEEK_CUR);
1136
15
            blk_size = size;
1137
15
        }
1138
1139
164k
        std::vector<std::byte> data(blk_size); // Zeroing of memory is intentional here
1140
164k
        filein.read(data);
1141
164k
        return data;
1142
164k
    } catch (const std::exception& e) {
1143
0
        LogError("Read from block file failed: %s for %s while reading raw block", e.what(), pos.ToString());
1144
0
        return util::Unexpected{ReadRawError::IO};
1145
0
    }
1146
164k
}
1147
1148
FlatFilePos BlockManager::WriteBlock(const CBlock& block, int nHeight)
1149
109k
{
1150
109k
    AssertLockHeld(::cs_main);
1151
109k
    const unsigned int block_size{static_cast<unsigned int>(GetSerializeSize(TX_WITH_WITNESS(block)))};
1152
109k
    FlatFilePos pos{FindNextBlockPos(block_size + STORAGE_HEADER_BYTES, nHeight, block.GetBlockTime())};
1153
109k
    if (pos.IsNull()) {
1154
0
        LogError("FindNextBlockPos failed for %s while writing block", pos.ToString());
1155
0
        return FlatFilePos();
1156
0
    }
1157
109k
    AutoFile file{OpenBlockFile(pos, /*fReadOnly=*/false)};
1158
109k
    if (file.IsNull()) {
1159
0
        LogError("OpenBlockFile failed for %s while writing block", pos.ToString());
1160
0
        m_opts.notifications.fatalError(_("Failed to write block."));
1161
0
        return FlatFilePos();
1162
0
    }
1163
109k
    {
1164
109k
        BufferedWriter fileout{file};
1165
1166
        // Write index header
1167
109k
        fileout << GetParams().MessageStart() << block_size;
1168
109k
        pos.nPos += STORAGE_HEADER_BYTES;
1169
        // Write block
1170
109k
        fileout << TX_WITH_WITNESS(block);
1171
109k
    }
1172
1173
109k
    if (file.fclose() != 0) {
1174
0
        LogError("Failed to close block file %s: %s", pos.ToString(), SysErrorString(errno));
1175
0
        m_opts.notifications.fatalError(_("Failed to close file when writing block."));
1176
0
        return FlatFilePos();
1177
0
    }
1178
1179
109k
    return pos;
1180
109k
}
1181
1182
static auto InitBlocksdirXorKey(const BlockManager::Options& opts)
1183
1.22k
{
1184
    // Bytes are serialized without length indicator, so this is also the exact
1185
    // size of the XOR-key file.
1186
1.22k
    std::array<std::byte, Obfuscation::KEY_SIZE> obfuscation{};
1187
1188
    // Consider this to be the first run if the blocksdir contains only hidden
1189
    // files (those which start with a .). Checking for a fully-empty dir would
1190
    // be too aggressive as a .lock file may have already been written.
1191
1.22k
    bool first_run = true;
1192
1.82k
    for (const auto& entry : fs::directory_iterator(opts.blocks_dir)) {
1193
1.82k
        const std::string path = fs::PathToString(entry.path().filename());
1194
1.82k
        if (!entry.is_regular_file() || !path.starts_with('.')) {
1195
771
            first_run = false;
1196
771
            break;
1197
771
        }
1198
1.82k
    }
1199
1200
1.22k
    if (opts.use_xor && first_run) {
1201
        // Only use random fresh key when the boolean option is set and on the
1202
        // very first start of the program.
1203
458
        FastRandomContext{}.fillrand(obfuscation);
1204
458
    }
1205
1206
1.22k
    const fs::path xor_key_path{opts.blocks_dir / "xor.dat"};
1207
1.22k
    if (fs::exists(xor_key_path)) {
1208
        // A pre-existing xor key file has priority.
1209
769
        AutoFile xor_key_file{fsbridge::fopen(xor_key_path, "rb")};
1210
769
        xor_key_file >> obfuscation;
1211
769
    } else {
1212
        // Create initial or missing xor key file
1213
460
        AutoFile xor_key_file{fsbridge::fopen(xor_key_path,
1214
#ifdef __MINGW64__
1215
            "wb" // Temporary workaround for https://github.com/bitcoin/bitcoin/issues/30210
1216
#else
1217
460
            "wbx"
1218
460
#endif
1219
460
        )};
1220
460
        xor_key_file << obfuscation;
1221
460
        if (xor_key_file.fclose() != 0) {
1222
0
            throw std::runtime_error{strprintf("Error closing XOR key file %s: %s",
1223
0
                                               fs::PathToString(xor_key_path),
1224
0
                                               SysErrorString(errno))};
1225
0
        }
1226
460
    }
1227
    // If the user disabled the key, it must be zero.
1228
1.22k
    if (!opts.use_xor && obfuscation != decltype(obfuscation){}) {
1229
1
        throw std::runtime_error{
1230
1
            strprintf("The blocksdir XOR-key can not be disabled when a random key was already stored! "
1231
1
                      "Stored key: '%s', stored path: '%s'.",
1232
1
                      HexStr(obfuscation), fs::PathToString(xor_key_path)),
1233
1
        };
1234
1
    }
1235
1.22k
    LogInfo("Using obfuscation key for blocksdir *.dat files (%s): '%s'\n", fs::PathToString(opts.blocks_dir), HexStr(obfuscation));
1236
1.22k
    return Obfuscation{obfuscation};
1237
1.22k
}
1238
1239
BlockManager::BlockManager(const util::SignalInterrupt& interrupt, Options opts)
1240
1.22k
    : m_prune_mode{opts.prune_target > 0},
1241
1.22k
      m_obfuscation{InitBlocksdirXorKey(opts)},
1242
1.22k
      m_opts{std::move(opts)},
1243
1.22k
      m_block_file_seq{FlatFileSeq{m_opts.blocks_dir, "blk", m_opts.fast_prune ? 0x4000 /* 16kB */ : BLOCKFILE_CHUNK_SIZE}},
1244
1.22k
      m_undo_file_seq{FlatFileSeq{m_opts.blocks_dir, "rev", UNDOFILE_CHUNK_SIZE}},
1245
1.22k
      m_interrupt{interrupt}
1246
1.22k
{
1247
1.22k
    m_block_tree_db = std::make_unique<BlockTreeDB>(m_opts.block_tree_db_params);
1248
1249
1.22k
    if (m_opts.block_tree_db_params.wipe_data) {
1250
15
        m_block_tree_db->WriteReindexing(true);
1251
15
        m_blockfiles_indexed = false;
1252
        // If we're reindexing in prune mode, wipe away unusable block files and all undo data files
1253
15
        if (m_prune_mode) {
1254
2
            CleanupBlockRevFiles();
1255
2
        }
1256
15
    }
1257
1.22k
}
1258
1259
class ImportingNow
1260
{
1261
    std::atomic<bool>& m_importing;
1262
1263
public:
1264
1.02k
    ImportingNow(std::atomic<bool>& importing) : m_importing{importing}
1265
1.02k
    {
1266
1.02k
        assert(m_importing == false);
1267
1.02k
        m_importing = true;
1268
1.02k
    }
1269
    ~ImportingNow()
1270
1.02k
    {
1271
1.02k
        assert(m_importing == true);
1272
1.02k
        m_importing = false;
1273
1.02k
    }
1274
};
1275
1276
void ImportBlocks(ChainstateManager& chainman, std::span<const fs::path> import_paths)
1277
1.02k
{
1278
1.02k
    ImportingNow imp{chainman.m_blockman.m_importing};
1279
1280
    // -reindex
1281
1.02k
    if (!chainman.m_blockman.m_blockfiles_indexed) {
1282
17
        int total_files{0};
1283
34
        while (fs::exists(chainman.m_blockman.GetBlockPosFilename(FlatFilePos(total_files, 0)))) {
1284
17
            total_files++;
1285
17
        }
1286
1287
        // Map of disk positions for blocks with unknown parent (only used for reindex);
1288
        // parent hash -> child disk position, multiple children can have the same parent.
1289
17
        std::multimap<uint256, FlatFilePos> blocks_with_unknown_parent;
1290
1291
31
        for (int nFile{0}; nFile < total_files; ++nFile) {
1292
17
            FlatFilePos pos(nFile, 0);
1293
17
            AutoFile file{chainman.m_blockman.OpenBlockFile(pos, /*fReadOnly=*/true)};
1294
17
            if (file.IsNull()) {
1295
0
                break; // This error is logged in OpenBlockFile
1296
0
            }
1297
17
            LogInfo("Reindexing block file blk%05u.dat (%d%% complete)...", (unsigned int)nFile, nFile * 100 / total_files);
1298
17
            chainman.LoadExternalBlockFile(file, &pos, &blocks_with_unknown_parent);
1299
17
            if (chainman.m_interrupt) {
1300
3
                LogInfo("Interrupt requested. Exit reindexing.");
1301
3
                return;
1302
3
            }
1303
17
        }
1304
14
        WITH_LOCK(::cs_main, chainman.m_blockman.m_block_tree_db->WriteReindexing(false));
1305
14
        chainman.m_blockman.m_blockfiles_indexed = true;
1306
14
        LogInfo("Reindexing finished");
1307
        // To avoid ending up in a situation without genesis block, re-try initializing (no-op if reindexing worked):
1308
14
        (void)chainman.LoadGenesisBlock();
1309
14
    }
1310
1311
    // -loadblock=
1312
1.02k
    for (const fs::path& path : import_paths) {
1313
1
        AutoFile file{fsbridge::fopen(path, "rb")};
1314
1
        if (!file.IsNull()) {
1315
1
            LogInfo("Importing blocks file %s...", fs::PathToString(path));
1316
1
            chainman.LoadExternalBlockFile(file);
1317
1
            if (chainman.m_interrupt) {
1318
0
                LogInfo("Interrupt requested. Exit block importing.");
1319
0
                return;
1320
0
            }
1321
1
        } else {
1322
0
            LogWarning("Could not open blocks file %s", fs::PathToString(path));
1323
0
        }
1324
1
    }
1325
1326
    // scan for better chains in the block chain database, that are not yet connected in the active best chain
1327
1.02k
    if (auto result = chainman.ActivateBestChains(); !result) {
1328
0
        chainman.GetNotifications().fatalError(util::ErrorString(result));
1329
0
    }
1330
    // End scope of ImportingNow
1331
1.02k
}
1332
1333
12
std::ostream& operator<<(std::ostream& os, const BlockfileType& type) {
1334
12
    switch(type) {
1335
0
        case BlockfileType::NORMAL: os << "normal"; break;
1336
12
        case BlockfileType::ASSUMED: os << "assumed"; break;
1337
0
        default: os.setstate(std::ios_base::failbit);
1338
12
    }
1339
12
    return os;
1340
12
}
1341
1342
12
std::ostream& operator<<(std::ostream& os, const BlockfileCursor& cursor) {
1343
12
    os << strprintf("BlockfileCursor(file_num=%d, undo_height=%d)", cursor.file_num, cursor.undo_height);
1344
12
    return os;
1345
12
}
1346
} // namespace node