Proton/vrclient_x64/flatapi.h

/* asm thunks for the flat (FnTable) API */

#ifdef __i386__

#include "cxx.h"
#include "pshpack1.h"
struct thunk
{
  BYTE mov_ecx;
  void *this;
  BYTE mov_edx;
  void *proc;
  BYTE jmp;
  LONG call_flat;
  BYTE nop;
};
#include "poppack.h"

extern void call_flat_method(void);

static inline void init_thunk( struct thunk *thunk, void *this, void *proc,
                               int param_count, BOOL has_floats, BOOL is_4th_float )
{
  thunk->mov_ecx = 0xb9;
  thunk->this = this;
  thunk->mov_edx = 0xba;
  thunk->proc = proc;
  thunk->jmp = 0xe9;
#ifdef __ASM_USE_THISCALL_WRAPPER
  thunk->call_flat = (char *)call_flat_method - (char *)(&thunk->call_flat + 1);
#else
  thunk->call_flat = (char *)proc - (char *)(&thunk->call_flat + 1);
#endif
}

#else

#include "pshpack1.h"
struct thunk
{
  BYTE mov_r10[2];
  void *this;
  BYTE mov_r11[2];
  void *proc;
  BYTE mov_rax[2];
  void *call_flat;
  BYTE jmp_rax[2];
};
#include "poppack.h"

static const struct thunk thunk_template =
{
    { 0x49, 0xba }, 0,        /* movq This, %r10 */
    { 0x49, 0xbb }, 0,        /* movq proc, %r11 */
    { 0x48, 0xb8 }, 0,        /* movq call_thunk, %rax */
    { 0xff, 0xe0 }            /* jmp *%rax */
};

typedef void (*pfn_call_flat_method)(void);

extern pfn_call_flat_method
get_call_flat_method_pfn( int param_count, BOOL has_floats, BOOL is_4th_float );

static inline void init_thunk( struct thunk *thunk, void *this, void *proc,
                               int param_count, BOOL has_floats, BOOL is_4th_float )
{
    *thunk = thunk_template;
    thunk->this = this;
    thunk->proc = proc;
    thunk->call_flat = get_call_flat_method_pfn(param_count, has_floats, is_4th_float);
}

#endif

static inline struct thunk *alloc_thunks( unsigned int count )
{
    return VirtualAlloc(NULL, count * sizeof(struct thunk),
        MEM_COMMIT | MEM_RESERVE, PAGE_EXECUTE_READWRITE);
}