Fix "Unexpected EOB" when reading crlf (#36)

jedireza · Mar 25, 2022 · 6f0f0f9 · 6f0f0f9
1 parent c7b20a1
commit 6f0f0f9
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/warc_reader.rs b/src/warc_reader.rs
@@ -288,9 +288,11 @@ impl<R: BufRead> StreamingIter<'_, R> {
 
         let mut crlfs = [0; 4];
 
-        match self.reader.read(&mut crlfs) {
-            Ok(4) => {}
-            Ok(_) => return Err(Error::UnexpectedEOB),
+        match self.reader.read_exact(&mut crlfs) {
+            Ok(()) => (),
+            Err(e) if e.kind() == std::io::ErrorKind::UnexpectedEof => {
+                return Err(Error::UnexpectedEOB)
+            }
             Err(io) => return Err(Error::ReadData(io)),
         }