segatools/mercuryhook/elisabeth.c

71 lines
1.6 KiB
C
Raw Normal View History

2021-12-30 05:51:44 +00:00
#include <initguid.h>
#include <windows.h>
#include <assert.h>
#include <stdbool.h>
#include <stdlib.h>
#include "mercuryhook/elisabeth.h"
2021-12-30 18:57:05 +00:00
#include "mercuryhook/mercury-dll.h"
#include "hook/table.h"
2021-12-30 18:57:05 +00:00
#include "hooklib/uart.h"
#include "hooklib/dll.h"
#include "hooklib/path.h"
2021-12-30 02:58:00 +00:00
#include "hooklib/setupapi.h"
#include "util/dprintf.h"
/* Hooks targeted DLLs dynamically loaded by elisabeth. */
static void dll_hook_insert_hooks(HMODULE target);
static FARPROC WINAPI my_GetProcAddress(HMODULE hModule, const char *name);
static FARPROC (WINAPI *next_GetProcAddress)(HMODULE hModule, const char *name);
2022-01-04 05:10:46 +00:00
static int my_USBIntLED_Init();
2021-12-30 02:58:00 +00:00
static const struct hook_symbol win32_hooks[] = {
{
.name = "GetProcAddress",
.patch = my_GetProcAddress,
.link = (void **) &next_GetProcAddress
}
};
2021-12-30 18:57:05 +00:00
HRESULT elisabeth_hook_init()
{
dll_hook_insert_hooks(NULL);
}
static void dll_hook_insert_hooks(HMODULE target)
{
hook_table_apply(
target,
"kernel32.dll",
2021-12-30 02:58:00 +00:00
win32_hooks,
_countof(win32_hooks));
}
FARPROC WINAPI my_GetProcAddress(HMODULE hModule, const char *name)
{
2021-12-30 02:58:00 +00:00
uintptr_t ordinal = (uintptr_t) name;
2021-12-30 05:51:44 +00:00
FARPROC result = next_GetProcAddress(hModule, name);
2021-12-30 02:58:00 +00:00
if (ordinal > 0xFFFF) {
/* Import by name */
2022-01-04 05:10:46 +00:00
if (strcmp(name, "USBIntLED_Init") == 0) {
result = my_USBIntLED_Init;
}
}
return result;
}
2022-01-04 05:10:46 +00:00
/* Intercept the call to initialize the LED board. */
static int my_USBIntLED_Init()
{
dprintf("Elisabeth: my_USBIntLED_Init hit!\n");
return 1;
}