rename project folder (:tf: commit log)

1 files changed, 213 insertions, 0 deletions

diff --git a/NorthstarDLL/serverchathooks.cpp b/NorthstarDLL/serverchathooks.cpp
new file mode 100644
index 00000000..145d691e
--- /dev/null
+++ b/NorthstarDLL/serverchathooks.cpp
@@ -0,0 +1,213 @@
+#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);
+})
\ No newline at end of file