chore: add more error info

Itsusinn · Jan 23, 2025 · 2385876 · 2385876
1 parent 2a96237
commit 2385876
Show file tree

Hide file tree

Showing 4 changed files with 52 additions and 7 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/tuic-server/Cargo.toml b/tuic-server/Cargo.toml
@@ -56,7 +56,7 @@ humantime-serde = "1"
 # Logging
 time = { version = "0.3", features = ["macros", "local-offset"] }
 humantime = { version = "2", default-features = false }
-tracing-subscriber = { version = "0.3", default-features = false, features = ["tracing-log", "std", "local-time","fmt"] }
+tracing-subscriber = { version = "0.3", default-features = false, features = ["tracing-log", "std", "local-time","fmt", "ansi"] }
 chrono = "0.4"
 tracing = "0.1"
 

diff --git a/tuic-server/src/connection/handle_stream.rs b/tuic-server/src/connection/handle_stream.rs
@@ -68,7 +68,9 @@ impl Connection {
                     addr = self.inner.remote_address(),
                     user = self.auth,
                 );
-                self.close();
+                self.close(&format!(
+                    "A serious error occurred at unidirectional stream pre-process stage {err}"
+                ));
             }
         }
     }
@@ -118,7 +120,9 @@ impl Connection {
                     addr = self.inner.remote_address(),
                     user = self.auth,
                 );
-                self.close();
+                self.close(&format!(
+                    "A serious error occurred at bidirectional stream pre-process stage {err}"
+                ));
             }
         }
     }
@@ -159,7 +163,9 @@ impl Connection {
                     addr = self.inner.remote_address(),
                     user = self.auth,
                 );
-                self.close();
+                self.close(&format!(
+                    "A serious error occurred at datagram pre-process stage {err}"
+                ));
             }
         }
     }

diff --git a/tuic-server/src/connection/mod.rs b/tuic-server/src/connection/mod.rs
@@ -158,7 +158,7 @@ impl Connection {
                     id = self.id(),
                     addr = self.inner.remote_address(),
                 );
-                self.close();
+                self.close("Authentication error");
             }
         }
     }
@@ -192,7 +192,7 @@ impl Connection {
         self.inner.close_reason().is_some()
     }
 
-    fn close(&self) {
-        self.inner.close(ERROR_CODE, &[]);
+    fn close(&self, reason: &str) {
+        self.inner.close(ERROR_CODE, reason.as_bytes());
     }
 }