[Python-checkins] bpo-36991: Fix incorrect exception escaping ZipFile.extract() (GH-13632)
Gregory P. Smith
webhook-mailer at python.org
Thu Sep 12 10:13:48 EDT 2019
https://github.com/python/cpython/commit/2f1b857562b0f1601c9019db74c29b7d7e21ac9f
commit: 2f1b857562b0f1601c9019db74c29b7d7e21ac9f
branch: master
author: Berker Peksag <berker.peksag at gmail.com>
committer: Gregory P. Smith <greg at krypto.org>
date: 2019-09-12T15:13:44+01:00
summary:
bpo-36991: Fix incorrect exception escaping ZipFile.extract() (GH-13632)
files:
A Misc/NEWS.d/next/Library/2019-09-12-14-52-38.bpo-36991.1OcSm8.rst
M Lib/test/test_zipfile.py
M Lib/zipfile.py
diff --git a/Lib/test/test_zipfile.py b/Lib/test/test_zipfile.py
index 99d599eac79a..f812188a7108 100644
--- a/Lib/test/test_zipfile.py
+++ b/Lib/test/test_zipfile.py
@@ -9,6 +9,7 @@
import sys
import time
import unittest
+import unittest.mock as mock
import zipfile
@@ -1766,6 +1767,16 @@ def test_seek_tell(self):
fp.seek(0, os.SEEK_SET)
self.assertEqual(fp.tell(), 0)
+ @requires_bz2
+ def test_decompress_without_3rd_party_library(self):
+ data = b'PK\x05\x06\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
+ zip_file = io.BytesIO(data)
+ with zipfile.ZipFile(zip_file, 'w', compression=zipfile.ZIP_BZIP2) as zf:
+ zf.writestr('a.txt', b'a')
+ with mock.patch('zipfile.bz2', None):
+ with zipfile.ZipFile(zip_file) as zf:
+ self.assertRaises(RuntimeError, zf.extract, 'a.txt')
+
def tearDown(self):
unlink(TESTFN)
unlink(TESTFN2)
diff --git a/Lib/zipfile.py b/Lib/zipfile.py
index c2c5b6a2a44a..97a5ebf753b5 100644
--- a/Lib/zipfile.py
+++ b/Lib/zipfile.py
@@ -703,6 +703,7 @@ def _get_compressor(compress_type, compresslevel=None):
def _get_decompressor(compress_type):
+ _check_compression(compress_type)
if compress_type == ZIP_STORED:
return None
elif compress_type == ZIP_DEFLATED:
diff --git a/Misc/NEWS.d/next/Library/2019-09-12-14-52-38.bpo-36991.1OcSm8.rst b/Misc/NEWS.d/next/Library/2019-09-12-14-52-38.bpo-36991.1OcSm8.rst
new file mode 100644
index 000000000000..c6fa852e94ad
--- /dev/null
+++ b/Misc/NEWS.d/next/Library/2019-09-12-14-52-38.bpo-36991.1OcSm8.rst
@@ -0,0 +1,2 @@
+Fixes a potential incorrect AttributeError exception escaping
+ZipFile.extract() in some unsupported input error situations.
More information about the Python-checkins
mailing list