controller/utils: Properly handle short reads on socket

1 files changed, 2 insertions, 2 deletions

diff --git a/controller/src/lib.rs b/controller/src/lib.rs
index ee02b3b..b53502f 100644
--- a/controller/src/lib.rs
+++ b/controller/src/lib.rs
@@ -35,7 +35,7 @@ async fn send_vectored(
 async fn receive_message(reader: &mut (impl AsyncReadExt + Unpin))
         -> io::Result<Option<RawMessage>> {
     let mut header = [0u8; 17];
-    if let Err(e) = reader.read(&mut header).await {
+    if let Err(e) = reader.read_exact(&mut header).await {
         if e.kind() == ErrorKind::UnexpectedEof { return Ok(None); }
         else { return Err(e); }
     }
@@ -46,7 +46,7 @@ async fn receive_message(reader: &mut (impl AsyncReadExt + Unpin))
 
     let mut payload = Vec::new();
     payload.resize(length, 0u8);
-    if let Err(e) = reader.read(&mut payload).await {
+    if let Err(e) = reader.read_exact(&mut payload).await {
         if e.kind() == ErrorKind::UnexpectedEof { return Ok(None); }
         else { return Err(e); }
     }