diff options
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/.gitignore | 1 | ||||
| -rw-r--r-- | utils/src/protocol.rs | 4 |
2 files changed, 3 insertions, 2 deletions
diff --git a/utils/.gitignore b/utils/.gitignore new file mode 100644 index 0000000..ea8c4bf --- /dev/null +++ b/utils/.gitignore @@ -0,0 +1 @@ +/target 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); } } |