1
0
mirror of synced 2024-11-30 16:24:27 +01:00
bananatools/taikohook/taiko-dll.c
2023-01-02 23:39:03 -05:00

106 lines
2.7 KiB
C

#include <windows.h>
#include <assert.h>
#include <stdlib.h>
#include "taikohook/taiko-dll.h"
#include "util/dll-bind.h"
#include "util/dprintf.h"
const struct dll_bind_sym taiko_dll_syms[] = {
{
.sym = "taiko_io_init",
.off = offsetof(struct taiko_dll, init),
}, {
.sym = "taiko_io_poll",
.off = offsetof(struct taiko_dll, poll),
}, {
.sym = "taiko_io_read_coin_counter",
.off = offsetof(struct taiko_dll, read_coin_counter),
}, {
.sym = "taiko_io_get_opbtns",
.off = offsetof(struct taiko_dll, get_opbtns),
}, {
.sym = "taiko_io_get_gamebtns",
.off = offsetof(struct taiko_dll, get_gamebtns),
}
};
struct taiko_dll taiko_dll;
HRESULT taiko_dll_init(const struct taiko_dll_config *cfg, HINSTANCE self)
{
uint16_t (*get_api_version)(void);
const struct dll_bind_sym *sym;
HINSTANCE owned;
HINSTANCE src;
HRESULT hr;
assert(cfg != NULL);
assert(self != NULL);
if (cfg->path[0] != L'\0') {
owned = LoadLibraryW(cfg->path);
if (owned == NULL) {
hr = HRESULT_FROM_WIN32(GetLastError());
dprintf("Taiko IO: Failed to load IO DLL: %lx: %S\n",
hr,
cfg->path);
goto end;
}
dprintf("Taiko IO: Using custom IO DLL: %S\n", cfg->path);
src = owned;
} else {
owned = NULL;
src = self;
}
get_api_version = (void *) GetProcAddress(src, "taiko_io_get_api_version");
if (get_api_version != NULL) {
taiko_dll.api_version = get_api_version();
} else {
taiko_dll.api_version = 0x0100;
dprintf("Custom IO DLL does not expose taiko_io_get_api_version, "
"assuming API version 1.0.\n"
"Please ask the developer to update their DLL.\n");
}
if (taiko_dll.api_version >= 0x0200) {
hr = E_NOTIMPL;
dprintf("Taiko IO: Custom IO DLL implements an unsupported "
"API version (%#04x). Please update Segatools.\n",
taiko_dll.api_version);
goto end;
}
sym = taiko_dll_syms;
hr = dll_bind(&taiko_dll, src, &sym, _countof(taiko_dll_syms));
if (FAILED(hr)) {
if (src != self) {
dprintf("Taiko IO: Custom IO DLL does not provide function "
"\"%s\". Please contact your IO DLL's developer for "
"further assistance.\n",
sym->sym);
goto end;
} else {
dprintf("Internal error: could not reflect \"%s\"\n", sym->sym);
}
}
owned = NULL;
end:
if (owned != NULL) {
FreeLibrary(owned);
}
return hr;
}