diff options
Diffstat (limited to 'src/codec')
| -rw-r--r-- | src/codec/decode.rs | 88 | ||||
| -rw-r--r-- | src/codec/encode.rs | 55 | ||||
| -rw-r--r-- | src/codec/error.rs | 22 |
3 files changed, 160 insertions, 5 deletions
diff --git a/src/codec/decode.rs b/src/codec/decode.rs index 65f08b5..1059811 100644 --- a/src/codec/decode.rs +++ b/src/codec/decode.rs @@ -1,7 +1,91 @@ -use std::io::{self, Read}; +use std::{io::Read, sync::Arc}; + +use crate::DEFAULT_BUFFER_LEN; + +macro_rules! impl_decode { + ($type: ty) => { + impl Decode for $type { + fn decode(reader: &mut dyn Read) -> crate::codec::error::Result<Self> + where + Self: Sized, + { + const BYTES: usize = (<$type>::BITS / 8) as usize; + let mut bytes = [0; BYTES]; + reader.read_exact(&mut bytes)?; + Ok(<$type>::from_be_bytes(bytes)) + } + } + }; +} pub trait Decode { - fn decode(reader: &impl Read) -> io::Result<Self> + fn decode(reader: &mut dyn Read) -> crate::codec::error::Result<Self> where Self: Sized; } + +impl Decode for bool { + fn decode(reader: &mut dyn Read) -> crate::codec::error::Result<Self> + where + Self: Sized, + { + let mut byte = [0_u8]; + reader.read_exact(&mut byte)?; + Ok(byte[0] != 0) + } +} + +impl Decode for Arc<[u8]> { + fn decode(reader: &mut dyn Read) -> crate::codec::error::Result<Self> { + let mut buf = Vec::with_capacity(DEFAULT_BUFFER_LEN); + reader.read_to_end(&mut buf)?; + Ok(Arc::<[u8]>::from(buf.into_boxed_slice())) + } +} + +impl<const S: usize> Decode for [u8; S] { + fn decode(reader: &mut dyn Read) -> crate::codec::error::Result<Self> + where + Self: Sized, + { + let mut slice = [0_u8; S]; + reader.read_exact(&mut slice)?; + Ok(slice) + } +} + +impl_decode!(u8); +impl_decode!(u16); +impl_decode!(u32); +impl_decode!(u64); +impl_decode!(u128); +impl_decode!(usize); + +impl_decode!(i8); +impl_decode!(i16); +impl_decode!(i32); +impl_decode!(i64); +impl_decode!(i128); +impl_decode!(isize); + +impl Decode for f64 { + fn decode(reader: &mut dyn Read) -> crate::codec::error::Result<Self> + where + Self: Sized, + { + let bits = u64::decode(reader)?; + let value = f64::from_bits(bits); + Ok(value) + } +} + +impl Decode for f32 { + fn decode(reader: &mut dyn Read) -> crate::codec::error::Result<Self> + where + Self: Sized, + { + let bits = u32::decode(reader)?; + let value = f32::from_bits(bits); + Ok(value) + } +} diff --git a/src/codec/encode.rs b/src/codec/encode.rs index f23913b..a40f551 100644 --- a/src/codec/encode.rs +++ b/src/codec/encode.rs @@ -1,6 +1,55 @@ -use std::io::{self, Write}; +use crate::codec::error::Error; +use std::{io::Write, sync::Arc}; + +macro_rules! impl_encode { + ($t: ty) => { + impl Encode for $t { + fn encode(self, buffer: &mut dyn Write) -> Result<usize, Error> { + let size = buffer.write(&self.to_be_bytes())?; + Ok(size) + } + } + }; +} pub trait Encode { - fn encode(&self, write: &mut dyn Write) -> io::Result<usize>; - fn encoded_size(&self) -> Option<usize>; + fn encode(self, buffer: &mut dyn Write) -> Result<usize, Error>; +} + +impl Encode for bool { + fn encode(self, buffer: &mut dyn Write) -> Result<usize, Error> { + let size = buffer.write(&[self as u8])?; + Ok(size) + } +} + +impl Encode for Arc<[u8]> { + fn encode(self, buffer: &mut dyn Write) -> Result<usize, Error> { + let len = buffer.write(&self)?; + Ok(len) + } } + +impl<const S: usize> Encode for [u8; S] { + fn encode(self, buffer: &mut dyn Write) -> Result<usize, Error> { + let len = buffer.write(&self)?; + Ok(len) + } +} + +impl_encode!(u8); +impl_encode!(u16); +impl_encode!(u32); +impl_encode!(u64); +impl_encode!(u128); +impl_encode!(usize); + +impl_encode!(i8); +impl_encode!(i16); +impl_encode!(i32); +impl_encode!(i64); +impl_encode!(i128); +impl_encode!(isize); + +impl_encode!(f32); +impl_encode!(f64); diff --git a/src/codec/error.rs b/src/codec/error.rs new file mode 100644 index 0000000..6639c98 --- /dev/null +++ b/src/codec/error.rs @@ -0,0 +1,22 @@ +use std::fmt::Display; + +#[derive(Debug)] +pub enum Error { + IoError(std::io::Error), +} + +impl std::error::Error for Error {} + +impl Display for Error { + fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { + write!(f, "{self:?}") + } +} + +pub type Result<T> = core::result::Result<T, Error>; + +impl From<std::io::Error> for Error { + fn from(value: std::io::Error) -> Self { + Self::IoError(value) + } +} |
