netsync: don't add peer if we're not current and the peer is not a sync

candidate Since we can use all the peers we could get for ibd, don't add peers that are not sync candidates when we're still not current.
btcsuite · Dec 9, 2024 · d5b78aa · d5b78aa
1 parent 6bae737
commit d5b78aa
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/netsync/manager.go b/netsync/manager.go
@@ -590,10 +590,17 @@ func (sm *SyncManager) handleNewPeerMsg(peer *peerpkg.Peer) {
 		return
 	}
 
-	log.Infof("New valid peer %s (%s)", peer, peer.UserAgent())
-
 	// Initialize the peer state.
 	isSyncCandidate := sm.isSyncCandidate(peer)
+	if !sm.current() && !isSyncCandidate {
+		log.Infof("New peer %s (%s) is not a valid sync candidate -- disconnecting",
+			peer, peer.UserAgent())
+		peer.Disconnect()
+		return
+	}
+
+	log.Infof("New valid peer %s (%s)", peer, peer.UserAgent())
+
 	sm.peerStates[peer] = &peerSyncState{
 		syncCandidate:   isSyncCandidate,
 		requestedTxns:   make(map[chainhash.Hash]struct{}),