WIP

tcp_relay_server/src/relay_ws.rs

@@ -36,6 +36,11 @@ impl Actor for RelayWS {
             loop {
                 match read_half.read(&mut buff).await {
                     Ok(l) => {
+                        if l == 0 {
+                            log::info!("Got empty read. Closing read end...");
+                            return;
+                        }
+
                         let to_send = DataForWebSocket(Vec::from(&buff[0..l]));
                         if let Err(e) = addr.send(to_send).await {
                             log::error!("Failed to send to websocket. Stopping now... {:?}", e);
@@ -69,6 +74,11 @@ impl StreamHandler<Result<ws::Message, ws::ProtocolError>> for RelayWS {
                     log::error!("Failed to forward some data, closing connection! {:?}", e);
                     ctx.stop();
                 }
+
+                if data.is_empty() {
+                    log::info!("Got empty binary message. Closing websocket...");
+                    ctx.stop();
+                }
             }
             _ => (),
         }