goldsrc-cheat/src/globals.c

#include <stdio.h>
#include <string.h>
#include <dlfcn.h>

#include "include/globals.h"
#include "include/sdk.h"
#include "include/util.h"

void* hw;
void** h_client;
DECL_INTF(cl_enginefunc_t, engine);
DECL_INTF(cl_clientfunc_t, client);
DECL_INTF(playermove_t, pmove);
DECL_INTF(engine_studio_api_t, enginestudio);
DECL_INTF(StudioModelRenderer_t, studiomodelrenderer);

/* Updated in CL_CreateMove hook */
cl_entity_t* localplayer = NULL;

bool globals_init(void) {
    /*
     * Get handler for hw.so
     *  RTLD_LAZY: If the symbol is never referenced, then it is never resolved.
     *  RTLD_NOLOAD: Don't load the shared object.
     */
    hw = dlopen("hw.so", RTLD_LAZY | RTLD_NOLOAD);

    /* Make sure it's a valid handler */
    if (!hw) {
        printf("hl-cheat: globals_init: can't open hw.so\n");
        return false;
    }

    h_client = (void**)dlsym(hw, "hClientDLL");

    if (!h_client) {
        printf("hl-cheat: globals_init: can't find hClientDLL\n");
        return false;
    }

    /* Get symbol addresses using dlsym and the handler we just opened */
    i_engine       = (cl_enginefunc_t*)dlsym(hw, "cl_enginefuncs");
    i_client       = (cl_clientfunc_t*)dlsym(hw, "cl_funcs");
    i_pmove        = *(playermove_t**)dlsym(hw, "pmove");
    i_enginestudio = (engine_studio_api_t*)dlsym(hw, "engine_studio_api");

    const char* SMR_STR   = "g_StudioRenderer"; /* For clang-format */
    i_studiomodelrenderer = *(StudioModelRenderer_t**)dlsym(*h_client, SMR_STR);

    if (!i_engine || !i_client || !i_pmove || !i_enginestudio ||
        !i_studiomodelrenderer) {
        printf("hl-cheat: globals_init: couldn't load some symbols\n");
        return false;
    }

    if (!unprotect_addr(i_studiomodelrenderer)) {
        printf("hl-cheat: globals_init: couldn't unprotect address of SMR\n");
        return false;
    }

    globals_store();

    return true;
}

void globals_store(void) {
    memcpy(&o_engine, i_engine, sizeof(cl_enginefunc_t));
    memcpy(&o_client, i_client, sizeof(cl_clientfunc_t));
}

void globals_restore(void) {
    memcpy(i_engine, &o_engine, sizeof(cl_enginefunc_t));
    memcpy(i_client, &o_client, sizeof(cl_clientfunc_t));
}