From 96e316ea16b7b915e02735457d5ac7495d3db305 Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Mon, 5 Dec 2022 20:29:05 +0100
Subject: serialization

---
 evc/src/ser.rs | 154 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 154 insertions(+)
 create mode 100644 evc/src/ser.rs

(limited to 'evc/src/ser.rs')

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) })
+    }
+}
-- 
cgit v1.2.3-70-g09d2