fixes

image_safety.py

@@ -0,0 +1,58 @@
+"""Ograniczona obsluga duzych obrazow Pillow.
+
+Pillow domyslnie ostrzega powyzej ~89 MP i zglasza blad powyzej ~179 MP.
+Dla skanow dopuszczamy konfigurowalny limit, ale PRZED dekodowaniem
+odrzucamy wszystko, co ten limit przekracza. Nie wylaczamy ochrony globalnie.
+
+Narzędzie CLI wykonuje operacje obrazow sekwencyjnie; prog PIL jest globalny,
+ale zmieniamy go tylko na czas kontrolowanego otwarcia i przywracamy po nim.
+"""
+from __future__ import annotations
+
+import warnings
+from contextlib import contextmanager
+
+from PIL import Image
+
+
+DEFAULT_MAX_IMAGE_PIXELS = 220_000_000  # 220 MP; obejmuje skany ~200 MP
+
+
+class ImagePixelLimitExceeded(ValueError):
+    """Obraz nie zostal zdekodowany, bo przekracza ustalony limit."""
+
+
+@contextmanager
+def safe_image_open(path, *, max_pixels=DEFAULT_MAX_IMAGE_PIXELS, check_pixels=True):
+    """Otworz naglowek obrazu z limitem, zanim Pillow zdekoduje piksele.
+
+    Ostrzezenie Pillow dot. progu zostaje zastapione JAWNYM sprawdzeniem
+    wielkosci (i statusem SKIP w callerze). Pozostale ostrzezenia pozostaja
+    widoczne. Dla samego odczytu wymiarow check_pixels=False nie dekoduje.
+    """
+    max_pixels = int(max_pixels)
+    if max_pixels <= 0:
+        raise ValueError('max_pixels musi byc wieksze od zera')
+    old_limit = Image.MAX_IMAGE_PIXELS
+    with warnings.catch_warnings():
+        warnings.filterwarnings('ignore', category=Image.DecompressionBombWarning)
+        Image.MAX_IMAGE_PIXELS = max_pixels
+        try:
+            try:
+                with Image.open(path) as image:
+                    pixels = image.width * image.height
+                    if check_pixels and pixels > max_pixels:
+                        raise ImagePixelLimitExceeded(
+                            f'{image.width}x{image.height} ({pixels / 1_000_000:.1f} MP) '
+                            f'przekracza --max-image-mpixels={max_pixels / 1_000_000:g}; '
+                            'pominieto dla ochrony pamieci RAM'
+                        )
+                    yield image
+            except Image.DecompressionBombError as exc:
+                # Bardzo duze obrazy Pillow moze odrzucic juz przy Image.open.
+                raise ImagePixelLimitExceeded(
+                    f'Pillow odrzucil obraz przekraczajacy limit '
+                    f'--max-image-mpixels={max_pixels / 1_000_000:g}: {exc}'
+                ) from exc
+        finally:
+            Image.MAX_IMAGE_PIXELS = old_limit