]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
gh-151981: Make tarfile._Stream.seek break at EOF (GH-151982)
authorPetr Viktorin <encukou@gmail.com>
Tue, 23 Jun 2026 13:13:30 +0000 (15:13 +0200)
committerGitHub <noreply@github.com>
Tue, 23 Jun 2026 13:13:30 +0000 (15:13 +0200)
Co-authored-by: Stan Ulbrych <stan@python.org>
Lib/tarfile.py
Lib/test/test_tarfile.py
Misc/NEWS.d/next/Security/2026-06-23-13-28-16.gh-issue-151981.xBHEcU.rst [new file with mode: 0644]

index 52c0d990d0a5f6bf1d935bd9bfc1b42265f619d1..58061b3e8c1ca4a69e32ae795419e1760e1810bd 100644 (file)
@@ -546,7 +546,9 @@ class _Stream:
         if pos - self.pos >= 0:
             blocks, remainder = divmod(pos - self.pos, self.bufsize)
             for i in range(blocks):
-                self.read(self.bufsize)
+                data = self.read(self.bufsize)
+                if not data:
+                    break
             self.read(remainder)
         else:
             raise StreamError("seeking backwards is not allowed")
index e0d06be57ccbb345824210d43082d4fd0169599d..fd6e3ce965461808284e6288d1d8e1889b4f20a2 100644 (file)
@@ -4878,6 +4878,22 @@ class TestExtractionFilters(unittest.TestCase):
         with self.check_context(arc.open(errorlevel='boo!'), filtererror_filter):
             self.expect_exception(TypeError)  # errorlevel is not int
 
+    @support.subTests('format', [tarfile.GNU_FORMAT, tarfile.PAX_FORMAT])
+    def test_getmembers_big_size(self, format):
+        # gh-151981: A loop in seek() for streaming files tried to read the
+        # declared number of blocks even at EOF
+        tinfo = tarfile.TarInfo("huge-file")
+        tinfo.size = 1 << 64
+        bio = io.BytesIO()
+        # Write header without data
+        bio.write(tinfo.tobuf(format))
+
+        # Reset & try to get contents
+        bio.seek(0)
+        with tarfile.open(fileobj=bio, mode="r|") as tar:
+            with self.assertRaises(tarfile.ReadError):
+                tar.getmembers()
+
 
 class OverwriteTests(archiver_tests.OverwriteTests, unittest.TestCase):
     testdir = os.path.join(TEMPDIR, "testoverwrite")
diff --git a/Misc/NEWS.d/next/Security/2026-06-23-13-28-16.gh-issue-151981.xBHEcU.rst b/Misc/NEWS.d/next/Security/2026-06-23-13-28-16.gh-issue-151981.xBHEcU.rst
new file mode 100644 (file)
index 0000000..2123ab8
--- /dev/null
@@ -0,0 +1,2 @@
+In :mod:`tarfile`, seeking a stream now stops when end of the stream is
+reached.