0 len check (#61)

Co-authored-by: mikemiles-dev <[email protected]>

Author: mikemiles-dev

Cargo.toml

@@ -1,7 +1,7 @@
 [package]
 name = "netflow_parser"
 description = "Parser for Netflow Cisco V5, V7, V9, IPFIX"
-version = "0.3.0"
+version = "0.3.1"
 edition = "2021"
 author = "[email protected]"
 license = "MIT OR Apache-2.0"

RELEASES.md

@@ -1,3 +1,6 @@
+# 0.3.1
+  * Added 0 length check when parsing template lengths.
+
 # 0.3.0
   * Reworked IPFIX + V9 Parsing.  Flowset length is now used.
   * Flow data field Counts are now correctly calculated.

SECURITY.md

@@ -4,6 +4,7 @@
 
 | Version | Supported          |
 | ------- | ------------------ |
+| 0.3.1   | :white_check_mark: |
 | 0.3.0   | :white_check_mark: |
 | 0.2.9   | :white_check_mark: |
 | 0.2.8   | :white_check_mark: |

src/variable_versions/ipfix.rs

@@ -287,11 +287,15 @@ fn parse_fields<'a, T: CommonTemplate>(
     let mut fields = vec![];
     let mut remaining = i;
 
-    let count: usize = i.len()
-        / template_fields
-            .iter()
-            .map(|m| m.field_length as usize)
-            .sum::<usize>();
+    let total_size = template_fields
+        .iter()
+        .map(|m| m.field_length as usize)
+        .sum::<usize>();
+
+    if total_size == 0 {
+        return Ok((&[], fields));
+    }
+    let count: usize = i.len() / total_size;
 
     let mut error = false;
 

src/variable_versions/v9.rs

@@ -370,6 +370,10 @@ fn parse_fields<'a>(
     };
     let mut remaining = i;
 
+    if template.get_total_size() == 0 {
+        return Ok((&[], fields));
+    }
+
     let count = i.len() as u16 / template.get_total_size();
 
     for _ in 0..count {