Skip to content

Commit 67bc59d

Browse files
serhiy-storchakahroncok
authored andcommitted
00422: pythongh-115133: Fix tests for XMLPullParser with Expat 2.6.0
Feeding the parser by too small chunks defers parsing to prevent CVE-2023-52425. Future versions of Expat may be more reactive. (cherry picked from commit 4a08e7b)
1 parent 159d1ad commit 67bc59d

File tree

2 files changed

+38
-22
lines changed

2 files changed

+38
-22
lines changed

Lib/test/test_xml_etree.py

+36-22
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
import io
1111
import operator
1212
import pickle
13+
import pyexpat
1314
import sys
1415
import types
1516
import unittest
@@ -97,6 +98,10 @@
9798
<document>&entity;</document>
9899
"""
99100

101+
fails_with_expat_2_6_0 = (unittest.expectedFailure
102+
if pyexpat.version_info >= (2, 6, 0) else
103+
lambda test: test)
104+
100105
class ModuleTest(unittest.TestCase):
101106
def test_sanity(self):
102107
# Import sanity.
@@ -1044,28 +1049,37 @@ def assert_event_tags(self, parser, expected):
10441049
self.assertEqual([(action, elem.tag) for action, elem in events],
10451050
expected)
10461051

1047-
def test_simple_xml(self):
1048-
for chunk_size in (None, 1, 5):
1049-
with self.subTest(chunk_size=chunk_size):
1050-
parser = ET.XMLPullParser()
1051-
self.assert_event_tags(parser, [])
1052-
self._feed(parser, "<!-- comment -->\n", chunk_size)
1053-
self.assert_event_tags(parser, [])
1054-
self._feed(parser,
1055-
"<root>\n <element key='value'>text</element",
1056-
chunk_size)
1057-
self.assert_event_tags(parser, [])
1058-
self._feed(parser, ">\n", chunk_size)
1059-
self.assert_event_tags(parser, [('end', 'element')])
1060-
self._feed(parser, "<element>text</element>tail\n", chunk_size)
1061-
self._feed(parser, "<empty-element/>\n", chunk_size)
1062-
self.assert_event_tags(parser, [
1063-
('end', 'element'),
1064-
('end', 'empty-element'),
1065-
])
1066-
self._feed(parser, "</root>\n", chunk_size)
1067-
self.assert_event_tags(parser, [('end', 'root')])
1068-
self.assertIsNone(parser.close())
1052+
def test_simple_xml(self, chunk_size=None):
1053+
parser = ET.XMLPullParser()
1054+
self.assert_event_tags(parser, [])
1055+
self._feed(parser, "<!-- comment -->\n", chunk_size)
1056+
self.assert_event_tags(parser, [])
1057+
self._feed(parser,
1058+
"<root>\n <element key='value'>text</element",
1059+
chunk_size)
1060+
self.assert_event_tags(parser, [])
1061+
self._feed(parser, ">\n", chunk_size)
1062+
self.assert_event_tags(parser, [('end', 'element')])
1063+
self._feed(parser, "<element>text</element>tail\n", chunk_size)
1064+
self._feed(parser, "<empty-element/>\n", chunk_size)
1065+
self.assert_event_tags(parser, [
1066+
('end', 'element'),
1067+
('end', 'empty-element'),
1068+
])
1069+
self._feed(parser, "</root>\n", chunk_size)
1070+
self.assert_event_tags(parser, [('end', 'root')])
1071+
self.assertIsNone(parser.close())
1072+
1073+
@fails_with_expat_2_6_0
1074+
def test_simple_xml_chunk_1(self):
1075+
self.test_simple_xml(chunk_size=1)
1076+
1077+
@fails_with_expat_2_6_0
1078+
def test_simple_xml_chunk_5(self):
1079+
self.test_simple_xml(chunk_size=5)
1080+
1081+
def test_simple_xml_chunk_22(self):
1082+
self.test_simple_xml(chunk_size=22)
10691083

10701084
def test_feed_while_iterating(self):
10711085
parser = ET.XMLPullParser()
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
Fix tests for :class:`~xml.etree.ElementTree.XMLPullParser` with Expat
2+
2.6.0.

0 commit comments

Comments
 (0)