Merge pull request #72 from AIBlockOfficial/bugfix_peer_limit_handshake

Bugfix peer limit handshake
AIBlockOfficial · May 21, 2024 · 46ec8e4 · 46ec8e4
2 parents 2c99a97 + 14145f3
commit 46ec8e4
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/comms_handler/node.rs b/src/comms_handler/node.rs
@@ -1059,7 +1059,15 @@ impl Node {
             peer.peer_type = Some(peer_type);
 
             if let Some(notify) = peer.notify_handshake_response.0.take() {
-                notify.send(()).unwrap();
+                match notify.send(()) {
+                    Ok(()) => {}
+                    Err(_) => {
+                        return Err(CommsError::PeerInvalidState(PeerInfo {
+                            node_type: Some(peer_type),
+                            address: Some(peer_addr),
+                        }));
+                    }
+                }
             }
         }