Add a BufRead implementation for CrcReader

Byron · Feb 7, 2016 · bb5d7c9 · bb5d7c9
1 parent 98874c3
commit bb5d7c9
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/crc.rs b/src/crc.rs
@@ -65,3 +65,15 @@ impl<R: Read> Read for CrcReader<R> {
         Ok(amt)
     }
 }
+
+impl<R: BufRead> BufRead for CrcReader<R> {
+    fn fill_buf(&mut self) -> io::Result<&[u8]> {
+        self.inner.fill_buf()
+    }
+    fn consume(&mut self, amt: usize) {
+        if let Ok(data) = self.inner.fill_buf() {
+            self.crc.update(&data[..amt]);
+        }
+        self.inner.consume(amt);
+    }
+}