commit a6296d34a478b4f697ea9db798146195075d496c
author INADA Naoki <methane@users.noreply.github.com> Thu Aug 24 14:55:17 2017 +0900
committer GitHub <noreply@github.com> Thu Aug 24 14:55:17 2017 +0900
tree 6a26d56297f7d85dd6a8f18bca96e0c4ffb60802
parent bf9075a0c55186d2f34df63e6c8512dd6414ff4b

bpo-31095: fix potential crash during GC (GH-2974)

Objects/setobject.c

diff --git a/Objects/setobject.c b/Objects/setobject.c
index 2347b9d..5c61bc7 100644
--- a/Objects/setobject.c
+++ b/Objects/setobject.c
@@ -553,6 +553,7 @@
     setentry *entry;
     Py_ssize_t used = so->used;
 
+    /* bpo-31095: UnTrack is needed before calling any callbacks */
     PyObject_GC_UnTrack(so);
     Py_TRASHCAN_SAFE_BEGIN(so)
     if (so->weakreflist != NULL)
@@ -809,6 +810,8 @@
 static void
 setiter_dealloc(setiterobject *si)
 {
+    /* bpo-31095: UnTrack is needed before calling any callbacks */
+    _PyObject_GC_UNTRACK(si);
     Py_XDECREF(si->si_set);
     PyObject_GC_Del(si);
 }