#6511: ZipFile will now raise BadZipfile when opening an empty or tiny file,

like it does for larger invalid files.
This commit is contained in:
Amaury Forgeot d'Arc 2009-07-28 22:15:30 +00:00
parent e55df1fa2a
commit 3e5b027a64
3 changed files with 17 additions and 1 deletions

View File

@ -703,6 +703,16 @@ class OtherTests(unittest.TestCase):
# quickly.
self.assertRaises(IOError, zipfile.ZipFile, TESTFN)
def test_empty_file_raises_BadZipFile(self):
f = open(TESTFN, 'w')
f.close()
self.assertRaises(zipfile.BadZipfile, zipfile.ZipFile, TESTFN)
f = open(TESTFN, 'w')
f.write("short file")
f.close()
self.assertRaises(zipfile.BadZipfile, zipfile.ZipFile, TESTFN)
def test_closed_zip_raises_RuntimeError(self):
# Verify that testzip() doesn't swallow inappropriate exceptions.
data = StringIO()

View File

@ -198,7 +198,10 @@ def _EndRecData(fpin):
# Check to see if this is ZIP file with no archive comment (the
# "end of central directory" structure should be the last item in the
# file if this is the case).
fpin.seek(-sizeEndCentDir, 2)
try:
fpin.seek(-sizeEndCentDir, 2)
except IOError:
return None
data = fpin.read()
if data[0:4] == stringEndArchive and data[-2:] == "\000\000":
# the signature is correct and there's no comment, unpack structure

View File

@ -354,6 +354,9 @@ Core and Builtins
Library
-------
- Issue #6511: ZipFile now raises BadZipfile (instead of an IOError) when
opening an empty or very small file.
- Issue #6553: Fixed a crash in cPickle.load(), when given a file-like object
containing incomplete data.