[RPC] setnetworkactive

spencerlievens · spencerlievens · commit e65aff602226 · 2018-01-06T20:29:23.000+01:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -8,6 +8,10 @@
 * Fix Boost 1.66 Compatibility
 * Net Overhaul and BTC Inlining
 * Bump Versions/Protocol
+* Update Tests
+* util: Add ParseUInt32 and ParseUInt64
+* [RPC] getmempoolancestors/getmempooldescendants
+* [RPC] setnetworkactive
 * Update CHANGELOG
 
 **Dynamic v2.2.0.0**
diff --git a/src/rpcclient.cpp b/src/rpcclient.cpp
@@ -128,6 +128,7 @@ static const CRPCConvertParam vRPCConvertParams[] =
     { "setban", 3 },
     { "getmempoolancestors", 1 },
     { "getmempooldescendants", 1 },
+    { "setnetworkactive", 0 },
     { "spork", 1 },
     { "voteraw", 1 },
     { "voteraw", 5 },
diff --git a/src/rpcnet.cpp b/src/rpcnet.cpp
@@ -577,6 +577,24 @@ UniValue clearbanned(const UniValue& params, bool fHelp)
     return NullUniValue;
 }
 
+UniValue setnetworkactive(const UniValue& params, bool fHelp)
+{
+    if (fHelp || params.size() != 1) {
+        throw std::runtime_error(
+            "setnetworkactive true|false\n"
+            "Disable/enable all p2p network activity."
+        );
+    }
+
+    if (!g_connman) {
+        throw JSONRPCError(RPC_CLIENT_P2P_DISABLED, "Error: Peer-to-peer functionality missing or disabled");
+    }
+
+    g_connman->SetNetworkActive(params[0].get_bool());
+
+    return g_connman->GetNetworkActive();
+}
+
 UniValue ntptime(const UniValue& params, bool fHelp)       
 {
     if (fHelp || params.size() > 1)
@@ -618,6 +636,7 @@ static const CRPCCommand commands[] =
     { "network",            "setban",                 &setban,                 true  },
     { "network",            "listbanned",             &listbanned,             true  },
     { "network",            "clearbanned",            &clearbanned,            true  },
+    { "network",            "setnetworkactive",       &setnetworkactive,       true, },
 };
 
 void RegisterNetRPCCommands(CRPCTable &tableRPC)
diff --git a/src/test/rpc_tests.cpp b/src/test/rpc_tests.cpp
@@ -90,6 +90,28 @@ BOOST_AUTO_TEST_CASE(rpc_rawparams)
     BOOST_CHECK_THROW(CallRPC(std::string("sendrawtransaction ")+rawtx+" extra"), std::runtime_error);
 }
 
+BOOST_AUTO_TEST_CASE(rpc_togglenetwork)
+{
+    UniValue r;
+
+    r = CallRPC("getnetworkinfo");
+    bool netState = find_value(r.get_obj(), "networkactive").get_bool();
+    BOOST_CHECK_EQUAL(netState, true);
+
+    BOOST_CHECK_NO_THROW(CallRPC("setnetworkactive false"));
+    r = CallRPC("getnetworkinfo");
+    int numConnection = find_value(r.get_obj(), "connections").get_int();
+    BOOST_CHECK_EQUAL(numConnection, 0);
+
+    netState = find_value(r.get_obj(), "networkactive").get_bool();
+    BOOST_CHECK_EQUAL(netState, false);
+
+    BOOST_CHECK_NO_THROW(CallRPC("setnetworkactive true"));
+    r = CallRPC("getnetworkinfo");
+    netState = find_value(r.get_obj(), "networkactive").get_bool();
+    BOOST_CHECK_EQUAL(netState, true);
+}
+
 BOOST_AUTO_TEST_CASE(rpc_rawsign)
 {
     UniValue r;
