首页 > 解决方案 > 通过 sys.set_trace 访问 Python 解释器的数据栈

问题描述


我正在使用sys.set_trace(..)对 python 代码执行字节码分析。
更具体地说,我设置了一个跟踪函数,每次执行字节码时都会触发。

对于我的分析,我需要获取带有操作码的字节码"STORE_ATTR"将在其上存储值的对象。
例如我有这个字节码:

Instruction(opname='STORE_ATTR', opcode=95, arg=9, argval='pos', argrepr='pos', offset=188, starts_line=None, is_jump_target=False)

看到这一点后,解释器将尝试从数据堆栈中弹出顶部元素,以便在其中存储一个值。我也有办法访问该对象吗?是否有一个类似于TOP()函数的 API,我可以从跟踪器中调用它并获取当前正在执行的帧的堆栈的顶部值?

标签: pythoninterpretercpython

解决方案


不幸的是,此类信息无法通过 Python 访问,但您可以使用 C API 进行访问。这里的另一个问题是,这高度依赖于您拥有的 Python 版本。以下适用于Python 3.8.5,但稍作修改也适用于其他版本。

等价物PyEval_SetTrace可以访问类的f_stacktop字段PyFrameObject。该字段由TOP()at使用ceval.c

您还可以在此处验证f_stacktop没有暴露给 Pythonframe类。

在这里,您可以看到一个可以访问当前帧的 TOS 的代码示例:

int tracefunc(PyObject *obj, PyFrameObject *frame, int what, PyObject *arg) {
    [...]
    PyObject **stack_pointer = frame->f_stacktop;
    PyObject *tos = stack_pointer[-1];
    [...]
}

推荐阅读