Get receiving data working

2017-08-07 12:48:50 -07:00
parent 71acfe3961
commit 6053ee059d
5 changed files with 86 additions and 39 deletions
--- a/src/proto/streams/recv.rs
+++ b/src/proto/streams/recv.rs
@@ -236,8 +236,12 @@ impl<P, B> Recv<P, B>
            .take_while(&mut self.buffer, |frame| frame.is_data());

        if frames.is_empty() {
-            stream.recv_task = Some(task::current());
-            Ok(Async::NotReady)
+            if stream.state.is_recv_closed() {
+                Ok(None.into())
+            } else {
+                stream.recv_task = Some(task::current());
+                Ok(Async::NotReady)
+            }
        } else {
            Ok(Some(Chunk {
                pending_recv: frames,