Subversion Repositories WoWGM

/****************************************************************************
*
*   Client.cpp
*
*   Program entry point (DllMain)
* 
***/

#include "pch.h"
#pragma hdrstop

#include "Client.h"
#include "ClientCommands.h"
#include "ClientDebugCommands.h"
#include <WowSvcs/ClientServices.h>
#include <Glue/CGlueMgr.h>
#include <Ui/GameUI.h>
#include <FrameScript/FrameScript.h>
#include <Object/ObjectClient/Player_C.h>


/****************************************************************************
*
*   Client memory addresses
*
***/

#define  CLIENTREGISTERCONSOLECOMMANDS    0x00401B60;
#define  CLIENTUNREGISTERCONSOLECOMMANDS  0x00402910;
#define  DESTROYHANDLERPLAYER             0x004066D0;
#define  INITIALIZEHANDLERPLAYER          0x004047E0;
#define  SCRIPT_GETBUILDINFO_PTR          0x004DBE60;
#define  WOWCLIENTINIT_PTR                0x00404130;


/****************************************************************************
*
*   Client function pointers
*
***/

void (*ClientRegisterConsoleCommandsPtr)() = *(void(*)())CLIENTREGISTERCONSOLECOMMANDS;

void (*ClientUnregisterConsoleCommandsPtr)()= *(void(*)())CLIENTUNREGISTERCONSOLECOMMANDS;

void (*DestroyHandlerPlayerPtr)() = *(void(*)())DESTROYHANDLERPLAYER;

void (*InitializeHandlerPlayerPtr)() = *(void(*)())INITIALIZEHANDLERPLAYER;

int (*Script_GetBuildInfoPtr)(lua_State*) = *(int(*)(lua_State *))SCRIPT_GETBUILDINFO_PTR;

void (*WowClientInitPtr)() = *(void(*)())WOWCLIENTINIT_PTR;


/****************************************************************************
*
*   Program entry point (DllMain)
*
***/

//===========================================================================
extern "C" BOOL WINAPI DllMain (HINSTANCE const instance,
                                DWORD const     reason,
                                LPVOID const    reserved) {

        switch (reason) {
                case DLL_PROCESS_ATTACH:
    {
                        WowGM::Initialize();
                        ConsoleCommandInitialize();
                        WowGM::InitializeHandlerPlayer();
                        WowGM::PlayerClientInitialize();
                        WowGM::ClientRegisterConsoleCommands();
                        WowGM::CGlueMgr::Initialize();
                        WowGM::CGGameUI::Initialize();
                        //ClientServices_Initialize();
                        WowGM::InstallGameConsoleCommands();
                        ConsoleAccessSetEnabled(TRUE);
                        break;
                }
                case DLL_THREAD_ATTACH:
                case DLL_THREAD_DETACH:
                        break;
                case DLL_PROCESS_DETACH:
    {
                        WowGM::DestroyHandlerPlayer();
                        WowGM::ClientUnregisterConsoleCommands();
      WowGM::CGlueMgr::Shutdown();
                        WowGM::CGGameUI::Shutdown();
                        //ClientServices_Destroy();
                        WowGM::UninstallGameConsoleCommands();
                        break;
                }
        }

        return TRUE;
}


/****************************************************************************
*
*   Private
*
***/

//===========================================================================
void WowGM::Initialize ()  {
        DetourRestoreAfterWith();
        FixInvalidPtrCheck();
}

//===========================================================================
void WowGM::FixInvalidPtrCheck () {
        // Allow callbacks outside of the .text section
        *((int*)0x00D415B8) = 0x00000001;
        *((int*)0x00D415BC) = 0x7FFFFFFF;
}

//===========================================================================
void InitializeHandlerPlayer () {
        InitializeHandlerPlayerPtr();

        ConsoleCommandExecute("run autoexec.wtf",FALSE);
}

//===========================================================================
void ClientRegisterConsoleCommands () {
        ClientRegisterConsoleCommandsPtr();

        g_password = CVar::Register("password",
                              "stored password for blizzcon, dev, qa, etc.",
                              NULL,
                              "",
                              NULL,
                              GAME,
                              false,
                              NULL,
                              false);

  ConsoleCommandRegister("Bug", CCommand_Bug, DEBUG, NOHELP);
  ConsoleCommandRegister("Suggestion", CCommand_Bug, DEBUG, NOHELP);
}

//===========================================================================
void DestroyHandlerPlayer () {
        DestroyHandlerPlayerPtr();
}

//===========================================================================
void ClientUnregisterConsoleCommands () {
        ClientUnregisterConsoleCommandsPtr();
  
  ConsoleCommandUnregister("Bug");
  ConsoleCommandUnregister("Suggestion");
}

//===========================================================================
int Script_GetBuildInfo (lua_State* L) {
        char* version = FrameScript_GetText("VERSION", -1, GENDER_NONE);
        _lua_pushstring(L, version);
#ifdef _DEBUG
        _lua_pushstring(L,"Debug");
#else

        _lua_pushstring(L, "GM");
#endif // #ifdef _DEBUG
        _lua_pushstring(L, "3.3.5");
        _lua_pushstring(L, "12340");
        _lua_pushstring(L, __DATE__);
        return 5;       // TODO: automatically return the number of arguments returned by Lua
}

//===========================================================================
void WowClientInit() {
        WowClientInitPtr();

        WowGM::CGlueMgr::SetLoginPassword();
        ConsoleCommandExecute("run autoexec.wtf",FALSE);
}


/****************************************************************************
*
*   Detours setup
*
***/

//===========================================================================
void WowGM::InitializeHandlerPlayer () {
        DETOUR_INIT;
        DETOUR_ATTACH(WowClientInitPtr,WowClientInit);
        DETOUR_ATTACH(InitializeHandlerPlayerPtr,::InitializeHandlerPlayer);
        DETOUR_ATTACH(Script_GetBuildInfoPtr, Script_GetBuildInfo);
        DETOUR_ATTACH(DestroyHandlerPlayerPtr,::DestroyHandlerPlayer);
        DETOUR_COMMIT;
}

//===========================================================================
void WowGM::ClientRegisterConsoleCommands () {
        DETOUR_INIT;
        DETOUR_ATTACH(ClientRegisterConsoleCommandsPtr,::ClientRegisterConsoleCommands);
        DETOUR_ATTACH(ClientUnregisterConsoleCommandsPtr,::ClientUnregisterConsoleCommands);
        DETOUR_COMMIT;
}

//===========================================================================
void WowGM::DestroyHandlerPlayer () {
        DETOUR_INIT;
        DETOUR_DETACH(WowClientInitPtr,WowClientInit);
        DETOUR_DETACH(InitializeHandlerPlayerPtr,::InitializeHandlerPlayer);
        DETOUR_DETACH(DestroyHandlerPlayerPtr,::DestroyHandlerPlayer);
        DETOUR_DETACH(Script_GetBuildInfoPtr, Script_GetBuildInfo);
        DETOUR_COMMIT;
}

//===========================================================================
void WowGM::ClientUnregisterConsoleCommands () {
        DETOUR_INIT;
        DETOUR_DETACH(ClientRegisterConsoleCommandsPtr,::ClientRegisterConsoleCommands);
        DETOUR_DETACH(ClientUnregisterConsoleCommandsPtr,::ClientUnregisterConsoleCommands);
        DETOUR_COMMIT;
}