initial commit

Signed-off-by: Toni Uhlig <matzeton@googlemail.com>

1 files changed, 78 insertions, 0 deletions

diff --git a/csgo_wh/src/DLLMain.cpp b/csgo_wh/src/DLLMain.cpp
new file mode 100755
index 0000000..b9d12cb
--- /dev/null
+++ b/csgo_wh/src/DLLMain.cpp
@@ -0,0 +1,78 @@
+#include "DLLMain.h"
+
+#include <cstdio>
+#include <d3d9.h>
+#include <stdarg.h>
+#include <windows.h>
+#include <psapi.h>
+#include <commctrl.h>
+
+#include "Hook.h"
+#include "D9DW.h"
+#include "Game.h"
+
+#define DXD9DEVICE_OFFSET 0x198298 + 0x44
+
+
+typedef __int32 (__stdcall* EndScene_t)(LPDIRECT3DDEVICE9);
+
+static LPDIRECT3DDEVICE9 pDevice = NULL;
+static bool bActive = true;
+static EndScene_t pEndScene = NULL;
+static D9DW ddraw;
+//static Game game;
+
+
+__int32 __stdcall hkEndScene(LPDIRECT3DDEVICE9 pDevice_t)
+{
+    asm volatile ("nop; nop; nop");
+    int retOrigEndScene = D3D_OK;
+    if(pDevice == NULL)
+    {
+        pDevice = pDevice_t;
+        //DEBUG_LOG("D3D9 Device (%X): %p", (UINT32)((UINT32) game.m_shaderapi.lpBaseOfDll + DXD9DEVICE_OFFSET), pDevice);
+        DEBUG_LOG("EndScene Arg: %p", pDevice_t);
+        DEBUG_FLUSH;
+        ddraw.Create(pDevice);
+    }
+    else
+    {
+        HRESULT c_ret = ddraw.checkDxDevice();
+        if (c_ret == D3D_OK)
+        {
+            //game.ReadCVars();
+            ddraw.Render(/* &game */ NULL, /* game.ReadEntities() */ false);
+        }
+        else pEndScene = NULL;
+        retOrigEndScene = (pEndScene != NULL ? pEndScene(pDevice_t) : D3D_OK);
+    }
+    asm volatile ("nop; nop; nop");
+    return retOrigEndScene;
+}
+
+DWORD WINAPI MainThread(void *arg)
+{
+    //game.Init();
+    Hook::hookEndScene(hkEndScene, &pEndScene, false);
+    while ( bActive )
+    {
+        Sleep(500);
+        if ( GetForegroundWindow() != FindWindow( 0, "Counter-Strike: Global Offensive" ) )
+        {
+            pDevice = NULL;
+            ddraw.Release();
+        }
+    }
+    return 0;
+}
+
+__declspec(dllexport) BOOL APIENTRY DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
+{
+    if (fdwReason == DLL_PROCESS_ATTACH)
+    {
+        DWORD dwThreadId;
+        DisableThreadLibraryCalls(hinstDLL);
+        CreateThread(NULL, 0, MainThread, NULL, 0, &dwThreadId);
+    }
+    return TRUE; // succesful
+}