Skip to content

Commit 123bfbb

Browse files
gpsheadsethmlarson
andcommitted
[3.10] gh-143916: Reject control characters in wsgiref.headers.Headers (GH-143917) (GH-143973)
gh-143916: Reject control characters in wsgiref.headers.Headers (GH-143917) * Add 'test.support' fixture for C0 control characters * gh-143916: Reject control characters in wsgiref.headers.Headers (cherry picked from commit f7fceed) (cherry picked from commit 22e4d55) Co-authored-by: Gregory P. Smith <68491+gpshead@users.noreply.github.com> Co-authored-by: Seth Michael Larson <seth@python.org>
1 parent f12346d commit 123bfbb

File tree

4 files changed

+23
-1
lines changed

4 files changed

+23
-1
lines changed

Lib/test/support/__init__.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2157,3 +2157,10 @@ def adjust_int_max_str_digits(max_digits):
21572157
yield
21582158
finally:
21592159
sys.set_int_max_str_digits(current)
2160+
2161+
2162+
def control_characters_c0() -> list[str]:
2163+
"""Returns a list of C0 control characters as strings.
2164+
C0 control characters defined as the byte range 0x00-0x1F, and 0x7F.
2165+
"""
2166+
return [chr(c) for c in range(0x00, 0x20)] + ["\x7F"]

Lib/test/test_wsgiref.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from unittest import mock
22
from test import support
3-
from test.support import socket_helper
3+
from test.support import socket_helper, control_characters_c0
44
from test.support import warnings_helper
55
from test.test_httpservers import NoLogRequestHandler
66
from unittest import TestCase
@@ -527,6 +527,16 @@ def testExtras(self):
527527
'\r\n'
528528
)
529529

530+
def testRaisesControlCharacters(self):
531+
headers = Headers()
532+
for c0 in control_characters_c0():
533+
self.assertRaises(ValueError, headers.__setitem__, f"key{c0}", "val")
534+
self.assertRaises(ValueError, headers.__setitem__, "key", f"val{c0}")
535+
self.assertRaises(ValueError, headers.add_header, f"key{c0}", "val", param="param")
536+
self.assertRaises(ValueError, headers.add_header, "key", f"val{c0}", param="param")
537+
self.assertRaises(ValueError, headers.add_header, "key", "val", param=f"param{c0}")
538+
539+
530540
class ErrorHandler(BaseCGIHandler):
531541
"""Simple handler subclass for testing BaseHandler"""
532542

Lib/wsgiref/headers.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
# existence of which force quoting of the parameter value.
1010
import re
1111
tspecials = re.compile(r'[ \(\)<>@,;:\\"/\[\]\?=]')
12+
_control_chars_re = re.compile(r'[\x00-\x1F\x7F]')
1213

1314
def _formatparam(param, value=None, quote=1):
1415
"""Convenience function to format and return a key=value pair.
@@ -41,6 +42,8 @@ def __init__(self, headers=None):
4142
def _convert_string_type(self, value):
4243
"""Convert/check value type."""
4344
if type(value) is str:
45+
if _control_chars_re.search(value):
46+
raise ValueError("Control characters not allowed in headers")
4447
return value
4548
raise AssertionError("Header names/values must be"
4649
" of type str (got {0})".format(repr(value)))
Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
Reject C0 control characters within wsgiref.headers.Headers fields, values,
2+
and parameters.

0 commit comments

Comments
 (0)