Add Hooks struct

danielkrupinski · Oct 17, 2023 · 9c343f2 · 9c343f2
1 parent daf6588
commit 9c343f2
Show file tree

Hide file tree

Showing 5 changed files with 32 additions and 11 deletions.
diff --git a/Source/Endpoints.h b/Source/Endpoints.h
@@ -30,15 +30,15 @@ inline int PeepEventsHook::SDL_PeepEvents(void* events, int numevents,
     hudFeatures().killfeedPreserver.run();
 
     GlobalContext::instance().panoramaGUI->run();
-    GlobalContext::instance().loopModeGameHook->update();
-    GlobalContext::instance().viewRenderHook->update();
+    GlobalContext::instance().hooks->loopModeGameHook.update();
+    GlobalContext::instance().hooks->viewRenderHook.update();
     visuals().scopeOverlayRemover.updatePanelVisibility();
 
     if (GlobalContext::instance().unloadFlag) {
         const auto originalPeepEvents = GlobalContext::instance().peepEventsHook.original;
         GlobalContext::instance().peepEventsHook.disable();
-        GlobalContext::instance().loopModeGameHook->forceUninstall();
-        GlobalContext::instance().viewRenderHook->forceUninstall();
+        GlobalContext::instance().hooks->loopModeGameHook.forceUninstall();
+        GlobalContext::instance().hooks->viewRenderHook.forceUninstall();
         GlobalContext::destroyInstance();
         return originalPeepEvents(events, numevents, action, minType, maxType);
     }
@@ -48,10 +48,10 @@ inline int PeepEventsHook::SDL_PeepEvents(void* events, int numevents,
 inline void* LoopModeGameHook::getWorldSession(cs2::CLoopModeGame* thisptr) noexcept
 {
     visuals().scopeOverlayRemover.getWorldSessionHook(RETURN_ADDRESS());
-    return GlobalContext::instance().loopModeGameHook->originalGetWorldSession(thisptr);
+    return GlobalContext::instance().hooks->loopModeGameHook.originalGetWorldSession(thisptr);
 }
 
 inline void ViewRenderHook::onRenderStart(cs2::CViewRender* thisptr) noexcept
 {
-    GlobalContext::instance().viewRenderHook->originalOnRenderStart(thisptr);
+    GlobalContext::instance().hooks->viewRenderHook.originalOnRenderStart(thisptr);
 }
diff --git a/Source/GlobalContext.h b/Source/GlobalContext.h
@@ -3,6 +3,7 @@
 #include <optional>
 
 #include "CS2/Constants/DllNames.h"
+#include "Hooks/Hooks.h"
 #include "Hooks/PeepEventsHook.h"
 #include "Hooks/ViewRenderHook.h"
 
@@ -23,8 +24,7 @@ struct GlobalContext {
     PatternFinder<PatternNotFoundLogger> clientPatternFinder{ DynamicLibrary{cs2::CLIENT_DLL}.getCodeSection().raw(), PatternNotFoundLogger{} };
     PatternFinder<PatternNotFoundLogger> panoramaPatternFinder{ DynamicLibrary{cs2::PANORAMA_DLL}.getCodeSection().raw(), PatternNotFoundLogger{} };
     std::optional<GameClassImplementations> gameClasses;
-    std::optional<LoopModeGameHook> loopModeGameHook;
-    std::optional<ViewRenderHook> viewRenderHook;
+    std::optional<Hooks> hooks;
     std::optional<Features> features;
     std::optional<PanoramaGUI> panoramaGUI;
 
@@ -52,9 +52,9 @@ struct GlobalContext {
             return;
 
         gameClasses.emplace();
-        loopModeGameHook.emplace(VmtLengthCalculator{ DynamicLibrary{cs2::CLIENT_DLL}.getCodeSection(), DynamicLibrary{cs2::CLIENT_DLL}.getVmtSection() });
-        viewRenderHook.emplace(VmtLengthCalculator{ DynamicLibrary{cs2::CLIENT_DLL}.getCodeSection(), DynamicLibrary{cs2::CLIENT_DLL}.getVmtSection() });
-        features.emplace(HudProvider{}, *loopModeGameHook);
+        const VmtLengthCalculator clientVmtLengthCalculator{ DynamicLibrary{cs2::CLIENT_DLL}.getCodeSection(), DynamicLibrary{cs2::CLIENT_DLL}.getVmtSection() };
+        hooks.emplace(clientVmtLengthCalculator);
+        features.emplace(HudProvider{}, hooks->loopModeGameHook, hooks->viewRenderHook);
         panoramaGUI.emplace(*features, unloadFlag);
 
         initializedFromGameThread = true;

diff --git a/Source/Hooks/Hooks.h b/Source/Hooks/Hooks.h
@@ -0,0 +1,17 @@
+#pragma once
+
+#include "LoopModeGameHook.h"
+#include "ViewRenderHook.h"
+
+#include <Vmt/VmtLengthCalculator.h>
+
+struct Hooks {
+    explicit Hooks(const VmtLengthCalculator& clientVmtLengthCalculator) noexcept
+        : loopModeGameHook{ clientVmtLengthCalculator }
+        , viewRenderHook{ clientVmtLengthCalculator }
+    {
+    }
+
+    LoopModeGameHook loopModeGameHook;
+    ViewRenderHook viewRenderHook;
+};
diff --git a/Source/Osiris.vcxproj b/Source/Osiris.vcxproj
@@ -68,6 +68,7 @@
     <ClInclude Include="Helpers\PatternNotFoundLogger.h" />
     <ClInclude Include="Helpers\PlantedC4Provider.h" />
     <ClInclude Include="Helpers\UnloadFlag.h" />
+    <ClInclude Include="Hooks\Hooks.h" />
     <ClInclude Include="Hooks\LoopModeGameHook.h" />
     <ClInclude Include="Hooks\PeepEventsHook.h" />
     <ClInclude Include="Hooks\ViewRenderHook.h" />

diff --git a/Source/Osiris.vcxproj.filters b/Source/Osiris.vcxproj.filters
@@ -531,6 +531,9 @@
     <ClInclude Include="Hooks\ViewRenderHook.h">
       <Filter>Hooks</Filter>
     </ClInclude>
+    <ClInclude Include="Hooks\Hooks.h">
+      <Filter>Hooks</Filter>
+    </ClInclude>
   </ItemGroup>
   <ItemGroup>
     <None Include="UI\Panorama\CreateGUI.js">