test+fix

py-pdf · May 11, 2024 · a37b73f · a37b73f
1 parent 770aaba
commit a37b73f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/pypdf/generic/_image_inline.py b/pypdf/generic/_image_inline.py
@@ -99,6 +99,9 @@ def extract_inline_A85(stream: StreamType) -> bytes:
             data += buf[: loc + 2]
             stream.seek(-len(buf) + loc + 2, 1)
             break
+        elif len(buf) == 2:  # end of buffer
+            data += buf
+            break
         data += buf[:-1]  # back by one char in case of in the middle of ~>
         stream.seek(-1, 1)
 

diff --git a/tests/test_generic.py b/tests/test_generic.py
@@ -1,5 +1,6 @@
 """Test the pypdf.generic module."""
 
+from base64 import a85encode
 from copy import deepcopy
 from io import BytesIO
 from pathlib import Path
@@ -1360,8 +1361,12 @@ def test_array_operators():
 def test_unitary_extract_inline_buffer_empty():
     with pytest.raises(PdfReadError):
         extract_inline_AHex(BytesIO())
+    with pytest.raises(PdfReadError):
+        extract_inline_AHex(BytesIO(4095 * b"00" + b"   "))
     with pytest.raises(PdfReadError):
         extract_inline_A85(BytesIO())
+    with pytest.raises(PdfReadError):
+        extract_inline_A85(BytesIO(a85encode(b"1")))
     with pytest.raises(PdfReadError):
         extract_inline_RL(BytesIO())