From c3dbde599b23d9e81692a0febc87d278c5210d91 Mon Sep 17 00:00:00 2001
From: Frank Bossen <fbossen@gmail.com>
Date: Wed, 17 Jul 2024 12:15:03 -0400
Subject: [PATCH] Port C code changes to Rust

---
 tools/input/annexb.rs   | 3 +++
 tools/input/section5.rs | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/tools/input/annexb.rs b/tools/input/annexb.rs
index 8b8943959..7a41bd5f9 100644
--- 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
index a0b2fdda0..afae4adf2 100644
--- 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 {