diff options
Diffstat (limited to 'utils/src')
-rw-r--r-- | utils/src/protocol.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/utils/src/protocol.rs b/utils/src/protocol.rs index 4820f79..fce168d 100644 --- a/utils/src/protocol.rs +++ b/utils/src/protocol.rs @@ -23,7 +23,7 @@ pub struct RawMessage { impl RawMessage { pub fn receive(reader: &mut BufReader<TcpStream>) -> io::Result<Option<Self>> { let mut header = [0u8; 17]; - if let Err(e) = reader.read(&mut header) { + if let Err(e) = reader.read_exact(&mut header) { if e.kind() == ErrorKind::UnexpectedEof { return Ok(None); } else { return Err(e); } } @@ -34,7 +34,7 @@ impl RawMessage { let mut payload = Vec::new(); payload.resize(length, 0u8); - if let Err(e) = reader.read(&mut payload) { + if let Err(e) = reader.read_exact(&mut payload) { if e.kind() == ErrorKind::UnexpectedEof { return Ok(None); } else { return Err(e); } } |