net: Add a simple function for waking the message handler

theuni · Fuzzbawls · commit 7e55dbf629c8 · 2020-09-02T00:41:59.000-07:00
This may be used publicly in the future
diff --git a/src/net.cpp b/src/net.cpp
@@ -1244,8 +1244,8 @@ void CConnman::ThreadSocketHandler()
                             if (!pnode->ReceiveMsgBytes(pchBuf, nBytes, notify))
                                 pnode->CloseSocketDisconnect();
                             RecordBytesRecv(nBytes);
-                            if(notify)
-                                condMsgProc.notify_one();
+                            if (notify)
+                                WakeMessageHandler();
                         } else if (nBytes == 0) {
                             // socket closed gracefully
                             if (!pnode->fDisconnect)
@@ -1304,6 +1304,11 @@ void CConnman::ThreadSocketHandler()
     }
 }
 
+void CConnman::WakeMessageHandler()
+{
+    condMsgProc.notify_one();
+}
+
 
 #ifdef USE_UPNP
 void ThreadMapPort()
diff --git a/src/net.h b/src/net.h
@@ -302,6 +302,8 @@ class CConnman
     void ThreadSocketHandler();
     void ThreadDNSAddressSeed();
 
+    void WakeMessageHandler();
+
     CNode* FindNode(const CNetAddr& ip);
     CNode* FindNode(const CSubNet& subNet);
     CNode* FindNode(const std::string& addrName);
