fix(fuzz): fix failing fuzzing after pdu encode/decode refactoring

Devolutions · Mar 21, 2024 · fa00ae7 · fa00ae7
1 parent 4da3643
commit fa00ae7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/crates/ironrdp-pdu/src/ber.rs b/crates/ironrdp-pdu/src/ber.rs
@@ -133,6 +133,7 @@ pub(crate) fn read_enumerated(stream: &mut ReadCursor<'_>, count: u8) -> PduResu
         return Err(invalid_message_err!("len", "invalid enumerated len"));
     }
 
+    ensure_size!(in: stream, size: 1);
     let enumerated = stream.read_u8();
     if enumerated == u8::MAX || enumerated + 1 > count {
         return Err(invalid_message_err!("enumerated", "invalid enumerated value"));