Use _PyFrame_GetLocalsArray

colesbury · Aug 13, 2024 · 5b9176b · 5b9176b
1 parent 44d7dff
commit 5b9176b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Python/gc.c b/Python/gc.c
@@ -537,10 +537,10 @@ visit_decref(PyObject *op, void *parent)
 int
 _PyGC_VisitFrameStack(_PyInterpreterFrame *frame, visitproc visit, void *arg)
 {
- _PyStackRef *i = frame->localsplus;
+ _PyStackRef *ref = _PyFrame_GetLocalsArray(frame);
  /* locals and stack */
- for (; i < frame->stackpointer; i++) {
- Py_VISIT(PyStackRef_AsPyObjectBorrow(*i));
+ for (; ref < frame->stackpointer; ref++) {
+ Py_VISIT(PyStackRef_AsPyObjectBorrow(*ref));
  }
  return 0;
 }

diff --git a/Python/gc_free_threading.c b/Python/gc_free_threading.c
@@ -954,7 +954,7 @@ visit_decref_unreachable(PyObject *op, void *data)
 int
 _PyGC_VisitFrameStack(_PyInterpreterFrame *frame, visitproc visit, void *arg)
 {
- _PyStackRef *ref = frame->localsplus;
+ _PyStackRef *ref = _PyFrame_GetLocalsArray(frame);
  /* locals and stack */
  for (; ref < frame->stackpointer; ref++) {
  // This is a bit tricky! We want to ignore deferred references when