From 4c389282e6ff9e0dbab516b861cf627a027da9eb Mon Sep 17 00:00:00 2001
From: Carl Lerche <me@carllerche.com>
Date: Thu, 6 Jul 2017 15:03:27 -0700
Subject: [PATCH] Fix bug with server reading preface

---
 src/server.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/server.rs b/src/server.rs
index ce2b257..4eb3e6c 100644
--- a/src/server.rs
+++ b/src/server.rs
@@ -80,7 +80,7 @@ impl<T: AsyncRead> Future for ReadPreface<T> {
 
     fn poll(&mut self) -> Poll<T, Self::Error> {
         let mut buf = [0; 24];
-        let rem = PREFACE.len() - self.pos;
+        let mut rem = PREFACE.len() - self.pos;
 
         while rem > 0 {
             let n = try_nb!(self.inner.as_mut().unwrap().read(&mut buf[..rem]));
@@ -91,6 +91,7 @@ impl<T: AsyncRead> Future for ReadPreface<T> {
             }
 
             self.pos += n;
+            rem -= n; // TODO test
         }
 
         Ok(Async::Ready(self.inner.take().unwrap()))