Fix png image plugin load_end func handle truncated file.

python-pillow · Jan 16, 2024 · 408541d · 408541d
1 parent 067c5f4
commit 408541d
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 1 deletion.
diff --git a/Tests/images/end_trunc_file.png b/Tests/images/end_trunc_file.png
diff --git a/Tests/test_file_png.py b/Tests/test_file_png.py
@@ -778,6 +778,14 @@ class MyStdOut:
         with Image.open(mystdout) as reloaded:
             assert_image_equal_tofile(reloaded, TEST_PNG_FILE)
 
+    def test_end_truncated_filed(self):
+        ImageFile.LOAD_TRUNCATED_IMAGES = True
+        try:
+            with Image.open("Tests/images/end_trunc_file.png") as im:
+                assert_image_equal_tofile(im, "Tests/images/end_trunc_file.png")
+        finally:
+            ImageFile.LOAD_TRUNCATED_IMAGES = False
+
 
 @pytest.mark.skipif(is_win32(), reason="Requires Unix or macOS")
 @skip_unless_feature("zlib")

diff --git a/src/PIL/PngImagePlugin.py b/src/PIL/PngImagePlugin.py
@@ -981,7 +981,14 @@ def load_end(self):
             except EOFError:
                 if cid == b"fdAT":
                     length -= 4
-                ImageFile._safe_read(self.fp, length)
+                try:
+                    ImageFile._safe_read(self.fp, length)
+                except OSError as e:
+                    if ImageFile.LOAD_TRUNCATED_IMAGES:
+                        logger.debug("%r %s %s (Truncated File Read)", cid, pos, length)
+                        break
+                    else:
+                        raise e
             except AttributeError:
                 logger.debug("%r %s %s (unknown)", cid, pos, length)
                 s = ImageFile._safe_read(self.fp, length)