- Add SharedData::Discard method

CorrM · Feb 23, 2023 · b38e0d1 · b38e0d1
1 parent 76538f3
commit b38e0d1
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 1 deletion.
diff --git a/examples/UnrealInternalExample/dllmain.cpp b/examples/UnrealInternalExample/dllmain.cpp
@@ -4,6 +4,8 @@
 
 #include "CleanCheat.h"
 
+#define GAME_VIEW_PORT_INDEX 0x02  // Maybe you need to change that number
+
 typedef void (__thiscall* ProcessEventType)(CG::UObject*, CG::UFunction*, void*);
 typedef void (__thiscall* PostRenderType)(CG::UGameViewportClient*, CG::UCanvas*);
 
@@ -64,6 +66,7 @@ void __stdcall PostRenderHook(CG::UGameViewportClient* gameViewportClient, CG::U
             Once = true;
 
             CleanCheat::Hook->Detour(reinterpret_cast<void**>(&OProcessEvent), reinterpret_cast<void*>(&ProcessEventHook));
+            LOG("Hook 'ProcessEvent' ... Success");
         }
     }
     catch (...)
@@ -134,7 +137,7 @@ void MainEntryPoint(HMODULE hModule)
     std::vector<CG::UGameViewportClient*> gameViewportClients = CG::UObject::FindObjects<CG::UGameViewportClient>();
     LOG("GameViewportClientCount: %d", static_cast<int>(gameViewportClients.size()));
 
-    CG::UGameViewportClient*& gameViewportClient = gameViewportClients[1]; // Maybe you need to change that number
+    CG::UGameViewportClient*& gameViewportClient = gameViewportClients[GAME_VIEW_PORT_INDEX];
     void** gameViewportClientVmt = *reinterpret_cast<void***>(gameViewportClient);
     LOG("PostRender     : 0x%llx", reinterpret_cast<uintptr_t>(gameViewportClientVmt[POST_RENDER_INDEX]));
 

diff --git a/src/CleanCheat/CleanCheatManager.h b/src/CleanCheat/CleanCheatManager.h
@@ -136,6 +136,9 @@ class CleanCheat final
         // Un hook all functions
         Hook->UnSwapAll();
         Hook->UnDetourAll();
+
+        // Shared data
+        SharedData->Discard();
 
         // Free memory
         DELETE_HEAP(Memory);

diff --git a/src/CleanCheat/SharedDataBase.h b/src/CleanCheat/SharedDataBase.h
@@ -6,10 +6,24 @@ ABSTRACT class SharedDataBase
 {
 public:
     virtual ~SharedDataBase() = default;
+
+protected:
+    /// <summary>
+    /// Called by Discard method
+    /// </summary>
+    virtual void OnDestroy() { }
 
 public:
     /// <summary>
     /// Called every tick
     /// </summary>
     virtual void Tick(TInit* data) = 0;
+
+    /// <summary>
+    /// Discard runner task
+    /// </summary>
+    void Discard()
+    {
+        OnDestroy();
+    }
 };