Close #17666: Fix reading gzip files with an extra field.
This commit is contained in:
parent
b0f9fa7a03
commit
7e69f0085e
|
@ -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:
|
||||
|
|
|
@ -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):
|
||||
|
|
Loading…
Reference in New Issue