From 327a23d82d6eb56afeac141257684df6d31ee769 Mon Sep 17 00:00:00 2001
From: moparisthebest <admin@moparisthebest.com>
Date: Tue, 17 Dec 2019 01:33:42 -0500
Subject: [PATCH] Put verbose byte count output behind verbose feature

---
 Cargo.toml | 1 +
 src/lib.rs | 6 +++---
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index ca8a516..7d04a58 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -23,6 +23,7 @@ include = [
 default = []
 tls = ["openssl"]
 openssl_vendored = ["openssl/vendored"]
+verbose = []
 
 [dependencies]
 openssl = { version = "0.10.26", optional = true }
diff --git a/src/lib.rs b/src/lib.rs
index 30e3bbf..ce39edb 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -111,14 +111,12 @@ impl<T: Write + Read + TryClone<T> + Send + 'static> TcpUdpPipe<T> {
     }
 
     fn send_udp(&mut self, len: usize) -> Result<()> {
+        #[cfg(feature = "verbose")]
         println!("udp got len: {}", len);
 
         self.buf[0] = ((len >> 8) & 0xFF) as u8;
         self.buf[1] = (len & 0xFF) as u8;
 
-        //let test_len = ((self.buf[0] as usize) << 8) + self.buf[1] as usize;
-        //println!("tcp sending test_len: {}", test_len);
-
         Ok(self.tcp_stream.write_all(&self.buf[..len + 2])?)
         // todo: do this? self.tcp_stream.flush()
     }
@@ -126,8 +124,10 @@ impl<T: Write + Read + TryClone<T> + Send + 'static> TcpUdpPipe<T> {
     pub fn tcp_to_udp(&mut self) -> Result<usize> {
         self.tcp_stream.read_exact(&mut self.buf[..2])?;
         let len = ((self.buf[0] as usize) << 8) + self.buf[1] as usize;
+        #[cfg(feature = "verbose")]
         println!("tcp expecting len: {}", len);
         self.tcp_stream.read_exact(&mut self.buf[..len])?;
+        #[cfg(feature = "verbose")]
         println!("tcp got len: {}", len);
         Ok(self.udp_socket.send(&self.buf[..len])?)