RPC: Limit generate command to regtest

It doesn't make sense on other networks; closes #191
muntorg · Apr 11, 2020 · 001d2dc · 001d2dc
1 parent 6b4d26a
commit 001d2dc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/rpc/mining.cpp b/src/rpc/mining.cpp
@@ -234,6 +234,9 @@ static UniValue generate(const JSONRPCRequest& request)
             + HelpExampleCli("generate", "11")
         );
 
+    if (!IsArgSet("-regtest"))
+        throw std::runtime_error("generate command only for regtest; for mainnet/testnet use setgenerate");
+
     int nGenerate = request.params[0].get_int();
     uint64_t nMaxTries = 1000000;
     if (request.params.size() > 1) {
@@ -475,6 +478,9 @@ static UniValue generatetoaddress(const JSONRPCRequest& request)
             + HelpExampleCli("generatetoaddress", "11 \"myaddress\"")
         );
 
+    if (!IsArgSet("-regtest"))
+        throw std::runtime_error("generatetoaddress command only for regtest; for mainnet/testnet use setgenerate");
+
     int nGenerate = request.params[0].get_int();
     uint64_t nMaxTries = 1000000;
     if (request.params.size() > 2) {