1 files changed, 5 insertions, 1 deletions
diff --git a/src/protocol.rs b/src/protocol.rs
index 6e70dbb..35681e1 100644
--- a/src/protocol.rs
+++ b/src/protocol.rs
@@ -1,5 +1,5 @@
 
-use std::net::{TcpStream, ToSocketAddrs};
+use std::net::{TcpStream, ToSocketAddrs, Shutdown};
 use std::time::Duration;
 use std::io::{Read, Write};
 use std::cmp;
@@ -395,4 +395,8 @@ impl DatConnection {
         trace!("\twas: {:?}", reg);
         Ok(reg)
     }
+
+    pub fn close(&mut self) {
+        self.tcp.shutdown(Shutdown::Both);
+    }
 }