如何通过libTooling（Clang C++ API）实现CPP转IR及相关AST操作？

Clang的libTooling、AST及CodeGen模块完全支持你要的功能，以下是分步实现的具体示例和说明：

一、从CPP文件生成AST #

通过ClangTool加载编译参数，结合自定义ASTFrontendAction即可获取完整AST。核心是实现ASTConsumer来处理翻译单元：

#include "clang/Tooling/Tooling.h"
#include "clang/AST/ASTConsumer.h"
#include "clang/AST/ASTContext.h"

class ASTPrinterConsumer : public clang::ASTConsumer {
public:
  explicit ASTPrinterConsumer(clang::ASTContext* Ctx) : Context(Ctx) {}
  void HandleTranslationUnit(clang::ASTContext& Ctx) override {
    // 这里可以遍历、分析AST节点，比如获取根节点TranslationUnitDecl
    auto* TU = Ctx.getTranslationUnitDecl();
    // 示例：打印所有顶层声明的名称
    for (auto* Decl : TU->decls()) {
      if (auto* NamedDecl = clang::dyn_cast<clang::NamedDecl>(Decl)) {
        llvm::errs() << "Top-level decl: " << NamedDecl->getNameAsString() << "\n";
      }
    }
  }
private:
  clang::ASTContext* Context;
};

class ASTPrinterAction : public clang::ASTFrontendAction {
public:
  std::unique_ptr<clang::ASTConsumer> CreateASTConsumer(
      clang::CompilerInstance& CI, llvm::StringRef InFile) override {
    return std::make_unique<ASTPrinterConsumer>(&CI.getASTContext());
  }
};

int main(int argc, const char** argv) {
  if (argc < 2) return 1;
  // 模拟Clang驱动的编译参数，需根据实际环境调整头文件路径、C++标准等
  std::vector<std::string> CompileArgs = {"-std=c++17", "-I/usr/include/c++/11"};
  CompileArgs.push_back(argv[1]);

  auto Compilations = clang::tooling::FixedCompilationDatabase::loadFromCommandLine(CompileArgs);
  clang::tooling::ClangTool Tool(Compilations, {argv[1]});
  return Tool.run(clang::tooling::newFrontendActionFactory<ASTPrinterAction>().get());
}

二、将AST转为内存形式的IR #

借助CodeGenerator类，在AST处理流程中触发LLVM IR的生成，直接得到内存中的llvm::Module对象：

#include "clang/Tooling/Tooling.h"
#include "clang/CodeGen/CodeGenAction.h"
#include "llvm/IR/LLVMContext.h"
#include "llvm/IR/Module.h"

class MemIRGenAction : public clang::ASTFrontendAction {
public:
  std::unique_ptr<clang::ASTConsumer> CreateASTConsumer(
      clang::CompilerInstance& CI, llvm::StringRef InFile) override {
    // 初始化LLVM上下文与代码生成器
    llvm::LLVMContext& LLVMCtx = llvm::getGlobalContext();
    CodeGen.reset(clang::CreateLLVMCodeGen(
        CI.getDiagnostics(), InFile.str(), CI.getHeaderSearchOpts(),
        CI.getPreprocessorOpts(), CI.getCodeGenOpts(), LLVMCtx));
    return CodeGen->CreateASTConsumer(CI, InFile);
  }

  void EndSourceFileAction() override {
    // 获取内存中的IR模块，可直接进行内存内操作（如优化、分析）
    std::unique_ptr<llvm::Module> IRModule = CodeGen->ReleaseModule();
    if (IRModule) {
      llvm::errs() << "Generated IR module name: " << IRModule->getName() << "\n";
      // 这里可以添加IR优化、分析逻辑
    }
  }

private:
  std::unique_ptr<clang::CodeGenerator> CodeGen;
};

int main(int argc, const char** argv) {
  if (argc < 2) return 1;
  std::vector<std::string> CompileArgs = {"-std=c++17", "-I/usr/include/c++/11", "-emit-llvm"};
  CompileArgs.push_back(argv[1]);

  auto Compilations = clang::tooling::FixedCompilationDatabase::loadFromCommandLine(CompileArgs);
  clang::tooling::ClangTool Tool(Compilations, {argv[1]});
  return Tool.run(clang::tooling::newFrontendActionFactory<MemIRGenAction>().get());
}

三、将AST转为可读文本形式的IR #

基于内存中的llvm::Module，使用llvm::raw_ostream将其输出为可读的LLVM IR文本，可写入文件或标准输出：

在上述MemIRGenAction的EndSourceFileAction中添加输出逻辑：

if (IRModule) {
  // 输出到标准输出
  IRModule->print(llvm::outs(), nullptr);
  
  // 输出到指定文件
  std::error_code EC;
  llvm::raw_fd_ostream OutFile("output.ll", EC, llvm::sys::fs::OF_None);
  if (!EC) {
    IRModule->print(OutFile, nullptr);
    OutFile.close();
  }
}

关键注意事项 #

• 编译参数必须匹配：要保证传入的头文件路径、C++标准、宏定义等参数与你使用clang++时一致，否则会出现AST生成失败或IR转换错误。
• 链接依赖库：编译时需要链接Clang和LLVM的相关库，比如clangTooling、clangAST、clangCodeGen、LLVMCore、LLVMIRPrinter等，可参考Clang官方CMake配置确定具体链接项。
• 官方示例参考：Clang自带的clang-check、clang-tidy等工具都是基于libTooling开发的，它们的源码是学习AST处理和代码生成的最佳参考。

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