chore: add an extra safeguard so as not to handle internal Messages t…

…hat were not actually internal Signed-off-by: ljedrz <[email protected]>
Flyman110 · Jan 13, 2021 · a09dbb7 · a09dbb7
1 parent 27a9f85
commit a09dbb7
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/network/src/lib.rs b/network/src/lib.rs
@@ -184,10 +184,14 @@ impl Server {
 
         match payload {
             Payload::ConnectingTo(remote_address, nonce) => {
-                self.peers.connecting_to_peer(remote_address, nonce)?;
+                if direction == Direction::Internal {
+                    self.peers.connecting_to_peer(remote_address, nonce)?;
+                }
             }
             Payload::ConnectedTo(remote_address, nonce) => {
-                self.peers.connected_to_peer(remote_address, nonce)?;
+                if direction == Direction::Internal {
+                    self.peers.connected_to_peer(remote_address, nonce)?;
+                }
             }
             Payload::Version(version) => {
                 self.peers.version_to_verack(&version)?;
@@ -225,7 +229,9 @@ impl Server {
                 self.blocks.received_sync(source.unwrap(), sync).await?;
             }
             Payload::Disconnect(addr) => {
-                self.peers.disconnected_from_peer(&addr)?;
+                if direction == Direction::Internal {
+                    self.peers.disconnected_from_peer(&addr)?;
+                }
             }
             Payload::GetPeers => {
                 self.peers.send_get_peers(source.unwrap());