diff --git a/src/main.rs b/src/main.rs index 1686cca..b78693d 100644 --- a/src/main.rs +++ b/src/main.rs @@ -14,7 +14,6 @@ use std::thread; use std::time::Duration; const TIMEOUT: u64 = 3 * 60 * 100; //3 minutes -static mut DEBUG: bool = false; fn print_usage(program: &str, opts: Options) { let program_path = std::path::PathBuf::from(program); @@ -52,9 +51,6 @@ fn main() { std::process::exit(-1); }); - unsafe { - DEBUG = matches.opt_present("d"); - } let local_port: i32 = matches.opt_str("l").unwrap().parse().unwrap(); let remote_port: i32 = matches.opt_str("r").unwrap().parse().unwrap(); let remote_host = matches.opt_str("h").unwrap(); @@ -73,8 +69,7 @@ fn forward(bind_addr: &str, local_port: i32, remote_host: &str, remote_port: i32 let remote_addr = format!("{}:{}", remote_host, remote_port); - let responder = local - .try_clone() + let responder = local.try_clone() .unwrap_or_else(|_| panic!("Failed to clone primary listening address socket {}", local.local_addr().unwrap())); let (main_sender, main_receiver) = channel::<(_, Vec)>(); thread::spawn(move || { @@ -82,8 +77,7 @@ fn forward(bind_addr: &str, local_port: i32, remote_host: &str, remote_port: i32 loop { let (dest, buf) = main_receiver.recv().unwrap(); let to_send = buf.as_slice(); - responder - .send_to(to_send, dest) + responder.send_to(to_send, dest) .unwrap_or_else(|_| panic!("Failed to forward response from upstream server to client {}", dest)); } }); @@ -171,9 +165,7 @@ fn forward(bind_addr: &str, local_port: i32, remote_host: &str, remote_port: i32 let to_send = buf[..num_bytes].to_vec(); match sender.send(to_send) { - Ok(_) => { - break; - } + Ok(_) => break, Err(_) => { if !ignore_failure { panic!("Failed to send message to datagram forwarder for client {}", client_id); @@ -187,4 +179,3 @@ fn forward(bind_addr: &str, local_port: i32, remote_host: &str, remote_port: i32 } } } -