From 95fdd19b0196c72af3ef2c5d6f6df8e349f03880 Mon Sep 17 00:00:00 2001
From: FunkyFr3sh <cc.red.alert.1@googlemail.com>
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;