Add -stopatheight for benchmarking

sipa · sipa · commit b297426c9670 · 2017-04-27T14:39:27.000-07:00
diff --git a/src/init.cpp b/src/init.cpp
@@ -436,6 +436,8 @@ std::string HelpMessage(HelpMessageMode mode)
         strUsage += HelpMessageOpt("-dropmessagestest=<n>", "Randomly drop 1 of every <n> network messages");
         strUsage += HelpMessageOpt("-fuzzmessagestest=<n>", "Randomly fuzz 1 of every <n> network messages");
         strUsage += HelpMessageOpt("-stopafterblockimport", strprintf("Stop running after importing blocks from disk (default: %u)", DEFAULT_STOPAFTERBLOCKIMPORT));
+        strUsage += HelpMessageOpt("-stopatheight", strprintf("Stop running after reaching the given height in the main chain (default: %u)", DEFAULT_STOPATHEIGHT));
+
         strUsage += HelpMessageOpt("-limitancestorcount=<n>", strprintf("Do not accept transactions if number of in-mempool ancestors is <n> or more (default: %u)", DEFAULT_ANCESTOR_LIMIT));
         strUsage += HelpMessageOpt("-limitancestorsize=<n>", strprintf("Do not accept transactions whose size with all in-mempool ancestors exceeds <n> kilobytes (default: %u)", DEFAULT_ANCESTOR_SIZE_LIMIT));
         strUsage += HelpMessageOpt("-limitdescendantcount=<n>", strprintf("Do not accept transactions if any ancestor would have <n> or more in-mempool descendants (default: %u)", DEFAULT_DESCENDANT_LIMIT));
diff --git a/src/validation.cpp b/src/validation.cpp
@@ -2537,6 +2537,9 @@ bool ActivateBestChain(CValidationState &state, const CChainParams& chainparams,
         return false;
     }
 
+    int nStopAtHeight = GetArg("-stopatheight", DEFAULT_STOPATHEIGHT);
+    if (nStopAtHeight && pindexNewTip->nHeight >= nStopAtHeight) StartShutdown();
+
     return true;
 }
 
diff --git a/src/validation.h b/src/validation.h
@@ -145,6 +145,9 @@ static const int MAX_UNCONNECTING_HEADERS = 10;
 
 static const bool DEFAULT_PEERBLOOMFILTERS = true;
 
+/** Default for -stopatheight */
+static const int DEFAULT_STOPATHEIGHT = 0;
+
 struct BlockHasher
 {
     size_t operator()(const uint256& hash) const { return hash.GetCheapHash(); }

Original file line number	Diff line number	Diff line change
`@@ -2537,6 +2537,9 @@ bool ActivateBestChain(CValidationState &state, const CChainParams& chainparams,`
`2537`	`2537`	`return false;`
`2538`	`2538`	`}`
`2539`	`2539`
	`2540`	`+ int nStopAtHeight = GetArg("-stopatheight", DEFAULT_STOPATHEIGHT);`
	`2541`	`+ if (nStopAtHeight && pindexNewTip->nHeight >= nStopAtHeight) StartShutdown();`
	`2542`	`+`
`2540`	`2543`	`return true;`
`2541`	`2544`	`}`
`2542`	`2545`
Original file line number	Diff line number	Diff line change
`@@ -145,6 +145,9 @@ static const int MAX_UNCONNECTING_HEADERS = 10;`
`145`	`145`
`146`	`146`	`static const bool DEFAULT_PEERBLOOMFILTERS = true;`
`147`	`147`
	`148`	`+/** Default for -stopatheight */`
	`149`	`+static const int DEFAULT_STOPATHEIGHT = 0;`
	`150`	`+`
`148`	`151`	`struct BlockHasher`
`149`	`152`	`{`
`150`	`153`	`size_t operator()(const uint256& hash) const { return hash.GetCheapHash(); }`