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 Future for ReadPreface { fn poll(&mut self) -> Poll { 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 Future for ReadPreface { } self.pos += n; + rem -= n; // TODO test } Ok(Async::Ready(self.inner.take().unwrap()))