diff options
Diffstat (limited to 'evc/src/ser.rs')
-rw-r--r-- | evc/src/ser.rs | 154 |
1 files changed, 154 insertions, 0 deletions
diff --git a/evc/src/ser.rs b/evc/src/ser.rs new file mode 100644 index 0000000..05072b2 --- /dev/null +++ b/evc/src/ser.rs @@ -0,0 +1,154 @@ +use std::io::{self, Read, Write}; + +pub trait Sink { + fn put<V: Ser>(&mut self, value: V) -> io::Result<()>; +} +pub trait Source { + fn get<V: Ser>(&mut self) -> io::Result<V>; +} + +impl<T: Write> Sink for T { + fn put<V: Ser>(&mut self, value: V) -> io::Result<()> { + value.write(self) + } +} +impl<T: Read> Source for T { + fn get<V: Ser>(&mut self) -> io::Result<V> { + V::read(self) + } +} + +pub trait Ser: Sized { + fn write(&self, sink: &mut impl Write) -> io::Result<()>; + fn read(source: &mut impl Read) -> io::Result<Self>; +} + +impl<A: Ser, B: Ser> Ser for (A, B) { + fn write(&self, sink: &mut impl Write) -> io::Result<()> { + self.0.write(sink)?; + self.1.write(sink)?; + Ok(()) + } + + fn read(source: &mut impl Read) -> io::Result<Self> { + Ok((A::read(source)?, B::read(source)?)) + } +} +impl<A: Ser, B: Ser, C: Ser> Ser for (A, B, C) { + fn write(&self, sink: &mut impl Write) -> io::Result<()> { + self.0.write(sink)?; + self.1.write(sink)?; + self.2.write(sink)?; + Ok(()) + } + + fn read(source: &mut impl Read) -> io::Result<Self> { + Ok((A::read(source)?, B::read(source)?, C::read(source)?)) + } +} + +impl<A: Ser, const N: usize> Ser for [A; N] { + fn write(&self, sink: &mut impl Write) -> io::Result<()> { + for e in self { + e.write(sink)?; + } + Ok(()) + } + + fn read(source: &mut impl Read) -> io::Result<Self> { + let mut k: [A; N] = unsafe { std::mem::zeroed() }; + for i in 0..N { + k[i] = A::read(source)?; + } + Ok(k) + } +} + +impl<T: Ser> Ser for Vec<T> { + fn write(&self, sink: &mut impl Write) -> io::Result<()> { + self.len().write(sink)?; + for e in self { + e.write(sink)?; + } + Ok(()) + } + + fn read(source: &mut impl Read) -> io::Result<Self> { + let mut v = vec![]; + for _ in 0..usize::read(source)? { + v.push(T::read(source)?) + } + Ok(v) + } +} + +impl Ser for u8 { + fn write(&self, sink: &mut impl Write) -> io::Result<()> { + sink.write_all(&[*self]) + } + fn read(source: &mut impl Read) -> io::Result<Self> { + let mut buf = [0u8; 1]; + source.read_exact(&mut buf)?; + Ok(buf[0]) + } +} +impl Ser for u16 { + fn write(&self, sink: &mut impl Write) -> io::Result<()> { + sink.write_all(&unsafe { std::mem::transmute_copy::<_, [u8; 2]>(self) }) + } + fn read(source: &mut impl Read) -> io::Result<Self> { + let mut buf = [0u8; 2]; + source.read_exact(&mut buf)?; + Ok(unsafe { std::mem::transmute_copy(&buf) }) + } +} +impl Ser for u32 { + fn write(&self, sink: &mut impl Write) -> io::Result<()> { + sink.write_all(&unsafe { std::mem::transmute_copy::<_, [u8; 4]>(self) }) + } + fn read(source: &mut impl Read) -> io::Result<Self> { + let mut buf = [0u8; 4]; + source.read_exact(&mut buf)?; + Ok(unsafe { std::mem::transmute_copy(&buf) }) + } +} +impl Ser for u64 { + fn write(&self, sink: &mut impl Write) -> io::Result<()> { + sink.write_all(&unsafe { std::mem::transmute_copy::<_, [u8; 8]>(self) }) + } + fn read(source: &mut impl Read) -> io::Result<Self> { + let mut buf = [0u8; 8]; + source.read_exact(&mut buf)?; + Ok(unsafe { std::mem::transmute_copy(&buf) }) + } +} +impl Ser for usize { + fn write(&self, sink: &mut impl Write) -> io::Result<()> { + sink.write_all(&unsafe { std::mem::transmute_copy::<_, [u8; 8]>(self) }) + } + fn read(source: &mut impl Read) -> io::Result<Self> { + let mut buf = [0u8; 8]; + source.read_exact(&mut buf)?; + Ok(unsafe { std::mem::transmute_copy(&buf) }) + } +} +impl Ser for f32 { + fn write(&self, sink: &mut impl Write) -> io::Result<()> { + sink.write_all(&unsafe { std::mem::transmute_copy::<_, [u8; 4]>(self) }) + } + fn read(source: &mut impl Read) -> io::Result<Self> { + let mut buf = [0u8; 4]; + source.read_exact(&mut buf)?; + Ok(unsafe { std::mem::transmute_copy(&buf) }) + } +} +impl Ser for f64 { + fn write(&self, sink: &mut impl Write) -> io::Result<()> { + sink.write_all(&unsafe { std::mem::transmute_copy::<_, [u8; 8]>(self) }) + } + fn read(source: &mut impl Read) -> io::Result<Self> { + let mut buf = [0u8; 8]; + source.read_exact(&mut buf)?; + Ok(unsafe { std::mem::transmute_copy(&buf) }) + } +} |