Kayori/segatools2
1
0
Fork 0
forked from Hay1tsme/segatools
Code Pull Requests Activity

added setupapi hook to elisabeth

Browse Source
This commit is contained in:
Hay1tsme 2021-12-29 21:58:00 -05:00 committed by Hay1tsme
parent 31d1ffe24f
commit 40ce4d1bb3
4 changed files with 27 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
hooklib/dll.c
View File

@ -284,6 +284,8 @@ static HMODULE WINAPI hook_LoadLibraryW(const wchar_t *name)
{
HMODULE result;
dprintf("hook_LoadLibraryW: Loading %S\n", name);
if (name == NULL) {
SetLastError(ERROR_INVALID_PARAMETER);

18
hooklib/setupapi.c
View File

@ -134,22 +134,27 @@ end:
return hr;
}
static void setupapi_hook_init(void)
static void setupapi_hook_init()
{
if (setupapi_initted) {
return;
}
hook_table_apply(
NULL,
"setupapi.dll",
setupapi_syms,
_countof(setupapi_syms));
setupapi_hook_insert_hooks(NULL);
InitializeCriticalSection(&setupapi_lock);
setupapi_initted = true;
}
void setupapi_hook_insert_hooks(HMODULE target)
{
hook_table_apply(
target,
"setupapi.dll",
setupapi_syms,
_countof(setupapi_syms));
}
static HDEVINFO WINAPI my_SetupDiGetClassDevsW(
const GUID *ClassGuid,
wchar_t *Enumerator,
@ -191,6 +196,7 @@ static BOOL WINAPI my_SetupDiEnumDeviceInterfaces(
DWORD MemberIndex,
SP_DEVICE_INTERFACE_DATA *DeviceInterfaceData)
{
dprintf("my_SetupDiEnumDeviceInterfaces hit!\n");
const struct setupapi_class *class_;
size_t i;

1
hooklib/setupapi.h
View File

@ -5,3 +5,4 @@
#include <stddef.h>
HRESULT setupapi_add_phantom_dev(const GUID *iface_class, const wchar_t *path);
void setupapi_hook_insert_hooks(HMODULE target);

34
mercuryhook/elisabeth.c
View File

@ -10,6 +10,7 @@
#include "hooklib/dll.h"
#include "hooklib/path.h"
#include "hooklib/setupapi.h"
#include "util/dprintf.h"
@ -21,7 +22,7 @@ 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 const struct hook_symbol elisabeth_hooks[] = {
static const struct hook_symbol win32_hooks[] = {
{
.name = "LoadLibraryW",
.patch = my_LoadLibraryW,
@ -39,14 +40,7 @@ static const wchar_t *target_modules[] = {
L"ftd2XX.dll",
};
static const char *target_functions[] = {
"FT_Read",
"FT_Write",
"USBIntLED_Init",
};
static const size_t target_modules_len = _countof(target_modules);
static const size_t target_functions_len = _countof(target_functions);
void elisabeth_hook_init()
{
@ -58,8 +52,8 @@ static void dll_hook_insert_hooks(HMODULE target)
hook_table_apply(
target,
"kernel32.dll",
elisabeth_hooks,
_countof(elisabeth_hooks));
win32_hooks,
_countof(win32_hooks));
}
static HMODULE WINAPI my_LoadLibraryW(const wchar_t *name)
@ -71,6 +65,8 @@ static HMODULE WINAPI my_LoadLibraryW(const wchar_t *name)
size_t name_len;
size_t target_module_len;
dprintf("Elisabeth: Trying to load %S\n", name);
if (name == NULL) {
SetLastError(ERROR_INVALID_PARAMETER);
@ -107,7 +103,7 @@ static HMODULE WINAPI my_LoadLibraryW(const wchar_t *name)
dprintf("Elisabeth: Loaded %S\n", target_module);
dll_hook_insert_hooks(result);
path_hook_insert_hooks(result);
setupapi_hook_insert_hooks(result);
}
}
@ -116,19 +112,13 @@ static HMODULE WINAPI my_LoadLibraryW(const wchar_t *name)
FARPROC WINAPI my_GetProcAddress(HMODULE hModule, const char *name)
{
uintptr_t ordinal;
uintptr_t ordinal = (uintptr_t) name;
FARPROC result = next_GetProcAddress(hModule, name);
FARPROC result = next_GetProcAddress(hModule, name);
for (size_t i = 0; i < target_functions_len; i++) {
ordinal = (uintptr_t) name;
if (ordinal > 0xFFFF) {
/* Import by name */
if (strcmp(target_functions[i], name) == 0)
dprintf("Elisabeth: GetProcAddress %s is %p\n", name, result);
}
if (ordinal > 0xFFFF) {
/* Import by name */
dprintf("Elisabeth: GetProcAddress %s is %p\n", name, result);
}
return result;