Close #17666: Fix reading gzip files with an extra field.

This commit is contained in:
Serhiy Storchaka 2013-04-08 22:35:02 +03:00
parent b0f9fa7a03
commit 7e69f0085e
3 changed files with 11 additions and 1 deletions

View File

@ -302,7 +302,8 @@ class GzipFile(io.BufferedIOBase):
if flag & FEXTRA:
# Read & discard the extra field, if present
self._read_exact(struct.unpack("<H", self._read_exact(2)))
extra_len, = struct.unpack("<H", self._read_exact(2))
self._read_exact(extra_len)
if flag & FNAME:
# Read and discard a null-terminated string containing the filename
while True:

View File

@ -403,6 +403,13 @@ class TestGzip(BaseTest):
with gzip.GzipFile(fileobj=io.BytesIO(truncated[:i])) as f:
self.assertRaises(EOFError, f.read, 1)
def test_read_with_extra(self):
# Gzip data with an extra field
gzdata = (b'\x1f\x8b\x08\x04\xb2\x17cQ\x02\xff'
b'\x05\x00Extra'
b'\x0bI-.\x01\x002\xd1Mx\x04\x00\x00\x00')
with gzip.GzipFile(fileobj=io.BytesIO(gzdata)) as f:
self.assertEqual(f.read(), b'Test')
class TestOpen(BaseTest):
def test_binary_modes(self):

View File

@ -21,6 +21,8 @@ Core and Builtins
Library
-------
- Issue #17666: Fix reading gzip files with an extra field.
- Issue #17502: Process DEFAULT values in mock side_effect that returns iterator.
Patch by Michael Foord.