1 files changed, 59 insertions, 0 deletions
diff --git a/src/packet/mod.rs b/src/packet/mod.rs
new file mode 100644
index 0000000..b918dc2
--- /dev/null
+++ b/src/packet/mod.rs
@@ -0,0 +1,59 @@
+// Copyright 2022-2024 Gabriel Bjørnager Jensen.
+
+use crate::checksum::Checksum;
+use crate::deserialise::{Deserialise, DStream};
+use crate::error::Error;
+use crate::serialise::{SStream, Serialise};
+
+use std::mem::size_of;
+
+#[derive(Clone, Debug)]
+pub struct Packet<T: Deserialise + Serialise> {
+	message:  T,
+	checksum: Checksum,
+}
+
+impl<T: Deserialise + Serialise> Packet<T> {
+	#[must_use]
+	pub fn encode(message: T) -> Box<[u8]> {
+		let checksum = Checksum::new(&message);
+		let packet = Self { message, checksum };
+
+		SStream::from(&packet).into()
+	}
+
+	pub fn decode(data: &[u8]) -> Result<T, Error> {
+		if data.len() > size_of::<Self>() {
+			return Err(Error::InvalidPackageLength { len: data.len(), ok_len: size_of::<Self>() });
+		}
+
+		let mut data = DStream::from(data);
+
+		let packet = Self::deserialise(&mut data)
+			.map_err(|e| Error::DeserialiseFailure { source: e })?;
+
+		let checksum = Checksum::new(&packet.message);
+
+		if packet.checksum != checksum {
+			return Err(Error::PackageChecksumMismatch { sum: packet.checksum, ok_sum: checksum });
+		}
+
+		Ok(packet.message)
+	}
+}
+
+impl<T: Deserialise + Serialise> Deserialise for Packet<T> {
+	fn deserialise(stream: &mut DStream) -> Result<Self, Box<dyn std::error::Error>> {
+		Ok(Self {
+			message:  Deserialise::deserialise(stream)?,
+			checksum: Deserialise::deserialise(stream)?,
+		})
+	}
+}
+
+impl<T: Deserialise + Serialise> Serialise for Packet<T> {
+	fn serialise(&self, stream: &mut SStream) {
+		self.message.serialise(stream);
+		self.checksum.serialise(stream);
+	}
+}