From df07d7b043eaba1f961d22883763ae77b7f68600 Mon Sep 17 00:00:00 2001 From: Ruben Beltran del Rio Date: Mon, 27 Jan 2025 22:36:43 +0100 Subject: Centralize logging --- src/proxy.rs | 21 +++++++++++++++++---- 1 file changed, 17 insertions(+), 4 deletions(-) (limited to 'src/proxy.rs') diff --git a/src/proxy.rs b/src/proxy.rs index b8d95cd..5678db9 100644 --- a/src/proxy.rs +++ b/src/proxy.rs @@ -238,11 +238,18 @@ fn handle_client(client_stream: TcpStream, configuration: &Arc) { } } - let debug_str = String::from_utf8_lossy(&buffer[..bytes_read]) + let debug_original = String::from_utf8_lossy(&buffer[..bytes_read]) .replace('\n', "\\n") .replace('\r', "\\r") .replace('\t', "\\t"); - debug!(">>> {}", debug_str); + + let debug_final = String::from_utf8_lossy(&command) + .replace('\n', "\\n") + .replace('\r', "\\r") + .replace('\t', "\\t"); + + debug!(">>> {debug_original}"); + debug!("### {debug_final}"); // Lock the TLS stream and write the data to server match tls_stream_clone.lock() { @@ -299,11 +306,17 @@ fn handle_client(client_stream: TcpStream, configuration: &Arc) { } } - let debug_str = String::from_utf8_lossy(&buffer[..bytes_read]) + let debug_original = String::from_utf8_lossy(&buffer[..bytes_read]) + .replace('\n', "\\n") + .replace('\r', "\\r") + .replace('\t', "\\t"); + + let debug_final = String::from_utf8_lossy(&command) .replace('\n', "\\n") .replace('\r', "\\r") .replace('\t', "\\t"); - debug!("<<< {}", debug_str); + debug!("<<< {debug_original}"); + debug!("### {debug_final}"); // Write decrypted data to client if client_writer.write_all(&command).is_err() { -- cgit