如何向LLVM IR函数传递数组并通过索引获取元素值

你的代码存在几个关键错误，导致无法正确获取数组元素：

1. 参数获取方式错误：第二个参数不能用arg1+1获取，LLVM的参数迭代器是迭代器类型，需用std::next或迭代器递增来获取下一个参数。
2. 数组元素访问指令误用：CreateExtractElement是针对**向量类型（VectorType）**的操作，而你传递的是指针类型（数组退化为指针），正确操作应该是先用CreateGEP生成元素指针，再用CreateLoad加载元素值。
3. 函数类型定义冗余：ArrayType::getInt32PtrTy(Context)等价于Type::getInt32PtrTy(Context)，直接使用后者更清晰。

修复后的完整代码如下：

#include "llvm/IR/Function.h"
#include "llvm/IR/IRBuilder.h"
#include "llvm/IR/Module.h"
#include "llvm/ExecutionEngine/ExecutionEngine.h"
#include "llvm/ExecutionEngine/EngineBuilder.h"
#include <memory>
#include <cstdint>
#include <cstdio>

using namespace llvm;

int main() {
  LLVMContext Context;
  std::unique_ptr<Module> Owner = std::make_unique<Module>("test", Context);
  Module &M = *Owner;

  // 定义函数类型：int(int32_t*, int)
  FunctionType *FuncType = FunctionType::get(
      Type::getInt32Ty(Context),
      {Type::getInt32PtrTy(Context), Type::getInt32Ty(Context)},
      false
  );
  Function *FooF = Function::Create(FuncType, Function::ExternalLinkage, "foo", M);

  // 遍历并命名参数（可选，方便调试）
  auto ArgIt = FooF->arg_begin();
  Argument *ArrayPtrArg = &*ArgIt;
  ArrayPtrArg->setName("array_ptr");
  ++ArgIt;
  Argument *IndexArg = &*ArgIt;
  IndexArg->setName("index");

  // 创建基本块
  BasicBlock *BB = BasicBlock::Create(Context, "EntryBlock", FooF);
  IRBuilder<> builder(BB);

  // 生成GEP指令获取元素指针：array_ptr[index]
  // 对于指针类型，第一个索引为0（表示不偏移指针本身），第二个索引为传入的数组下标
  Value *Indices[] = {
      builder.getInt32(0),
      IndexArg
  };
  Value *ElementPtr = builder.CreateGEP(Type::getInt32Ty(Context), ArrayPtrArg, Indices, "element_ptr");

  // 加载元素值
  Value *ElementVal = builder.CreateLoad(ElementPtr, "element_val");

  // 返回结果
  builder.CreateRet(ElementVal);

  // 创建JIT并执行函数
  ExecutionEngine* EE = EngineBuilder(std::move(Owner)).create();
  using FunctionPtr = int(*)(int32_t *, int);
  FunctionPtr func = reinterpret_cast<FunctionPtr>(EE->getFunctionAddress("foo"));
  
  int32_t array[3] = {1,2,3};
  int first = func(array, 0);  // 返回1
  int second = func(array, 1); // 返回2
  printf("array[0] = %d, array[1] = %d\n", first, second);

  return 0;
}

关键修复点说明： #

• 参数获取：通过迭代器递增的方式正确获取第二个参数，避免错误的指针算术操作。
• GEP指令：CreateGEP用于计算数组元素的内存地址，是LLVM中访问指针指向元素的标准方式。
• Load指令：通过GEP得到元素指针后，用CreateLoad从内存中读取元素值，完成数组元素的访问。

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