Injector.cpp

//
// Created by emiel on 6/03/21.
//
//
#include "Injector.h"
#include "Module.h"
#include <iostream>
#include "vtablehook.h"
#include "hooks/Hooks.h"
#include "hexdump.h"
#include "NetVar.h"
#include "netvars.h"
#include "DrawTools.h"

IBaseClientDLL *g_clientdll   = nullptr;
void *g_cvar                  = nullptr;
vgui::IPanel *g_panels        = nullptr;
vgui::ISurface *g_surface     = nullptr;
IVEngineClient *g_engineClient  = nullptr;
IClientEntityList *g_entityList = nullptr;
IInputSystem *g_inputSystem     = nullptr;
IVRenderView *g_renderView = nullptr;
IVModelRender *g_modelRender = nullptr;
IVModelInfo *g_modelInfo = nullptr;
IMaterialSystem *g_materialSystem = nullptr;

bool draw::enabled = false;
int draw::screenWidth = -1;
int draw::screenHeight = -1;

/**
 * Grabs interface pointer with given name from given module and exits the process if not found.
 * @param module
 * @param name
 * @return pointer to given interface of given module
 */

template <typename T>
T grabCriticalInterface(const std::shared_ptr<Module>& module, const char *name)
{
    auto interface = module->CreateInterface(name, nullptr);
    if (!interface)
    {
        printf("interface %s not found!\n", name);
        exit(1);
    }
    return static_cast<T>(interface);
}

void Injector::inject()
{
    printf("Loading injector!\n");

    auto client_module  = Module::grab("tf/bin/client.so");
    auto vstdlib_module = Module::grab("bin/libvstdlib.so");
    auto vgui2_module   = Module::grab("bin/vgui2.so");
    auto surface_module = Module::grab("bin/vguimatsurface.so");
    auto engine_module  = Module::grab("bin/engine.so");
    auto input_module   = Module::grab("bin/inputsystem.so");
    auto mat_sys        = Module::grab("bin/materialsystem.so");


    auto interfaceregs_symb =  client_module->getSymbol("s_pInterfaceRegs");

    if (interfaceregs_symb)
    {
        client_module->close();
        printf("s_pInterfaceRegs found! Using this instead of CreateInterface.\n");
    }
    else
    {

        g_clientdll = grabCriticalInterface<IBaseClientDLL*>(client_module, "VClient017");
        g_panels    = grabCriticalInterface<vgui::IPanel*>(vgui2_module, "VGUI_Panel009");
        g_cvar      = grabCriticalInterface<void*>(vstdlib_module, "VEngineCvar004");
        g_surface   = grabCriticalInterface<vgui::ISurface*>(surface_module, "VGUI_Surface030");
        g_engineClient = grabCriticalInterface<IVEngineClient*>(engine_module, "VEngineClient014");
        g_entityList   = grabCriticalInterface<IClientEntityList*>(client_module, "VClientEntityList003");
        g_inputSystem  = grabCriticalInterface<IInputSystem*>(input_module, "InputSystemVersion001");
        g_modelRender  = grabCriticalInterface<IVModelRender*>(engine_module, "VEngineModel016");
        g_modelInfo    = grabCriticalInterface<IVModelInfo*>(engine_module, "VModelInfoClient006");
        g_renderView   = grabCriticalInterface<IVRenderView*>(engine_module, "VEngineRenderView014");
        g_materialSystem = grabCriticalInterface<IMaterialSystem*>(mat_sys, "VMaterialSystem081");

        auto clientVMT  = *(void***)g_clientdll;
        auto clientMode = **(void***)((char*)clientVMT[10] + 1);

        engine_module->walkInterfaces();

        gNetvars.init();
        netvar.Init();

        INSTANTIATE_HOOK(g_panels, Panel_PaintTraverse);
        INSTANTIATE_HOOK(clientMode, ClientMode_CreateMove);
        INSTANTIATE_HOOK(g_modelRender, DrawModelExecute);
        INSTANTIATE_HOOK(g_engineClient->GetBSPTreeQuery(), ListLeavesInBox);

        engine_module->findSignature("");
    }

    printf("Loaded successfully\n");


}