Match all 8 bytes of PNG signature

h2non · Sep 24, 2024 · 6553f0f · 6553f0f
1 parent 0c7f219
commit 6553f0f
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/filetype/types/image.py b/filetype/types/image.py
@@ -133,12 +133,15 @@ def __init__(self):
         )
 
     def match(self, buf):
-        return (len(buf) > 3 and
+        return (len(buf) > 8 and
                 buf[0] == 0x89 and
                 buf[1] == 0x50 and
                 buf[2] == 0x4E and
-                buf[3] == 0x47)
-
+                buf[3] == 0x47 and
+                buf[4] == 0x0D and
+                buf[5] == 0x0A and
+                buf[6] == 0x1A and
+                buf[7] == 0x0A)
 
 class Gif(Type):
     """