AcceptBlock() - not accept blocks from a forked chain that exceed the…

… max reorg limit
PIVX-Project · Jan 30, 2019 · 4219339 · 4219339
1 parent 47759e1
commit 4219339
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/main.cpp b/src/main.cpp
@@ -4560,15 +4560,22 @@ bool AcceptBlock(CBlock& block, CValidationState& state, CBlockIndex** ppindex,
             const bool hasPIVInputs = !pivInputs.empty();
             const bool hasZPIVInputs = !zPIVInputs.empty();
 
+            int readBlock = 0;
             vector<CBigNum> vBlockSerials;
             CBlock bl;
             // Go backwards on the forked chain up to the split
             do {
+                // Check if the forked chain is longer than the max reorg limit
+                if(readBlock == Params().MaxReorganizationDepth()){
+                    // TODO: Remove this chain from disk.
+                    return error("%s: forked chain longer than maximum reorg limit", __func__);
+                }
+
                 if(!ReadBlockFromDisk(bl, prev))
                     // Previous block not on disk
                     return error("%s: previous block %s not on disk", __func__, prev->GetBlockHash().GetHex());
-
-
+                // Increase amount of read blocks
+                readBlock++;
                 // Loop through every input from said block
                 for (CTransaction t : bl.vtx) {
                     for (CTxIn in: t.vin) {