如何在不使用LLVM Pass的前提下通过Clang对C/C++代码插入指令？

Great question! 不用LLVM Pass在Clang内部给C/C++代码插桩确实是个实用的需求，而且完全可行——核心是利用Clang的AST（抽象语法树）遍历与修改能力，通过自定义前端动作（Frontend Action）来实现。下面给你拆解具体的实现思路和可落地的示例：

Clang的前端编译流程会把源码解析成AST，我们可以通过自定义ASTFrontendAction和ASTConsumer来遍历AST节点，精准定位到你想要插桩的位置（比如变量初始化语句之后），然后手动构建新的AST节点（也就是你要插入的指令），最后让Clang基于修改后的AST生成新的代码。

步骤1：精准定位目标AST节点 #

比如你要在int num = 0;之后插入代码，需要匹配对应的变量声明节点（VarDecl），可以用Clang的AST Matchers来做精准筛选，示例Matcher规则如下：

auto targetMatcher = varDecl(
    hasName("num"),                    // 匹配变量名为num
    hasAncestor(functionDecl(hasName("main"))), // 限定在main函数内
    hasInitializer(integerLiteral(equals(0)))   // 匹配初始化值为0的情况
).bind("targetVar");

步骤2：构建并插入自定义AST节点 #

找到目标节点后，我们需要在它的后续位置插入新的语句（比如if(num !=0) { ... }）。这一步需要手动构建AST节点并替换原有的语句块，示例代码片段如下（基于Clang的C++ API）：

// 自定义AST消费者，负责遍历和修改AST
class InsertCodeConsumer : public ASTConsumer {
public:
  explicit InsertCodeConsumer(ASTContext &Ctx) : Context(Ctx) {}

  void HandleTranslationUnit(ASTContext &Ctx) override {
    // 定义Matcher的回调逻辑
    class InsertCallback : public MatchFinder::MatchCallback {
    public:
      InsertCallback(ASTContext &Context) : Ctx(Context) {}

      void run(const MatchFinder::MatchResult &Result) override {
        // 获取匹配到的目标变量声明
        if (const auto *targetVar = Result.Nodes.getNodeAs<VarDecl>("targetVar")) {
          // 获取变量所在的语句块
          if (const auto *parentBlock = dyn_cast<CompoundStmt>(targetVar->getParent())) {
            ASTContext &ctx = Result.Context;
            SourceManager &sm = ctx.getSourceManager();
            SourceLocation loc = targetVar->getLocation();

            // 1. 构建if条件：num != 0
            Expr *varRef = DeclRefExpr::Create(ctx, targetVar->getDeclNameInfo(), nullptr,
                                               targetVar->getType(), VK_LValue, loc);
            Expr *zeroVal = IntegerLiteral::Create(ctx, APInt(32, 0), ctx.IntTy, loc);
            Expr *condExpr = BinaryOperator::Create(ctx, varRef, zeroVal, BO_NE, ctx.BoolTy,
                                                   VK_RValue, OK_Ordinary, loc);

            // 2. 构建if体内的自定义代码（示例：打印提示）
            StringLiteral *hintStr = StringLiteral::Create(ctx, "num is not zero!", loc);
            Expr *printArg = ImplicitCastExpr::Create(ctx, ctx.getPointerType(ctx.CharTy),
                                                     CK_NoOp, hintStr, nullptr, VK_RValue);
            DeclRefExpr *printfRef = DeclRefExpr::Create(ctx, DeclarationNameInfo(&ctx.Idents.get("printf")),
                                                       nullptr, ctx.getFunctionType(ctx.IntTy, {ctx.getPointerType(ctx.CharTy)}, false),
                                                       VK_RValue, loc);
            CallExpr *printCall = CallExpr::Create(ctx, printfRef, {printArg}, ctx.IntTy, VK_RValue, loc);
            Stmt *printStmt = ExprStmt::Create(ctx, printCall);

            // 3. 构建if语句的复合块
            SmallVector<Stmt*, 1> bodyStmts;
            bodyStmts.push_back(printStmt);
            CompoundStmt *ifBody = CompoundStmt::Create(ctx, bodyStmts, loc, loc);

            // 4. 构建完整的if语句
            IfStmt *newIfStmt = IfStmt::Create(ctx, nullptr, condExpr, ifBody, nullptr, loc);

            // 5. 将if语句插入到变量声明之后
            SmallVector<Stmt*, 8> newBlockStmts;
            for (auto *stmt : parentBlock->body()) {
              newBlockStmts.push_back(stmt);
              if (stmt == targetVar) {
                newBlockStmts.push_back(newIfStmt);
              }
            }

            // 6. 替换原有的语句块
            CompoundStmt *newBlock = CompoundStmt::Create(ctx, newBlockStmts, parentBlock->getBeginLoc(), parentBlock->getEndLoc());
            ctx.ReplaceStmt(const_cast<CompoundStmt*>(parentBlock), newBlock);
          }
        }
      }
    private:
      ASTContext &Ctx;
    };

    // 注册Matcher并执行AST匹配
    MatchFinder finder;
    InsertCallback callback(Context);
    finder.addMatcher(targetMatcher, &callback);
    finder.matchAST(Context);
  }

private:
  ASTContext &Context;
};

// 自定义Frontend Action，用于加载我们的AST消费者
class InsertCodeAction : public ASTFrontendAction {
public:
  std::unique_ptr<ASTConsumer> CreateASTConsumer(CompilerInstance &CI, StringRef File) override {
    return std::make_unique<InsertCodeConsumer>(CI.getASTContext());
  }
};

// 注册这个自定义Action，方便Clang调用
static FrontendActionRegistry::Add<InsertCodeAction> X("insert-custom-code", "Insert custom code into target location");

步骤3：编译插件并使用 #

1. 用CMake编译上述代码为Clang插件（需要链接Clang的相关库），示例CMake配置：

cmake_minimum_required(VERSION 3.13)
project(InsertCodePlugin)

find_package(LLVM REQUIRED CONFIG)
find_package(Clang REQUIRED CONFIG)

include_directories(${LLVM_INCLUDE_DIRS} ${CLANG_INCLUDE_DIRS})
add_definitions(${LLVM_DEFINITIONS} ${CLANG_DEFINITIONS})

add_library(InsertCodePlugin MODULE InsertCode.cpp)
target_link_libraries(InsertCodePlugin PRIVATE clangAST clangASTMatchers clangFrontend clangTooling)

2. 用Clang加载插件处理你的源码：

clang -Xclang -load -Xclang ./libInsertCodePlugin.so -Xclang -add-plugin -Xclang insert-custom-code your_source.c

其他可选方案 #

• Clang Rewriter：如果不需要严格的AST层面修改，只是做文本插桩，可以用Clang的Rewriter工具，结合AST遍历找到源码位置后直接插入文本代码，实现更简单但精度稍弱。
• 预处理钩子：如果插桩逻辑固定，也可以结合自定义宏和Clang的预处理扩展能力，但灵活性不如AST方法。

内容的提问来源于stack exchange，提问作者user9478968