src/utils/read.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

use std::io::{Read, Result as IoResult};

pub trait ReadUtils {
    fn read_arr<const S: usize>(&mut self) -> IoResult<[u8; S]>;

    fn read_vec(&mut self, size: usize) -> IoResult<Vec<u8>>;
}

impl<R: Read> ReadUtils for R {
    #[inline]
    fn read_arr<const S: usize>(&mut self) -> IoResult<[u8; S]> {
        let mut arr = [0; S];
        self.read_exact(&mut arr)?;
        Ok(arr)
    }

    #[inline]
    fn read_vec(&mut self, size: usize) -> IoResult<Vec<u8>> {
        let mut vec = vec![0; size];
        self.read_exact(&mut vec)?;
        Ok(vec)
    }
}