Port C code changes to Rust

memorysafety · Jul 17, 2024 · c3dbde5 · c3dbde5
1 parent 82ad54c
commit c3dbde5
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
@@ -104,6 +104,9 @@ unsafe fn parse_obu_header(
     buf_size -= 1;
     if extension_flag != 0 {
         buf = buf.offset(1);
+        if buf_size == 0 {
+            return -1;
+        }
         buf_size -= 1;
     }
     if has_size_flag != 0 {