From b02d72887afd4498b03cdd767ca46676fb150622 Mon Sep 17 00:00:00 2001
From: jvoisin
Date: Sun, 6 May 2018 21:58:31 +0200
Subject: Test for faulty files, and document how MAT2 is behaving wrt. them

---
 src/images.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src/images.py')

diff --git a/src/images.py b/src/images.py
index 7c1abaa..6cc3dfe 100644
--- a/src/images.py
+++ b/src/images.py
@@ -20,6 +20,13 @@ class PNGParser(abstract.AbstractParser):
             'Compression', 'Filter', 'Interlace', 'BackgroundColor', 'ImageSize',
             'Megapixels', 'ImageHeight'}
 
+    def __init__(self, filename):
+        super().__init__(filename)
+        try:  # better fail here than later
+            cairo.ImageSurface.create_from_png(self.filename)
+        except MemoryError:
+            raise ValueError
+
     def get_meta(self):
         out = subprocess.check_output(['/usr/bin/exiftool', '-json', self.filename])
         meta = json.loads(out.decode('utf-8'))[0]
-- 
cgit v1.3