summaryrefslogtreecommitdiffstats
path: root/Objects/iterobject.c
diff options
context:
space:
mode:
authorIonite <dev@ionite.io>2023-02-25 03:50:53 (GMT)
committerGitHub <noreply@github.com>2023-02-25 03:50:53 (GMT)
commit9f472f81bc2636eda212c12796b8d6581783bcef (patch)
tree55e89ad5142c112bc8ff0c63af7c5a62b5f6f1df /Objects/iterobject.c
parent3e80d21b7673edf70753e14d88907c60bc6970c3 (diff)
downloadcpython-9f472f81bc2636eda212c12796b8d6581783bcef.zip
cpython-9f472f81bc2636eda212c12796b8d6581783bcef.tar.gz
cpython-9f472f81bc2636eda212c12796b8d6581783bcef.tar.bz2
[3.10] gh-101765: Fix SystemError / segmentation fault in iter `__reduce__` when internal access of `builtins.__dict__` exhausts the iterator (GH-101769) (#102229)
(cherry picked from commit 54dfa14c5a94b893b67a4d9e9e403ff538ce9023)
Diffstat (limited to 'Objects/iterobject.c')
-rw-r--r--Objects/iterobject.c22
1 files changed, 16 insertions, 6 deletions
diff --git a/Objects/iterobject.c b/Objects/iterobject.c
index e493e41..980c04b 100644
--- a/Objects/iterobject.c
+++ b/Objects/iterobject.c
@@ -104,11 +104,16 @@ PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(
static PyObject *
iter_reduce(seqiterobject *it, PyObject *Py_UNUSED(ignored))
{
+ PyObject *iter = _PyEval_GetBuiltinId(&PyId_iter);
+
+ /* _PyEval_GetBuiltinId can invoke arbitrary code,
+ * call must be before access of iterator pointers.
+ * see issue #101765 */
+
if (it->it_seq != NULL)
- return Py_BuildValue("N(O)n", _PyEval_GetBuiltinId(&PyId_iter),
- it->it_seq, it->it_index);
+ return Py_BuildValue("N(O)n", iter, it->it_seq, it->it_index);
else
- return Py_BuildValue("N(())", _PyEval_GetBuiltinId(&PyId_iter));
+ return Py_BuildValue("N(())", iter);
}
PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
@@ -243,11 +248,16 @@ calliter_iternext(calliterobject *it)
static PyObject *
calliter_reduce(calliterobject *it, PyObject *Py_UNUSED(ignored))
{
+ PyObject *iter = _PyEval_GetBuiltinId(&PyId_iter);
+
+ /* _PyEval_GetBuiltinId can invoke arbitrary code,
+ * call must be before access of iterator pointers.
+ * see issue #101765 */
+
if (it->it_callable != NULL && it->it_sentinel != NULL)
- return Py_BuildValue("N(OO)", _PyEval_GetBuiltinId(&PyId_iter),
- it->it_callable, it->it_sentinel);
+ return Py_BuildValue("N(OO)", iter, it->it_callable, it->it_sentinel);
else
- return Py_BuildValue("N(())", _PyEval_GetBuiltinId(&PyId_iter));
+ return Py_BuildValue("N(())", iter);
}
static PyMethodDef calliter_methods[] = {