如何通过Clang C++ API获取AST节点关联指针及节点的唯一十六进制编码

一、获取与Clang AST节点关联的指针 #

在使用Clang的RecursiveASTVisitor遍历AST时，每个访问方法的参数本身就是对应AST节点的指针：

• 比如处理变量声明时，VisitVarDecl(VarDecl *VD)中的VD就是该VarDecl节点的指针；
• 处理赋值表达式时，VisitBinaryOperator(BinaryOperator *BO)中的BO就是该赋值节点的指针。

如果需要把指针转换成可用于标识的数值（比如后续关联操作的键），可以将其转为无符号整数类型：

#include <cstdint>

// 将VarDecl节点的指针转为整数形式（对应dump里的十六进制编码）
uintptr_t nodeIdentifier = reinterpret_cast<uintptr_t>(VD);

二、获取AST Dump中的十六进制编码并实现节点关联 #

你在AST Dump里看到的十六进制编码，本质就是AST节点对象的内存地址——也就是上面提到的nodeIdentifier的值。不需要找什么隐藏的存储字段或调用特定函数，直接用节点的指针就能拿到。

要实现变量引用与之前的使用/赋值操作关联，可以参考这个思路：

1. 遍历AST时，维护一个映射表（比如std::map<uintptr_t, std::vector<Stmt*>>），以节点的指针数值为键，存储该变量对应的所有历史操作节点；
2. 遇到DeclRefExpr（变量引用节点）时，通过DeclRefExpr->getDecl()拿到对应的变量声明节点，再用该声明节点的指针值去映射表中查找关联的操作；
3. 遇到赋值表达式（BinaryOperator且操作符为BO_Assign）时，提取左值对应的变量声明节点，把当前赋值节点存入映射表的对应条目里。

给你一个简单的代码片段示例：

#include "clang/AST/ASTVisitor.h"
#include "clang/AST/Stmt.h"
#include "clang/AST/Decl.h"
#include <map>
#include <vector>
#include "llvm/Support/Format.h"

class MyASTVisitor : public clang::RecursiveASTVisitor<MyASTVisitor> {
private:
  // 存储变量声明指针 -> 关联操作节点列表的映射
  std::map<uintptr_t, std::vector<clang::Stmt*>> varOperationMap;

public:
  bool VisitBinaryOperator(clang::BinaryOperator *BO) {
    if (BO->isAssignmentOp()) {
      // 提取赋值左值对应的变量声明
      if (auto *declRef = llvm::dyn_cast<clang::DeclRefExpr>(BO->getLHS()->IgnoreParenCasts())) {
        if (auto *varDecl = llvm::dyn_cast<clang::VarDecl>(declRef->getDecl())) {
          uintptr_t varKey = reinterpret_cast<uintptr_t>(varDecl);
          varOperationMap[varKey].push_back(BO);
          // 可以打印节点地址，和dump格式对比
          llvm::errs() << "记录赋值操作，变量节点地址：" << llvm::format("%p", varDecl) << "\n";
        }
      }
    }
    return true;
  }

  bool VisitDeclRefExpr(clang::DeclRefExpr *DRE) {
    if (auto *varDecl = llvm::dyn_cast<clang::VarDecl>(DRE->getDecl())) {
      uintptr_t varKey = reinterpret_cast<uintptr_t>(varDecl);
      // 查找该变量的历史操作
      auto it = varOperationMap.find(varKey);
      if (it != varOperationMap.end()) {
        llvm::errs() << "找到变量引用，关联的历史操作数量：" << it->second.size() << "\n";
        // 在这里实现你的计算逻辑
      }
    }
    return true;
  }
};

补充提示：

• 遍历期间Clang不会销毁AST节点，所以用指针值作为唯一标识是安全的；
• 如果需要输出和AST Dump完全一致的十六进制字符串，可以用llvm::format("%p", nodePtr)来生成（需要包含llvm/Support/Format.h）。

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