fix: handle socket error on address query

Banyc · Dec 19, 2024 · 575cb76 · 575cb76
1 parent 2bcaa96
commit 575cb76
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/protocol/src/stream/streams/tcp_mux.rs b/protocol/src/stream/streams/tcp_mux.rs
@@ -89,9 +89,17 @@ where
                         }
                     };
                     counter!("stream.tcp_mux.tcp.accepts").increment(1);
-                    let addr = SocketAddrPair {
-                        local_addr: stream.local_addr().unwrap(),
-                        peer_addr: stream.peer_addr().unwrap(),
+                    let addr = || -> io::Result<SocketAddrPair> {
+                        Ok(SocketAddrPair {
+                            local_addr: stream.local_addr()?,
+                            peer_addr: stream.peer_addr()?,
+                        })
+                    };
+                    let addr = match addr() {
+                        Ok(addr) => addr,
+                        Err(_) => {
+                            continue;
+                        }
                     };
                     let (r, w) = stream.into_split();
                     let (_, accepter) = spawn_mux_no_reconnection(r, w, server_mux_config(), &mut self.mux);