mirror of https://github.com/python/cpython
gh-83245: Raise BadZipFile instead of ValueError when reading a corrupt ZIP file (GH-32291)
Co-authored-by: Serhiy Storchaka <storchaka@gmail.com>
This commit is contained in:
parent
6a6f823ea7
commit
202ed2506c
|
@ -1740,6 +1740,17 @@ class OtherTests(unittest.TestCase):
|
|||
fp.write("short file")
|
||||
self.assertRaises(zipfile.BadZipFile, zipfile.ZipFile, TESTFN)
|
||||
|
||||
def test_negative_central_directory_offset_raises_BadZipFile(self):
|
||||
# Zip file containing an empty EOCD record
|
||||
buffer = bytearray(b'PK\x05\x06' + b'\0'*18)
|
||||
|
||||
# Set the size of the central directory bytes to become 1,
|
||||
# causing the central directory offset to become negative
|
||||
for dirsize in 1, 2**32-1:
|
||||
buffer[12:16] = struct.pack('<L', dirsize)
|
||||
f = io.BytesIO(buffer)
|
||||
self.assertRaises(zipfile.BadZipFile, zipfile.ZipFile, f)
|
||||
|
||||
def test_closed_zip_raises_ValueError(self):
|
||||
"""Verify that testzip() doesn't swallow inappropriate exceptions."""
|
||||
data = io.BytesIO()
|
||||
|
|
|
@ -1381,6 +1381,8 @@ class ZipFile:
|
|||
print("given, inferred, offset", offset_cd, inferred, concat)
|
||||
# self.start_dir: Position of start of central directory
|
||||
self.start_dir = offset_cd + concat
|
||||
if self.start_dir < 0:
|
||||
raise BadZipFile("Bad offset for central directory")
|
||||
fp.seek(self.start_dir, 0)
|
||||
data = fp.read(size_cd)
|
||||
fp = io.BytesIO(data)
|
||||
|
|
|
@ -0,0 +1,2 @@
|
|||
:class:`zipfile.ZipFile` now raises :exc:`zipfile.BadZipFile` instead of ``ValueError`` when reading a
|
||||
corrupt zip file in which the central directory offset is negative.
|
Loading…
Reference in New Issue