tokio-rustls: Add to README and clean code (#15)

* tokio-rustls: Add to README and clean code

* cargo fmt

README.md

@@ -5,6 +5,7 @@
 This crate contains a collection of Tokio based TLS libraries.
 
 - [`tokio-native-tls`](tokio-native-tls)
+- [`tokio-rustls`](tokio-rustls)
 
 ## Getting Help
 

tokio-rustls/src/common/handshake.rs

@@ -47,7 +47,13 @@ where
     fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
         let this = self.get_mut();
 
-        if let MidHandshake::Handshaking(mut stream) = mem::replace(this, MidHandshake::End) {
+        let mut stream =
+            if let MidHandshake::Handshaking(stream) = mem::replace(this, MidHandshake::End) {
+                stream
+            } else {
+                panic!("unexpected polling after handshake")
+            };
+
         if !stream.skip_handshake() {
             let (state, io, session) = stream.get_mut();
             let mut tls_stream = Stream::new(io, session).set_eof(!state.readable());
@@ -56,9 +62,7 @@ where
                 ( $e:expr ) => {
                     match $e {
                         Poll::Ready(Ok(_)) => (),
-                            Poll::Ready(Err(err)) => {
+                        Poll::Ready(Err(err)) => return Poll::Ready(Err((err, stream.into_io()))),
-                                return Poll::Ready(Err((err, stream.into_io())))
-                            }
                         Poll::Pending => {
                             *this = MidHandshake::Handshaking(stream);
                             return Poll::Pending;
@@ -77,8 +81,5 @@ where
         }
 
         Poll::Ready(Ok(stream))
-        } else {
-            panic!("unexpected polling after handshake")
-        }
     }
 }

tokio-rustls/src/common/mod.rs

@@ -96,17 +96,6 @@ impl<'a, IO: AsyncRead + AsyncWrite + Unpin, S: Session> Stream<'a, IO, S> {
         Pin::new(self)
     }
 
-    pub fn process_new_packets(&mut self, cx: &mut Context) -> io::Result<()> {
-        self.session.process_new_packets().map_err(|err| {
-            // In case we have an alert to send describing this error,
-            // try a last-gasp write -- but don't predate the primary
-            // error.
-            let _ = self.write_io(cx);
-
-            io::Error::new(io::ErrorKind::InvalidData, err)
-        })
-    }
-
     pub fn read_io(&mut self, cx: &mut Context) -> Poll<io::Result<usize>> {
         struct Reader<'a, 'b, T> {
             io: &'a mut T,
@@ -130,6 +119,15 @@ impl<'a, IO: AsyncRead + AsyncWrite + Unpin, S: Session> Stream<'a, IO, S> {
             Err(err) => return Poll::Ready(Err(err)),
         };
 
+        self.session.process_new_packets().map_err(|err| {
+            // In case we have an alert to send describing this error,
+            // try a last-gasp write -- but don't predate the primary
+            // error.
+            let _ = self.write_io(cx);
+
+            io::Error::new(io::ErrorKind::InvalidData, err)
+        })?;
+
         Poll::Ready(Ok(n))
     }
 
@@ -218,10 +216,7 @@ impl<'a, IO: AsyncRead + AsyncWrite + Unpin, S: Session> Stream<'a, IO, S> {
             while !self.eof && self.session.wants_read() {
                 match self.read_io(cx) {
                     Poll::Ready(Ok(0)) => self.eof = true,
-                    Poll::Ready(Ok(n)) => {
+                    Poll::Ready(Ok(n)) => rdlen += n,
-                        rdlen += n;
-                        self.process_new_packets(cx)?;
-                    }
                     Poll::Pending => {
                         read_would_block = true;
                         break;
@@ -267,7 +262,7 @@ impl<'a, IO: AsyncRead + AsyncWrite + Unpin, S: Session> AsyncRead for Stream<'a
                         self.eof = true;
                         break;
                     }
-                    Poll::Ready(Ok(_)) => self.process_new_packets(cx)?,
+                    Poll::Ready(Ok(_)) => (),
                     Poll::Pending => {
                         would_block = true;
                         break;