fixed regtest+ds issues

2015-02-02 11:10:17 -07:00 · 2015-02-02 11:10:17 -07:00 · 1f828f45ec
commit 1f828f45ec
parent 2992907d91
1 changed files with 10 additions and 3 deletions
--- a/src/net.cpp
+++ b/src/net.cpp
@ -434,9 +434,16 @@ CNode* FindNode(std::string addrName)
 CNode* FindNode(const CService& addr)
 {
    LOCK(cs_vNodes);
-    BOOST_FOREACH(CNode* pnode, vNodes)
-        if(pnode->addr == addr)
-            return (pnode);
+    BOOST_FOREACH(CNode* pnode, vNodes){
+        if(RegTest()){
+            //if using regtest, just check the IP
+            if((CNetAddr)pnode->addr == (CNetAddr)addr)
+                return (pnode);
+        } else {
+            if(pnode->addr == addr)
+                return (pnode);
+        }
+    }
    return NULL;
 }