LLVM中PtrUseVisitor类的正确使用方法及实现示例咨询

PtrUseVisitor采用和InstVisitor一致的CRTP设计模式，需先自定义派生类继承后使用，完整可运行的实现示例如下：

1. 自定义PtrUseVisitor派生类 #

#include "llvm/Analysis/PtrUseVisitor.h"
#include "llvm/IR/Instructions.h"
#include "llvm/Support/raw_ostream.h"

using namespace llvm;

class MyPtrUseVisitor : public PtrUseVisitor<MyPtrUseVisitor> {
public:
  // 核心回调：每遍历到一个指针使用者就会触发该方法
  bool visitUse(Use &U, unsigned Depth) {
    Instruction *User = dyn_cast<Instruction>(U.getUser());
    if (!User) return false;
    errs() << "指针使用者（深度" << Depth << "）：" << *User << "\n";
    // 返回false表示继续遍历所有使用，返回true可提前终止遍历
    return false;
  }

  // 可选：可以像InstVisitor一样重载特定指令的回调方法
  void visitStoreInst(StoreInst &SI) {
    errs() << "识别到Store类型的指针使用：" << SI << "\n";
  }
};

2. 结合InstVisitor的调用实现 #

class MyInstVisitor : public InstVisitor<MyInstVisitor> {
public:
  void visitLoadInst(LoadInst &I) {
    errs() << "定位到目标Load指令：" << I << "\n";
    Value *PtrOp = I.getPointerOperand();
    if (!PtrOp->getType()->isPointerTy()) return;
    
    // 实例化自定义访问器，调用visitPtr遍历该指针的所有使用者
    MyPtrUseVisitor PtrVisitor;
    PtrVisitor.visitPtr(PtrOp);
    errs() << "============================\n";
  }
};

// 入口调用逻辑
void visitor(Function &F) {
  MyInstVisitor MAV;
  MAV.visit(F);
}

常见问题说明 #

• PtrUseVisitor是模板基类，无法直接实例化，模板参数必须填写自定义派生类的类型，不可直接填Instruction
• visitPtr是成员方法，需通过派生类实例调用，不可直接调用基类静态方法
• 若需要限制遍历深度，可以在派生类初始化时传入最大深度参数，默认会递归遍历所有层级的指针使用

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