Reject messages with zero tags

int08h · Mar 20, 2018 · 1b21bbc · 1b21bbc
1 parent 1ce57a1
commit 1b21bbc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/error.rs b/src/error.rs
@@ -34,6 +34,9 @@ pub enum Error {
     /// Offset within message was not 32-bit aligned 
     InvalidOffsetAlignment(u32),
 
+    /// Invalid number of tags specified 
+    InvalidNumTags(u32),
+
     /// Otherwise invalid request
     InvalidRequest,
 }

diff --git a/src/message.rs b/src/message.rs
@@ -47,6 +47,10 @@ impl RtMessage {
         let mut msg = Cursor::new(bytes);
 
         let num_tags = msg.read_u32::<LittleEndian>()?;
+        if num_tags == 0 {
+            return Err(Error::InvalidNumTags(0));
+        }
+
         let mut rt_msg = RtMessage::new(num_tags);
 
         if num_tags == 1 {