aboutsummaryrefslogtreecommitdiff
path: root/src/net.cpp
diff options
context:
space:
mode:
authorVasil Dimov <vd@FreeBSD.org>2021-10-25 11:03:58 +0200
committerVasil Dimov <vd@FreeBSD.org>2021-11-18 13:39:10 +0100
commitf52b6b2d9f482353821da0ef4c485c402a396c8d (patch)
tree71100049a61ca06f6eebb7570d2ef765ac1f4f01 /src/net.cpp
parentc7eb19ec8302e6a5abd89c0566540c2c862e9121 (diff)
downloadbitcoin-f52b6b2d9f482353821da0ef4c485c402a396c8d.tar.xz
net: split CConnman::SocketHandler()
`CConnman::SocketHandler()` does 3 things: 1. Check sockets for readiness 2. Process ready listening sockets 3. Process ready connected sockets Split the processing (2. and 3.) into separate methods to make the code easier to grasp. Also, move the processing of listening sockets after the processing of connected sockets to make it obvious that there is no dependency and also explicitly release the snapshot before dealing with listening sockets - it is only necessary for the connected sockets part.
Diffstat (limited to 'src/net.cpp')
-rw-r--r--src/net.cpp51
1 files changed, 34 insertions, 17 deletions
diff --git a/src/net.cpp b/src/net.cpp
index 610a795c64..652bdb36b6 100644
--- a/src/net.cpp
+++ b/src/net.cpp
@@ -1502,28 +1502,33 @@ void CConnman::SocketEvents(const std::vector<CNode*>& nodes,
void CConnman::SocketHandler()
{
- const NodesSnapshot snap{*this, /*shuffle=*/false};
+ std::set<SOCKET> recv_set;
+ std::set<SOCKET> send_set;
+ std::set<SOCKET> error_set;
- std::set<SOCKET> recv_set, send_set, error_set;
- SocketEvents(snap.Nodes(), recv_set, send_set, error_set);
+ {
+ const NodesSnapshot snap{*this, /*shuffle=*/false};
- if (interruptNet) return;
+ // Check for the readiness of the already connected sockets and the
+ // listening sockets in one call ("readiness" as in poll(2) or
+ // select(2)). If none are ready, wait for a short while and return
+ // empty sets.
+ SocketEvents(snap.Nodes(), recv_set, send_set, error_set);
- //
- // Accept new connections
- //
- for (const ListenSocket& hListenSocket : vhListenSocket)
- {
- if (hListenSocket.socket != INVALID_SOCKET && recv_set.count(hListenSocket.socket) > 0)
- {
- AcceptConnection(hListenSocket);
- }
+ // Service (send/receive) each of the already connected nodes.
+ SocketHandlerConnected(snap.Nodes(), recv_set, send_set, error_set);
}
- //
- // Service each socket
- //
- for (CNode* pnode : snap.Nodes()) {
+ // Accept new connections from listening sockets.
+ SocketHandlerListening(recv_set);
+}
+
+void CConnman::SocketHandlerConnected(const std::vector<CNode*>& nodes,
+ const std::set<SOCKET>& recv_set,
+ const std::set<SOCKET>& send_set,
+ const std::set<SOCKET>& error_set)
+{
+ for (CNode* pnode : nodes) {
if (interruptNet)
return;
@@ -1607,6 +1612,18 @@ void CConnman::SocketHandler()
}
}
+void CConnman::SocketHandlerListening(const std::set<SOCKET>& recv_set)
+{
+ for (const ListenSocket& listen_socket : vhListenSocket) {
+ if (interruptNet) {
+ return;
+ }
+ if (listen_socket.socket != INVALID_SOCKET && recv_set.count(listen_socket.socket) > 0) {
+ AcceptConnection(listen_socket);
+ }
+ }
+}
+
void CConnman::ThreadSocketHandler()
{
SetSyscallSandboxPolicy(SyscallSandboxPolicy::NET);