AimGate/Hooks.cpp at main · SAPNXTDOOR/AimGate · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
#include <windows.h>
#include "InputState.h"

// Hooks
HHOOK keyboardHook = NULL;
HHOOK mouseHook = NULL;

extern bool g_enabled;


/* =========================
   KEYBOARD HOOK
   ========================= */
LRESULT CALLBACK KeyboardProc(int nCode, WPARAM wParam, LPARAM lParam) {
    if (nCode == HC_ACTION) {
        KBDLLHOOKSTRUCT* k = (KBDLLHOOKSTRUCT*)lParam;

        // 🔥 CRITICAL: ignore injected input from SendInput
        if (k->flags & LLKHF_INJECTED) {
            return CallNextHookEx(NULL, nCode, wParam, lParam);
        }

        bool isKeyDown = (wParam == WM_KEYDOWN || wParam == WM_SYSKEYDOWN);
        bool isKeyUp   = (wParam == WM_KEYUP   || wParam == WM_SYSKEYUP);

        if (isKeyDown || isKeyUp) {

            switch (k->vkCode) {

            case 'W':
                if (isKeyDown && !pW) pW = true;
                else if (isKeyUp)     pW = false;
                break;

            case 'A':
                if (isKeyDown && !pA) pA = true;
                else if (isKeyUp)     pA = false;
                break;

            case 'S':
                if (isKeyDown && !pS) pS = true;
                else if (isKeyUp)     pS = false;
                break;

            case 'D':
                if (isKeyDown && !pD) pD = true;
                else if (isKeyUp)     pD = false;
                break;

            default:
                break;
            }

            // Update logical state (Mouse Override only)
            Update();

            // Block original WASD input
            if (g_enabled &&
                (k->vkCode == 'W' || k->vkCode == 'A' ||
                k->vkCode == 'S' || k->vkCode == 'D')) {
                return 1;
            }
        }
    }

    return CallNextHookEx(NULL, nCode, wParam, lParam);
}

/* =========================
   MOUSE HOOK
   ========================= */
LRESULT CALLBACK MouseProc(int nCode, WPARAM wParam, LPARAM lParam) {
    if (nCode == HC_ACTION) {

        if (wParam == WM_LBUTTONDOWN) {
            pFire = true;
            Update();
        }
        else if (wParam == WM_LBUTTONUP) {
            pFire = false;
            Update();
        }
    }

    return CallNextHookEx(NULL, nCode, wParam, lParam);
}

/* =========================
   INSTALL HOOKS
   ========================= */
void InstallKeyboardHook() {
    keyboardHook = SetWindowsHookEx(
        WH_KEYBOARD_LL,
        KeyboardProc,
        NULL,
        0
    );
}

void InstallMouseHook() {
    mouseHook = SetWindowsHookEx(
        WH_MOUSE_LL,
        MouseProc,
        NULL,
        0
    );
}