Port C code changes to Rust

memorysafety · Jul 17, 2024 · ce56fd9 · ce56fd9
1 parent 82ad54c
commit ce56fd9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/tools/input/annexb.rs b/tools/input/annexb.rs
@@ -125,6 +125,9 @@ unsafe fn parse_obu_header(
     buf = buf.offset(1);
     buf_size -= 1;
     if extension_flag != 0 {
+        if buf_size == 0 {
+            return -1;
+        }
         buf = buf.offset(1);
         buf_size -= 1;
     }

diff --git a/tools/input/section5.rs b/tools/input/section5.rs
@@ -103,6 +103,9 @@ unsafe fn parse_obu_header(
     buf = buf.offset(1);
     buf_size -= 1;
     if extension_flag != 0 {
+        if buf_size == 0 {
+            return -1;
+        }
         buf = buf.offset(1);
         buf_size -= 1;
     }