[#52] fix(server): do not crash during unsuccessful handshake (#53)

M0dEx · Jun 6, 2024 · 33ad071 · 33ad071
1 parent c700056
commit 33ad071
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "quincy"
-version = "0.8.2"
+version = "0.8.3"
 authors = ["Jakub Kubík <[email protected]>"]
 license = "MIT"
 description = "QUIC-based VPN"

diff --git a/src/server.rs b/src/server.rs
@@ -108,12 +108,20 @@ impl QuincyServer {
             tokio::select! {
                 // New connections
                 Some(handshake) = endpoint.accept() => {
+                    let client_ip = handshake.remote_address().ip();
+
                     debug!(
                         "Received incoming connection from '{}'",
-                        handshake.remote_address().ip()
+                        client_ip
                     );
 
-                    let quic_connection = handshake.await?;
+                    let quic_connection = match handshake.await {
+                        Ok(connection) => connection,
+                        Err(e) => {
+                            warn!("Connection handshake with client '{client_ip}' failed: {e}");
+                            continue;
+                        }
+                    };
 
                     let connection = QuincyConnection::new(
                         quic_connection,