character selection

1 files changed, 5 insertions, 2 deletions

diff --git a/server/src/main.rs b/server/src/main.rs
index c4df346e..45ee39dd 100644
--- a/server/src/main.rs
+++ b/server/src/main.rs
@@ -11,9 +11,9 @@ use tokio::{
 };
 use tokio_tungstenite::tungstenite::Message;
 use undercooked::{
+    data::build_gamedata,
     game::Game,
     protocol::{PacketC, PacketS},
-    data::build_gamedata,
 };
 
 #[tokio::main]
@@ -94,7 +94,10 @@ async fn main() -> Result<()> {
             }
             r = ws_listener.accept() => {
                 let (sock, addr) = r?;
-                let sock = tokio_tungstenite::accept_async(sock).await?;
+                let Ok(sock) = tokio_tungstenite::accept_async(sock).await else {
+                    warn!("invalid ws handshake");
+                    continue
+                };
                 let (mut write, mut read) = sock.split();
                 let game = game.clone();
                 let mut rx = rx.resubscribe();