From 1eedfba81c5b73fe1d0697cca08038a41a3b921d Mon Sep 17 00:00:00 2001
From: Pierre Hubert <pierre.git@communiquons.org>
Date: Tue, 30 Aug 2022 14:19:30 +0200
Subject: [PATCH] Stopped infinite loop

---
 tcp_relay_client/src/relay_client.rs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tcp_relay_client/src/relay_client.rs b/tcp_relay_client/src/relay_client.rs
index d8a7e74..02d0da9 100644
--- a/tcp_relay_client/src/relay_client.rs
+++ b/tcp_relay_client/src/relay_client.rs
@@ -41,9 +41,15 @@ async fn relay_connection(ws_url: String, socket: TcpStream) {
         loop {
             match tcp_read.read(&mut buff).await {
                 Ok(s) => {
+                    if s == 0 {
+                        log::info!("Got empty read TCP buffer. Stopping...");
+                        break;
+                    }
+
                     if let Err(e) = ws_write.send(Message::Binary(Vec::from(&buff[0..s]))).await {
                         log::error!(
                             "Failed to write to WS connection! {:?} Exiting TCP read -> WS write loop...",e);
+                        break;
                     }
                 }
                 Err(e) => {