github/reth
1
1
Fork 0
mirror of https://github.com/paradigmxyz/reth.git synced 2026-01-29 17:18:08 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(net): add disconnect reason (#222)

Browse Source
This commit is contained in:
Matthias Seitz
2022-11-17 15:48:50 +01:00
committed by GitHub
parent 84ef65dfb7
commit b4098b9558
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
crates/net/network/src/manager.rs
View File

@@ -31,7 +31,10 @@ use crate::{
};
use futures::{Future, StreamExt};
use parking_lot::Mutex;
use reth_eth_wire::capability::{Capabilities, CapabilityMessage};
use reth_eth_wire::{
capability::{Capabilities, CapabilityMessage},
DisconnectReason,
};
use reth_interfaces::provider::BlockProvider;
use reth_primitives::PeerId;
use std::{
@@ -237,7 +240,9 @@ where
// reject message in POS
if self.handle.mode().is_stake() {
// connections to peers which send invalid messages should be terminated
self.swarm.sessions_mut().disconnect(peer_id, None);
self.swarm
.sessions_mut()
.disconnect(peer_id, Some(DisconnectReason::SubprotocolSpecific));
} else {
only_pow(self);
}