1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
use byteorder::{BigEndian, ByteOrder, LittleEndian, ReadBytesExt, WriteBytesExt}; use bytes::BytesMut; use std::io; use std::io::{Cursor, Write}; pub enum IOReadErrorValue { NotEnoughBytes, IO(io::Error), } pub struct IOReadError { pub value: IOReadErrorValue, } impl From<IOReadErrorValue> for IOReadError { fn from(val: IOReadErrorValue) -> Self { IOReadError { value: val } } } impl From<io::Error> for IOReadError { fn from(error: io::Error) -> Self { IOReadError { value: IOReadErrorValue::IO(error), } } } pub struct Reader { buffer: BytesMut, } impl Reader { pub fn new(input: BytesMut) -> Reader { Reader { buffer: input } } pub fn extend_from_slice(&mut self, extend: &[u8]) { self.buffer.extend_from_slice(extend) } pub fn read_bytes(&mut self, bytes_num: usize) -> Result<BytesMut, IOReadError> { if self.buffer.len() < bytes_num { return Err(IOReadError { value: IOReadErrorValue::NotEnoughBytes, }); } Ok(self.buffer.split_to(bytes_num)) } pub fn read_bytes_cursor(&mut self, bytes_num: usize) -> Result<Cursor<BytesMut>, IOReadError> { let tmp_bytes = self.read_bytes(bytes_num)?; let tmp_cursor = Cursor::new(tmp_bytes); Ok(tmp_cursor) } pub fn read_u8(&mut self) -> Result<u8, IOReadError> { let mut cursor = self.read_bytes_cursor(1)?; Ok(cursor.read_u8()?) } pub fn read_u24<T: ByteOrder>(&mut self) -> Result<u32, IOReadError> { let mut cursor = self.read_bytes_cursor(3)?; let val = cursor.read_u24::<T>()?; Ok(val) } pub fn read_u32<T: ByteOrder>(&mut self) -> Result<u32, IOReadError> { let mut cursor = self.read_bytes_cursor(4)?; let val = cursor.read_u32::<T>()?; Ok(val) } }