initial commit

1 files changed, 34 insertions, 0 deletions

diff --git a/NorthstarDedicatedTest/hookutils.h b/NorthstarDedicatedTest/hookutils.h
new file mode 100644
index 00000000..dd3b15b1
--- /dev/null
+++ b/NorthstarDedicatedTest/hookutils.h
@@ -0,0 +1,34 @@
+#pragma once
+#include <vector>
+
+// Sets an area of memory as writeable until the TempReadWrite object goes out of scope
+class TempReadWrite
+{
+private:
+    DWORD m_origProtection;
+    void* m_ptr;
+
+public:
+    TempReadWrite(void* ptr);
+    ~TempReadWrite();
+};
+
+// Enables all hooks created with the HookEnabler object when it goes out of scope and handles hook errors
+class HookEnabler
+{
+private:
+    struct HookTarget
+    {
+        char* targetName;
+        LPVOID targetAddress;
+    };
+
+    std::vector<HookTarget*> m_hookTargets;
+
+public: 
+    void CreateHook(LPVOID ppTarget, LPVOID ppDetour, LPVOID* ppOriginal, const char* targetName = nullptr);
+    ~HookEnabler();
+};
+
+// macro to call HookEnabler::CreateHook with the hook's name
+#define ENABLER_CREATEHOOK(enabler, ppTarget, ppDetour, ppOriginal) enabler.CreateHook(ppTarget, ppDetour, ppOriginal, #ppDetour)
\ No newline at end of file