Add graceful error handling for less than 2 bytes packet (#34)

altugbakan · Dec 7, 2024 · 9d0debe · 9d0debe
1 parent bd92df2
commit 9d0debe
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/packet.rs b/src/packet.rs
@@ -58,6 +58,9 @@ pub enum Packet {
 impl Packet {
     /// Deserializes a [`u8`] slice into a [`Packet`].
     pub fn deserialize(buf: &[u8]) -> Result<Packet, Box<dyn Error>> {
+        if buf.len() < 2 {
+            return Err("Buffer too short to serialize".into());
+        }
         let opcode = Opcode::from_u16(Convert::to_u16(&buf[0..=1])?)?;
 
         match opcode {