咨询：如何在LLVM Clang中实现变量创建前的代码注入功能

嘿，我刚好折腾过类似的LLVM Pass开发，给你一步步拆解怎么实现你要的功能：

因为Clang会把C++源码转换成LLVM IR（中间表示），直接在IR层面插入函数调用比修改源码更可靠，也更符合LLVM的设计思路。核心是用FunctionPass遍历每个函数，找到局部变量对应的AllocaInst（变量分配指令），然后在它前面插入foo()的调用指令。

1. 搭建LLVM Pass开发环境 #

确保你已经安装了LLVM开发环境（带源码或预编译的开发包），然后创建一个新的Pass项目，用LLVM的CMake模板配置编译。

2. 编写Pass代码 #

下面是一个完整的Pass实现，会在所有局部变量（包括指针）分配前插入foo()：

#include "llvm/IR/Function.h"
#include "llvm/Pass.h"
#include "llvm/IR/IRBuilder.h"
#include "llvm/Support/raw_ostream.h"

using namespace llvm;

namespace {
// 定义我们的Pass，继承自FunctionPass
struct InsertFooBeforeVars : public FunctionPass {
  static char ID;
  InsertFooBeforeVars() : FunctionPass(ID) {}

  bool runOnFunction(Function &F) override {
    Module *module = F.getParent();
    LLVMContext &ctx = module->getContext();

    // 第一步：获取或声明foo函数的原型（如果模块里还没有的话）
    FunctionType *fooFuncType = FunctionType::get(
      Type::getVoidTy(ctx),  // 返回值是void
      false                  // 无参数
    );
    Function *fooFunc = module->getFunction("foo");
    if (!fooFunc) {
      // 如果模块里没找到foo，就创建一个外部声明
      fooFunc = Function::Create(
        fooFuncType,
        Function::ExternalLinkage,
        "foo",
        module
      );
    }

    // 第二步：遍历函数里的每个基本块和指令
    for (auto &basicBlock : F) {
      for (auto &inst : basicBlock) {
        // 识别局部变量分配指令：AllocaInst对应源码里的局部变量/指针创建
        if (AllocaInst *allocInst = dyn_cast<AllocaInst>(&inst)) {
          // 在当前分配指令前插入foo()调用
          IRBuilder<> builder(allocInst);  // 把builder定位到allocInst前面
          builder.CreateCall(fooFunc);

          // 可选：打印日志确认插入成功
          errs() << "Inserted foo() before variable: " << *allocInst << "\n";
        }
      }
    }

    return true;  // 返回true表示我们修改了IR
  }
};
} // 匿名命名空间结束

// 注册Pass
char InsertFooBeforeVars::ID = 0;
static RegisterPass<InsertFooBeforeVars> X(
  "insert-foo-before-vars",
  "Insert foo() call before every local variable allocation",
  false,  // 不修改CFG
  false   // 不是分析Pass
);

3. 编译&使用Pass #

• 用CMake编译这个Pass成动态库（比如libInsertFooBeforeVars.so）
• 用Clang加载Pass处理你的代码：

  clang -Xclang -load -Xclang ./libInsertFooBeforeVars.so your_source.cpp -o output
  

  处理后的代码对应的IR里，每个alloca指令前都会有call void @foo()，最终生成的可执行文件会在变量创建前调用foo()。

• 全局变量处理：上面的代码只处理局部变量，如果需要在全局变量初始化前插入调用，需要遍历模块的GlobalVariable列表，然后在模块的初始化函数（比如llvm.global_ctors）里添加调用。
• 特定变量过滤：如果你只想给特定类型/名字的变量插入调用，可以在识别AllocaInst后，增加判断逻辑（比如通过allocInst->getAllocatedType()判断类型，或者通过变量的Debug信息关联源码里的名字）。
• foo函数的参数：如果foo()需要参数，修改FunctionType的参数列表即可，比如FunctionType::get(Type::getVoidTy(ctx), {Type::getInt32Ty(ctx)}, false)表示接受一个int参数。

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