pycparser 检索函数调用层次结构

Question

有没有办法使用 pycparser 获取函数调用堆栈？

例如，如果我有这样的代码：

static int Func_1(int *val)
{
    unsigned int data_1 = 0;
    int result;
    result = Func_2(val,
                    data_1
                   );
    result = Func_3(val,
                    result
                   );                 
    return result;
}

所以使用FuncDefVisiter我可以检索Func_1，使用FuncCallVisiter我可以检索Func_2和Func_3

但是，我如何检索从Func_1中调用例如Func_2的信息？

Answer 1

您可以编写一个查找FuncDef节点的访问者，并在这些节点上运行另一个访问者以查找所有FuncCall节点。例如：

class FuncDefVisitor(c_ast.NodeVisitor):
    def visit_FuncDef(self, node):
        # Here node is a FuncDef node for the function definition.
        v = FuncCallVisitor()
        v.visit(node)
class FuncCallVisitor(c_ast.NodeVisitor):
    def visit_FuncCall(self, node):
        # do something

在visit_FuncDef中，我们创建了一个新的调用访问者并在node上运行它，这意味着它只在函数定义的AST节点内运行。它将在代码中找到Func_2和Func_3，但不会在其他函数中找到 func 调用。

---

也就是说，请注意，您只会从中获取静态调用图。在 C 中，函数可以通过函数指针调用其他人，这在一般情况下不是静态知道的。