From 39accf7bd3d0d2cb7e783f48ed11824e524d30a2 Mon Sep 17 00:00:00 2001
From: Revertron <r@revertron.com>
Date: Thu, 1 Apr 2021 15:43:19 +0200
Subject: [PATCH] Small fix in p2p.

---
 src/p2p/network.rs | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/p2p/network.rs b/src/p2p/network.rs
index d198c63..70b7435 100644
--- a/src/p2p/network.rs
+++ b/src/p2p/network.rs
@@ -171,9 +171,17 @@ fn handle_connection_event(context: Arc<Mutex<Context>>, peers: &mut Peers, regi
 
     if event.is_readable() {
         let data = {
-            let peer = peers.get_mut_peer(&event.token()).expect("Error getting peer for connection");
-            let mut stream = peer.get_stream();
-            read_message(&mut stream)
+            let token = event.token();
+            match peers.get_mut_peer(&token) {
+                None => {
+                    error!("Error getting peer for connection {}", token.0);
+                    return false;
+                }
+                Some(peer) => {
+                    let mut stream = peer.get_stream();
+                    read_message(&mut stream)
+                }
+            }
         };
 
         if data.is_ok() {