[Python-checkins] cpython (3.3): Fix possible NULL pointer dereferences in testcapi module

[christian.heimes]

http://hg.python.org/cpython/rev/d49f65ff4f3c
changeset:   84831:d49f65ff4f3c
branch:      3.3
parent:      84829:aedcf7ef8489
user:        Christian Heimes <christian at cheimes.de>
date:        Fri Jul 26 14:52:18 2013 +0200
summary:
  Fix possible NULL pointer dereferences in testcapi module
CID 1058280
CID 1058282
CID 1058284

files:
  Modules/_testcapimodule.c |  8 +++++++-
  1 files changed, 7 insertions(+), 1 deletions(-)


diff --git a/Modules/_testcapimodule.c b/Modules/_testcapimodule.c
--- a/Modules/_testcapimodule.c
+++ b/Modules/_testcapimodule.c
@@ -120,6 +120,10 @@
 
     for (i = 0; i < count; i++) {
         v = PyLong_FromLong(i);
+        if (v == NULL) {
+            Py_DECREF(v);
+            return -1;
+        }
         if (PyDict_SetItem(dict, v, v) < 0) {
             Py_DECREF(v);
             return -1;
@@ -1642,6 +1646,8 @@
 
     for (i = 0; i < Py_ARRAY_LENGTH(testcases); ++i) {
         PyObject *plong = PyLong_FromLong(testcases[i].input);
+        if (plong == NULL)
+            return NULL;
         size_t nbits = _PyLong_NumBits(plong);
         int sign = _PyLong_Sign(plong);
 
@@ -2234,7 +2240,7 @@
     gettimeofday(&start, NULL);
     for(i=0; i < 10000000; i++) {
         result = PyNumber_Add(op1, op1);
-        Py_DECREF(result);
+        Py_XDECREF(result);
     }
     gettimeofday(&stop, NULL);
     Py_DECREF(op1);

-- 
Repository URL: http://hg.python.org/cpython