SF patch #1020188: Use Py_CLEAR where necessary to avoid crashes

(Contributed by Dima Dorfman)

3 files changed, 6 insertions, 14 deletions

diff --git a/Objects/enumobject.c b/Objects/enumobject.c
index ebe7e7c..aac88db 100644
--- a/Objects/enumobject.c
+++ b/Objects/enumobject.c
@@ -230,10 +230,7 @@ reversed_next(reversedobject *ro)
 			PyErr_Clear();
 	}
 	ro->index = -1;
-	if (ro->seq != NULL) {
-		Py_DECREF(ro->seq);
-		ro->seq = NULL;
-	}
+	Py_CLEAR(ro->seq);
 	return NULL;
 }
 
diff --git a/Objects/genobject.c b/Objects/genobject.c
index 66a5106..174c697 100644
--- a/Objects/genobject.c
+++ b/Objects/genobject.c
@@ -51,8 +51,7 @@ gen_iternext(PyGenObject *gen)
 	 * may keep a chain of frames alive or it could create a reference
 	 * cycle. */
 	assert(f->f_back != NULL);
-	Py_DECREF(f->f_back);
-	f->f_back = NULL;
+	Py_CLEAR(f->f_back);
 
 	/* If the generator just returned (as opposed to yielding), signal
 	 * that the generator is exhausted. */
diff --git a/Objects/iterobject.c b/Objects/iterobject.c
index 25e4e11..b414cc2 100644
--- a/Objects/iterobject.c
+++ b/Objects/iterobject.c
@@ -192,18 +192,14 @@ calliter_iternext(calliterobject *it)
 				return result; /* Common case, fast path */
 			Py_DECREF(result);
 			if (ok > 0) {
-				Py_DECREF(it->it_callable);
-				it->it_callable = NULL;
-				Py_DECREF(it->it_sentinel);
-				it->it_sentinel = NULL;
+				Py_CLEAR(it->it_callable);
+				Py_CLEAR(it->it_sentinel);
 			}
 		}
 		else if (PyErr_ExceptionMatches(PyExc_StopIteration)) {
 			PyErr_Clear();
-			Py_DECREF(it->it_callable);
-			it->it_callable = NULL;
-			Py_DECREF(it->it_sentinel);
-			it->it_sentinel = NULL;
+			Py_CLEAR(it->it_callable);
+			Py_CLEAR(it->it_sentinel);
 		}
 	}
 	return NULL;