Commit c5a45669 authored by Hynek Schlawack's avatar Hynek Schlawack

#15377: Make posixpath.join() more strict when checking for str/bytes mix

Based on a patch by Nick Coghlan.
parent eb3e62f1
...@@ -83,10 +83,11 @@ def join(a, *p): ...@@ -83,10 +83,11 @@ def join(a, *p):
else: else:
path += sep + b path += sep + b
except TypeError: except TypeError:
strs = [isinstance(s, str) for s in (a, ) + p] valid_types = all(isinstance(s, (str, bytes, bytearray))
if any(strs) and not all(strs): for s in (a, ) + p)
if valid_types:
# Must have a mixture of text and binary data
raise TypeError("Can't mix strings and bytes in path components.") raise TypeError("Can't mix strings and bytes in path components.")
else:
raise raise
return path return path
......
import unittest import unittest
from test import support, test_genericpath from test import support, test_genericpath
import itertools
import posixpath import posixpath
import os import os
import sys import sys
...@@ -56,18 +57,21 @@ class PosixPathTest(unittest.TestCase): ...@@ -56,18 +57,21 @@ class PosixPathTest(unittest.TestCase):
self.assertEqual(posixpath.join(b"/foo/", b"bar/", b"baz/"), self.assertEqual(posixpath.join(b"/foo/", b"bar/", b"baz/"),
b"/foo/bar/baz/") b"/foo/bar/baz/")
# Check for friendly str/bytes mixing message def check_error_msg(list_of_args, msg):
for args in [[b'bytes', 'str'], """Check posixpath.join raises friendly TypeErrors."""
[bytearray(b'bytes'), 'str']]: for args in (item for perm in list_of_args
for _ in range(2): for item in itertools.permutations(perm)):
with self.assertRaises(TypeError) as cm: with self.assertRaises(TypeError) as cm:
posixpath.join(*args) posixpath.join(*args)
self.assertEqual( self.assertEqual(msg, cm.exception.args[0])
"Can't mix strings and bytes in path components.",
cm.exception.args[0]
)
args.reverse() # check both orders
check_error_msg([[b'bytes', 'str'], [bytearray(b'bytes'), 'str']],
"Can't mix strings and bytes in path components.")
# regression, see #15377
with self.assertRaises(TypeError) as cm:
os.path.join(None, 'str')
self.assertNotEqual("Can't mix strings and bytes in path components.",
cm.exception.args[0])
def test_split(self): def test_split(self):
self.assertEqual(posixpath.split("/foo/bar"), ("/foo", "bar")) self.assertEqual(posixpath.split("/foo/bar"), ("/foo", "bar"))
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment