mirror/lighthouse-pulse
1
0
Fork 0
mirror of https://gitlab.com/pulsechaincom/lighthouse-pulse.git synced 2025-01-12 05:50:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

More stringent dialing (#2363)

Browse Source 
* More stringent dialing

* Cover cached enr dialing
This commit is contained in:
Age Manning 2021-05-26 14:21:44 +10:00 committed by GitHub
parent 5d9a1bc558
commit 55aada006f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
beacon_node/eth2_libp2p/src/peer_manager/mod.rs
View File

@ -668,18 +668,16 @@ impl<TSpec: EthSpec> PeerManager<TSpec> {
.cached_enrs() .cached_enrs()
.filter_map(|(peer_id, enr)| { .filter_map(|(peer_id, enr)| {
let peers = self.network_globals.peers.read(); let peers = self.network_globals.peers.read();
if predicate(enr) if predicate(enr) && peers.should_dial(peer_id) {
&& !peers.is_connected_or_dialing(peer_id)
&& !peers.is_banned(peer_id)
{
Some(*peer_id) Some(*peer_id)
} else { } else {
None None
} }
}) })
.collect(); .collect();
for peer in &peers_to_dial { for peer_id in &peers_to_dial {
self.dial_peer(peer); debug!(self.log, "Dialing cached ENR peer"; "peer_id" => %peer_id);
self.dial_peer(peer_id);
} }
} }
@ -698,16 +696,7 @@ impl<TSpec: EthSpec> PeerManager<TSpec> {
// we attempt a connection if this peer is a subnet peer or if the max peer count // we attempt a connection if this peer is a subnet peer or if the max peer count
// is not yet filled (including dialing peers) // is not yet filled (including dialing peers)
if (min_ttl.is_some() || connected_or_dialing + to_dial_peers.len() < self.max_peers) if (min_ttl.is_some() || connected_or_dialing + to_dial_peers.len() < self.max_peers)
&& !self && self.network_globals.peers.read().should_dial(&peer_id)
.network_globals
.peers
.read()
.is_connected_or_dialing(&peer_id)
&& !self
.network_globals
.peers
.read()
.is_banned_or_disconnected(&peer_id)
{ {
// This should be updated with the peer dialing. In fact created once the peer is // This should be updated with the peer dialing. In fact created once the peer is
// dialed // dialed

12
beacon_node/eth2_libp2p/src/peer_manager/peerdb.rs
View File

@ -153,6 +153,16 @@ impl<TSpec: EthSpec> PeerDB<TSpec> {
) )
} }
/// Returns true if the peer should be dialed. This checks the connection state and the
/// score state and determines if the peer manager should dial this peer.
pub fn should_dial(&self, peer_id: &PeerId) -> bool {
matches!(
self.connection_status(peer_id),
Some(PeerConnectionStatus::Disconnected { .. })
| Some(PeerConnectionStatus::Unknown { .. })
) && !self.is_banned_or_disconnected(peer_id)
}
/// Returns true if the peer is synced at least to our current head. /// Returns true if the peer is synced at least to our current head.
pub fn is_synced(&self, peer_id: &PeerId) -> bool { pub fn is_synced(&self, peer_id: &PeerId) -> bool {
match self.peers.get(peer_id).map(|info| &info.sync_status) { match self.peers.get(peer_id).map(|info| &info.sync_status) {
@ -341,7 +351,7 @@ impl<TSpec: EthSpec> PeerDB<TSpec> {
} }
if let Err(e) = info.dialing_peer() { if let Err(e) = info.dialing_peer() {
error!(self.log, "{}", e); error!(self.log, "{}", e; "peer_id" => %peer_id);
} }
} }