From faef84e1ae8dce1153dfb695a249b977948c814c Mon Sep 17 00:00:00 2001 From: Inori Date: Sat, 21 Dec 2019 04:53:14 -0500 Subject: [PATCH] always alias MiniFFI class as Win32API --- binding/miniffi-binding.cpp | 474 ++++++++++++++++++------------------ 1 file changed, 239 insertions(+), 235 deletions(-) diff --git a/binding/miniffi-binding.cpp b/binding/miniffi-binding.cpp index 759b1a5a..88092f4e 100644 --- a/binding/miniffi-binding.cpp +++ b/binding/miniffi-binding.cpp @@ -1,17 +1,16 @@ // Most of the MiniFFI class was taken from Ruby 1.8's Win32API.c, // it's just as basic but should work fine for the moment -#include #include "fake-api.h" +#include #include "binding-util.h" -#define _T_VOID 0 -#define _T_NUMBER 1 -#define _T_POINTER 2 -#define _T_INTEGER 3 -#define _T_BOOL 4 - +#define _T_VOID 0 +#define _T_NUMBER 1 +#define _T_POINTER 2 +#define _T_INTEGER 3 +#define _T_BOOL 4 // Might need to let MiniFFI.initialize set calling convention // as an optional arg, this won't work with everything @@ -24,7 +23,10 @@ // L O N G, but variables won't get set up correctly otherwise // should allow for __fastcalls (macOS likes these) and whatever else -typedef PREFABI void* (*MINIFFI_FUNC)(unsigned long,unsigned long,unsigned long,unsigned long,unsigned long,unsigned long,unsigned long,unsigned long); +typedef PREFABI void *(*MINIFFI_FUNC)(unsigned long, unsigned long, + unsigned long, unsigned long, + unsigned long, unsigned long, + unsigned long, unsigned long); // MiniFFI class, also named Win32API on Windows // Uses LoadLibrary/GetProcAddress on Windows, dlopen/dlsym everywhere else @@ -35,254 +37,256 @@ DEF_TYPE_CUSTOMFREE(MiniFFI, SDL_UnloadObject); DEF_ALLOCFUNC_CUSTOMFREE(MiniFFI, SDL_UnloadObject); #endif -static void* -MiniFFI_GetFunctionHandle(void *libhandle, const char *func) -{ +static void *MiniFFI_GetFunctionHandle(void *libhandle, const char *func) { #ifdef USE_FAKEAPI -#define CAPTURE(n) if (!strcmp(#n,func)) return (void*)&MKXP_##n - CAPTURE(GetCurrentThreadId); - CAPTURE(GetWindowThreadProcessId); - CAPTURE(FindWindowEx); - CAPTURE(GetForegroundWindow); - CAPTURE(GetClientRect); - CAPTURE(GetCursorPos); - CAPTURE(ScreenToClient); - CAPTURE(SetWindowPos); - CAPTURE(SetWindowTextA); - CAPTURE(GetWindowRect); - CAPTURE(RegisterHotKey); - CAPTURE(SetWindowLong); - CAPTURE(GetKeyboardState); -#ifndef __WIN32__ - // Functions only needed on Linux and macOS go here - CAPTURE(RtlMoveMemory); - CAPTURE(LoadLibrary); - CAPTURE(FreeLibrary); - CAPTURE(GetAsyncKeyState); - CAPTURE(GetSystemPowerStatus); - CAPTURE(ShowWindow); - CAPTURE(GetSystemMetrics); - CAPTURE(SetCapture); - CAPTURE(ReleaseCapture); - CAPTURE(ShowCursor); - CAPTURE(GetPrivateProfileString); - CAPTURE(GetUserDefaultLangID); - CAPTURE(GetUserName); +#define CAPTURE(n) \ + if (!strcmp(#n, func)) \ + return (void *)&MKXP_##n + CAPTURE(GetCurrentThreadId); + CAPTURE(GetWindowThreadProcessId); + CAPTURE(FindWindowEx); + CAPTURE(GetForegroundWindow); + CAPTURE(GetClientRect); + CAPTURE(GetCursorPos); + CAPTURE(ScreenToClient); + CAPTURE(SetWindowPos); + CAPTURE(SetWindowTextA); + CAPTURE(GetWindowRect); + CAPTURE(RegisterHotKey); + CAPTURE(SetWindowLong); + CAPTURE(GetKeyboardState); +#ifndef __WIN32__ + // Functions only needed on Linux and macOS go here + CAPTURE(RtlMoveMemory); + CAPTURE(LoadLibrary); + CAPTURE(FreeLibrary); + CAPTURE(GetAsyncKeyState); + CAPTURE(GetSystemPowerStatus); + CAPTURE(ShowWindow); + CAPTURE(GetSystemMetrics); + CAPTURE(SetCapture); + CAPTURE(ReleaseCapture); + CAPTURE(ShowCursor); + CAPTURE(GetPrivateProfileString); + CAPTURE(GetUserDefaultLangID); + CAPTURE(GetUserName); #endif #endif - if (!libhandle) return 0; - return SDL_LoadFunction(libhandle, func); + if (!libhandle) + return 0; + return SDL_LoadFunction(libhandle, func); } // MiniFFI.new(library, function[, imports[, exports]]) // Yields itself in blocks -RB_METHOD(MiniFFI_initialize) -{ - VALUE libname, func, imports, exports; - rb_scan_args(argc, argv, "22", &libname, &func, &imports, &exports); - SafeStringValue(libname); - SafeStringValue(func); - void *hlib = SDL_LoadObject(RSTRING_PTR(libname)); - setPrivateData(self, hlib); - void *hfunc = MiniFFI_GetFunctionHandle(hlib, RSTRING_PTR(func)); +RB_METHOD(MiniFFI_initialize) { + VALUE libname, func, imports, exports; + rb_scan_args(argc, argv, "22", &libname, &func, &imports, &exports); + SafeStringValue(libname); + SafeStringValue(func); + void *hlib = SDL_LoadObject(RSTRING_PTR(libname)); + setPrivateData(self, hlib); + void *hfunc = MiniFFI_GetFunctionHandle(hlib, RSTRING_PTR(func)); #ifdef __WIN32__ - if (hlib && !hfunc) - { - VALUE func_a = rb_str_new3(func); - func_a = rb_str_cat(func_a, "A", 1); - hfunc = SDL_LoadFunction(hlib, RSTRING_PTR(func_a)); - } + if (hlib && !hfunc) { + VALUE func_a = rb_str_new3(func); + func_a = rb_str_cat(func_a, "A", 1); + hfunc = SDL_LoadFunction(hlib, RSTRING_PTR(func_a)); + } #endif - if (!hfunc) - rb_raise(rb_eRuntimeError, SDL_GetError()); - - - rb_iv_set(self, "_func", ULONG2NUM((unsigned long)hfunc)); - rb_iv_set(self, "_funcname", func); - rb_iv_set(self, "_libname", libname); - - VALUE ary_imports = rb_ary_new(); - VALUE *entry; - switch (TYPE(imports)) - { - case T_NIL: - break; - case T_ARRAY: - entry = RARRAY_PTR(imports); - for (int i = 0; i < RARRAY_LEN(imports); i++) - { - SafeStringValue(entry[i]); - switch (*(char*)RSTRING_PTR(entry[i])) - { - case 'N': case 'n': case 'L': case 'l': - rb_ary_push(ary_imports, INT2FIX(_T_NUMBER)); - break; - - case 'P': case 'p': - rb_ary_push(ary_imports, INT2FIX(_T_POINTER)); - break; - - case 'I': case 'i': - rb_ary_push(ary_imports, INT2FIX(_T_INTEGER)); - break; - - case 'B': case 'b': - rb_ary_push(ary_imports, INT2FIX(_T_BOOL)); - break; - } - } - break; - default: - SafeStringValue(imports); - const char *s = RSTRING_PTR(imports); - for (int i = 0; i < RSTRING_LEN(imports); i++) - { - switch (*s++) - { - case 'N': case 'n': case 'L': case 'l': - rb_ary_push(ary_imports, INT2FIX(_T_NUMBER)); - break; - - case 'P': case 'p': - rb_ary_push(ary_imports, INT2FIX(_T_POINTER)); - break; - - case 'I': case 'i': - rb_ary_push(ary_imports, INT2FIX(_T_INTEGER)); - break; - - case 'B': case 'b': - rb_ary_push(ary_imports, INT2FIX(_T_BOOL)); - break; - } - } - break; + if (!hfunc) + rb_raise(rb_eRuntimeError, "%s", SDL_GetError()); + + rb_iv_set(self, "_func", ULONG2NUM((unsigned long)hfunc)); + rb_iv_set(self, "_funcname", func); + rb_iv_set(self, "_libname", libname); + + VALUE ary_imports = rb_ary_new(); + VALUE *entry; + switch (TYPE(imports)) { + case T_NIL: + break; + case T_ARRAY: + entry = RARRAY_PTR(imports); + for (int i = 0; i < RARRAY_LEN(imports); i++) { + SafeStringValue(entry[i]); + switch (*(char *)RSTRING_PTR(entry[i])) { + case 'N': + case 'n': + case 'L': + case 'l': + rb_ary_push(ary_imports, INT2FIX(_T_NUMBER)); + break; + + case 'P': + case 'p': + rb_ary_push(ary_imports, INT2FIX(_T_POINTER)); + break; + + case 'I': + case 'i': + rb_ary_push(ary_imports, INT2FIX(_T_INTEGER)); + break; + + case 'B': + case 'b': + rb_ary_push(ary_imports, INT2FIX(_T_BOOL)); + break; + } } - - if (8 < RARRAY_LEN(ary_imports)) - rb_raise(rb_eRuntimeError, "too many parameters: %ld\n", RARRAY_LEN(ary_imports)); - - rb_iv_set(self, "_imports", ary_imports); - int ex; - if (NIL_P(exports)) - { - ex = _T_VOID; + break; + default: + SafeStringValue(imports); + const char *s = RSTRING_PTR(imports); + for (int i = 0; i < RSTRING_LEN(imports); i++) { + switch (*s++) { + case 'N': + case 'n': + case 'L': + case 'l': + rb_ary_push(ary_imports, INT2FIX(_T_NUMBER)); + break; + + case 'P': + case 'p': + rb_ary_push(ary_imports, INT2FIX(_T_POINTER)); + break; + + case 'I': + case 'i': + rb_ary_push(ary_imports, INT2FIX(_T_INTEGER)); + break; + + case 'B': + case 'b': + rb_ary_push(ary_imports, INT2FIX(_T_BOOL)); + break; + } } - else - { - SafeStringValue(exports); - switch(*RSTRING_PTR(exports)) - { - case 'V': case 'v': - ex = _T_VOID; - break; - - case 'N': case 'n': case 'L': case 'l': - ex = _T_NUMBER; - break; - - case 'P': case 'p': - ex = _T_POINTER; - break; - - case 'I': case 'i': - ex = _T_INTEGER; - break; - - case 'B': case 'b': - ex = _T_BOOL; - break; - } + break; + } + + if (8 < RARRAY_LEN(ary_imports)) + rb_raise(rb_eRuntimeError, "too many parameters: %ld\n", + RARRAY_LEN(ary_imports)); + + rb_iv_set(self, "_imports", ary_imports); + int ex; + if (NIL_P(exports)) { + ex = _T_VOID; + } else { + SafeStringValue(exports); + switch (*RSTRING_PTR(exports)) { + case 'V': + case 'v': + ex = _T_VOID; + break; + + case 'N': + case 'n': + case 'L': + case 'l': + ex = _T_NUMBER; + break; + + case 'P': + case 'p': + ex = _T_POINTER; + break; + + case 'I': + case 'i': + ex = _T_INTEGER; + break; + + case 'B': + case 'b': + ex = _T_BOOL; + break; } - rb_iv_set(self, "_exports", INT2FIX(ex)); - if (rb_block_given_p()) rb_yield(self); - return Qnil; + } + rb_iv_set(self, "_exports", INT2FIX(ex)); + if (rb_block_given_p()) + rb_yield(self); + return Qnil; } -RB_METHOD(MiniFFI_call) -{ - struct { - unsigned long params[8]; - } param; +RB_METHOD(MiniFFI_call) { + struct { + unsigned long params[8]; + } param; #define params param.params - VALUE func = rb_iv_get(self, "_func"); - VALUE own_imports = rb_iv_get(self, "_imports"); - VALUE own_exports = rb_iv_get(self, "_exports"); - MINIFFI_FUNC ApiFunction = (MINIFFI_FUNC)NUM2ULONG(func); - VALUE args; - int items = rb_scan_args(argc, argv, "0*", &args); - int nimport = RARRAY_LEN(own_imports); - if (items != nimport) - rb_raise(rb_eRuntimeError, "wrong number of parameters: expected %d, got %d", - nimport, items); - - for (int i = 0; i < nimport; i++) - { - VALUE str = rb_ary_entry(args, i); - unsigned long lParam = 0; - switch(FIX2INT(rb_ary_entry(own_imports, i))) - { - case _T_POINTER: - if (NIL_P(str)) - { - lParam = 0; - } - else if (FIXNUM_P(str)) - { - lParam = NUM2ULONG(str); - } - else - { - StringValue(str); - rb_str_modify(str); - lParam = (unsigned long)RSTRING_PTR(str); - } - break; - - case _T_BOOL: - lParam = RTEST(rb_ary_entry(args, i)); - break; - - case _T_NUMBER: case _T_INTEGER: default: - lParam = NUM2ULONG(rb_ary_entry(args, i)); - break; - } - params[i] = lParam; - } - unsigned long ret = (unsigned long)ApiFunction(params[0],params[1],params[2],params[3],params[4],params[5],params[6],params[7]); - - switch (FIX2INT(own_exports)) - { - case _T_NUMBER: case _T_INTEGER: - return ULONG2NUM(ret); - - case _T_POINTER: - return rb_str_new_cstr((char*)ret); - - case _T_BOOL: - return rb_bool_new(ret); - - case _T_VOID: default: - return ULONG2NUM(0); + VALUE func = rb_iv_get(self, "_func"); + VALUE own_imports = rb_iv_get(self, "_imports"); + VALUE own_exports = rb_iv_get(self, "_exports"); + MINIFFI_FUNC ApiFunction = (MINIFFI_FUNC)NUM2ULONG(func); + VALUE args; + int items = rb_scan_args(argc, argv, "0*", &args); + int nimport = RARRAY_LEN(own_imports); + if (items != nimport) + rb_raise(rb_eRuntimeError, + "wrong number of parameters: expected %d, got %d", nimport, items); + + for (int i = 0; i < nimport; i++) { + VALUE str = rb_ary_entry(args, i); + unsigned long lParam = 0; + switch (FIX2INT(rb_ary_entry(own_imports, i))) { + case _T_POINTER: + if (NIL_P(str)) { + lParam = 0; + } else if (FIXNUM_P(str)) { + lParam = NUM2ULONG(str); + } else { + StringValue(str); + rb_str_modify(str); + lParam = (unsigned long)RSTRING_PTR(str); + } + break; + + case _T_BOOL: + lParam = RTEST(rb_ary_entry(args, i)); + break; + + case _T_NUMBER: + case _T_INTEGER: + default: + lParam = NUM2ULONG(rb_ary_entry(args, i)); + break; } + params[i] = lParam; + } + unsigned long ret = + (unsigned long)ApiFunction(params[0], params[1], params[2], params[3], + params[4], params[5], params[6], params[7]); + + switch (FIX2INT(own_exports)) { + case _T_NUMBER: + case _T_INTEGER: + return ULONG2NUM(ret); + + case _T_POINTER: + return rb_str_new_cstr((char *)ret); + + case _T_BOOL: + return rb_bool_new(ret); + + case _T_VOID: + default: + return ULONG2NUM(0); + } } - -void -MiniFFIBindingInit() -{ - VALUE cMiniFFI = rb_define_class("MiniFFI", rb_cObject); +void MiniFFIBindingInit() { + VALUE cMiniFFI = rb_define_class("MiniFFI", rb_cObject); #ifndef OLD_RUBY - rb_define_alloc_func(cMiniFFI, classAllocate<&MiniFFIType>); + rb_define_alloc_func(cMiniFFI, classAllocate<&MiniFFIType>); #else - rb_define_alloc_func(cMiniFFI, MiniFFIAllocate); + rb_define_alloc_func(cMiniFFI, MiniFFIAllocate); #endif - _rb_define_method(cMiniFFI, "initialize", MiniFFI_initialize); - _rb_define_method(cMiniFFI, "call", MiniFFI_call); - rb_define_alias(cMiniFFI, "Call", "call"); + _rb_define_method(cMiniFFI, "initialize", MiniFFI_initialize); + _rb_define_method(cMiniFFI, "call", MiniFFI_call); + rb_define_alias(cMiniFFI, "Call", "call"); -#if defined(__WIN32__) || defined(USE_FAKEAPI) - rb_define_const(rb_cObject, "Win32API", cMiniFFI); -#endif + rb_define_const(rb_cObject, "Win32API", cMiniFFI); }