Enforce correct port on mainnet for DIP3 MNs (#2576)

dashpay · Dec 21, 2018 · 03a6865 · 03a6865
1 parent 3f26ed7
commit 03a6865
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/evo/providertx.cpp b/src/evo/providertx.cpp
@@ -27,6 +27,15 @@ static bool CheckService(const uint256& proTxHash, const ProTx& proTx, CValidati
         return state.DoS(10, false, REJECT_INVALID, "bad-protx-addr");
     }
 
+    int mainnetDefaultPort = Params(CBaseChainParams::MAIN).GetDefaultPort();
+    if (Params().NetworkIDString() == CBaseChainParams::MAIN) {
+        if (proTx.addr.GetPort() != mainnetDefaultPort) {
+            return state.DoS(10, false, REJECT_INVALID, "bad-protx-addr-port");
+        }
+    } else if (proTx.addr.GetPort() == mainnetDefaultPort) {
+        return state.DoS(10, false, REJECT_INVALID, "bad-protx-addr-port");
+    }
+
     if (!proTx.addr.IsIPv4()) {
         return state.DoS(10, false, REJECT_INVALID, "bad-protx-addr");
     }