mirror of
https://github.com/python/cpython.git
synced 2025-08-31 14:07:50 +00:00
gh-123270: Replaced SanitizedNames with a more surgical fix. (#123354)
Applies changes from zipp 3.20.1 and jaraco/zipp#124
This commit is contained in:
parent
7e38e6745d
commit
2231286d78
3 changed files with 87 additions and 71 deletions
|
@ -5,6 +5,7 @@ import pathlib
|
|||
import pickle
|
||||
import stat
|
||||
import sys
|
||||
import time
|
||||
import unittest
|
||||
import zipfile
|
||||
import zipfile._path
|
||||
|
@ -592,7 +593,11 @@ class TestPath(unittest.TestCase):
|
|||
|
||||
def test_malformed_paths(self):
|
||||
"""
|
||||
Path should handle malformed paths.
|
||||
Path should handle malformed paths gracefully.
|
||||
|
||||
Paths with leading slashes are not visible.
|
||||
|
||||
Paths with dots are treated like regular files.
|
||||
"""
|
||||
data = io.BytesIO()
|
||||
zf = zipfile.ZipFile(data, "w")
|
||||
|
@ -601,11 +606,40 @@ class TestPath(unittest.TestCase):
|
|||
zf.writestr("../parent.txt", b"content")
|
||||
zf.filename = ''
|
||||
root = zipfile.Path(zf)
|
||||
assert list(map(str, root.iterdir())) == [
|
||||
'one-slash.txt',
|
||||
'two-slash.txt',
|
||||
'parent.txt',
|
||||
]
|
||||
assert list(map(str, root.iterdir())) == ['../']
|
||||
assert root.joinpath('..').joinpath('parent.txt').read_bytes() == b'content'
|
||||
|
||||
def test_unsupported_names(self):
|
||||
"""
|
||||
Path segments with special characters are readable.
|
||||
|
||||
On some platforms or file systems, characters like
|
||||
``:`` and ``?`` are not allowed, but they are valid
|
||||
in the zip file.
|
||||
"""
|
||||
data = io.BytesIO()
|
||||
zf = zipfile.ZipFile(data, "w")
|
||||
zf.writestr("path?", b"content")
|
||||
zf.writestr("V: NMS.flac", b"fLaC...")
|
||||
zf.filename = ''
|
||||
root = zipfile.Path(zf)
|
||||
contents = root.iterdir()
|
||||
assert next(contents).name == 'path?'
|
||||
assert next(contents).name == 'V: NMS.flac'
|
||||
assert root.joinpath('V: NMS.flac').read_bytes() == b"fLaC..."
|
||||
|
||||
def test_backslash_not_separator(self):
|
||||
"""
|
||||
In a zip file, backslashes are not separators.
|
||||
"""
|
||||
data = io.BytesIO()
|
||||
zf = zipfile.ZipFile(data, "w")
|
||||
zf.writestr(DirtyZipInfo.for_name("foo\\bar", zf), b"content")
|
||||
zf.filename = ''
|
||||
root = zipfile.Path(zf)
|
||||
(first,) = root.iterdir()
|
||||
assert not first.is_dir()
|
||||
assert first.name == 'foo\\bar'
|
||||
|
||||
@pass_alpharep
|
||||
def test_interface(self, alpharep):
|
||||
|
@ -613,3 +647,30 @@ class TestPath(unittest.TestCase):
|
|||
|
||||
zf = zipfile.Path(alpharep)
|
||||
assert isinstance(zf, Traversable)
|
||||
|
||||
|
||||
class DirtyZipInfo(zipfile.ZipInfo):
|
||||
"""
|
||||
Bypass name sanitization.
|
||||
"""
|
||||
|
||||
def __init__(self, filename, *args, **kwargs):
|
||||
super().__init__(filename, *args, **kwargs)
|
||||
self.filename = filename
|
||||
|
||||
@classmethod
|
||||
def for_name(cls, name, archive):
|
||||
"""
|
||||
Construct the same way that ZipFile.writestr does.
|
||||
|
||||
TODO: extract this functionality and re-use
|
||||
"""
|
||||
self = cls(filename=name, date_time=time.localtime(time.time())[:6])
|
||||
self.compress_type = archive.compression
|
||||
self.compress_level = archive.compresslevel
|
||||
if self.filename.endswith('/'): # pragma: no cover
|
||||
self.external_attr = 0o40775 << 16 # drwxrwxr-x
|
||||
self.external_attr |= 0x10 # MS-DOS directory flag
|
||||
else:
|
||||
self.external_attr = 0o600 << 16 # ?rw-------
|
||||
return self
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue