From b4098b955858dbe47cd84ffedcdec4e1e9cc4742 Mon Sep 17 00:00:00 2001
From: Matthias Seitz <matthias.seitz@outlook.de>
Date: Thu, 17 Nov 2022 15:48:50 +0100
Subject: [PATCH] chore(net): add disconnect reason (#222)

---
 crates/net/network/src/manager.rs | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/crates/net/network/src/manager.rs b/crates/net/network/src/manager.rs
index d73a4b213f..7ea95e9ff9 100644
--- a/crates/net/network/src/manager.rs
+++ b/crates/net/network/src/manager.rs
@@ -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);
         }