From 95fdd19b0196c72af3ef2c5d6f6df8e349f03880 Mon Sep 17 00:00:00 2001 From: FunkyFr3sh Date: Sat, 1 Dec 2018 06:28:46 +0100 Subject: [PATCH] remove VirtualQuery --- src/dinput.c | 13 +++++-------- src/mouse.c | 10 +++------- 2 files changed, 8 insertions(+), 15 deletions(-) diff --git a/src/dinput.c b/src/dinput.c index 155b1d9..5185c53 100644 --- a/src/dinput.c +++ b/src/dinput.c @@ -15,17 +15,14 @@ static PROC HookFunc(PROC *orgFunc, PROC newFunc) { PROC org = *orgFunc; DWORD oldProtect; - MEMORY_BASIC_INFORMATION mbi; - if (VirtualQuery(orgFunc, &mbi, sizeof(MEMORY_BASIC_INFORMATION))) + if (VirtualProtect(orgFunc, sizeof(PROC), PAGE_EXECUTE_READWRITE, &oldProtect)) { - if (VirtualProtect(mbi.BaseAddress, mbi.RegionSize, PAGE_READWRITE, &oldProtect)) - { - *orgFunc = newFunc; - VirtualProtect(mbi.BaseAddress, mbi.RegionSize, oldProtect, &oldProtect); - return org; - } + *orgFunc = newFunc; + VirtualProtect(orgFunc, sizeof(PROC), oldProtect, &oldProtect); + return org; } + return 0; } diff --git a/src/mouse.c b/src/mouse.c index b6b13a7..154ca3f 100644 --- a/src/mouse.c +++ b/src/mouse.c @@ -180,15 +180,11 @@ void HookIAT(HMODULE hMod, char *moduleName, char *functionName, PROC newFunctio _stricmp((const char *)pImport->Name, functionName) == 0) { DWORD oldProtect; - MEMORY_BASIC_INFORMATION mbi; - if (VirtualQuery(&pFirstThunk->u1.Function, &mbi, sizeof(MEMORY_BASIC_INFORMATION))) + if (VirtualProtect(&pFirstThunk->u1.Function, sizeof(DWORD), PAGE_READWRITE, &oldProtect)) { - if (VirtualProtect(mbi.BaseAddress, mbi.RegionSize, PAGE_READWRITE, &oldProtect)) - { - pFirstThunk->u1.Function = (DWORD)newFunction; - VirtualProtect(mbi.BaseAddress, mbi.RegionSize, oldProtect, &oldProtect); - } + pFirstThunk->u1.Function = (DWORD)newFunction; + VirtualProtect(&pFirstThunk->u1.Function, sizeof(DWORD), oldProtect, &oldProtect); } break;