从函数调用 asmjit 中检索 ptr

Question

我正在尝试使用 AsmJit 生成函数调用，并向其传递 char*。这个 char* 本身是从另一个函数调用中检索到的。我试过这个：

typedef
const char* getStr();

const char* getStrImpl()  {
    return "hello pie";
}

void use_str_impl(int id, const char* c_str) {
    // do stuff...
}

int main() {
    JitRuntime rt;
    CodeHolder code;
    code.init(rt.getCodeInfo());

    X86Compiler c(&code);

    auto jitted_func = c.addFunc(FuncSignature0<const char*>(code.getCodeInfo().getCdeclCallConv()));
    auto err = c.getLastError();

    auto call = c.call((uint64_t) fooFuncImpl, FuncSignature0<intptr_t>());
    X86Gpd res(call->getRet().getId());

    auto call2 = c.call((uint64_t) send_input, FuncSignature2<void, int, intptr_t>());
    err = !call2->setArg(0, Imm(42));
    err = !call2->setArg(1, res);

    c.ret();

    c.endFunc();
    err = c.finalize();

    if(err) return 0;

    size_t size = code.getCodeSize();
    VMemMgr vm;

    void* p = vm.alloc(size);
    if (!p) return 0;

    code.relocate(p);
    auto fun = (entrypoint*) p;
    fun();
}

事实证明，这不会为第二个参数或对 setArg 的第二次调用生成任何指令。我还尝试使用 .newIntPtr 并使用移动指令将调用结果移动到位。但这生成了 dec 并添加了对我和我的小组装经验毫无意义的指令。做这种事情的正确方法是什么？

顺便说一句，我正在使用 AsmJit 下一个分支。

Answer 1

我已对您的示例进行了一些修改并提出了一些意见。

更好地使用 JitRuntime：

JitRuntime rt;
size_t size = code.getCodeSize();
VMemMgr vm;
....
void* p = vm.alloc(size);
if (!p) return 0;
code.relocate(p);
auto fun = (entrypoint*) p;

你使用 JitRuntime 只是为了设置 CodeHolder 的参数，但后来避免了它并自己为函数分配了内存。虽然这是一个有效的用例，但大多数人并不这样做。在大多数情况下，使用运行时的 add() 就足够了。

CCFuncCall::getRet() 的使用无效：

X86Gpd res(call->getRet().getId());

此时 call 节点没有分配任何 return 寄存器，因此它会 return 一个无效的 ID。如果你需要创建一个虚拟寄存器，你总是必须调用编译器的newSomething()。 AsmJit 的编译器提供了 API 来在运行时检查这种情况，如果你不确定：

// Would print 0
printf("%d", (int)c.isVirtRegValid(call->getRet().getId()));

解决方案是创建一个新的虚拟寄存器并将其分配给函数的 return 值。赋值return值需要一个索引（就像赋值一个参数），原因是某些函数可能return多个值（如32位模式下的64位值），使用0作为索引就足够了大多数时候。

X86Gp reg = c.newIntPtr("reg");
call->setRet(0, reg);

您可以验证 getRet() 功能：

X86Gp reg = c.newIntPtr("reg");
assert(call->getRet(0).isNone());
call->setRet(0, reg);
assert(call->getRet(0) == reg);

完整的工作示例：

#include <stdio.h>
#include <asmjit/asmjit.h>

const char* func_a() {
  printf("func_a(): Called\n");
  return "hello pie";
}

void func_b(int id, const char* c_str) {
  printf("func_b(%d, %s): Called\n", id, c_str);
}

int main() {
  using namespace asmjit;

  JitRuntime rt;

  CodeHolder code;
  code.init(rt.getCodeInfo());

  X86Compiler c(&code);
  X86Gp reg = c.newIntPtr("reg");

  // Compilation step...
  c.addFunc(FuncSignature0<void>(code.getCodeInfo().getCdeclCallConv()));

  auto call_a = c.call((uint64_t)func_a, FuncSignature0<intptr_t>());
  call_a->setRet(0, reg);

  auto call_b = c.call((uint64_t)func_b, FuncSignature2<void, int, intptr_t>());
  call_b->setArg(0, Imm(42));
  call_b->setArg(1, reg);

  c.ret();
  c.endFunc();

  // Finalize does the following:
  //   - allocates virtual registers
  //   - inserts prolog / epilog
  //   - assembles to CodeHolder
  auto err = c.finalize();
  if (err) {
    printf("COMPILER FAILED: %s\b", DebugUtils::errorAsString(err));
    return 1;
  }

  typedef void (*EntryPoint)(void);
  EntryPoint entry;

  // Adds function to the runtime. Should be freed by rt.release().
  // Function is valid until the runtime is valid if not released.
  err = rt.add(&entry, &code);
  if (err) {
    printf("RUNTIME FAILED: %s\b", DebugUtils::errorAsString(err));
    return 1;
  }

  entry();
  return 0;
}

Answer 2

我正在尝试创建一个接收和 returns 双精度值的函数。对于调用方法，我使用了 Mem 的方法。最后我需要将结果保存在变量 xmm1.

我无法识别错误。正弦函数被正确调用。但是对于最终的汇编器生成错误。

        JitRuntime rt;
        CodeHolder code;
        code.init(rt.codeInfo());

        asmjit::x86::Compiler cc(&code);
        asmjit::x86::Gp reg = cc.newIntPtr("reg");

        asmjit::Zone zonee(1024);
        asmjit::ConstPool constPool(&zonee);
        asmjit::Label constPoolLabel = cc.newLabel();


        // Compilation step...
        // c.addFunc(asmjit::FuncSignatureT<void>(code.codeInfo().getCdeclCallConv()));
        cc.addFunc(asmjit::FuncSignatureT<void>());


        auto call_a = cc.call((uint64_t)func_a, FuncSignatureT<intptr_t>());
        call_a->setRet(0, reg);

        auto call_b = cc.call((uint64_t)func_b, FuncSignatureT<void, int, intptr_t>());
        call_b->setArg(0, Imm(42));
        call_b->setArg(1, reg);

        auto seno = [&](double value) {
            size_t valueOffset;
            double seno = static_cast<double_t>(std::sin(value));
            cout << " seno " << seno << endl;
            constPool.add(&seno, sizeof(double), valueOffset);
            return asmjit::x86::ptr(constPoolLabel, valueOffset);
        };

        asmjit::x86::Mem mem;
        double test = 180.5;
        auto call_c = cc.call(seno(test), asmjit::FuncSignatureT<double_t>());
        call_c->setArg(0, asmjit::Imm(test));
        call_c->_setRet(0, mem);

        cc.movsd(asmjit::x86::xmm1, mem);

        cc.ret();
        cc.endFunc();

        // Finalize does the following:
        //   - allocates virtual registers
        //   - inserts prolog / epilog
        //   - assembles to CodeHolder
        auto err = cc.finalize();
        if (err) {
           printf("COMPILER FAILED: %s\b", DebugUtils::errorAsString(err));
           return;
        }

        typedef void (*EntryPoint)(void);
        EntryPoint entry;

        // Adds function to the runtime. Should be freed by rt.release().
        // Function is valid until the runtime is valid if not released.
        err = rt.add(&entry, &code);
        if (err) {
           printf("RUNTIME FAILED: %s\b", DebugUtils::errorAsString(err));
           return;
        }

        entry();
        return;

Answer 3

也许内存对象应该与某个内存地址相关？

Mem mem = qword_ptr ((uint64_t) &test);

从函数调用 asmjit 中检索 ptr

Retrieve ptr from function call asmjit

c++

asmjit

更好地使用 JitRuntime：

CCFuncCall::getRet() 的使用无效：

完整的工作示例：