summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorNick Pope <nick@nickpope.me.uk>2021-02-16 10:14:17 +0000
committerCarlton Gibson <carlton.gibson@noumenal.es>2021-02-18 10:21:04 +0100
commit326a926beef869d3341bc9ef737887f0449b6b71 (patch)
tree1900a31204c7bc47fd115d533516474fd1fb3985 /tests
parentad36388406195ecddb319fea862298e6c8d27d3b (diff)
[3.0.x] Fixed CVE-2021-23336 -- Fixed web cache poisoning via django.utils.http.limited_parse_qsl().
Diffstat (limited to 'tests')
-rw-r--r--tests/handlers/test_exception.py2
-rw-r--r--tests/requests/test_data_upload_settings.py8
-rw-r--r--tests/utils_tests/test_http.py52
3 files changed, 54 insertions, 8 deletions
diff --git a/tests/handlers/test_exception.py b/tests/handlers/test_exception.py
index 7afd4acc6b..0c1e763990 100644
--- a/tests/handlers/test_exception.py
+++ b/tests/handlers/test_exception.py
@@ -6,7 +6,7 @@ from django.test.client import FakePayload
class ExceptionHandlerTests(SimpleTestCase):
def get_suspicious_environ(self):
- payload = FakePayload('a=1&a=2;a=3\r\n')
+ payload = FakePayload('a=1&a=2&a=3\r\n')
return {
'REQUEST_METHOD': 'POST',
'CONTENT_TYPE': 'application/x-www-form-urlencoded',
diff --git a/tests/requests/test_data_upload_settings.py b/tests/requests/test_data_upload_settings.py
index f60f1850ea..44897cc9fa 100644
--- a/tests/requests/test_data_upload_settings.py
+++ b/tests/requests/test_data_upload_settings.py
@@ -11,7 +11,7 @@ TOO_MUCH_DATA_MSG = 'Request body exceeded settings.DATA_UPLOAD_MAX_MEMORY_SIZE.
class DataUploadMaxMemorySizeFormPostTests(SimpleTestCase):
def setUp(self):
- payload = FakePayload('a=1&a=2;a=3\r\n')
+ payload = FakePayload('a=1&a=2&a=3\r\n')
self.request = WSGIRequest({
'REQUEST_METHOD': 'POST',
'CONTENT_TYPE': 'application/x-www-form-urlencoded',
@@ -117,7 +117,7 @@ class DataUploadMaxNumberOfFieldsGet(SimpleTestCase):
request = WSGIRequest({
'REQUEST_METHOD': 'GET',
'wsgi.input': BytesIO(b''),
- 'QUERY_STRING': 'a=1&a=2;a=3',
+ 'QUERY_STRING': 'a=1&a=2&a=3',
})
request.GET['a']
@@ -126,7 +126,7 @@ class DataUploadMaxNumberOfFieldsGet(SimpleTestCase):
request = WSGIRequest({
'REQUEST_METHOD': 'GET',
'wsgi.input': BytesIO(b''),
- 'QUERY_STRING': 'a=1&a=2;a=3',
+ 'QUERY_STRING': 'a=1&a=2&a=3',
})
request.GET['a']
@@ -168,7 +168,7 @@ class DataUploadMaxNumberOfFieldsMultipartPost(SimpleTestCase):
class DataUploadMaxNumberOfFieldsFormPost(SimpleTestCase):
def setUp(self):
- payload = FakePayload("\r\n".join(['a=1&a=2;a=3', '']))
+ payload = FakePayload("\r\n".join(['a=1&a=2&a=3', '']))
self.request = WSGIRequest({
'REQUEST_METHOD': 'POST',
'CONTENT_TYPE': 'application/x-www-form-urlencoded',
diff --git a/tests/utils_tests/test_http.py b/tests/utils_tests/test_http.py
index aa9f194a8a..0ec14a6819 100644
--- a/tests/utils_tests/test_http.py
+++ b/tests/utils_tests/test_http.py
@@ -3,14 +3,16 @@ import unittest
from datetime import datetime
from unittest import mock
+from django.core.exceptions import TooManyFieldsSent
from django.test import SimpleTestCase, ignore_warnings
from django.utils.datastructures import MultiValueDict
from django.utils.deprecation import RemovedInDjango40Warning
from django.utils.http import (
base36_to_int, escape_leading_slashes, http_date, int_to_base36,
- is_safe_url, is_same_domain, parse_etags, parse_http_date, quote_etag,
- url_has_allowed_host_and_scheme, urlencode, urlquote, urlquote_plus,
- urlsafe_base64_decode, urlsafe_base64_encode, urlunquote, urlunquote_plus,
+ is_safe_url, is_same_domain, limited_parse_qsl, parse_etags,
+ parse_http_date, quote_etag, url_has_allowed_host_and_scheme, urlencode,
+ urlquote, urlquote_plus, urlsafe_base64_decode, urlsafe_base64_encode,
+ urlunquote, urlunquote_plus,
)
@@ -359,3 +361,47 @@ class EscapeLeadingSlashesTests(unittest.TestCase):
for url, expected in tests:
with self.subTest(url=url):
self.assertEqual(escape_leading_slashes(url), expected)
+
+
+# Backport of unit tests for urllib.parse.parse_qsl() from Python 3.8.8.
+# Copyright (C) 2021 Python Software Foundation (see LICENSE.python).
+class ParseQSLBackportTests(unittest.TestCase):
+ def test_parse_qsl(self):
+ tests = [
+ ('', []),
+ ('&', []),
+ ('&&', []),
+ ('=', [('', '')]),
+ ('=a', [('', 'a')]),
+ ('a', [('a', '')]),
+ ('a=', [('a', '')]),
+ ('&a=b', [('a', 'b')]),
+ ('a=a+b&b=b+c', [('a', 'a b'), ('b', 'b c')]),
+ ('a=1&a=2', [('a', '1'), ('a', '2')]),
+ (';a=b', [(';a', 'b')]),
+ ('a=a+b;b=b+c', [('a', 'a b;b=b c')]),
+ ]
+ for original, expected in tests:
+ with self.subTest(original):
+ result = limited_parse_qsl(original, keep_blank_values=True)
+ self.assertEqual(result, expected, 'Error parsing %r' % original)
+ expect_without_blanks = [v for v in expected if len(v[1])]
+ result = limited_parse_qsl(original, keep_blank_values=False)
+ self.assertEqual(result, expect_without_blanks, 'Error parsing %r' % original)
+
+ def test_parse_qsl_encoding(self):
+ result = limited_parse_qsl('key=\u0141%E9', encoding='latin-1')
+ self.assertEqual(result, [('key', '\u0141\xE9')])
+ result = limited_parse_qsl('key=\u0141%C3%A9', encoding='utf-8')
+ self.assertEqual(result, [('key', '\u0141\xE9')])
+ result = limited_parse_qsl('key=\u0141%C3%A9', encoding='ascii')
+ self.assertEqual(result, [('key', '\u0141\ufffd\ufffd')])
+ result = limited_parse_qsl('key=\u0141%E9-', encoding='ascii')
+ self.assertEqual(result, [('key', '\u0141\ufffd-')])
+ result = limited_parse_qsl('key=\u0141%E9-', encoding='ascii', errors='ignore')
+ self.assertEqual(result, [('key', '\u0141-')])
+
+ def test_parse_qsl_field_limit(self):
+ with self.assertRaises(TooManyFieldsSent):
+ limited_parse_qsl('&'.join(['a=a'] * 11), fields_limit=10)
+ limited_parse_qsl('&'.join(['a=a'] * 10), fields_limit=10)