diff options
| author | Martin Panter <vadmium+py@gmail.com> | 2016-10-01 02:45:17 (GMT) |
|---|---|---|
| committer | Martin Panter <vadmium+py@gmail.com> | 2016-10-01 02:45:17 (GMT) |
| commit | 38317d3318e05832a864d413aa744a48a8975dce (patch) | |
| tree | 8757c1d2116ed1d20aa28b02e0b11f05e9bc3412 /Lib/test/test_bz2.py | |
| parent | 34b9d14be642be80e9ac10c59004570a7efa5804 (diff) | |
| download | cpython-38317d3318e05832a864d413aa744a48a8975dce.zip cpython-38317d3318e05832a864d413aa744a48a8975dce.tar.gz cpython-38317d3318e05832a864d413aa744a48a8975dce.tar.bz2 | |
Issue #28275: Clean up to avoid use-after-free after bzip decompress failure
Diffstat (limited to 'Lib/test/test_bz2.py')
| -rw-r--r-- | Lib/test/test_bz2.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/Lib/test/test_bz2.py b/Lib/test/test_bz2.py index a1e4b8d..478921a 100644 --- a/Lib/test/test_bz2.py +++ b/Lib/test/test_bz2.py @@ -821,6 +821,12 @@ class BZ2DecompressorTest(BaseTest): out.append(bzd.decompress(self.DATA[300:])) self.assertEqual(b''.join(out), self.TEXT) + def test_failure(self): + bzd = BZ2Decompressor() + self.assertRaises(Exception, bzd.decompress, self.BAD_DATA * 30) + # Previously, a second call could crash due to internal inconsistency + self.assertRaises(Exception, bzd.decompress, self.BAD_DATA * 30) + class CompressDecompressTest(BaseTest): def testCompress(self): data = bz2.compress(self.TEXT) |