diff options
author | Donald Stufft <donald@stufft.io> | 2015-03-29 20:43:23 (GMT) |
---|---|---|
committer | Donald Stufft <donald@stufft.io> | 2015-03-29 20:43:23 (GMT) |
commit | d90f8d10e088657593fa753ecacab95845d378aa (patch) | |
tree | 826e1017d337826a9b6818539d954ca4c1332c2b /Lib | |
parent | 1058cda38f1b409c4d52eef236f4915df592a112 (diff) | |
download | cpython-d90f8d10e088657593fa753ecacab95845d378aa.zip cpython-d90f8d10e088657593fa753ecacab95845d378aa.tar.gz cpython-d90f8d10e088657593fa753ecacab95845d378aa.tar.bz2 |
Closes #23801 - Ignore entire preamble to multipart in cgi.FieldStorage
Diffstat (limited to 'Lib')
-rwxr-xr-x | Lib/cgi.py | 9 | ||||
-rw-r--r-- | Lib/test/test_cgi.py | 19 |
2 files changed, 26 insertions, 2 deletions
@@ -693,8 +693,13 @@ class FieldStorage: raise ValueError("%s should return bytes, got %s" \ % (self.fp, type(first_line).__name__)) self.bytes_read += len(first_line) - # first line holds boundary ; ignore it, or check that - # b"--" + ib == first_line.strip() ? + + # Ensure that we consume the file until we've hit our inner boundary + while (first_line.strip() != (b"--" + self.innerboundary) and + first_line): + first_line = self.fp.readline() + self.bytes_read += len(first_line) + while True: parser = FeedParser() hdr_text = b"" diff --git a/Lib/test/test_cgi.py b/Lib/test/test_cgi.py index 1127dd1..d2c326b 100644 --- a/Lib/test/test_cgi.py +++ b/Lib/test/test_cgi.py @@ -248,6 +248,25 @@ class CgiTests(unittest.TestCase): got = getattr(fs.list[x], k) self.assertEqual(got, exp) + def test_fieldstorage_multipart_leading_whitespace(self): + env = { + 'REQUEST_METHOD': 'POST', + 'CONTENT_TYPE': 'multipart/form-data; boundary={}'.format(BOUNDARY), + 'CONTENT_LENGTH': '560'} + # Add some leading whitespace to our post data that will cause the + # first line to not be the innerboundary. + fp = BytesIO(b"\r\n" + POSTDATA.encode('latin-1')) + fs = cgi.FieldStorage(fp, environ=env, encoding="latin-1") + self.assertEqual(len(fs.list), 4) + expect = [{'name':'id', 'filename':None, 'value':'1234'}, + {'name':'title', 'filename':None, 'value':''}, + {'name':'file', 'filename':'test.txt', 'value':b'Testing 123.\n'}, + {'name':'submit', 'filename':None, 'value':' Add '}] + for x in range(len(fs.list)): + for k, exp in expect[x].items(): + got = getattr(fs.list[x], k) + self.assertEqual(got, exp) + def test_fieldstorage_multipart_non_ascii(self): #Test basic FieldStorage multipart parsing env = {'REQUEST_METHOD':'POST', |