44 lines
1.0 KiB
C++
44 lines
1.0 KiB
C++
#include "sleephook.h"
|
|
|
|
#include "avs/game.h"
|
|
#include "util/detour.h"
|
|
|
|
// settings
|
|
static DWORD SLEEPHOOK_MS_MAX;
|
|
static DWORD SLEEPHOOK_MS_REPLACE;
|
|
|
|
static decltype(Sleep) *Sleep_orig = nullptr;
|
|
static decltype(SleepEx) *SleepEx_orig = nullptr;
|
|
|
|
static VOID WINAPI Sleep_hook(DWORD dwMilliseconds) {
|
|
if (dwMilliseconds > SLEEPHOOK_MS_MAX) {
|
|
dwMilliseconds = SLEEPHOOK_MS_REPLACE;
|
|
}
|
|
|
|
Sleep_orig(dwMilliseconds);
|
|
}
|
|
|
|
static DWORD WINAPI SleepEx_hook(DWORD dwMilliseconds, BOOL bAltertable) {
|
|
if (dwMilliseconds > SLEEPHOOK_MS_MAX) {
|
|
dwMilliseconds = SLEEPHOOK_MS_REPLACE;
|
|
}
|
|
|
|
return SleepEx_orig(dwMilliseconds, bAltertable);
|
|
}
|
|
|
|
void hooks::sleep::init(DWORD ms_max, DWORD ms_replace, HMODULE module) {
|
|
|
|
// auto module
|
|
if (!module) {
|
|
module = avs::game::DLL_INSTANCE;
|
|
}
|
|
|
|
// settings
|
|
SLEEPHOOK_MS_MAX = ms_max;
|
|
SLEEPHOOK_MS_REPLACE = ms_replace;
|
|
|
|
// hook functions
|
|
Sleep_orig = detour::iat_try("Sleep", Sleep_hook, module);
|
|
SleepEx_orig = detour::iat_try("SleepEx", SleepEx_hook, module);
|
|
}
|