From ba17cec5e157a2c0314d8a47fb06128d062d5ad3 Mon Sep 17 00:00:00 2001
From: Klesti Selimaj <klesti.selimay@icloud.com>
Date: Fri, 28 Aug 2026 14:16:58 +0200
Subject: [PATCH] Improved a few things

---
 src/crypto.rs | 2 +-
 src/server.rs | 9 ++++++---
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/crypto.rs b/src/crypto.rs
index 31cdbed..b1dd985 100644
--- a/src/crypto.rs
+++ b/src/crypto.rs
@@ -151,7 +151,7 @@ pub async fn crypto_handshake(
         .transpose()?
         .ok_or(anyhow::anyhow!("Failed to get client x key"))?
     else {
-        return Err(anyhow::anyhow!(""));
+        return Err(anyhow::anyhow!("WS message is not binary"));
     };
 
     let client_pubkey = X25519Public::from(*raw_pubkey.as_array().ok_or(anyhow::anyhow!(
diff --git a/src/server.rs b/src/server.rs
index 8436298..083d01f 100644
--- a/src/server.rs
+++ b/src/server.rs
@@ -74,9 +74,12 @@ impl Server {
         let s = self.clone();
 
         ws.on_upgrade(move |socket: WebSocket| async move {
-            let Ok(client) = crate::crypto::crypto_handshake(&s, socket).await else {
-                eprintln!("Filed to initialize crypto");
-                return;
+            let client = match crate::crypto::crypto_handshake(&s, socket).await {
+                Ok(client) => client,
+                Err(err) => {
+                    eprintln!("Failed to initialize crypto: {err}");
+                    return;
+                }
             };
 
             match UserConnections::initialize(&s, &client).await {