3cad5e
From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
3cad5e
From: Lumir Balhar <lbalhar@redhat.com>
3cad5e
Date: Thu, 25 May 2023 10:03:57 +0200
3cad5e
Subject: [PATCH] 00399: gh-102153: Start stripping C0 control and space chars
3cad5e
 in `urlsplit` (GH-102508) (#104575)
3cad5e
3cad5e
* gh-102153: Start stripping C0 control and space chars in `urlsplit` (GH-102508)
3cad5e
3cad5e
`urllib.parse.urlsplit` has already been respecting the WHATWG spec a bit GH-25595.
3cad5e
3cad5e
This adds more sanitizing to respect the "Remove any leading C0 control or space from input" [rule](https://url.spec.whatwg.org/GH-url-parsing:~:text=Remove%20any%20leading%20and%20trailing%20C0%20control%20or%20space%20from%20input.) in response to [CVE-2023-24329](https://nvd.nist.gov/vuln/detail/CVE-2023-24329).
3cad5e
3cad5e
Backported to Python 2 from Python 3.12.
3cad5e
3cad5e
Co-authored-by: Illia Volochii <illia.volochii@gmail.com>
3cad5e
Co-authored-by: Gregory P. Smith [Google] <greg@krypto.org>
3cad5e
Co-authored-by: Lumir Balhar <lbalhar@redhat.com>
3cad5e
---
3cad5e
 Lib/test/test_urlparse.py | 57 +++++++++++++++++++++++++++++++++++++++
3cad5e
 Lib/urlparse.py           | 10 +++++++
3cad5e
 2 files changed, 67 insertions(+)
3cad5e
3cad5e
diff --git a/Lib/test/test_urlparse.py b/Lib/test/test_urlparse.py
3cad5e
index 16eefed56f6..419e9c2bdcc 100644
3cad5e
--- a/Lib/test/test_urlparse.py
3cad5e
+++ b/Lib/test/test_urlparse.py
3cad5e
@@ -666,7 +666,64 @@ class UrlParseTestCase(unittest.TestCase):
3cad5e
             self.assertEqual(p.scheme, "https")
3cad5e
             self.assertEqual(p.geturl(), "https://www.python.org/javascript:alert('msg')/?query=something#fragment")
3cad5e
 
3cad5e
+    def test_urlsplit_strip_url(self):
3cad5e
+        noise = "".join([chr(i) for i in range(0, 0x20 + 1)])
3cad5e
+        base_url = "http://User:Pass@www.python.org:080/doc/?query=yes#frag"
3cad5e
 
3cad5e
+        url = noise.decode("utf-8") + base_url
3cad5e
+        p = urlparse.urlsplit(url)
3cad5e
+        self.assertEqual(p.scheme, "http")
3cad5e
+        self.assertEqual(p.netloc, "User:Pass@www.python.org:080")
3cad5e
+        self.assertEqual(p.path, "/doc/")
3cad5e
+        self.assertEqual(p.query, "query=yes")
3cad5e
+        self.assertEqual(p.fragment, "frag")
3cad5e
+        self.assertEqual(p.username, "User")
3cad5e
+        self.assertEqual(p.password, "Pass")
3cad5e
+        self.assertEqual(p.hostname, "www.python.org")
3cad5e
+        self.assertEqual(p.port, 80)
3cad5e
+        self.assertEqual(p.geturl(), base_url)
3cad5e
+
3cad5e
+        url = noise + base_url.encode("utf-8")
3cad5e
+        p = urlparse.urlsplit(url)
3cad5e
+        self.assertEqual(p.scheme, b"http")
3cad5e
+        self.assertEqual(p.netloc, b"User:Pass@www.python.org:080")
3cad5e
+        self.assertEqual(p.path, b"/doc/")
3cad5e
+        self.assertEqual(p.query, b"query=yes")
3cad5e
+        self.assertEqual(p.fragment, b"frag")
3cad5e
+        self.assertEqual(p.username, b"User")
3cad5e
+        self.assertEqual(p.password, b"Pass")
3cad5e
+        self.assertEqual(p.hostname, b"www.python.org")
3cad5e
+        self.assertEqual(p.port, 80)
3cad5e
+        self.assertEqual(p.geturl(), base_url.encode("utf-8"))
3cad5e
+
3cad5e
+        # Test that trailing space is preserved as some applications rely on
3cad5e
+        # this within query strings.
3cad5e
+        query_spaces_url = "https://www.python.org:88/doc/?query=    "
3cad5e
+        p = urlparse.urlsplit(noise.decode("utf-8") + query_spaces_url)
3cad5e
+        self.assertEqual(p.scheme, "https")
3cad5e
+        self.assertEqual(p.netloc, "www.python.org:88")
3cad5e
+        self.assertEqual(p.path, "/doc/")
3cad5e
+        self.assertEqual(p.query, "query=    ")
3cad5e
+        self.assertEqual(p.port, 88)
3cad5e
+        self.assertEqual(p.geturl(), query_spaces_url)
3cad5e
+
3cad5e
+        p = urlparse.urlsplit("www.pypi.org ")
3cad5e
+        # That "hostname" gets considered a "path" due to the
3cad5e
+        # trailing space and our existing logic...  YUCK...
3cad5e
+        # and re-assembles via geturl aka unurlsplit into the original.
3cad5e
+        # django.core.validators.URLValidator (at least through v3.2) relies on
3cad5e
+        # this, for better or worse, to catch it in a ValidationError via its
3cad5e
+        # regular expressions.
3cad5e
+        # Here we test the basic round trip concept of such a trailing space.
3cad5e
+        self.assertEqual(urlparse.urlunsplit(p), "www.pypi.org ")
3cad5e
+
3cad5e
+        # with scheme as cache-key
3cad5e
+        url = "//www.python.org/"
3cad5e
+        scheme = noise.decode("utf-8") + "https" + noise.decode("utf-8")
3cad5e
+        for _ in range(2):
3cad5e
+            p = urlparse.urlsplit(url, scheme=scheme)
3cad5e
+            self.assertEqual(p.scheme, "https")
3cad5e
+            self.assertEqual(p.geturl(), "https://www.python.org/")
3cad5e
 
3cad5e
     def test_attributes_bad_port(self):
3cad5e
         """Check handling of non-integer ports."""
3cad5e
diff --git a/Lib/urlparse.py b/Lib/urlparse.py
3cad5e
index 6cc40a8d2fb..0f03a7cc4a9 100644
3cad5e
--- a/Lib/urlparse.py
3cad5e
+++ b/Lib/urlparse.py
3cad5e
@@ -26,6 +26,10 @@ scenarios for parsing, and for backward compatibility purposes, some
3cad5e
 parsing quirks from older RFCs are retained. The testcases in
3cad5e
 test_urlparse.py provides a good indicator of parsing behavior.
3cad5e
 
3cad5e
+The WHATWG URL Parser spec should also be considered.  We are not compliant with
3cad5e
+it either due to existing user code API behavior expectations (Hyrum's Law).
3cad5e
+It serves as a useful guide when making changes.
3cad5e
+
3cad5e
 """
3cad5e
 
3cad5e
 import re
3cad5e
@@ -63,6 +67,10 @@ scheme_chars = ('abcdefghijklmnopqrstuvwxyz'
3cad5e
                 '0123456789'
3cad5e
                 '+-.')
3cad5e
 
3cad5e
+# Leading and trailing C0 control and space to be stripped per WHATWG spec.
3cad5e
+# == "".join([chr(i) for i in range(0, 0x20 + 1)])
3cad5e
+_WHATWG_C0_CONTROL_OR_SPACE = '\x00\x01\x02\x03\x04\x05\x06\x07\x08\t\n\x0b\x0c\r\x0e\x0f\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f '
3cad5e
+
3cad5e
 # Unsafe bytes to be removed per WHATWG spec
3cad5e
 _UNSAFE_URL_BYTES_TO_REMOVE = ['\t', '\r', '\n']
3cad5e
 
3cad5e
@@ -201,6 +209,8 @@ def urlsplit(url, scheme='', allow_fragments=True):
3cad5e
     (e.g. netloc is a single string) and we don't expand % escapes."""
3cad5e
     url = _remove_unsafe_bytes_from_url(url)
3cad5e
     scheme = _remove_unsafe_bytes_from_url(scheme)
3cad5e
+    url = url.lstrip(_WHATWG_C0_CONTROL_OR_SPACE)
3cad5e
+    scheme = scheme.strip(_WHATWG_C0_CONTROL_OR_SPACE)
3cad5e
     allow_fragments = bool(allow_fragments)
3cad5e
     key = url, scheme, allow_fragments, type(url), type(scheme)
3cad5e
     cached = _parse_cache.get(key, None)