get rid of unwrap

paradigmxyz · Jun 25, 2024 · ce008f3 · ce008f3
1 parent bb04772
commit ce008f3
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/crates/net/network/src/session/mod.rs b/crates/net/network/src/session/mod.rs
@@ -1020,7 +1020,17 @@ async fn authenticate_stream(
         }
 
         let (multiplex_stream, their_status) =
-            multiplex_stream.into_eth_satellite_stream(status, fork_filter).await.unwrap();
+            match multiplex_stream.into_eth_satellite_stream(status, fork_filter).await {
+                Ok((multiplex_stream, their_status)) => (multiplex_stream, their_status),
+                Err(err) => {
+                    return PendingSessionEvent::Disconnected {
+                        remote_addr,
+                        session_id,
+                        direction,
+                        error: Some(PendingSessionHandshakeError::Eth(err)),
+                    }
+                }
+            };
 
         (multiplex_stream.into(), their_status)
     };