Issue #16228: Fix a crash in the json module where a list changes size while it is being encoded.

Patch by Serhiy Storchaka.
This commit is contained in:
Antoine Pitrou 2012-11-01 19:52:06 +01:00
parent 42d5c41522
commit 9f69e79c45
3 changed files with 14 additions and 7 deletions

View File

@ -19,6 +19,14 @@ class TestDump:
{2: 3.0, 4.0: 5, False: 1, 6: True}, sort_keys=True), {2: 3.0, 4.0: 5, False: 1, 6: True}, sort_keys=True),
'{"false": 1, "2": 3.0, "4.0": 5, "6": true}') '{"false": 1, "2": 3.0, "4.0": 5, "6": true}')
# Issue 16228: Crash on encoding resized list
def test_encode_mutated(self):
a = [object()] * 10
def crasher(obj):
del a[-1]
self.assertEqual(self.dumps(a, default=crasher),
'[null, null, null, null, null]')
class TestPyDump(TestDump, PyTest): pass class TestPyDump(TestDump, PyTest): pass
class TestCDump(TestDump, CTest): pass class TestCDump(TestDump, CTest): pass

View File

@ -143,6 +143,9 @@ Core and Builtins
Library Library
------- -------
- Issue #16228: Fix a crash in the json module where a list changes size
while it is being encoded. Patch by Serhiy Storchaka.
- Issue #14897: Enhance error messages of struct.pack and - Issue #14897: Enhance error messages of struct.pack and
struct.pack_into. Patch by Matti Mäki. struct.pack_into. Patch by Matti Mäki.

View File

@ -1585,8 +1585,6 @@ encoder_listencode_list(PyEncoderObject *s, PyObject *rval, PyObject *seq, Py_ss
static PyObject *empty_array = NULL; static PyObject *empty_array = NULL;
PyObject *ident = NULL; PyObject *ident = NULL;
PyObject *s_fast = NULL; PyObject *s_fast = NULL;
Py_ssize_t num_items;
PyObject **seq_items;
Py_ssize_t i; Py_ssize_t i;
if (open_array == NULL || close_array == NULL || empty_array == NULL) { if (open_array == NULL || close_array == NULL || empty_array == NULL) {
@ -1600,8 +1598,7 @@ encoder_listencode_list(PyEncoderObject *s, PyObject *rval, PyObject *seq, Py_ss
s_fast = PySequence_Fast(seq, "_iterencode_list needs a sequence"); s_fast = PySequence_Fast(seq, "_iterencode_list needs a sequence");
if (s_fast == NULL) if (s_fast == NULL)
return -1; return -1;
num_items = PySequence_Fast_GET_SIZE(s_fast); if (PySequence_Fast_GET_SIZE(s_fast) == 0) {
if (num_items == 0) {
Py_DECREF(s_fast); Py_DECREF(s_fast);
return PyList_Append(rval, empty_array); return PyList_Append(rval, empty_array);
} }
@ -1622,7 +1619,6 @@ encoder_listencode_list(PyEncoderObject *s, PyObject *rval, PyObject *seq, Py_ss
} }
} }
seq_items = PySequence_Fast_ITEMS(s_fast);
if (PyList_Append(rval, open_array)) if (PyList_Append(rval, open_array))
goto bail; goto bail;
if (s->indent != Py_None) { if (s->indent != Py_None) {
@ -1634,8 +1630,8 @@ encoder_listencode_list(PyEncoderObject *s, PyObject *rval, PyObject *seq, Py_ss
buf += newline_indent buf += newline_indent
*/ */
} }
for (i = 0; i < num_items; i++) { for (i = 0; i < PySequence_Fast_GET_SIZE(s_fast); i++) {
PyObject *obj = seq_items[i]; PyObject *obj = PySequence_Fast_GET_ITEM(s_fast, i);
if (i) { if (i) {
if (PyList_Append(rval, s->item_separator)) if (PyList_Append(rval, s->item_separator))
goto bail; goto bail;