Add weakref support generator-iterators.
Part of fixing SF bug #591704.
diff --git a/Python/ceval.c b/Python/ceval.c
index 9df336e..50ea9c3 100644
--- a/Python/ceval.c
+++ b/Python/ceval.c
@@ -94,6 +94,9 @@
/* True if generator is being executed. */
int gi_running;
+
+ /* List of weak reference. */
+ PyObject *gi_weakreflist;
} genobject;
static PyObject *
@@ -106,6 +109,7 @@
}
gen->gi_frame = f;
gen->gi_running = 0;
+ gen->gi_weakreflist = NULL;
_PyObject_GC_TRACK(gen);
return (PyObject *)gen;
}
@@ -120,6 +124,8 @@
gen_dealloc(genobject *gen)
{
_PyObject_GC_UNTRACK(gen);
+ if (gen->gi_weakreflist != NULL)
+ PyObject_ClearWeakRefs((PyObject *) gen);
Py_DECREF(gen->gi_frame);
PyObject_GC_Del(gen);
}
@@ -205,7 +211,7 @@
(traverseproc)gen_traverse, /* tp_traverse */
0, /* tp_clear */
0, /* tp_richcompare */
- 0, /* tp_weaklistoffset */
+ offsetof(genobject, gi_weakreflist), /* tp_weaklistoffset */
(getiterfunc)gen_getiter, /* tp_iter */
(iternextfunc)gen_iternext, /* tp_iternext */
0, /* tp_methods */