Ensure disconnecting peers are added to the peerdb (#2451)
This commit is contained in:
parent
059d9ec1b1
commit
381befbf82
@ -319,7 +319,7 @@ impl<TSpec: EthSpec> PeerManager<TSpec> {
|
||||
self.network_globals
|
||||
.peers
|
||||
.write()
|
||||
.notify_disconnecting(&peer_id, true);
|
||||
.notify_disconnecting(peer_id, true);
|
||||
return;
|
||||
}
|
||||
|
||||
@ -339,7 +339,7 @@ impl<TSpec: EthSpec> PeerManager<TSpec> {
|
||||
self.network_globals
|
||||
.peers
|
||||
.write()
|
||||
.notify_disconnecting(&peer_id, false);
|
||||
.notify_disconnecting(peer_id, false);
|
||||
return;
|
||||
}
|
||||
|
||||
@ -1023,7 +1023,7 @@ impl<TSpec: EthSpec> PeerManager<TSpec> {
|
||||
|
||||
let mut peer_db = self.network_globals.peers.write();
|
||||
for peer_id in disconnecting_peers {
|
||||
peer_db.notify_disconnecting(&peer_id, false);
|
||||
peer_db.notify_disconnecting(peer_id, false);
|
||||
self.events.push(PeerManagerEvent::DisconnectPeer(
|
||||
peer_id,
|
||||
GoodbyeReason::TooManyPeers,
|
||||
|
@ -475,10 +475,11 @@ impl<TSpec: EthSpec> PeerDB<TSpec> {
|
||||
|
||||
/// Notifies the peer manager that the peer is undergoing a normal disconnect. Optionally tag
|
||||
/// the peer to be banned after the disconnect.
|
||||
pub fn notify_disconnecting(&mut self, peer_id: &PeerId, to_ban_afterwards: bool) {
|
||||
if let Some(info) = self.peers.get_mut(peer_id) {
|
||||
info.disconnecting(to_ban_afterwards);
|
||||
}
|
||||
pub fn notify_disconnecting(&mut self, peer_id: PeerId, to_ban_afterwards: bool) {
|
||||
self.peers
|
||||
.entry(peer_id)
|
||||
.or_default()
|
||||
.disconnecting(to_ban_afterwards);
|
||||
}
|
||||
|
||||
/// Marks a peer to be disconnected and then banned.
|
||||
|
Loading…
Reference in New Issue
Block a user