From 5ba895c1dbda283acf91f9360e28522f6393757d Mon Sep 17 00:00:00 2001
From: lilydjwg <lilydjwg@gmail.com>
Date: Sun, 2 Aug 2015 14:44:27 +0800
Subject: [PATCH] close client connections on error

---
 src/relay/tcprelay/local.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/relay/tcprelay/local.rs b/src/relay/tcprelay/local.rs
index d84bdb16..3fef283c 100644
--- a/src/relay/tcprelay/local.rs
+++ b/src/relay/tcprelay/local.rs
@@ -240,11 +240,13 @@ impl TcpRelayLocal {
                                 Ok(0) => {
                                     error!("Unexpected EOF while reading initialize vector");
                                     debug!("Already read: {:?}", &iv[..total_len]);
+                                    let _ = local_writer.get_ref().shutdown(Shutdown::Both);
                                     return;
                                 },
                                 Ok(n) => total_len += n,
                                 Err(err) => {
                                     error!("Error while reading initialize vector: {:?}", err);
+                                    let _ = local_writer.get_ref().shutdown(Shutdown::Both);
                                     return;
                                 }
                             }