diff options
author | Serhiy Storchaka <storchaka@gmail.com> | 2013-01-22 15:01:59 (GMT) |
---|---|---|
committer | Serhiy Storchaka <storchaka@gmail.com> | 2013-01-22 15:01:59 (GMT) |
commit | 7c3922f44c226eac29a497648bbc3cc8702905a8 (patch) | |
tree | 395bde42013572a48d3cf30ddc6d28fdd0cc1cb7 /Lib/test/test_gzip.py | |
parent | fc6e8aabf58d748369e0d3b08495ac35a67d2870 (diff) | |
download | cpython-7c3922f44c226eac29a497648bbc3cc8702905a8.zip cpython-7c3922f44c226eac29a497648bbc3cc8702905a8.tar.gz cpython-7c3922f44c226eac29a497648bbc3cc8702905a8.tar.bz2 |
Issue #1159051: GzipFile now raises EOFError when reading a corrupted file
with truncated header or footer.
Added tests for reading truncated gzip and bzip2 files.
Diffstat (limited to 'Lib/test/test_gzip.py')
-rw-r--r-- | Lib/test/test_gzip.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/Lib/test/test_gzip.py b/Lib/test/test_gzip.py index ced226f..ba9d7da 100644 --- a/Lib/test/test_gzip.py +++ b/Lib/test/test_gzip.py @@ -365,6 +365,21 @@ class TestGzip(unittest.TestCase): datac = gzip.compress(data) self.assertEqual(gzip.decompress(datac), data) + def test_read_truncated(self): + data = data1*50 + # Drop the CRC (4 bytes) and file size (4 bytes). + truncated = gzip.compress(data)[:-8] + with gzip.GzipFile(fileobj=io.BytesIO(truncated)) as f: + self.assertRaises(EOFError, f.read) + with gzip.GzipFile(fileobj=io.BytesIO(truncated)) as f: + self.assertEqual(f.read(len(data)), data) + self.assertRaises(EOFError, f.read, 1) + # Incomplete 10-byte header. + for i in range(2, 10): + with gzip.GzipFile(fileobj=io.BytesIO(truncated[:i])) as f: + self.assertRaises(EOFError, f.read, 1) + + def test_main(verbose=None): support.run_unittest(TestGzip) |