Subversion Repositories WoWGM

/****************************************************************************
*
*   Client.cpp
*
*   Program entry point (DllMain)
*
*   Written by Tristan Cormier
*   11.20.24
* 
***/

#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>


//===========================================================================
BOOL CCommand_Bug(char const* command, char const* args);


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

#define  CLIENTREGISTERCONSOLECOMMANDS    0x00401B60
#define  COMMONMAINDESTROY                0x004066D0
#define  COMMONMAININITIALIZE             0x004047E0
#define  SCRIPT_GETBUILDINFO              0x004DBE60
#define  WOWCLIENTDESTROY                 0x00402910
#define  WOWCLIENTINIT                    0x00404130


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

void  (*ClientRegisterConsoleCommands)()  = *(void(*)())CLIENTREGISTERCONSOLECOMMANDS;
void  (*CommonMainDestroy)()              = *(void(*)())COMMONMAINDESTROY;
void  (*CommonMainInitialize)()           = *(void(*)())COMMONMAININITIALIZE;
int   (*Script_GetBuildInfo)(lua_State*)  = *(int(*)(lua_State *))SCRIPT_GETBUILDINFO;
void  (*WowClientDestroy)()               = *(void(*)())WOWCLIENTDESTROY;
void  (*WowClientInit)()                  = *(void(*)())WOWCLIENTINIT;


/****************************************************************************
*
*   Callbacks
*
***/

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

        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 CommonMainDestroyProc () {
        CommonMainDestroy();
}
//===========================================================================
void CommonMainInitializeProc () {
  ConsoleAccessSetEnabled(TRUE);
        CommonMainInitialize();
}

//===========================================================================
static int Script_GetBuildInfoProc (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;
}

//===========================================================================
void WowClientDestroyProc () {
  ConsoleCommandUnregister("Bug");
  ConsoleCommandUnregister("Suggestion");
  WowClientDestroy();
}

//===========================================================================
void WowClientInitProc () {
        WowClientInit();
  WowGM::CGlueMgr::SetLoginPassword();
  ConsoleCommandExecute("run autoexec.wtf",FALSE);
}


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

//===========================================================================
void RegisterClientCallbacks () {
  DETOUR_INIT;
  DETOUR_ATTACH(CommonMainDestroy, CommonMainDestroyProc);
  DETOUR_ATTACH(CommonMainInitialize, CommonMainInitializeProc);
  DETOUR_ATTACH(ClientRegisterConsoleCommands, ClientRegisterConsoleCommandsProc);
  DETOUR_ATTACH(Script_GetBuildInfo, Script_GetBuildInfoProc);
  DETOUR_ATTACH(WowClientDestroy, WowClientDestroyProc);
  DETOUR_ATTACH(WowClientInit, WowClientInitProc);
  DETOUR_COMMIT;
}

//===========================================================================
void UnregisterClientCallbacks () {
  DETOUR_INIT;
  DETOUR_DETACH(CommonMainDestroy, CommonMainDestroyProc);
  DETOUR_DETACH(CommonMainInitialize, CommonMainInitializeProc);
  DETOUR_DETACH(ClientRegisterConsoleCommands, ClientRegisterConsoleCommandsProc);
  DETOUR_DETACH(Script_GetBuildInfo, Script_GetBuildInfoProc);
  DETOUR_DETACH(WowClientDestroy, WowClientDestroyProc);
  DETOUR_DETACH(WowClientInit, WowClientInitProc);
  DETOUR_COMMIT;
}

//===========================================================================
BOOL CCommand_Bug (char const* command, char const* args) {
  //TODO:
  //ASSERT(command);

  unsigned int category;

  if (!SStrCmpI(command, "bug"))
    category = 0;
  else if (!SStrCmpI(command, "suggestion"))
    category = 1;

  if (ClientServices_Report(category, args))
    ConsolePrintf("%s submitted", command);
  else
    ConsolePrintf("%s submission failed", command);

  return TRUE;
}


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

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

        switch (reason) {
                case DLL_PROCESS_ATTACH:
    {
      // Allow callbacks outside of the .text section
      *((int*)0x00D415B8) = 0x00000001;
      *((int*)0x00D415BC) = 0x7FFFFFFF;

      // Initialize the Detour library
      DetourRestoreAfterWith();

      RegisterClientCallbacks();
                        ConsoleCommandInitialize();
                        CGlueMgr::Initialize();
                        CGGameUI::Initialize();
      //TODO:
                        //ClientServices_Initialize();
                        WowGM::InstallGameConsoleCommands();
                        break;
                }
                case DLL_THREAD_ATTACH:
                case DLL_THREAD_DETACH:
                        break;
                case DLL_PROCESS_DETACH:
    {
      UnregisterClientCallbacks();
      //TODO:
      WowGM::CGlueMgr::Shutdown();
                        CGGameUI::Shutdown();
                        //ClientServices_Destroy();
                        WowGM::UninstallGameConsoleCommands();
                        break;
                }
        }

        return TRUE;
}