使用 RecursiveASTVisitor 访问 Clang AST 时确定 Stmt 的父函数节点答案

【问题标题】：Determine parent function node of a Stmt when visiting Clang AST using RecursiveASTVisitor使用 RecursiveASTVisitor 访问 Clang AST 时确定 Stmt 的父函数节点
【发布时间】：2014-08-20 17:35:25
【问题描述】：

我正在学习如何使用 clang 的 libtooling 构建解析 C 的工具。

我正在使用RecursiveASTVisitor-inherited 类，因此它的所有遍历和访问方法都可用。我想知道在覆盖方法VisitStmt(Stmt *s)时是否可以确定语句的父函数节点。例如这里，我可以从s 获得FunctionDecl* 吗？

谢谢。

【问题讨论】：

标签： c++ c++11 clang llvm-clang

【解决方案1】：

我没有clang方面的专业知识，但你不能只存储你在上次VisitDecl调用中收到的值吗？

喜欢：

class FooVisitor : public RecursiveASTVisitor<FooVisitor> {
 public:
  explicit FooVisitor(ASTContext* context)
    : context_(context), current_func_(nullptr) {}

  bool VisitDecl(Decl *decl) {
     if (decl->isFunctionOrFunctionTemplate())
        current_func_ = decl->getAsFunction();
     return true;
  }

  bool VisitStmt(Stmt* stmt) {
     do_something(current_func_, stmt);
     return true;
  }
 private:
  ASTContext* context_;
  FunctionDecl* current_func_;
};

我还没有编译它，所以可能需要一些修正，但它应该能说明这个概念。

【讨论】：

谢谢，这可能是个主意。我只想知道 Clang AST 是如何连接的。
@TrúcNguyễnLâm：“Clang AST 的连接程度”是什么意思？

【解决方案2】：

由于你持有 ASTContext，我认为你可以通过 ASTContext::getParent() 访问 Node 的父母。

【讨论】：

【解决方案3】：

我认为这段代码是你想要的：

    const Stmt* ST = str;

    while (true) {
        //get parents
        const auto& parents = pContext->getParents(*ST);
        if ( parents.empty() ) {
            llvm::errs() << "Can not find parent\n";
            return false;
        }
        llvm::errs() << "find parent size=" << parents.size() << "\n";
        ST = parents[0].get<Stmt>();
        if (!ST)
            return false;
        ST->dump();
        if (isa<CompoundStmt>(ST))
            break;
    }

【讨论】：

【解决方案4】：

可能有点晚了。如果您将 decls 和 Stmts 的解析添加到@xawi2000，那么我想您会到达顶部节点。例如，你可能有

函数声明

compoundStmt

变量声明

Stmt

在这种情况下，您需要通过 stmt 进行解析，然后通过 decl 来获取 functionDecl。只解析过stmts，可能不会给出这种情况

【讨论】：