mirror of
https://github.com/RustPython/RustPython.git
synced 2026-10-28 18:59:04 +08:00
Update tarfile.py to 3.14.5 (#7879)
This commit is contained in:
1 parent
451bdcc9da
commit
f26016049d
2 files changed
+44
-6
No files matched your search
Vendored
+25
-4
@@ -1278,6 +1278,20 @@ class TarInfo(object):
|
||||
@classmethod
|
||||
def frombuf(cls, buf, encoding, errors):
|
||||
"""Construct a TarInfo object from a 512 byte bytes object.
|
||||
|
||||
To support the old v7 tar format AREGTYPE headers are
|
||||
transformed to DIRTYPE headers if their name ends in '/'.
|
||||
"""
|
||||
return cls._frombuf(buf, encoding, errors)
|
||||
|
||||
@classmethod
|
||||
def _frombuf(cls, buf, encoding, errors, *, dircheck=True):
|
||||
"""Construct a TarInfo object from a 512 byte bytes object.
|
||||
|
||||
If ``dircheck`` is set to ``True`` then ``AREGTYPE`` headers will
|
||||
be normalized to ``DIRTYPE`` if the name ends in a trailing slash.
|
||||
``dircheck`` must be set to ``False`` if this function is called
|
||||
on a follow-up header such as ``GNUTYPE_LONGNAME``.
|
||||
"""
|
||||
if len(buf) == 0:
|
||||
raise EmptyHeaderError("empty header")
|
||||
@@ -1308,7 +1322,7 @@ class TarInfo(object):
|
||||
|
||||
# Old V7 tar format represents a directory as a regular
|
||||
# file with a trailing slash.
|
||||
if obj.type == AREGTYPE and obj.name.endswith("/"):
|
||||
if dircheck and obj.type == AREGTYPE and obj.name.endswith("/"):
|
||||
obj.type = DIRTYPE
|
||||
|
||||
# The old GNU sparse format occupies some of the unused
|
||||
@@ -1343,8 +1357,15 @@ class TarInfo(object):
|
||||
"""Return the next TarInfo object from TarFile object
|
||||
tarfile.
|
||||
"""
|
||||
return cls._fromtarfile(tarfile)
|
||||
|
||||
@classmethod
|
||||
def _fromtarfile(cls, tarfile, *, dircheck=True):
|
||||
"""
|
||||
See dircheck documentation in _frombuf().
|
||||
"""
|
||||
buf = tarfile.fileobj.read(BLOCKSIZE)
|
||||
obj = cls.frombuf(buf, tarfile.encoding, tarfile.errors)
|
||||
obj = cls._frombuf(buf, tarfile.encoding, tarfile.errors, dircheck=dircheck)
|
||||
obj.offset = tarfile.fileobj.tell() - BLOCKSIZE
|
||||
return obj._proc_member(tarfile)
|
||||
|
||||
@@ -1402,7 +1423,7 @@ class TarInfo(object):
|
||||
|
||||
# Fetch the next header and process it.
|
||||
try:
|
||||
next = self.fromtarfile(tarfile)
|
||||
next = self._fromtarfile(tarfile, dircheck=False)
|
||||
except HeaderError as e:
|
||||
raise SubsequentHeaderError(str(e)) from None
|
||||
|
||||
@@ -1537,7 +1558,7 @@ class TarInfo(object):
|
||||
|
||||
# Fetch the next header.
|
||||
try:
|
||||
next = self.fromtarfile(tarfile)
|
||||
next = self._fromtarfile(tarfile, dircheck=False)
|
||||
except HeaderError as e:
|
||||
raise SubsequentHeaderError(str(e)) from None
|
||||
|
||||
|
||||
Vendored
+19
-2
@@ -38,8 +38,6 @@ try:
|
||||
import lzma
|
||||
except ImportError:
|
||||
lzma = None
|
||||
# XXX: RUSTPYTHON; xz is not supported yet
|
||||
lzma = None
|
||||
try:
|
||||
from compression import zstd
|
||||
except ImportError:
|
||||
@@ -1236,6 +1234,25 @@ class LongnameTest:
|
||||
self.assertIsNotNone(tar.getmember(longdir))
|
||||
self.assertIsNotNone(tar.getmember(longdir.removesuffix('/')))
|
||||
|
||||
def test_longname_file_not_directory(self):
|
||||
# Test reading a longname file and ensure it is not handled as a directory
|
||||
# Issue #141707
|
||||
buf = io.BytesIO()
|
||||
with tarfile.open(mode='w', fileobj=buf, format=self.format) as tar:
|
||||
ti = tarfile.TarInfo()
|
||||
ti.type = tarfile.AREGTYPE
|
||||
ti.name = ('a' * 99) + '/' + ('b' * 3)
|
||||
tar.addfile(ti)
|
||||
|
||||
expected = {t.name: t.type for t in tar.getmembers()}
|
||||
|
||||
buf.seek(0)
|
||||
with tarfile.open(mode='r', fileobj=buf) as tar:
|
||||
actual = {t.name: t.type for t in tar.getmembers()}
|
||||
|
||||
self.assertEqual(expected, actual)
|
||||
|
||||
|
||||
class GNUReadTest(LongnameTest, ReadTest, unittest.TestCase):
|
||||
|
||||
subdir = "gnu"
|
||||
|
||||
Reference in new issue
Block a user