go - JITed LLVM 代码如何回调到 Go 函数中？

Question

我正在编写代码以使用 LLVM Go bindings 从自定义 VM 字节码生成 LLVM 字节码;然后代码被 JITed 并在进程内执行。

自定义 VM 字节码有几个不能直接在 LLVM 中实现的操作，因为它们需要更改外部状态；这些操作码的功能以 Go 函数的形式实现。

有excellent guides关于开始从 Go 生成 LLVM 字节码，但没有解决回调或导出函数的问题。是否可以生成 LLVM 指令来回调 Go 函数？如果是，怎么办？

我已经尝试了下面@arrowd 描述的方法，但它似乎不起作用。源代码，改编自 Felix Angell 的博客文章:

package main

import (
    "C"
    "fmt"
    "llvm.org/llvm/final/bindings/go/llvm"
)

// export AddInts
func AddInts(arg1, arg2 int) int {
    return arg1 + arg2;
}

func main() {
    // setup our builder and module
    builder := llvm.NewBuilder()
    mod := llvm.NewModule("my_module")

    // create our function prologue
    main := llvm.FunctionType(llvm.Int32Type(), []llvm.Type{}, false)
    llvm.AddFunction(mod, "main", main)
    block := llvm.AddBasicBlock(mod.NamedFunction("main"), "entry")
    builder.SetInsertPoint(block, block.FirstInstruction())

    // int a = 32
    a := builder.CreateAlloca(llvm.Int32Type(), "a")
    builder.CreateStore(llvm.ConstInt(llvm.Int32Type(), 32, false), a)

    // int b = 16
    b := builder.CreateAlloca(llvm.Int32Type(), "b")
    builder.CreateStore(llvm.ConstInt(llvm.Int32Type(), 16, false), b)

    // return a + b
    bVal := builder.CreateLoad(b, "b_val")
    aVal := builder.CreateLoad(a, "a_val")
    addIntsType := llvm.FunctionType(llvm.Int32Type(), []llvm.Type{llvm.Int32Type(), llvm.Int32Type()}, false)
    addInts := llvm.AddFunction(mod, "AddInts", addIntsType)
    call := builder.CreateCall(addInts, []llvm.Value{aVal, bVal}, "AddInts")
    builder.CreateRet(call)

    // verify it's all good
    if ok := llvm.VerifyModule(mod, llvm.ReturnStatusAction); ok != nil {
        fmt.Println(ok.Error())
    }
    mod.Dump()

    // create our exe engine
    engine, err := llvm.NewExecutionEngine(mod)
    if err != nil {
        fmt.Println(err.Error())
    }

    // run the function!
    funcResult := engine.RunFunction(mod.NamedFunction("main"), []llvm.GenericValue{})
    fmt.Printf("%d\n", funcResult.Int(false))
}

返回:

; ModuleID = 'my_module'

define i32 @main() {
entry:
  %a = alloca i32
  store i32 32, i32* %a
  %b = alloca i32
  store i32 16, i32* %b
  %b_val = load i32* %b
  %a_val = load i32* %a
  %AddInts = call i32 @AddInts(i32 %a_val, i32 %b_val)
  ret i32 %AddInts
}

declare i32 @AddInts(i32, i32)
LLVM ERROR: Tried to execute an unknown external function: AddInts
exit status 1

Answer 1

根据 this answer您可以通过添加“导出”注释将 Go 函数导出为 C 函数。如果您为您的函数执行此操作，LLVM JIT 应该能够在运行时链接期间解析此符号。答案没有提供调用该导出函数的 C 代码，但假设它只是 goProgressCB(args); 那么您可以为该函数创建一个 CallInst 并且JIT 它。

已更新。

看看this answer .好像最近的 Go 可以生成 C 头文件，让你调用 Go 返回。有了它，您可以使用 clang -S -emit-llvm 或 clang -march=cpp 编译它们，并获得对 LLVM API 的文本 IR 或 C++ 调用，这将构成调用说明。