add support for custom convars and concommands

1 files changed, 68 insertions, 0 deletions

diff --git a/NorthstarDedicatedTest/sourceinterface.cpp b/NorthstarDedicatedTest/sourceinterface.cpp
new file mode 100644
index 00000000..cedfad3c
--- /dev/null
+++ b/NorthstarDedicatedTest/sourceinterface.cpp
@@ -0,0 +1,68 @@
+#include "pch.h"
+#include "sourceinterface.h"
+#include "hooks.h"
+#include "hookutils.h"
+
+#include "sourceconsole.h"
+#include "context.h"
+#include "convar.h"
+#include <iostream>
+
+// really wanted to do a modular callback system here but honestly couldn't be bothered so hardcoding stuff for now: todo later
+
+CreateInterfaceFn clientCreateInterfaceOriginal;
+void* ClientCreateInterfaceHook(const char* pName, int* pReturnCode)
+{
+	void* ret = clientCreateInterfaceOriginal(pName, pReturnCode);
+
+	std::cout << "CreateInterface CLIENT " << pName << std::endl;
+	if (!strcmp(pName, "GameClientExports001"))
+		InitialiseConsoleOnInterfaceCreation();
+
+	return ret;
+}
+
+CreateInterfaceFn serverCreateInterfaceOriginal;
+void* ServerCreateInterfaceHook(const char* pName, int* pReturnCode)
+{
+	void* ret = serverCreateInterfaceOriginal(pName, pReturnCode);
+
+	std::cout << "CreateInterface SERVER " << pName << std::endl;
+
+	return ret;
+}
+
+CreateInterfaceFn engineCreateInterfaceOriginal;
+void* EngineCreateInterfaceHook(const char* pName, int* pReturnCode)
+{
+	void* ret = engineCreateInterfaceOriginal(pName, pReturnCode);
+
+	std::cout << "CreateInterface ENGINE " << pName << std::endl;
+
+	return ret;
+}
+
+void HookClientCreateInterface(HMODULE baseAddress)
+{
+	HookEnabler hook;
+	ENABLER_CREATEHOOK(hook, GetProcAddress(baseAddress, "CreateInterface"), &ClientCreateInterfaceHook, reinterpret_cast<LPVOID*>(&clientCreateInterfaceOriginal));
+}
+
+void HookServerCreateInterface(HMODULE baseAddress)
+{
+	HookEnabler hook;
+	ENABLER_CREATEHOOK(hook, GetProcAddress(baseAddress, "CreateInterface"), &ServerCreateInterfaceHook, reinterpret_cast<LPVOID*>(&serverCreateInterfaceOriginal));
+}
+
+void HookEngineCreateInterface(HMODULE baseAddress)
+{
+	HookEnabler hook;
+	ENABLER_CREATEHOOK(hook, GetProcAddress(baseAddress, "CreateInterface"), &EngineCreateInterfaceHook, reinterpret_cast<LPVOID*>(&engineCreateInterfaceOriginal));
+}
+
+void InitialiseInterfaceCreationHooks()
+{
+	AddDllLoadCallback("client.dll", HookClientCreateInterface);
+	AddDllLoadCallback("server.dll", HookServerCreateInterface);
+	AddDllLoadCallback("engine.dll", HookEngineCreateInterface);
+}
\ No newline at end of file