Skip to content

Commit 66673f1

Browse files
committed
net: fix race condition in self-connect detection
Initiating an outbound network connection currently involves the following steps after the socket connection is established (see `CConnman::OpenNetworkConnection` method): 1. set up node state 2. queue VERSION message 3. add new node to vector `m_nodes` If we connect to ourself, it can happen that the sent VERSION message (step 2) is received and processed locally *before* the node object is added to the connection manager's `m_nodes` vector (step 3). In this case, the self-connect remains undiscovered, as the detection doesn't find the outbound peer in `m_nodes` yet (see `CConnman::CheckIncomingNonce`). Fix this by swapping the order of 2. and 3., by taking the `PushNodeVersion` call out of `InitializeNode` and doing that in the `SendMessages` method instead, which is only called for `CNode` instances in `m_nodes`. Thanks go to vasild, mzumsande, dergoegge and sipa for suggestions on how to fix this.
1 parent 1c11089 commit 66673f1

File tree

3 files changed

+16
-5
lines changed

3 files changed

+16
-5
lines changed

src/net.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -991,7 +991,7 @@ class NetEventsInterface
991991
/** Mutex for anything that is only accessed via the msg processing thread */
992992
static Mutex g_msgproc_mutex;
993993

994-
/** Initialize a peer (setup state, queue any initial messages) */
994+
/** Initialize a peer (setup state) */
995995
virtual void InitializeNode(CNode& node, ServiceFlags our_services) = 0;
996996

997997
/** Handle removal of a peer (clear state) */

src/net_processing.cpp

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -243,6 +243,9 @@ struct Peer {
243243
* Most peers use headers-first syncing, which doesn't use this mechanism */
244244
uint256 m_continuation_block GUARDED_BY(m_block_inv_mutex) {};
245245

246+
/** Set to true once initial VERSION message was sent (only relevant for outbound peers). */
247+
bool m_outbound_version_message_sent GUARDED_BY(NetEventsInterface::g_msgproc_mutex){false};
248+
246249
/** This peer's reported block height when we connected */
247250
std::atomic<int> m_starting_height{-1};
248251

@@ -1677,9 +1680,6 @@ void PeerManagerImpl::InitializeNode(CNode& node, ServiceFlags our_services)
16771680
LOCK(m_peer_mutex);
16781681
m_peer_map.emplace_hint(m_peer_map.end(), nodeid, peer);
16791682
}
1680-
if (!node.IsInboundConn()) {
1681-
PushNodeVersion(node, *peer);
1682-
}
16831683
}
16841684

16851685
void PeerManagerImpl::ReattemptInitialBroadcast(CScheduler& scheduler)
@@ -5326,6 +5326,10 @@ bool PeerManagerImpl::ProcessMessages(CNode* pfrom, std::atomic<bool>& interrupt
53265326
PeerRef peer = GetPeerRef(pfrom->GetId());
53275327
if (peer == nullptr) return false;
53285328

5329+
// For outbound connections, ensure that the initial VERSION message
5330+
// has been sent first before processing any incoming messages
5331+
if (!pfrom->IsInboundConn() && !peer->m_outbound_version_message_sent) return false;
5332+
53295333
{
53305334
LOCK(peer->m_getdata_requests_mutex);
53315335
if (!peer->m_getdata_requests.empty()) {
@@ -5817,6 +5821,12 @@ bool PeerManagerImpl::SendMessages(CNode* pto)
58175821
// disconnect misbehaving peers even before the version handshake is complete.
58185822
if (MaybeDiscourageAndDisconnect(*pto, *peer)) return true;
58195823

5824+
// Initiate version handshake for outbound connections
5825+
if (!pto->IsInboundConn() && !peer->m_outbound_version_message_sent) {
5826+
PushNodeVersion(*pto, *peer);
5827+
peer->m_outbound_version_message_sent = true;
5828+
}
5829+
58205830
// Don't send anything until the version handshake is complete
58215831
if (!pto->fSuccessfullyConnected || pto->fDisconnect)
58225832
return true;

src/test/util/net.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,8 @@ void ConnmanTestMsg::Handshake(CNode& node,
2828
auto& connman{*this};
2929

3030
peerman.InitializeNode(node, local_services);
31-
FlushSendBuffer(node); // Drop the version message added by InitializeNode.
31+
peerman.SendMessages(&node);
32+
FlushSendBuffer(node); // Drop the version message added by SendMessages.
3233

3334
CSerializedNetMsg msg_version{
3435
NetMsg::Make(NetMsgType::VERSION,

0 commit comments

Comments
 (0)