use deku::prelude::*;
use std::io::{self, Cursor, Read, Seek, SeekFrom};
struct ReaderCrc<R: Read + Seek> {
reader: R,
pub cache: Vec<u8>,
}
impl<R: Read + Seek> ReaderCrc<R> {
pub fn new(reader: R) -> Self {
Self {
reader,
cache: vec![],
}
}
}
impl<R: Read + Seek> Read for ReaderCrc<R> {
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
let n = self.reader.read(buf);
self.cache.extend_from_slice(buf);
n
}
}
impl<R: Read + Seek> Seek for ReaderCrc<R> {
fn seek(&mut self, pos: SeekFrom) -> std::io::Result<u64> {
self.reader.seek(pos)
}
}
#[derive(Debug, DekuRead)]
pub struct DekuStruct {
pub a: u8,
pub b: u8,
}
fn main() {
let data = vec![0x01, 0x02];
let input = Cursor::new(&data);
let mut reader = ReaderCrc::new(input);
let (_, s) = DekuStruct::from_reader((&mut reader, 0)).unwrap();
assert_eq!(reader.cache, data);
assert_eq!(s.a, 1);
assert_eq!(s.b, 2);
}