refactor

src/lib.rs

@@ -75,9 +75,7 @@ impl EncryptedStream {
     pub fn read_packet(&mut self) -> Result<Vec<u8>, EncryptionError> {
         // the first bytes should be the nonce
         let mut nonce_bytes = [0u8; 12];
-        if let Err(err) = self.socket.read(&mut nonce_bytes) {
+        self.socket.read(&mut nonce_bytes).map_err(|err| IO(err))?;
-            return Err(IO(err));
-        }
 
         self.reset(nonce_bytes);
 

src/main.rs

@@ -1,7 +1,6 @@
 use hermes::EncryptedStream;
 use std::env::args;
-use std::io::Write;
+use std::net::TcpListener;
-use std::net::{Shutdown, TcpListener};
 use std::process::exit;
 
 fn main() {
@@ -27,10 +26,6 @@ fn main() {
                     )
                     .unwrap();
                 println!("successfully wrote data");
-
-                encrypted.flush().unwrap();
-                encrypted.shutdown(Shutdown::Both).unwrap();
-                println!("closed connection");
             }
         }
         "c" => {