Implement build_function_asm

Uses ASMJIT to build function at startup
This commit is contained in:
Nekotekina 2018-05-14 23:06:17 +03:00
parent fd8aae5725
commit fd525ae1cf
3 changed files with 88 additions and 7 deletions

View file

@ -1,5 +1,61 @@
#pragma once
#define ASMJIT_STATIC
#define ASMJIT_DEBUG
#include "asmjit.h"
#include <functional>
namespace asmjit
{
// Should only be used to build global functions
JitRuntime& get_global_runtime();
// Emit xbegin and adjacent loop
void build_transaction_enter(X86Assembler& c, Label abort);
// Emit xabort and return zero
void build_transaction_abort(X86Assembler& c, unsigned char code);
}
// Build runtime function with asmjit::X86Assembler
template <typename FT, typename F>
FT build_function_asm(F&& builder)
{
using namespace asmjit;
auto& rt = get_global_runtime();
CodeHolder code;
code.init(rt.getCodeInfo());
code._globalHints = asmjit::CodeEmitter::kHintOptimizedAlign;
std::array<X86Gp, 4> args;
#ifdef _WIN32
args[0] = x86::rcx;
args[1] = x86::rdx;
args[2] = x86::r8;
args[3] = x86::r9;
#else
args[0] = x86::rdi;
args[1] = x86::rsi;
args[2] = x86::rdx;
args[3] = x86::rcx;
#endif
X86Assembler compiler(&code);
builder(std::ref(compiler), args);
FT result;
if (rt.add(&result, &code))
{
return nullptr;
}
return result;
}
#ifdef LLVM_AVAILABLE
#include <memory>