prevent panic on invalid compressed stringtable size

fixes #2
demostf · May 22, 2021 · 4d87f5b · 4d87f5b
1 parent e2a631c
commit 4d87f5b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/demo/message/stringtable.rs b/src/demo/message/stringtable.rs
@@ -48,6 +48,12 @@ impl<'a> Parse<'a> for CreateStringTableMessage<'a> {
             let decompressed_size: u32 = table_data.read()?;
             let compressed_size: u32 = table_data.read()?;
 
+            if compressed_size < 4 {
+                return Err(ParseError::InvalidDemo(
+                    "Invalid compressed string table size",
+                ));
+            }
+
             let magic = table_data.read_string(Some(4))?;
 
             if magic != "SNAP" {