Question

我试图遍历所有函数定义并从中提取信息。我必须遍历函数体中的所有语句，并根据类型执行特定的函数。

目前，我有一个难看的if-else块。有没有更优雅的方法可以做到这一点？

void FunctionMatcher::processStatement(const clang::Stmt *statement) {
    string type = statement->getStmtClassName();
    if (type == "ReturnStmt") {
        auto rs = dyn_cast<const ReturnStmt *>(statement);
        processReturnStmt(rs);
    } else if (type == "WhileStmt") {
        auto ws = dyn_cast<WhileStmt>(statement);
        processWhileStmt(ws);
    } else if (type == "ForStmt") {
        auto fs = dyn_cast<const ForStmt *>(statement);
        processForStmt(fs);
    } else if (type == "IfStmt") {
        auto is = dyn_cast<const IfStmt *>(statement);
        processIfStmt(is);
    } else if (type == "SwitchStmt") {
        auto ss = dyn_cast<const SwitchStmt *>(statement);
        processSwitchStmt(ss);
    } else if (type == "CompoundStmt") {
        auto cs = dyn_cast<const CompoundStmt *>(statement);
        for (auto child : cs->children())
            processStatement(child);
    } else {
      // ...
    }

Answer 1

通过浏览clang :: TextNodeDumper的代码，我找到了一个解决方案。显然Clang有自己的访客来进行声明，声明等... 简单的例子：

class StatementVisitor : public ConstStmtVisitor<StatementVisitor> {

public:
    StatementVisitor();

    void Visit(const Stmt *Node) {
        ConstStmtVisitor<StatementVisitor>::Visit(Node);
    }

    void VisitIfStmt(const IfStmt *Node) {
        llvm::outs() << " An if statement yay!\n";
    }

    void VisitWhileStmt(const WhileStmt *Node) {
        llvm::outs() << " A While statement yay!\n";
    }
};

Answer 2

您可以使用RecursiveASTVisitor

它以递归方式遍历给定代码中的所有语句

class MyASTVisitor : public RecursiveASTVisitor<MyASTVisitor>
{
    public:
    bool VisitFunctionDecl(FunctionDecl* f)
    {
        ...
    }

    bool VisitIfStmt(IfStmt* IF)
    {
        ...
    }

    bool VisitForStmt(ForStmt* FS)
    {
        ...
    }

    bool VisitWhileStmt(WhileStmt* WS)
    {
        ...
    }
}

有没有一种优雅的方式遍历Clang AST语句？

2 个答案: