如何通过LLVM API将IR模块转为ARM64汇编并存储到模块中？

核心流程与关键代码 #

要实现IR到ARM64汇编的转换并存储，需严格遵循LLVM的代码生成流水线，以下是可直接复用的实现逻辑：

1. 初始化目标环境与目标机器 #

首先确保正确初始化ARM64目标，避免因目标未注册导致代码生成失败：

#include "llvm/Target/TargetMachine.h"
#include "llvm/Target/TargetRegistry.h"
#include "llvm/CodeGen/TargetPassConfig.h"
#include "llvm/IR/LegacyPassManager.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Support/raw_ostream.h"

using namespace llvm;

// ARM64通用目标三元组，可根据场景调整（如aarch64-linux-gnu）
const char* TargetTriple = "aarch64-unknown-linux-gnu";

2. 创建目标机器实例 #

通过目标三元组获取ARM64 TargetMachine，这是代码生成的核心载体：

std::string Error;
const Target* Target = TargetRegistry::lookupTarget(TargetTriple, Error);
if (!Target) {
  errs() << Error;
  return nullptr; // 按需处理错误
}

// 设置CPU与特性（generic适配通用ARM64架构）
TargetOptions Options;
std::string CPUStr = "generic";
std::string FeaturesStr;

TargetMachine* TM = Target->createTargetMachine(
  TargetTriple, CPUStr, FeaturesStr, Options, Reloc::Model::PIC_
);
if (!TM) {
  errs() << "Failed to create target machine\n";
  return nullptr;
}

3. 配置PassManager并生成汇编到文件 #

关键是正确使用addPassesToEmitFile，并确保运行PassManager完成转换——如果之前仅能打印到终端，大概率是未正确绑定输出文件流：

// 指定输出汇编文件名
std::string OutputFilename = "output.s";
std::error_code EC;
raw_fd_ostream OutFile(OutputFilename, EC, sys::fs::OF_None);
if (EC) {
  errs() << "Could not open output file: " << EC.message() << "\n";
  return nullptr;
}

// 创建旧版PassManager（LLVM15+可改用NewPM，旧版更兼容传统流程）
legacy::PassManager PM;

// 设置输出类型为汇编文件
if (TM->addPassesToEmitFile(PM, OutFile, nullptr, CodeGenFileType::AssemblyFile)) {
  errs() << "TargetMachine can't emit a file of this type\n";
  return nullptr;
}

// 运行PassManager，完成IR到汇编的转换并写入文件
PM.run(*YourIRModule);
OutFile.close(); // 手动关闭确保内容写入磁盘

4. 获取汇编字符串（内存中的“模块”） #

若需求是返回内存中的汇编字符串而非写入文件，可替换为raw_string_ostream：

std::string AssemblyStr;
raw_string_ostream StrStream(AssemblyStr);

legacy::PassManager PM;
if (TM->addPassesToEmitFile(PM, StrStream, nullptr, CodeGenFileType::AssemblyFile)) {
  errs() << "TargetMachine can't emit assembly string\n";
  return nullptr;
}

PM.run(*YourIRModule);
StrStream.flush();
// 此时AssemblyStr即为ARM64汇编代码，可作为函数返回值

常见问题排查 #

• 目标未注册：链接LLVM时需包含ARM64相关库（如LLVMARM64CodeGen、LLVMARM64AsmPrinter），否则lookupTarget会失败。
• PassManager未运行：必须调用PM.run(*Module)触发代码生成，仅调用addPassesToEmitFile不会自动执行转换。
• 输出流未正确处理：使用文件流时需确保文件成功打开，Pass运行后手动关闭流更可靠。
• 重定位模型错误：根据场景选择合适模型，嵌入式场景用Reloc::Static，Linux用户态用Reloc::PIC_。

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