From 2e6e7efdbf28e74f9ca22b3d4e70185f8f6ca118 Mon Sep 17 00:00:00 2001
From: Paul Lecuq <paul@paulbsd.com>
Date: Tue, 26 Dec 2023 11:11:38 +0100
Subject: [PATCH] hotfix on ws connections

---
 src/websocket.rs | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/src/websocket.rs b/src/websocket.rs
index c5a733f..d23aab7 100644
--- a/src/websocket.rs
+++ b/src/websocket.rs
@@ -50,11 +50,16 @@ pub async fn websocketpubsub(
                             continue;
                         }
                     };
-                    if tosend.ipdata.clone().unwrap().hostname != gethostname(true)
-                        || tosend.msgtype == "init".to_string()
-                    {
-                        let txps = txpubsub.read().await;
-                        txps.send(tosend).await.unwrap();
+                    match tosend.ipdata.clone() {
+                        Some(o) => {
+                            if o.hostname != gethostname(true)
+                                || tosend.msgtype == "init".to_string()
+                            {
+                                let txps = txpubsub.read().await;
+                                txps.send(tosend).await.unwrap();
+                            }
+                        }
+                        None => {}
                     }
                 }
                 Err(e) => {
@@ -111,7 +116,9 @@ pub async fn send_to_ipbl_websocket(
             }
         };
     } else {
+        println!("can't write to socket");
         ws.close(None).unwrap_or(());
+        sleep_s(1);
         return false;
     };
 
@@ -125,6 +132,8 @@ pub async fn send_to_ipbl_websocket(
             }
         };
     } else {
+        println!("can't read from socket");
+        sleep_s(1);
         ws.close(None).unwrap_or(());
         return false;
     };