From 15540a34c27679cdddaaf8acc2f35498fee7a3be Mon Sep 17 00:00:00 2001
From: Toralf Wittner <tw@dtex.org>
Date: Thu, 17 May 2018 15:29:27 +0200
Subject: [PATCH] Add missing check for read of 0 bytes. (#194)

Forgot this one in PR #178.
---
 mplex/src/read.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/mplex/src/read.rs b/mplex/src/read.rs
index b7b4df13..47459f6f 100644
--- a/mplex/src/read.rs
+++ b/mplex/src/read.rs
@@ -101,6 +101,7 @@ fn block_on_wrong_stream<T: AsyncRead, Buf: Array<Item = u8>>(
     });
 
     let mut out_consumed = 0;
+    let mut stream_eof = false;
     if let Some((tasks, cache)) = lock.open_streams
         .entry(substream_id)
         .or_insert_with(|| SubstreamMetadata::new_open())
@@ -130,6 +131,10 @@ fn block_on_wrong_stream<T: AsyncRead, Buf: Array<Item = u8>>(
 
             match lock.stream.read(buf_prefix) {
                 Ok(consumed) => {
+                    if consumed == 0 && !buf_prefix.is_empty() {
+                        stream_eof = true
+                    }
+
                     let new_remaining = remaining_bytes - consumed;
 
                     assert!(cache.extend_from_slice(&buf_prefix[..consumed]));
@@ -158,6 +163,10 @@ fn block_on_wrong_stream<T: AsyncRead, Buf: Array<Item = u8>>(
         }
     }
 
+    if stream_eof {
+        lock.close()
+    }
+
     Ok(out_consumed)
 }