mirror of
https://github.com/geode-sdk/geode.git
synced 2024-12-21 05:12:29 -05:00
141 lines
No EOL
3.6 KiB
C++
141 lines
No EOL
3.6 KiB
C++
#include "platform_macro.h"
|
|
#if defined(TARGET_ARCH_X64)
|
|
|
|
#include "dobby_internal.h"
|
|
|
|
#include "core/modules/assembler/assembler-x64.h"
|
|
|
|
#include "TrampolineBridge/ClosureTrampolineBridge/common-bridge-handler.h"
|
|
|
|
using namespace zz;
|
|
using namespace zz::x64;
|
|
|
|
static void *closure_bridge = NULL;
|
|
|
|
void *get_closure_bridge() {
|
|
// if already initialized, just return.
|
|
if (closure_bridge)
|
|
return closure_bridge;
|
|
|
|
// Check if enable the inline-assembly closure_bridge_template
|
|
#if ENABLE_CLOSURE_BRIDGE_TEMPLATE
|
|
|
|
extern void closure_bridge_tempate();
|
|
closure_bridge = closure_bridge_template;
|
|
|
|
#else
|
|
|
|
// otherwise, use the Assembler build the closure_bridge
|
|
#define _ turbo_assembler_.
|
|
#define __ turbo_assembler_.GetCodeBuffer()->
|
|
|
|
char *pushfq = (char *)"\x9c";
|
|
char *popfq = (char *)"\x9d";
|
|
|
|
TurboAssembler turbo_assembler_(0);
|
|
|
|
// save flags register
|
|
__ EmitBuffer(pushfq, 1);
|
|
// align rsp 16-byte
|
|
_ sub(rsp, Immediate(8, 32));
|
|
|
|
// general register
|
|
_ sub(rsp, Immediate(16 * 8, 32));
|
|
_ mov(Address(rsp, 8 * 0), rax);
|
|
_ mov(Address(rsp, 8 * 1), rbx);
|
|
_ mov(Address(rsp, 8 * 2), rcx);
|
|
_ mov(Address(rsp, 8 * 3), rdx);
|
|
_ mov(Address(rsp, 8 * 4), rbp);
|
|
_ mov(Address(rsp, 8 * 5), rsp);
|
|
_ mov(Address(rsp, 8 * 6), rdi);
|
|
_ mov(Address(rsp, 8 * 7), rsi);
|
|
_ mov(Address(rsp, 8 * 8), r8);
|
|
_ mov(Address(rsp, 8 * 9), r9);
|
|
_ mov(Address(rsp, 8 * 10), r10);
|
|
_ mov(Address(rsp, 8 * 11), r11);
|
|
_ mov(Address(rsp, 8 * 12), r12);
|
|
_ mov(Address(rsp, 8 * 13), r13);
|
|
_ mov(Address(rsp, 8 * 14), r14);
|
|
_ mov(Address(rsp, 8 * 15), r15);
|
|
|
|
// save origin sp
|
|
_ mov(rax, rsp);
|
|
_ add(rax, Immediate(8 + 8 + 8 + 16 * 8, 32));
|
|
_ sub(rsp, Immediate(2 * 8, 32));
|
|
_ mov(Address(rsp, 8), rax);
|
|
|
|
// ======= Jump to UnifiedInterface Bridge Handle =======
|
|
|
|
// prepare args
|
|
// @rdi: data_address
|
|
// @rsi: RegisterContext stack address
|
|
_ mov(rdi, rsp);
|
|
_ mov(rsi, Address(rsp, 8 + 8 + 16 * 8 + 2 * 8));
|
|
|
|
// [!!!] As we can't detect the sp is aligned or not, check if need stack align
|
|
{
|
|
// mov rax, rsp
|
|
__ EmitBuffer((void *)"\x48\x89\xE0", 3);
|
|
// and rax, 0xF
|
|
__ EmitBuffer((void *)"\x48\x83\xE0\x0F", 4);
|
|
// cmp rax, 0x0
|
|
__ EmitBuffer((void *)"\x48\x83\xF8\x00", 4);
|
|
// jnz [stack_align_call_bridge]
|
|
__ EmitBuffer((void *)"\x75\x15", 2);
|
|
}
|
|
|
|
// LABEL: call_bridge
|
|
_ CallFunction(ExternalReference((void *)intercept_routing_common_bridge_handler));
|
|
|
|
// jmp [restore_stack_register]
|
|
__ EmitBuffer((void *)"\xE9\x12\x00\x00\x00", 5);
|
|
|
|
// LABEL: stack_align_call_bridge
|
|
// push rax
|
|
__ EmitBuffer((void *)"\x50", 1);
|
|
_ CallFunction(ExternalReference((void *)intercept_routing_common_bridge_handler));
|
|
// pop rax
|
|
__ EmitBuffer((void *)"\x58", 1);
|
|
|
|
// ======= RegisterContext Restore =======
|
|
|
|
// restore sp placeholder stack
|
|
_ add(rsp, Immediate(2 * 8, 32));
|
|
|
|
// general register
|
|
_ pop(rax);
|
|
_ pop(rbx);
|
|
_ pop(rcx);
|
|
_ pop(rdx);
|
|
_ pop(rbp);
|
|
_ add(rsp, Immediate(8, 32)); // => pop rsp
|
|
_ pop(rdi);
|
|
_ pop(rsi);
|
|
_ pop(r8);
|
|
_ pop(r9);
|
|
_ pop(r10);
|
|
_ pop(r11);
|
|
_ pop(r12);
|
|
_ pop(r13);
|
|
_ pop(r14);
|
|
_ pop(r15);
|
|
|
|
// align rsp 16-byte
|
|
_ add(rsp, Immediate(8, 32));
|
|
// restore flags register
|
|
__ EmitBuffer(popfq, 1);
|
|
|
|
// trick: use the 'carry_data' stack(remain at closure trampoline) placeholder, as the return address
|
|
_ ret();
|
|
|
|
_ RelocBind();
|
|
|
|
AssemblyCodeChunk *code = AssemblyCodeBuilder::FinalizeFromTurboAssembler(&turbo_assembler_);
|
|
closure_bridge = (void *)code->raw_instruction_start();
|
|
|
|
DLOG(0, "[closure bridge] Build the closure bridge at %p", closure_bridge);
|
|
#endif
|
|
return (void *)closure_bridge;
|
|
}
|
|
|
|
#endif |