From 5f5e911d10b7a24e2e8e716f7d6f585b81743a68 Mon Sep 17 00:00:00 2001
From: Rob McMullen <feedback@playermissile.com>
Date: Mon, 16 Jul 2018 07:03:47 -0700
Subject: [PATCH] Added check for incomplete file in container unpacking

---
 atrcopy/container.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/atrcopy/container.py b/atrcopy/container.py
index 499dfe2..dbb8f66 100644
--- a/atrcopy/container.py
+++ b/atrcopy/container.py
@@ -23,7 +23,10 @@ class DiskImageContainer:
 
     def __unpack_raw_data(self, data):
         raw = data.tobytes()
-        unpacked = self.unpack_bytes(raw)
+        try:
+            unpacked = self.unpack_bytes(raw)
+        except EOFError as e:
+            raise errors.InvalidContainer(e)
         return to_numpy(unpacked)
 
     def unpack_bytes(self, byte_data):