Fix comparison tool by asking for blocks more aggressively

1 files changed, 6 insertions, 1 deletions

diff --git a/src/main.h b/src/main.h
index 76de47071e..2a67747b47 100644
--- a/src/main.h
+++ b/src/main.h
@@ -946,13 +946,15 @@ private:
         MODE_ERROR,   // run-time error
     } mode;
     int nDoS;
+    bool corruptionPossible;
 public:
     CValidationState() : mode(MODE_VALID), nDoS(0) {}
-    bool DoS(int level, bool ret = false) {
+    bool DoS(int level, bool ret = false, bool corruptionIn = false) {
         if (mode == MODE_ERROR)
             return ret;
         nDoS += level;
         mode = MODE_INVALID;
+        corruptionPossible = corruptionIn;
         return ret;
     }
     bool Invalid(bool ret = false) {
@@ -982,6 +984,9 @@ public:
         }
         return false;
     }
+    bool CorruptionPossible() {
+        return corruptionPossible;
+    }
 };
 
 /** An in-memory indexed chain of blocks. */