From f2ee34ffc2b82e2f70b58d5b8766e17296df59d7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marc-Andr=C3=A9=20Lureau?= <marcandre.lureau@redhat.com>
Date: Thu, 28 Mar 2024 21:22:28 +0400
Subject: [PATCH] feat(server): add some debug statements
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Remove 'main label too.

Signed-off-by: Marc-André Lureau <marcandre.lureau@redhat.com>
---
 crates/ironrdp-server/src/server.rs | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/crates/ironrdp-server/src/server.rs b/crates/ironrdp-server/src/server.rs
index 818d39ae..8146bbc7 100644
--- a/crates/ironrdp-server/src/server.rs
+++ b/crates/ironrdp-server/src/server.rs
@@ -252,6 +252,7 @@ impl RdpServer {
 
         self.static_channels = result.static_channels;
         for (_type_id, channel, channel_id) in self.static_channels.iter_mut() {
+            debug!(?channel, ?channel_id, "Start");
             let Some(channel_id) = channel_id else {
                 continue;
             };
@@ -313,10 +314,11 @@ impl RdpServer {
 
         let mut display_updates = self.display.updates().await?;
 
-        'main: loop {
+        loop {
             tokio::select! {
                 frame = framed.read_pdu() => {
                     let Ok((action, bytes)) = frame else {
+                        debug!(?frame, "disconnecting");
                         break;
                     };
 
@@ -330,7 +332,8 @@ impl RdpServer {
                             match self.handle_x224(&mut framed, result.io_channel_id, result.user_channel_id, &bytes).await {
                                 Ok(disconnect) => {
                                     if disconnect {
-                                        break 'main;
+                                        debug!("Got disconnect request");
+                                        break;
                                     }
                                 },
 
@@ -363,6 +366,8 @@ impl RdpServer {
             }
         }
 
+        debug!("End of client loop");
+
         Ok(())
     }