Add missing check of PyDict_SetItem()'s return value

CID 486659

Add missing check of PyDict_SetItem()'s return value
CID 486659
3f95bdcd · Christian Heimes · b578735d · 97cb67b9 · 3f95bdcd
Commit 3f95bdcd authored Jul 20, 2013 by Christian Heimes
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

Modules/_testcapimodule.c Modules/_testcapimodule.c +4 -1

No files found.
--- a/Modules/_testcapimodule.c
+++ b/Modules/_testcapimodule.c
@@ -120,7 +120,10 @@ test_dict_inner(int count)

    for (i = 0; i < count; i++) {
        v = PyLong_FromLong(i);
-        PyDict_SetItem(dict, v, v);
+        if (PyDict_SetItem(dict, v, v) < 0) {
+            Py_DECREF(v);
+            return -1;
+        }
        Py_DECREF(v);
    }