diff options
author | Ruben Vorderman <r.h.p.vorderman@lumc.nl> | 2021-11-19 18:07:05 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-11-19 18:07:05 (GMT) |
commit | 0ff3d95b9875805ac03aeffc37ae4458ce3b8ac0 (patch) | |
tree | 6f82877c128a0bab19926e8fe4b3a9d7eb3716b2 | |
parent | 7e44dc0ba768451f287a541cd1c85f7d87a41561 (diff) | |
download | cpython-0ff3d95b9875805ac03aeffc37ae4458ce3b8ac0.zip cpython-0ff3d95b9875805ac03aeffc37ae4458ce3b8ac0.tar.gz cpython-0ff3d95b9875805ac03aeffc37ae4458ce3b8ac0.tar.bz2 |
bpo-45507: EOFErrors should be thrown for truncated gzip members (GH-29029)
-rw-r--r-- | Lib/gzip.py | 3 | ||||
-rw-r--r-- | Lib/test/test_gzip.py | 8 | ||||
-rw-r--r-- | Misc/NEWS.d/next/Library/2021-10-18-14-00-01.bpo-45507.lDotNV.rst | 1 |
3 files changed, 12 insertions, 0 deletions
diff --git a/Lib/gzip.py b/Lib/gzip.py index ac17810..6773ea3 100644 --- a/Lib/gzip.py +++ b/Lib/gzip.py @@ -603,6 +603,9 @@ def decompress(data): do = zlib.decompressobj(wbits=-zlib.MAX_WBITS) # Read all the data except the header decompressed = do.decompress(data[fp.tell():]) + if not do.eof or len(do.unused_data) < 8: + raise EOFError("Compressed file ended before the end-of-stream " + "marker was reached") crc, length = struct.unpack("<II", do.unused_data[:8]) if crc != zlib.crc32(decompressed): raise BadGzipFile("CRC check failed") diff --git a/Lib/test/test_gzip.py b/Lib/test/test_gzip.py index f86e767..aa66d2f 100644 --- a/Lib/test/test_gzip.py +++ b/Lib/test/test_gzip.py @@ -562,6 +562,14 @@ class TestGzip(BaseTest): datac = gzip.compress(data) self.assertEqual(gzip.decompress(datac), data) + def test_decompress_truncated_trailer(self): + compressed_data = gzip.compress(data1) + self.assertRaises(EOFError, gzip.decompress, compressed_data[:-4]) + + def test_decompress_missing_trailer(self): + compressed_data = gzip.compress(data1) + self.assertRaises(EOFError, gzip.decompress, compressed_data[:-8]) + def test_read_truncated(self): data = data1*50 # Drop the CRC (4 bytes) and file size (4 bytes). diff --git a/Misc/NEWS.d/next/Library/2021-10-18-14-00-01.bpo-45507.lDotNV.rst b/Misc/NEWS.d/next/Library/2021-10-18-14-00-01.bpo-45507.lDotNV.rst new file mode 100644 index 0000000..a69ccda --- /dev/null +++ b/Misc/NEWS.d/next/Library/2021-10-18-14-00-01.bpo-45507.lDotNV.rst @@ -0,0 +1 @@ +Add tests for truncated/missing trailers in gzip.decompress implementation. |