[Fixed] shutdown should only flush io

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "tokio-rustls"
-version = "0.1.6"
+version = "0.1.7"
 authors = ["quininer kel <[email protected]>"]
 license = "MIT/Apache-2.0"
 repository = "https://github.com/quininer/tokio-rustls"

README.md

@@ -7,11 +7,11 @@ Asynchronous TLS/SSL streams for [Tokio](https://tokio.rs/) using
 ### Basic Structure of a Client
 
 ```rust
-// ...
-
 use rustls::ClientConfig;
 use tokio_rustls::ClientConfigExt;
 
+// ...
+
 let mut config = ClientConfig::new();
 config.root_store.add_trust_anchors(&webpki_roots::ROOTS);
 let config = Arc::new(config);

examples/client.rs

@@ -78,7 +78,7 @@ fn main() {
     #[cfg(unix)]
     let stdin = File::new_nb(StdFile(stdin.lock())).unwrap()
         .into_io(&handle).unwrap();
-    
+
     #[cfg(unix)]
     let stdout = stdout();
 

examples/server.rs

@@ -63,7 +63,7 @@ fn main() {
                     let (reader, writer) = stream.split();
                     io::copy(reader, writer)
                 })
-                .map(move |(n, _, _)| println!("Echo: {} - {}", n, addr))
+                .map(move |(n, ..)| println!("Echo: {} - {}", n, addr))
                 .map_err(move |err| println!("Error: {:?} - {}", err, addr));
             handle.spawn(done);
 
@@ -72,11 +72,11 @@ fn main() {
             let done = arc_config.accept_async(stream)
                 .and_then(|stream| io::write_all(
                     stream,
-                    "HTTP/1.0 200 ok\r\n\
+                    &b"HTTP/1.0 200 ok\r\n\
                     Connection: close\r\n\
                     Content-length: 12\r\n\
                     \r\n\
-                    Hello world!".as_bytes()
+                    Hello world!"[..]
                 ))
                 .and_then(|(stream, _)| io::flush(stream))
                 .map(move |_| println!("Accept: {}", addr))

src/lib.rs

@@ -227,7 +227,7 @@ impl<S, C> io::Write for TlsStream<S, C>
         while self.session.wants_write() {
             self.session.write_tls(&mut self.io)?;
         }
-        Ok(())
+        self.io.flush()
     }
 }
 
@@ -244,7 +244,10 @@ impl<S, C> AsyncWrite for TlsStream<S, C>
 {
     fn shutdown(&mut self) -> Poll<(), io::Error> {
         self.session.send_close_notify();
-        try_nb!(io::Write::flush(self));
+        while self.session.wants_write() {
+            try_nb!(self.session.write_tls(&mut self.io));
+        }
+        try_nb!(self.io.flush());
         self.io.shutdown()
     }
 }