Improve the resilience of MAT2 wrt. corrupted PNG

1 files changed, 4 insertions, 0 deletions

diff --git a/libmat2/images.py b/libmat2/images.py
index 837f015..03121e1 100644
--- a/libmat2/images.py
+++ b/libmat2/images.py
@@ -62,6 +62,10 @@ class PNGParser(_ImageParser):
 
     def __init__(self, filename):
         super().__init__(filename)
+
+        if imghdr.what(filename) != 'png':
+            raise ValueError
+
         try:  # better fail here than later
             cairo.ImageSurface.create_from_png(self.filename)
         except MemoryError: