From 86e3de4ad695e10a9169699ab02dfdcf5dc6b2c8 Mon Sep 17 00:00:00 2001 From: FunkyFr3sh Date: Mon, 27 Feb 2023 18:25:28 +0100 Subject: [PATCH] hook PeekMessageA for EV Nova --- inc/dd.h | 1 + inc/hook.h | 6 ++- inc/winapi_hooks.h | 1 + src/config.c | 4 +- src/hook.c | 2 + src/winapi_hooks.c | 96 ++++++++++++++++++++++++++++++++++++++++++++++ 6 files changed, 108 insertions(+), 2 deletions(-) diff --git a/inc/dd.h b/inc/dd.h index 0607d42..b164176 100644 --- a/inc/dd.h +++ b/inc/dd.h @@ -176,6 +176,7 @@ typedef struct CNCDDRAW SPEEDLIMITER flip_limiter; DWORD gui_thread_id; BOOL rgb555; + BOOL hook_peekmessage; } CNCDDRAW; diff --git a/inc/hook.h b/inc/hook.h index 21a8393..c9c4400 100644 --- a/inc/hook.h +++ b/inc/hook.h @@ -9,7 +9,7 @@ #define SKIP_HOOK3 0x00000002l typedef struct HOOKLISTDATA { char function_name[32]; PROC new_function; PROC* function; DWORD flags; } HOOKLISTDATA; -typedef struct HOOKLIST { char module_name[32]; HOOKLISTDATA data[26]; } HOOKLIST; +typedef struct HOOKLIST { char module_name[32]; HOOKLISTDATA data[27]; } HOOKLIST; typedef BOOL(WINAPI* GETCURSORPOSPROC)(LPPOINT); typedef BOOL(WINAPI* CLIPCURSORPROC)(const RECT*); @@ -45,6 +45,9 @@ typedef int (WINAPI* STRETCHDIBITSPROC)( HDC, int, int, int, int, int, int, int, int, const VOID*, const BITMAPINFO*, UINT, DWORD); typedef HHOOK(WINAPI* SETWINDOWSHOOKEXAPROC)(int, HOOKPROC, HINSTANCE, DWORD); +typedef BOOL(WINAPI* PEEKMESSAGEAPROC)(LPMSG, HWND, UINT, UINT, UINT); + + typedef int (WINAPI* GETDEVICECAPSPROC)(HDC, int); typedef HMODULE(WINAPI* LOADLIBRARYAPROC)(LPCSTR); typedef HMODULE(WINAPI* LOADLIBRARYWPROC)(LPCWSTR); @@ -82,6 +85,7 @@ extern STRETCHBLTPROC real_StretchBlt; extern SETDIBITSTODEVICEPROC real_SetDIBitsToDevice; extern STRETCHDIBITSPROC real_StretchDIBits; extern SETWINDOWSHOOKEXAPROC real_SetWindowsHookExA; +extern PEEKMESSAGEAPROC real_PeekMessageA; extern GETDEVICECAPSPROC real_GetDeviceCaps; extern LOADLIBRARYAPROC real_LoadLibraryA; extern LOADLIBRARYWPROC real_LoadLibraryW; diff --git a/inc/winapi_hooks.h b/inc/winapi_hooks.h index 686e446..f84cf40 100644 --- a/inc/winapi_hooks.h +++ b/inc/winapi_hooks.h @@ -30,6 +30,7 @@ BOOL WINAPI fake_ShowWindow(HWND hWnd, int nCmdShow); HWND WINAPI fake_GetTopWindow(HWND hWnd); HWND WINAPI fake_GetForegroundWindow(void); HHOOK WINAPI fake_SetWindowsHookExA(int idHook, HOOKPROC lpfn, HINSTANCE hmod, DWORD dwThreadId); +BOOL WINAPI fake_PeekMessageA(LPMSG lpMsg, HWND hWnd, UINT wMsgFilterMin, UINT wMsgFilterMax, UINT wRemoveMsg); int WINAPI fake_GetDeviceCaps(HDC hdc, int index); BOOL WINAPI fake_StretchBlt( diff --git a/src/config.c b/src/config.c index ebc1f16..d3ff152 100644 --- a/src/config.c +++ b/src/config.c @@ -56,6 +56,7 @@ void cfg_load() g_ddraw->limit_bltfast = cfg_get_bool("limit_bltfast", FALSE); g_ddraw->opengl_core = cfg_get_bool("opengl_core", FALSE); g_ddraw->rgb555 = cfg_get_bool("rgb555", FALSE); + g_ddraw->hook_peekmessage = cfg_get_bool("hook_peekmessage", FALSE); cfg_get_string("screenshotdir", ".\\Screenshots\\", g_ddraw->screenshot_dir, sizeof(g_ddraw->screenshot_dir)); if (g_ddraw->locktopleft) @@ -378,6 +379,7 @@ static void cfg_create_ini() "fixmousehook=false\n" "bpp=0\n" "rgb555=false\n" + "hook_peekmessage=false\n" "\n" "\n" "\n" @@ -740,7 +742,7 @@ static void cfg_create_ini() "[EV Nova]\n" "renderer=opengl\n" "devmode=true\n" - "resizable=false\n" + "hook_peekmessage=true\n" "rgb555=true\n" "keytogglefullscreen=0x46\n" "\n" diff --git a/src/hook.c b/src/hook.c index 2a2a8d5..4ce22fd 100644 --- a/src/hook.c +++ b/src/hook.c @@ -46,6 +46,7 @@ STRETCHBLTPROC real_StretchBlt = StretchBlt; SETDIBITSTODEVICEPROC real_SetDIBitsToDevice = SetDIBitsToDevice; STRETCHDIBITSPROC real_StretchDIBits = StretchDIBits; SETWINDOWSHOOKEXAPROC real_SetWindowsHookExA = SetWindowsHookExA; +PEEKMESSAGEAPROC real_PeekMessageA = PeekMessageA; GETDEVICECAPSPROC real_GetDeviceCaps = GetDeviceCaps; LOADLIBRARYAPROC real_LoadLibraryA = LoadLibraryA; LOADLIBRARYWPROC real_LoadLibraryW = LoadLibraryW; @@ -84,6 +85,7 @@ static HOOKLIST g_hooks[] = { "ShowWindow", (PROC)fake_ShowWindow, (PROC*)&real_ShowWindow, 0 }, { "GetTopWindow", (PROC)fake_GetTopWindow, (PROC*)&real_GetTopWindow, 0 }, { "GetForegroundWindow", (PROC)fake_GetForegroundWindow, (PROC*)&real_GetForegroundWindow, 0 }, + { "PeekMessageA", (PROC)fake_PeekMessageA, (PROC*)&real_PeekMessageA, 0 }, { "", NULL, NULL, 0 } } }, diff --git a/src/winapi_hooks.c b/src/winapi_hooks.c index e82c09b..daa9ebb 100644 --- a/src/winapi_hooks.c +++ b/src/winapi_hooks.c @@ -553,6 +553,102 @@ HHOOK WINAPI fake_SetWindowsHookExA(int idHook, HOOKPROC lpfn, HINSTANCE hmod, D return real_SetWindowsHookExA(idHook, lpfn, hmod, dwThreadId); } +BOOL WINAPI fake_PeekMessageA(LPMSG lpMsg, HWND hWnd, UINT wMsgFilterMin, UINT wMsgFilterMax, UINT wRemoveMsg) +{ + BOOL result = real_PeekMessageA(lpMsg, hWnd, wMsgFilterMin, wMsgFilterMax, wRemoveMsg); + + if (result) + { + switch (lpMsg->message) + { + case WM_LBUTTONUP: + case WM_RBUTTONUP: + case WM_MBUTTONUP: + { + if (!g_ddraw->devmode && !g_mouse_locked) + { + int x = GET_X_LPARAM(lpMsg->lParam); + int y = GET_Y_LPARAM(lpMsg->lParam); + + if (x > g_ddraw->render.viewport.x + g_ddraw->render.viewport.width || + x < g_ddraw->render.viewport.x || + y > g_ddraw->render.viewport.y + g_ddraw->render.viewport.height || + y < g_ddraw->render.viewport.y) + { + x = g_ddraw->width / 2; + y = g_ddraw->height / 2; + } + else + { + x = (DWORD)((x - g_ddraw->render.viewport.x) * g_ddraw->render.unscale_w); + y = (DWORD)((y - g_ddraw->render.viewport.y) * g_ddraw->render.unscale_h); + } + + InterlockedExchange((LONG*)&g_ddraw->cursor.x, x); + InterlockedExchange((LONG*)&g_ddraw->cursor.y, y); + + mouse_lock(); + return FALSE; + } + /* fall through for lParam */ + } + /* down messages are ignored if we have no cursor lock */ + case WM_XBUTTONDBLCLK: + case WM_XBUTTONDOWN: + case WM_XBUTTONUP: + case WM_MOUSEWHEEL: + case WM_MOUSEHOVER: + case WM_LBUTTONDBLCLK: + case WM_MBUTTONDBLCLK: + case WM_RBUTTONDBLCLK: + case WM_LBUTTONDOWN: + case WM_RBUTTONDOWN: + case WM_MBUTTONDOWN: + case WM_MOUSEMOVE: + { + if (!g_ddraw->devmode && !g_mouse_locked) + { + return FALSE; + } + + int x = max(GET_X_LPARAM(lpMsg->lParam) - g_ddraw->mouse.x_adjust, 0); + int y = max(GET_Y_LPARAM(lpMsg->lParam) - g_ddraw->mouse.y_adjust, 0); + + if (g_ddraw->adjmouse) + { + if (g_ddraw->vhack && !g_ddraw->devmode) + { + POINT pt = { 0, 0 }; + fake_GetCursorPos(&pt); + + x = pt.x; + y = pt.y; + } + else + { + x = (DWORD)(roundf(x * g_ddraw->render.unscale_w)); + y = (DWORD)(roundf(y * g_ddraw->render.unscale_h)); + } + } + + x = min(x, g_ddraw->width - 1); + y = min(y, g_ddraw->height - 1); + + InterlockedExchange((LONG*)&g_ddraw->cursor.x, x); + InterlockedExchange((LONG*)&g_ddraw->cursor.y, y); + + lpMsg->lParam = MAKELPARAM(x, y); + fake_GetCursorPos(&lpMsg->pt); + + break; + } + + } + } + + return result; +} + int WINAPI fake_GetDeviceCaps(HDC hdc, int index) { if (g_ddraw &&