Max/djhackersdev_bemanitools
1
0
Fork 0
Code Issues 72 Pull Requests 15 Actions Packages Projects Releases 20 Wiki Activity

Use process_hijack_startup instead of hooking an early function

Browse Source
This commit is contained in:
QCDLZCLW3K
2022-05-30 02:00:49 +00:00
committed by 33c17f40
parent 9cfc6ee887
commit 348fbb15d7
1 changed files with 7 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/ddrhookx/dllmain.c
+7 -17
View File
@@ -23,7 +23,7 @@
#include "ddrhookx/master.h" #include "ddrhookx/master.h"
#include "hook/iohook.h" #include "hook/iohook.h"
#include "hook/table.h" #include "hook/process.h"
#include "hooklib/app.h" #include "hooklib/app.h"
#include "hooklib/rs232.h" #include "hooklib/rs232.h"
@@ -48,23 +48,15 @@
bool standard_def; bool standard_def;
bool _15khz; bool _15khz;
static DWORD STDCALL my_main();
static DWORD(STDCALL *real_main)();
static const hook_d3d9_irp_handler_t ddrhookx_d3d9_handlers[] = { static const hook_d3d9_irp_handler_t ddrhookx_d3d9_handlers[] = {
ddrhookx_d3d9_irp_handler, ddrhookx_d3d9_irp_handler,
}; };
static DWORD STDCALL my_GetModuleFileNameA(HMODULE hModule, LPSTR lpFilename, DWORD nSize);
static DWORD(STDCALL *real_GetModuleFileNameA)(HMODULE hModule, LPSTR lpFilename, DWORD nSize);
static bool ddrhookx_init_check = false; static bool ddrhookx_init_check = false;
static const struct hook_symbol init_hook_syms[] = {
{
.name = "GetModuleFileNameA",
.patch = my_GetModuleFileNameA,
.link = (void **) &real_GetModuleFileNameA,
},
};
static void ddrhookx_setup_d3d9_hooks( static void ddrhookx_setup_d3d9_hooks(
const struct ddrhookx_config_gfx *config_gfx) const struct ddrhookx_config_gfx *config_gfx)
{ {
@@ -77,8 +69,7 @@ static void ddrhookx_setup_d3d9_hooks(
hook_d3d9_init(ddrhookx_d3d9_handlers, lengthof(ddrhookx_d3d9_handlers)); hook_d3d9_init(ddrhookx_d3d9_handlers, lengthof(ddrhookx_d3d9_handlers));
} }
static DWORD STDCALL static DWORD STDCALL my_main()
my_GetModuleFileNameA(HMODULE hModule, LPSTR lpFilename, DWORD nSize)
{ {
bool ok; bool ok;
struct cconfig *config; struct cconfig *config;
@@ -176,7 +167,7 @@ my_GetModuleFileNameA(HMODULE hModule, LPSTR lpFilename, DWORD nSize)
log_info("--- End ddrhookx GetModuleFileNameA ---"); log_info("--- End ddrhookx GetModuleFileNameA ---");
skip: skip:
return real_GetModuleFileNameA(hModule, lpFilename, nSize); return real_main();
} }
BOOL WINAPI DllMain(HMODULE self, DWORD reason, void *ctx) BOOL WINAPI DllMain(HMODULE self, DWORD reason, void *ctx)
@@ -184,8 +175,7 @@ BOOL WINAPI DllMain(HMODULE self, DWORD reason, void *ctx)
if (reason == DLL_PROCESS_ATTACH) { if (reason == DLL_PROCESS_ATTACH) {
log_to_writer(log_writer_debug, NULL); log_to_writer(log_writer_debug, NULL);
hook_table_apply( process_hijack_startup(my_main, &real_main);
NULL, "kernel32.dll", init_hook_syms, lengthof(init_hook_syms));
master_insert_hooks(NULL); master_insert_hooks(NULL);
filesystem_hook_init(); filesystem_hook_init();