forked from Hay1tsme/segatools
LED board done
This commit is contained in:
Hay1tsme
parent
cc0f8dc595
commit
8702cc9e99
@ -20,24 +20,11 @@
|
||||
/* Hooks targeted DLLs dynamically loaded by elisabeth. */
|
||||
|
||||
static void dll_hook_insert_hooks(HMODULE target);
|
||||
static HMODULE WINAPI my_LoadLibraryW(const wchar_t *name);
|
||||
static HMODULE (WINAPI *next_LoadLibraryW)(const wchar_t *name);
|
||||
static FARPROC WINAPI my_GetProcAddress(HMODULE hModule, const char *name);
|
||||
static FARPROC (WINAPI *next_GetProcAddress)(HMODULE hModule, const char *name);
|
||||
static HRESULT elisabeth_handle_irp(struct irp *irp);
|
||||
static HRESULT elisabeth_handle_irp_locked(struct irp *irp);
|
||||
|
||||
static CRITICAL_SECTION elisabeth_lock;
|
||||
static struct uart elisabeth_uart;
|
||||
static uint8_t elisabeth_written_bytes[520];
|
||||
static uint8_t elisabeth_readable_bytes[520];
|
||||
static int my_USBIntLED_Init();
|
||||
|
||||
static const struct hook_symbol win32_hooks[] = {
|
||||
{
|
||||
.name = "LoadLibraryW",
|
||||
.patch = my_LoadLibraryW,
|
||||
.link = (void **) &next_LoadLibraryW,
|
||||
},
|
||||
{
|
||||
.name = "GetProcAddress",
|
||||
.patch = my_GetProcAddress,
|
||||
@ -45,90 +32,9 @@ static const struct hook_symbol win32_hooks[] = {
|
||||
}
|
||||
};
|
||||
|
||||
static const wchar_t *target_modules[] = {
|
||||
L"USBIntLED.DLL"
|
||||
};
|
||||
|
||||
static const size_t target_modules_len = _countof(target_modules);
|
||||
|
||||
HRESULT elisabeth_hook_init()
|
||||
{
|
||||
dll_hook_insert_hooks(NULL);
|
||||
setupapi_add_phantom_dev(&elisabeth_guid, L"$ftdi");
|
||||
|
||||
InitializeCriticalSection(&elisabeth_lock);
|
||||
|
||||
uart_init(&elisabeth_uart, 1);
|
||||
elisabeth_uart.written.bytes = elisabeth_written_bytes;
|
||||
elisabeth_uart.written.nbytes = sizeof(elisabeth_written_bytes);
|
||||
elisabeth_uart.readable.bytes = elisabeth_readable_bytes;
|
||||
elisabeth_uart.readable.nbytes = sizeof(elisabeth_readable_bytes);
|
||||
|
||||
return iohook_push_handler(elisabeth_handle_irp);
|
||||
}
|
||||
|
||||
static HRESULT elisabeth_handle_irp(struct irp *irp)
|
||||
{
|
||||
HRESULT hr;
|
||||
|
||||
assert(irp != NULL);
|
||||
|
||||
if (!uart_match_irp(&elisabeth_uart, irp)) {
|
||||
return iohook_invoke_next(irp);
|
||||
}
|
||||
|
||||
EnterCriticalSection(&elisabeth_lock);
|
||||
hr = elisabeth_handle_irp_locked(irp);
|
||||
LeaveCriticalSection(&elisabeth_lock);
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
static HRESULT elisabeth_handle_irp_locked(struct irp *irp)
|
||||
{
|
||||
//union elisabeth_req_any req;
|
||||
struct iobuf req_iobuf;
|
||||
HRESULT hr;
|
||||
|
||||
if (irp->op == IRP_OP_OPEN) {
|
||||
dprintf("Elisabeth: Starting backend\n");
|
||||
hr = mercury_dll.elisabeth_init();
|
||||
|
||||
if (FAILED(hr)) {
|
||||
dprintf("Elisabeth: Backend error: %x\n", (int) hr);
|
||||
|
||||
return hr;
|
||||
}
|
||||
}
|
||||
|
||||
hr = uart_handle_irp(&elisabeth_uart, irp);
|
||||
|
||||
if (FAILED(hr) || irp->op != IRP_OP_WRITE) {
|
||||
return hr;
|
||||
}
|
||||
|
||||
for (;;) {
|
||||
|
||||
//req_iobuf.bytes = req.bytes;
|
||||
//req_iobuf.nbytes = sizeof(req.bytes);
|
||||
//req_iobuf.pos = 0;
|
||||
|
||||
/*hr = elisabeth_frame_decode(&req_iobuf, &elisabeth_uart.written);
|
||||
|
||||
if (hr != S_OK) {
|
||||
if (FAILED(hr)) {
|
||||
dprintf("Elisabeth: Deframe error: %x\n", (int) hr);
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
hr = elisabeth_req_dispatch(&req);
|
||||
|
||||
if (FAILED(hr)) {
|
||||
dprintf("Elisabeth: Processing error: %x\n", (int) hr);
|
||||
}*/
|
||||
}
|
||||
}
|
||||
|
||||
static void dll_hook_insert_hooks(HMODULE target)
|
||||
@ -140,58 +46,6 @@ static void dll_hook_insert_hooks(HMODULE target)
|
||||
_countof(win32_hooks));
|
||||
}
|
||||
|
||||
static HMODULE WINAPI my_LoadLibraryW(const wchar_t *name)
|
||||
{
|
||||
const wchar_t *name_end;
|
||||
const wchar_t *target_module;
|
||||
bool already_loaded;
|
||||
HMODULE result;
|
||||
size_t name_len;
|
||||
size_t target_module_len;
|
||||
|
||||
if (name == NULL) {
|
||||
SetLastError(ERROR_INVALID_PARAMETER);
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
// Check if the module is already loaded
|
||||
already_loaded = GetModuleHandleW(name) != NULL;
|
||||
|
||||
// Must call the next handler so the DLL reference count is incremented
|
||||
result = next_LoadLibraryW(name);
|
||||
|
||||
if (!already_loaded && result != NULL) {
|
||||
name_len = wcslen(name);
|
||||
|
||||
for (size_t i = 0; i < target_modules_len; i++) {
|
||||
target_module = target_modules[i];
|
||||
target_module_len = wcslen(target_module);
|
||||
|
||||
// Check if the newly loaded library is at least the length of
|
||||
// the name of the target module
|
||||
if (name_len < target_module_len) {
|
||||
continue;
|
||||
}
|
||||
|
||||
name_end = &name[name_len - target_module_len];
|
||||
|
||||
// Check if the name of the newly loaded library is one of the
|
||||
// modules the path hooks should be injected into
|
||||
if (_wcsicmp(name_end, target_module) != 0) {
|
||||
continue;
|
||||
}
|
||||
|
||||
dprintf("Elisabeth: Loaded %S\n", target_module);
|
||||
|
||||
dll_hook_insert_hooks(result);
|
||||
setupapi_hook_insert_hooks(result);
|
||||
}
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
FARPROC WINAPI my_GetProcAddress(HMODULE hModule, const char *name)
|
||||
{
|
||||
uintptr_t ordinal = (uintptr_t) name;
|
||||
@ -200,8 +54,17 @@ FARPROC WINAPI my_GetProcAddress(HMODULE hModule, const char *name)
|
||||
|
||||
if (ordinal > 0xFFFF) {
|
||||
/* Import by name */
|
||||
//dprintf("Elisabeth: GetProcAddress %s is %p\n", name, result);
|
||||
if (strcmp(name, "USBIntLED_Init") == 0) {
|
||||
result = my_USBIntLED_Init;
|
||||
}
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
/* Intercept the call to initialize the LED board. */
|
||||
static int my_USBIntLED_Init()
|
||||
{
|
||||
dprintf("Elisabeth: my_USBIntLED_Init hit!\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
@ -1,11 +1,3 @@
|
||||
#pragma once
|
||||
#include <initguid.h>
|
||||
|
||||
DEFINE_GUID(
|
||||
elisabeth_guid,
|
||||
0x219D0508,
|
||||
0x57A8,
|
||||
0x4FF5,
|
||||
0x97, 0x0A1, 0x0BD, 0x86, 0x58, 0x7C, 0x6C, 0x7E);
|
||||
|
||||
HRESULT elisabeth_hook_init();
|
||||
|
||||
@ -22,8 +22,8 @@ const struct dll_bind_sym mercury_dll_syms[] = {
|
||||
.sym = "mercury_io_get_gamebtns",
|
||||
.off = offsetof(struct mercury_dll, get_gamebtns),
|
||||
}, {
|
||||
.sym = "mercury_io_elisabeth_init",
|
||||
.off = offsetof(struct mercury_dll, elisabeth_init),
|
||||
.sym = "mercury_io_touch_init",
|
||||
.off = offsetof(struct mercury_dll, touch_init),
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
@ -10,7 +10,7 @@ struct mercury_dll {
|
||||
HRESULT (*poll)(void);
|
||||
void (*get_opbtns)(uint8_t *opbtn);
|
||||
void (*get_gamebtns)(uint16_t *player1, uint16_t *player2);
|
||||
HRESULT (*elisabeth_init)(void);
|
||||
HRESULT (*touch_init)(void);
|
||||
};
|
||||
|
||||
struct mercury_dll_config {
|
||||
|
||||
@ -15,6 +15,6 @@ EXPORTS
|
||||
mercury_io_get_api_version
|
||||
mercury_io_get_gamebtns
|
||||
mercury_io_get_opbtns
|
||||
mercury_io_elisabeth_init
|
||||
mercury_io_touch_init
|
||||
mercury_io_init
|
||||
mercury_io_poll
|
||||
|
||||
@ -48,7 +48,7 @@ void mercury_io_get_gamebtns(uint16_t *player1, uint16_t *player2)
|
||||
|
||||
}
|
||||
|
||||
HRESULT mercury_io_elisabeth_init(void)
|
||||
HRESULT mercury_io_touch_init(void)
|
||||
{
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
@ -66,4 +66,4 @@ void mercury_io_get_opbtns(uint8_t *opbtn);
|
||||
|
||||
void mercury_io_get_gamebtns(uint16_t *player1, uint16_t *player2);
|
||||
|
||||
HRESULT mercury_io_elisabeth_init(void);
|
||||
HRESULT mercury_io_touch_init(void);
|
||||
|
||||
Loading…
Reference in New Issue
Block a user