diff --git a/hkbhook/unity.c b/hkbhook/unity.c index 8c0aac1..b1e8344 100644 --- a/hkbhook/unity.c +++ b/hkbhook/unity.c @@ -10,6 +10,7 @@ #include "hooklib/reg.h" #include "hook/procaddr.h" #include "hooklib/serial.h" +#include "platform/clock.h" #include "util/dprintf.h" @@ -96,6 +97,7 @@ static HMODULE WINAPI my_LoadLibraryW(const wchar_t *name) path_hook_insert_hooks(result); reg_hook_insert_hooks(result); proc_addr_insert_hooks(result); + clock_hook_insert_hooks(result); serial_hook_apply_hooks(result); iohook_apply_hooks(result); } diff --git a/mai2hook/unity.c b/mai2hook/unity.c index 4c82b07..6715465 100644 --- a/mai2hook/unity.c +++ b/mai2hook/unity.c @@ -10,6 +10,7 @@ #include "hooklib/reg.h" #include "hook/procaddr.h" #include "hooklib/serial.h" +#include "platform/clock.h" #include "util/dprintf.h" @@ -107,6 +108,7 @@ static HMODULE WINAPI my_LoadLibraryW(const wchar_t *name) path_hook_insert_hooks(result); reg_hook_insert_hooks(result); proc_addr_insert_hooks(result); + clock_hook_insert_hooks(result); serial_hook_apply_hooks(result); iohook_apply_hooks(result); } diff --git a/mu3hook/unity.c b/mu3hook/unity.c index efefc32..62695b9 100644 --- a/mu3hook/unity.c +++ b/mu3hook/unity.c @@ -6,6 +6,7 @@ #include "hooklib/dll.h" #include "hooklib/path.h" +#include "platform/clock.h" #include "util/dprintf.h" @@ -88,6 +89,7 @@ static HMODULE WINAPI my_LoadLibraryW(const wchar_t *name) dll_hook_insert_hooks(result); path_hook_insert_hooks(result); + clock_hook_insert_hooks(result); } } diff --git a/platform/clock.c b/platform/clock.c index b67b111..e163961 100644 --- a/platform/clock.c +++ b/platform/clock.c @@ -4,6 +4,7 @@ #include #include "hook/table.h" +#include "hook/procaddr.h" #include "platform/clock.h" @@ -19,7 +20,9 @@ static BOOL WINAPI my_SetTimeZoneInformation(TIME_ZONE_INFORMATION *tzinfo); static BOOL (WINAPI * next_GetSystemTimeAsFileTime)(FILETIME *out); static int64_t clock_current_day; +static bool clock_timezone; static bool clock_time_warp; +static bool clock_writeable; static const struct hook_symbol clock_base_hook_syms[] = { { @@ -158,7 +161,7 @@ static BOOL WINAPI my_GetSystemTime(SYSTEMTIME *out) return ok; } -#if 0 +#if defined(LOG_CLOCK) static int last_second; if (out->wSecond != last_second) { @@ -225,35 +228,41 @@ HRESULT clock_hook_init(const struct clock_config *cfg) { assert(cfg != NULL); + clock_timezone = cfg->timezone; clock_time_warp = cfg->timewarp; + clock_writeable = cfg->writeable; - if (cfg->timezone || cfg->timewarp || !cfg->writeable) { + clock_hook_insert_hooks(NULL); + + return S_OK; +} + +void clock_hook_insert_hooks(HMODULE target) { + if (clock_timezone || clock_time_warp || !clock_writeable) { /* All the clock hooks require the core GSTAFT hook to be installed */ /* Note the ! up there btw. */ hook_table_apply( - NULL, + target, "kernel32.dll", clock_base_hook_syms, _countof(clock_base_hook_syms)); } - if (cfg->timezone) { + if (clock_timezone) { hook_table_apply( - NULL, + target, "kernel32.dll", clock_read_hook_syms, _countof(clock_read_hook_syms)); } - if (!cfg->writeable) { + if (!clock_writeable) { /* Install hook if this config parameter is FALSE! */ hook_table_apply( - NULL, + target, "kernel32.dll", clock_write_hook_syms, _countof(clock_write_hook_syms)); } - - return S_OK; } diff --git a/platform/clock.h b/platform/clock.h index 4d67754..6d8c201 100644 --- a/platform/clock.h +++ b/platform/clock.h @@ -11,3 +11,4 @@ struct clock_config { }; HRESULT clock_hook_init(const struct clock_config *cfg); +void clock_hook_insert_hooks(HMODULE target);