black_ops_1/dllmain.cpp at main · crvvdev/black_ops_1 · 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
#include "includes.hpp"
#include <MinHook.h>

using namespace Hooks;
bool g_minHookInitialized = false;

bool InitializeD3D(DWORD_PTR **d3dDeviceVTable)
{
    assert(d3dDeviceVTable);

    IDirect3D9 *d3d9 = Direct3DCreate9(D3D_SDK_VERSION);
    if (!d3d9)
    {
        return false;
    }

    D3DPRESENT_PARAMETERS d3dpp{};
    d3dpp.Windowed = TRUE;
    d3dpp.SwapEffect = D3DSWAPEFFECT_DISCARD;
    d3dpp.BackBufferFormat = D3DFMT_UNKNOWN;

    IDirect3DDevice9 *d3dDevice = nullptr;
    HRESULT hr = d3d9->CreateDevice(D3DADAPTER_DEFAULT, D3DDEVTYPE_HAL, GetDesktopWindow(),
                                    D3DCREATE_SOFTWARE_VERTEXPROCESSING, &d3dpp, &d3dDevice);
    if (FAILED(hr) || !d3dDevice)
    {
        d3d9->Release();
        return false;
    }

    *d3dDeviceVTable = *(DWORD_PTR **)(d3dDevice);

    return true;
}

void OnDetach()
{
    if (!g_minHookInitialized)
    {
        return;
    }

    MH_DisableHook(MH_ALL_HOOKS);
    MH_Uninitialize();
}

DWORD WINAPI StartRoutine(LPVOID)
{
    DWORD_PTR *d3dDeviceVTable = nullptr;

    if (!InitializeD3D(&d3dDeviceVTable))
    {
        return EXIT_FAILURE;
    }

    /*
        Aim 360 Patch
    */
    Tools::Patch((void *)AIM_360, "\xEB", 1);

    MH_Initialize();

    MH_CreateHook(reinterpret_cast<PVOID>(d3dDeviceVTable[16]), &hkReset, reinterpret_cast<LPVOID *>(&oReset));
    MH_CreateHook(reinterpret_cast<PVOID>(d3dDeviceVTable[42]), &hkEndScene, reinterpret_cast<LPVOID *>(&oEndScene));

    // R_EndFrame is the procedure which the game renders the current frame/scene
    MH_CreateHook(reinterpret_cast<PVOID>(R_ENDFRAME), &hkR_EndFrame, reinterpret_cast<LPVOID *>(&oR_EndFrame));

    // CL_WritePacket is one of the procedures which the game process player input
    MH_CreateHook(reinterpret_cast<PVOID>(CL_WRITEPACKET), &hkCL_WritePacket,
                  reinterpret_cast<LPVOID *>(&oCL_WritePacket));

    // CG_PredictPlayerState is one of the procedures which the game process player input as well
    /* MH_CreateHook(reinterpret_cast<PVOID>(CG_PREDICTPLAYERSTATE), &hkCG_PredictPlayerState,
                   reinterpret_cast<LPVOID *>(&oCG_PredictPlayerState));*/

    MH_EnableHook(MH_ALL_HOOKS);

    g_minHookInitialized = true;

    return EXIT_SUCCESS;
}

BOOL WINAPI DllMain(HMODULE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
{
    UNREFERENCED_PARAMETER(lpvReserved);

    switch (fdwReason)
    {
    case DLL_PROCESS_ATTACH: {
        Globals::g_thisModule = hinstDLL;

        // Create a new thread to bootstrap our hooks
        CreateThread(nullptr, 0, StartRoutine, nullptr, 0, nullptr);
        break;
    }
    case DLL_PROCESS_DETACH: {
        OnDetach();
        break;
    }
    }
    return true;
}