Make initial block index loading faster

firoorg · Mar 8, 2024 · 6bcd96e · 6bcd96e
1 parent 8bd4abd
commit 6bcd96e
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 2 deletions.
diff --git a/src/txdb.cpp b/src/txdb.cpp
@@ -356,6 +356,16 @@ bool CBlockTreeDB::LoadBlockIndexGuts(boost::function<CBlockIndex*(const uint256
     pcursor->Seek(std::make_pair(DB_BLOCK_INDEX, uint256()));
 
     // Load mapBlockIndex
+
+    // We need to check PoW for the last N blocks. To do so we can't just save a pointer to the last block and go back 
+    // from it because of possible forks. This multimap is used to track the most recent blocks (by height) saved in 
+    // the block index on disk
+    std::multimap<int, CBlockIndex*> lastNBlocks;
+    // lowest height of all the elements in lastNBlocks
+    int firstInLastNBlocksHeight = 0;
+
+    bool fCheckPoWForAllBlocks = GetBoolArg("-fullblockindexcheck", DEFAULT_FULL_BLOCKINDEX_CHECK);
+
     while (pcursor->Valid()) {
         boost::this_thread::interruption_point();
         std::pair<char, uint256> key;
@@ -406,8 +416,22 @@ bool CBlockTreeDB::LoadBlockIndexGuts(boost::function<CBlockIndex*(const uint256
 
                 pindexNew->activeDisablingSporks = diskindex.activeDisablingSporks;
 
-                if (!CheckProofOfWork(pindexNew->GetBlockPoWHash(), pindexNew->nBits, consensusParams))
-                    return error("LoadBlockIndex(): CheckProofOfWork failed: %s", pindexNew->ToString());
+                if (fCheckPoWForAllBlocks) {
+                    if (!CheckProofOfWork(pindexNew->GetBlockPoWHash(), pindexNew->nBits, consensusParams))
+                        return error("LoadBlockIndex(): CheckProofOfWork failed: %s", pindexNew->ToString());
+                }
+                else {
+                    if (pindexNew->nHeight >= firstInLastNBlocksHeight) {
+                        lastNBlocks.insert(std::pair<int, CBlockIndex*>(pindexNew->nHeight, pindexNew));
+                        if (lastNBlocks.size() > DEFAULT_BLOCKINDEX_NUMBER_OF_BLOCKS_TO_CHECK) {
+                            // pop the first element from the map
+                            auto firstElement = lastNBlocks.begin();
+                            auto elementToPop = firstElement++;
+                            lastNBlocks.erase(elementToPop);
+                            firstInLastNBlocksHeight = firstElement->first;
+                        }
+                    }
+                }
 
                 pcursor->Next();
             } else {
@@ -418,6 +442,14 @@ bool CBlockTreeDB::LoadBlockIndexGuts(boost::function<CBlockIndex*(const uint256
         }
     }
 
+    if (!fCheckPoWForAllBlocks) {
+        // delayed check for all the blocks
+        for (const auto &blockIndex: lastNBlocks) {
+            if (!CheckProofOfWork(blockIndex.second->GetBlockPoWHash(), blockIndex.second->nBits, consensusParams))
+                return error("LoadBlockIndex(): CheckProofOfWork failed: %s", blockIndex.second->ToString());
+        }
+    }
+
     return true;
 }
 

diff --git a/src/txdb.h b/src/txdb.h
@@ -43,6 +43,11 @@ static const int64_t nMaxBlockDBAndTxIndexCache = 1024;
 //! Max memory allocated to coin DB specific cache (MiB)
 static const int64_t nMaxCoinsDBCache = 8;
 
+//! By default don't check block index PoW on client startup
+static const bool DEFAULT_FULL_BLOCKINDEX_CHECK = false;
+//! If not doing full check of block index, check only N of the latest blocks
+static const int DEFAULT_BLOCKINDEX_NUMBER_OF_BLOCKS_TO_CHECK = 10000;
+
 struct CDiskTxPos : public CDiskBlockPos
 {
     unsigned int nTxOffset; // after header