almost fully replaced hooking lib

1 files changed, 185 insertions, 212 deletions

diff --git a/NorthstarDLL/serverchathooks.cpp b/NorthstarDLL/serverchathooks.cpp
index 145d691e..b9246f83 100644
--- a/NorthstarDLL/serverchathooks.cpp
+++ b/NorthstarDLL/serverchathooks.cpp
@@ -1,213 +1,186 @@
-#include "pch.h"
-#include "serverchathooks.h"
-#include "serverauthentication.h"
-#include "squirrel.h"
-#include "miscserverscript.h"
-
-#include <rapidjson/document.h>
-#include <rapidjson/stringbuffer.h>
-#include <rapidjson/writer.h>
-
-
-class CServerGameDLL;
-class CBasePlayer;
-
-class CRecipientFilter
-{
-	char unknown[58];
-};
-
-CServerGameDLL* g_pServerGameDLL;
-
-typedef void(__fastcall* CServerGameDLL__OnReceivedSayTextMessageType)(
-	CServerGameDLL* self, unsigned int senderPlayerId, const char* text, int channelId);
-CServerGameDLL__OnReceivedSayTextMessageType CServerGameDLL__OnReceivedSayTextMessage;
-CServerGameDLL__OnReceivedSayTextMessageType CServerGameDLL__OnReceivedSayTextMessageHookBase;
-
-typedef CBasePlayer*(__fastcall* UTIL_PlayerByIndexType)(int playerIndex);
-UTIL_PlayerByIndexType UTIL_PlayerByIndex;
-
-typedef void(__fastcall* CRecipientFilter__ConstructType)(CRecipientFilter* self);
-CRecipientFilter__ConstructType CRecipientFilter__Construct;
-
-typedef void(__fastcall* CRecipientFilter__DestructType)(CRecipientFilter* self);
-CRecipientFilter__DestructType CRecipientFilter__Destruct;
-
-typedef void(__fastcall* CRecipientFilter__AddAllPlayersType)(CRecipientFilter* self);
-CRecipientFilter__AddAllPlayersType CRecipientFilter__AddAllPlayers;
-
-typedef void(__fastcall* CRecipientFilter__AddRecipientType)(CRecipientFilter* self, const CBasePlayer* player);
-CRecipientFilter__AddRecipientType CRecipientFilter__AddRecipient;
-
-typedef void(__fastcall* CRecipientFilter__MakeReliableType)(CRecipientFilter* self);
-CRecipientFilter__MakeReliableType CRecipientFilter__MakeReliable;
-
-typedef void(__fastcall* UserMessageBeginType)(CRecipientFilter* filter, const char* messagename);
-UserMessageBeginType UserMessageBegin;
-
-typedef void(__fastcall* MessageEndType)();
-MessageEndType MessageEnd;
-
-typedef void(__fastcall* MessageWriteByteType)(int iValue);
-MessageWriteByteType MessageWriteByte;
-
-typedef void(__fastcall* MessageWriteStringType)(const char* sz);
-MessageWriteStringType MessageWriteString;
-
-typedef void(__fastcall* MessageWriteBoolType)(bool bValue);
-MessageWriteBoolType MessageWriteBool;
-
-bool isSkippingHook = false;
-
-static void CServerGameDLL__OnReceivedSayTextMessageHook(CServerGameDLL* self, unsigned int senderPlayerId, const char* text, bool isTeam)
-{
-	// MiniHook doesn't allow calling the base function outside of anywhere but the hook function.
-	// To allow bypassing the hook, isSkippingHook can be set.
-	if (isSkippingHook)
-	{
-		isSkippingHook = false;
-		CServerGameDLL__OnReceivedSayTextMessageHookBase(self, senderPlayerId, text, isTeam);
-		return;
-	}
-
-	void* sender = GetPlayerByIndex(senderPlayerId - 1);
-
-	// check chat ratelimits
-	if (!g_ServerAuthenticationManager->CheckPlayerChatRatelimit(sender))
-	{
-		return;
-	}
-
-	if (g_pServerSquirrel->setupfunc("CServerGameDLL_ProcessMessageStartThread") != SQRESULT_ERROR)
-	{
-		g_pServerSquirrel->pushinteger(g_pServerSquirrel->sqvm2, (int)senderPlayerId - 1);
-		g_pServerSquirrel->pushstring(g_pServerSquirrel->sqvm2, text);
-		g_pServerSquirrel->pushbool(g_pServerSquirrel->sqvm2, isTeam);
-		g_pServerSquirrel->call(g_pServerSquirrel->sqvm2, 3);
-	}
-	else
-		CServerGameDLL__OnReceivedSayTextMessageHookBase(self, senderPlayerId, text, isTeam);
-}
-
-void ChatSendMessage(unsigned int playerIndex, const char* text, bool isteam)
-{
-	isSkippingHook = true;
-	CServerGameDLL__OnReceivedSayTextMessage(
-		g_pServerGameDLL,
-		// Ensure the first bit isn't set, since this indicates a custom message
-		(playerIndex + 1) & CUSTOM_MESSAGE_INDEX_MASK,
-		text,
-		isteam);
-}
-
-void ChatBroadcastMessage(int fromPlayerIndex, int toPlayerIndex, const char* text, bool isTeam, bool isDead, CustomMessageType messageType)
-{
-	CBasePlayer* toPlayer = NULL;
-	if (toPlayerIndex >= 0)
-	{
-		toPlayer = UTIL_PlayerByIndex(toPlayerIndex + 1);
-		if (toPlayer == NULL)
-			return;
-	}
-
-	// Build a new string where the first byte is the message type
-	char sendText[256];
-	sendText[0] = (char)messageType;
-	strncpy(sendText + 1, text, 255);
-	sendText[255] = 0;
-
-	// Anonymous custom messages use playerId=0, non-anonymous ones use a player ID with the first bit set
-	unsigned int fromPlayerId = fromPlayerIndex < 0 ? 0 : ((fromPlayerIndex + 1) | CUSTOM_MESSAGE_INDEX_BIT);
-
-	CRecipientFilter filter;
-	CRecipientFilter__Construct(&filter);
-	if (toPlayer == NULL)
-	{
-		CRecipientFilter__AddAllPlayers(&filter);
-	}
-	else
-	{
-		CRecipientFilter__AddRecipient(&filter, toPlayer);
-	}
-	CRecipientFilter__MakeReliable(&filter);
-
-	UserMessageBegin(&filter, "SayText");
-	MessageWriteByte(fromPlayerId);
-	MessageWriteString(sendText);
-	MessageWriteBool(isTeam);
-	MessageWriteBool(isDead);
-	MessageEnd();
-
-	CRecipientFilter__Destruct(&filter);
-}
-
-// void function NSSendMessage( int playerIndex, string text, bool isTeam )
-SQRESULT SQ_SendMessage(void* sqvm)
-{
-	int playerIndex = g_pServerSquirrel->getinteger(sqvm, 1);
-	const char* text = g_pServerSquirrel->getstring(sqvm, 2);
-	bool isTeam = g_pServerSquirrel->getbool(sqvm, 3);
-
-	ChatSendMessage(playerIndex, text, isTeam);
-
-	return SQRESULT_NULL;
-}
-
-// void function NSBroadcastMessage( int fromPlayerIndex, int toPlayerIndex, string text, bool isTeam, bool isDead, int messageType )
-SQRESULT SQ_BroadcastMessage(void* sqvm)
-{
-	int fromPlayerIndex = g_pServerSquirrel->getinteger(sqvm, 1);
-	int toPlayerIndex = g_pServerSquirrel->getinteger(sqvm, 2);
-	const char* text = g_pServerSquirrel->getstring(sqvm, 3);
-	bool isTeam = g_pServerSquirrel->getbool(sqvm, 4);
-	bool isDead = g_pServerSquirrel->getbool(sqvm, 5);
-	int messageType = g_pServerSquirrel->getinteger(sqvm, 6);
-
-	if (messageType < 1)
-	{
-		g_pServerSquirrel->raiseerror(sqvm, fmt::format("Invalid message type {}", messageType).c_str());
-		return SQRESULT_ERROR;
-	}
-
-	ChatBroadcastMessage(fromPlayerIndex, toPlayerIndex, text, isTeam, isDead, (CustomMessageType)messageType);
-
-	return SQRESULT_NULL;
-}
-
-ON_DLL_LOAD("engine.dll", EngineServerChatHooks, [](HMODULE baseAddress)
-{
-	g_pServerGameDLL = (CServerGameDLL*)((char*)baseAddress + 0x13F0AA98);
-})
-
-ON_DLL_LOAD_RELIESON("server.dll", ServerChatHooks, ServerSquirrel, [](HMODULE baseAddress)
-{
-	CServerGameDLL__OnReceivedSayTextMessage = (CServerGameDLL__OnReceivedSayTextMessageType)((char*)baseAddress + 0x1595C0);
-	UTIL_PlayerByIndex = (UTIL_PlayerByIndexType)((char*)baseAddress + 0x26AA10);
-	CRecipientFilter__Construct = (CRecipientFilter__ConstructType)((char*)baseAddress + 0x1E9440);
-	CRecipientFilter__Destruct = (CRecipientFilter__DestructType)((char*)baseAddress + 0x1E9700);
-	CRecipientFilter__AddAllPlayers = (CRecipientFilter__AddAllPlayersType)((char*)baseAddress + 0x1E9940);
-	CRecipientFilter__AddRecipient = (CRecipientFilter__AddRecipientType)((char*)baseAddress + 0x1E9b30);
-	CRecipientFilter__MakeReliable = (CRecipientFilter__MakeReliableType)((char*)baseAddress + 0x1EA4E0);
-
-	UserMessageBegin = (UserMessageBeginType)((char*)baseAddress + 0x15C520);
-	MessageEnd = (MessageEndType)((char*)baseAddress + 0x158880);
-	MessageWriteByte = (MessageWriteByteType)((char*)baseAddress + 0x158A90);
-	MessageWriteString = (MessageWriteStringType)((char*)baseAddress + 0x158D00);
-	MessageWriteBool = (MessageWriteBoolType)((char*)baseAddress + 0x158A00);
-
-	HookEnabler hook;
-	ENABLER_CREATEHOOK(
-		hook,
-		CServerGameDLL__OnReceivedSayTextMessage,
-		&CServerGameDLL__OnReceivedSayTextMessageHook,
-		reinterpret_cast<LPVOID*>(&CServerGameDLL__OnReceivedSayTextMessageHookBase));
-
-	// Chat sending functions
-	g_pServerSquirrel->AddFuncRegistration("void", "NSSendMessage", "int playerIndex, string text, bool isTeam", "", SQ_SendMessage);
-	g_pServerSquirrel->AddFuncRegistration(
-		"void",
-		"NSBroadcastMessage",
-		"int fromPlayerIndex, int toPlayerIndex, string text, bool isTeam, bool isDead, int messageType",
-		"",
-		SQ_BroadcastMessage);
+#include "pch.h"

+#include "serverchathooks.h"

+#include "serverauthentication.h"

+#include "squirrel.h"

+#include "miscserverscript.h"

+

+#include <rapidjson/document.h>

+#include <rapidjson/stringbuffer.h>

+#include <rapidjson/writer.h>

+

+AUTOHOOK_INIT()

+

+class CServerGameDLL;

+class CBasePlayer;

+

+class CRecipientFilter

+{

+	char unknown[58];

+};

+

+CServerGameDLL* g_pServerGameDLL;

+

+void(__fastcall* CServerGameDLL__OnReceivedSayTextMessage)(

+	CServerGameDLL* self, unsigned int senderPlayerId, const char* text, int channelId);

+

+CBasePlayer*(__fastcall* UTIL_PlayerByIndex)(int playerIndex);

+void(__fastcall* CRecipientFilter__Construct)(CRecipientFilter* self);

+void(__fastcall* CRecipientFilter__Destruct)(CRecipientFilter* self);

+void(__fastcall* CRecipientFilter__AddAllPlayers)(CRecipientFilter* self);

+void(__fastcall* CRecipientFilter__AddRecipient)(CRecipientFilter* self, const CBasePlayer* player);

+void(__fastcall* CRecipientFilter__MakeReliable)(CRecipientFilter* self);

+

+void(__fastcall* UserMessageBegin)(CRecipientFilter* filter, const char* messagename);

+void(__fastcall* MessageEnd)();

+void(__fastcall* MessageWriteByte)(int iValue);

+void(__fastcall* MessageWriteString)(const char* sz);

+void(__fastcall* MessageWriteBool)(bool bValue);

+

+bool bShouldCallSayTextHook = false;

+

+AUTOHOOK(_CServerGameDLL__OnReceivedSayTextMessageHook, server.dll + 0x1595C0,

+void,, (CServerGameDLL* self, unsigned int senderPlayerId, const char* text, bool isTeam))

+{

+	// MiniHook doesn't allow calling the base function outside of anywhere but the hook function.

+	// To allow bypassing the hook, isSkippingHook can be set.

+	if (bShouldCallSayTextHook)

+	{

+		bShouldCallSayTextHook = false;

+		_CServerGameDLL__OnReceivedSayTextMessageHook(self, senderPlayerId, text, isTeam);

+		return;

+	}

+

+	void* sender = GetPlayerByIndex(senderPlayerId - 1);

+

+	// check chat ratelimits

+	if (!g_ServerAuthenticationManager->CheckPlayerChatRatelimit(sender))

+		return;

+

+	if (g_pServerSquirrel->setupfunc("CServerGameDLL_ProcessMessageStartThread") != SQRESULT_ERROR)

+	{

+		g_pServerSquirrel->pushinteger(g_pServerSquirrel->sqvm2, (int)senderPlayerId - 1);

+		g_pServerSquirrel->pushstring(g_pServerSquirrel->sqvm2, text);

+		g_pServerSquirrel->pushbool(g_pServerSquirrel->sqvm2, isTeam);

+		g_pServerSquirrel->call(g_pServerSquirrel->sqvm2, 3);

+	}

+	else

+		_CServerGameDLL__OnReceivedSayTextMessageHook(self, senderPlayerId, text, isTeam);

+}

+

+void ChatSendMessage(unsigned int playerIndex, const char* text, bool isteam)

+{

+	bShouldCallSayTextHook = true;

+	CServerGameDLL__OnReceivedSayTextMessage(

+		g_pServerGameDLL,

+		// Ensure the first bit isn't set, since this indicates a custom message

+		(playerIndex + 1) & CUSTOM_MESSAGE_INDEX_MASK,

+		text,

+		isteam);

+}

+

+void ChatBroadcastMessage(int fromPlayerIndex, int toPlayerIndex, const char* text, bool isTeam, bool isDead, CustomMessageType messageType)

+{

+	CBasePlayer* toPlayer = NULL;

+	if (toPlayerIndex >= 0)

+	{

+		toPlayer = UTIL_PlayerByIndex(toPlayerIndex + 1);

+		if (toPlayer == NULL)

+			return;

+	}

+

+	// Build a new string where the first byte is the message type

+	char sendText[256];

+	sendText[0] = (char)messageType;

+	strncpy(sendText + 1, text, 255);

+	sendText[255] = 0;

+

+	// Anonymous custom messages use playerId=0, non-anonymous ones use a player ID with the first bit set

+	unsigned int fromPlayerId = fromPlayerIndex < 0 ? 0 : ((fromPlayerIndex + 1) | CUSTOM_MESSAGE_INDEX_BIT);

+

+	CRecipientFilter filter;

+	CRecipientFilter__Construct(&filter);

+	if (toPlayer == NULL)

+	{

+		CRecipientFilter__AddAllPlayers(&filter);

+	}

+	else

+	{

+		CRecipientFilter__AddRecipient(&filter, toPlayer);

+	}

+	CRecipientFilter__MakeReliable(&filter);

+

+	UserMessageBegin(&filter, "SayText");

+	MessageWriteByte(fromPlayerId);

+	MessageWriteString(sendText);

+	MessageWriteBool(isTeam);

+	MessageWriteBool(isDead);

+	MessageEnd();

+

+	CRecipientFilter__Destruct(&filter);

+}

+

+// void function NSSendMessage( int playerIndex, string text, bool isTeam )

+SQRESULT SQ_SendMessage(void* sqvm)

+{

+	int playerIndex = g_pServerSquirrel->getinteger(sqvm, 1);

+	const char* text = g_pServerSquirrel->getstring(sqvm, 2);

+	bool isTeam = g_pServerSquirrel->getbool(sqvm, 3);

+

+	ChatSendMessage(playerIndex, text, isTeam);

+

+	return SQRESULT_NULL;

+}

+

+// void function NSBroadcastMessage( int fromPlayerIndex, int toPlayerIndex, string text, bool isTeam, bool isDead, int messageType )

+SQRESULT SQ_BroadcastMessage(void* sqvm)

+{

+	int fromPlayerIndex = g_pServerSquirrel->getinteger(sqvm, 1);

+	int toPlayerIndex = g_pServerSquirrel->getinteger(sqvm, 2);

+	const char* text = g_pServerSquirrel->getstring(sqvm, 3);

+	bool isTeam = g_pServerSquirrel->getbool(sqvm, 4);

+	bool isDead = g_pServerSquirrel->getbool(sqvm, 5);

+	int messageType = g_pServerSquirrel->getinteger(sqvm, 6);

+

+	if (messageType < 1)

+	{

+		g_pServerSquirrel->raiseerror(sqvm, fmt::format("Invalid message type {}", messageType).c_str());

+		return SQRESULT_ERROR;

+	}

+

+	ChatBroadcastMessage(fromPlayerIndex, toPlayerIndex, text, isTeam, isDead, (CustomMessageType)messageType);

+

+	return SQRESULT_NULL;

+}

+

+ON_DLL_LOAD("engine.dll", EngineServerChatHooks, [](HMODULE baseAddress)

+{

+	g_pServerGameDLL = (CServerGameDLL*)((char*)baseAddress + 0x13F0AA98);

+})

+

+ON_DLL_LOAD_RELIESON("server.dll", ServerChatHooks, ServerSquirrel, [](HMODULE baseAddress)

+{

+	AUTOHOOK_DISPATCH_MODULE(server.dll)

+

+	CServerGameDLL__OnReceivedSayTextMessage = (void(__fastcall*)(CServerGameDLL*, unsigned int, const char*, int))((char*)baseAddress + 0x1595C0);

+	UTIL_PlayerByIndex = (CBasePlayer*(__fastcall*)(int))((char*)baseAddress + 0x26AA10);

+	CRecipientFilter__Construct = (void(__fastcall*)(CRecipientFilter*))((char*)baseAddress + 0x1E9440);

+	CRecipientFilter__Destruct = (void(__fastcall*)(CRecipientFilter*))((char*)baseAddress + 0x1E9700);

+	CRecipientFilter__AddAllPlayers = (void(__fastcall*)(CRecipientFilter*))((char*)baseAddress + 0x1E9940);

+	CRecipientFilter__AddRecipient = (void(__fastcall*)(CRecipientFilter*, const CBasePlayer*))((char*)baseAddress + 0x1E9b30);

+	CRecipientFilter__MakeReliable = (void(__fastcall*)(CRecipientFilter*))((char*)baseAddress + 0x1EA4E0);

+

+	UserMessageBegin = (void(__fastcall*)(CRecipientFilter*, const char*))((char*)baseAddress + 0x15C520);

+	MessageEnd = (void(__fastcall*)())((char*)baseAddress + 0x158880);

+	MessageWriteByte = (void(__fastcall*)(int))((char*)baseAddress + 0x158A90);

+	MessageWriteString = (void(__fastcall*)(const char*))((char*)baseAddress + 0x158D00);

+	MessageWriteBool = (void(__fastcall*)(bool))((char*)baseAddress + 0x158A00);

+

+	// Chat sending functions

+	g_pServerSquirrel->AddFuncRegistration("void", "NSSendMessage", "int playerIndex, string text, bool isTeam", "", SQ_SendMessage);

+	g_pServerSquirrel->AddFuncRegistration(

+		"void",

+		"NSBroadcastMessage",

+		"int fromPlayerIndex, int toPlayerIndex, string text, bool isTeam, bool isDead, int messageType",

+		"",

+		SQ_BroadcastMessage);

 })
\ No newline at end of file