From 27f19834c2c43e303e8bcd4303afe1ef6be8d6af Mon Sep 17 00:00:00 2001
From: wcampbell <wcampbell1995@gmail.com>
Date: Thu, 3 Aug 2023 19:58:39 -0400
Subject: [PATCH] Add check

---
 src/container.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/container.rs b/src/container.rs
index a07b05bc..5c13aaa8 100644
--- a/src/container.rs
+++ b/src/container.rs
@@ -101,6 +101,9 @@ impl<R: Read> Container<R> {
     #[inline]
     pub fn read_bytes(&mut self, amt: usize, buf: &mut [u8]) -> Result<ContainerRet, DekuError> {
         if self.leftover.is_empty() {
+            if buf.len() < amt {
+                return Err(DekuError::Incomplete(NeedSize::new(amt * 8)));
+            }
             if let Err(e) = self.inner.read_exact(&mut buf[..amt]) {
                 if e.kind() == io::ErrorKind::UnexpectedEof {
                     return Err(DekuError::Incomplete(NeedSize::new(amt * 8)));