From b810441ff5a88e8d9e136022bab9553aaa78f782 Mon Sep 17 00:00:00 2001 From: "Y. T. Chung" Date: Thu, 9 Jul 2015 20:25:27 +0800 Subject: [PATCH] opt code --- src/relay/tcprelay/local.rs | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/src/relay/tcprelay/local.rs b/src/relay/tcprelay/local.rs index 3a2ca9ca..d84bdb16 100644 --- a/src/relay/tcprelay/local.rs +++ b/src/relay/tcprelay/local.rs @@ -167,15 +167,10 @@ impl TcpRelayLocal { let header = socks5::TcpResponseHeader::new(socks5::Reply::Succeeded, socks5::Address::SocketAddress(sockname)); trace!("Send header to client {:?}", header); - if let Err(err) = header.write_to(&mut local_writer) { + if let Err(err) = header.write_to(&mut local_writer).and(local_writer.flush()) { error!("Error occurs while writing header to local stream: {:?}", err); return; } - - if let Err(err) = local_writer.flush() { - error!("Error occurs while flushing local writer: {:?}", err); - return; - } } // Send initialize vector to remote and create encryptor