From: Victor Stinner <victor.stinner@gmail.com>
Date: Wed, 17 Jul 2013 23:12:35 +0000 (+0200)
Subject: Issue #18408: Fix array_tolist(), handle PyList_SetItem() failure
X-Git-Tag: v3.4.0a1~158
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=4755beac3cb918a64af956b3fda3ebb5ee85c96b;p=python

Issue #18408: Fix array_tolist(), handle PyList_SetItem() failure
---

diff --git a/Modules/arraymodule.c b/Modules/arraymodule.c
index 6b37cacafe..f5706ddc41 100644
--- a/Modules/arraymodule.c
+++ b/Modules/arraymodule.c
@@ -1027,7 +1027,7 @@ array_contains(arrayobject *self, PyObject *v)
     for (i = 0, cmp = 0 ; cmp == 0 && i < Py_SIZE(self); i++) {
         PyObject *selfi = getarrayitem((PyObject *)self, i);
         if (selfi == NULL)
-            return NULL;
+            return -1;
         cmp = PyObject_RichCompareBool(selfi, v, Py_EQ);
         Py_DECREF(selfi);
     }
@@ -1405,13 +1405,16 @@ array_tolist(arrayobject *self, PyObject *unused)
         return NULL;
     for (i = 0; i < Py_SIZE(self); i++) {
         PyObject *v = getarrayitem((PyObject *)self, i);
-        if (v == NULL) {
-            Py_DECREF(list);
-            return NULL;
-        }
-        PyList_SetItem(list, i, v);
+        if (v == NULL)
+            goto error;
+        if (PyList_SetItem(list, i, v) < 0)
+            goto error;
     }
     return list;
+
+error:
+    Py_DECREF(list);
+    return NULL;
 }
 
 PyDoc_STRVAR(tolist_doc,