]> granicus.if.org Git - python/commitdiff
Merged revisions 76319 via svnmerge from
authorBenjamin Peterson <benjamin@python.org>
Mon, 16 Nov 2009 00:36:18 +0000 (00:36 +0000)
committerBenjamin Peterson <benjamin@python.org>
Mon, 16 Nov 2009 00:36:18 +0000 (00:36 +0000)
svn+ssh://pythondev@svn.python.org/python/branches/py3k

........
  r76319 | benjamin.peterson | 2009-11-15 18:34:25 -0600 (Sun, 15 Nov 2009) | 4 lines

  fix one visible and several possible refleaks in rangeobject.c

  In some cases, the code was just reordered to allow for less decrefing.
........

Objects/rangeobject.c

index c8b10eb6a8691c61f56f8c4555aaef3a6a675b4b..d7c2fb91649b3f021c133b7b8001419197599f8e 100644 (file)
@@ -476,6 +476,7 @@ int_range_iter(long start, long stop, long step)
     it->step = step;
     ulen = get_len_of_range(start, stop, step);
     if (ulen > (unsigned long)LONG_MAX) {
+        Py_DECREF(it);
         PyErr_SetString(PyExc_OverflowError,
                         "range too large to represent as a range_iterator");
         return NULL;
@@ -527,16 +528,14 @@ longrangeiter_next(longrangeiterobject *r)
     if (!one)
         return NULL;
 
-    product = PyNumber_Multiply(r->index, r->step);
-    if (!product) {
-        Py_DECREF(one);
-        return NULL;
-    }
-
     new_index = PyNumber_Add(r->index, one);
     Py_DECREF(one);
-    if (!new_index) {
-        Py_DECREF(product);
+    if (!new_index)
+        return NULL;
+
+    product = PyNumber_Multiply(r->index, r->step);
+    if (!product) {
+        Py_DECREF(new_index);
         return NULL;
     }
 
@@ -546,6 +545,9 @@ longrangeiter_next(longrangeiterobject *r)
         Py_DECREF(r->index);
         r->index = new_index;
     }
+    else {
+        Py_DECREF(new_index);
+    }
 
     return result;
 }
@@ -724,6 +726,9 @@ long_range:
     if (!len)
         goto create_failure;
 
+    /* Steal reference to len. */
+    it->len = len;
+
     one = PyLong_FromLong(1);
     if (!one)
         goto create_failure;
@@ -745,24 +750,16 @@ long_range:
         goto create_failure;
 
     it->step = PyNumber_Negative(range->step);
-    if (!it->step) {
-        Py_DECREF(it->start);
+    if (!it->step)
         goto create_failure;
-    }
-
-    /* Steal reference to len. */
-    it->len = len;
 
     it->index = PyLong_FromLong(0);
-    if (!it->index) {
-        Py_DECREF(it);
-        return NULL;
-    }
+    if (!it->index)
+        goto create_failure;
 
     return (PyObject *)it;
 
 create_failure:
-    Py_XDECREF(len);
-    PyObject_Del(it);
+    Py_DECREF(it);
     return NULL;
 }