diff options
Diffstat (limited to 'NorthstarDLL/serverchathooks.cpp')
| -rw-r--r-- | NorthstarDLL/serverchathooks.cpp | 209 |
1 files changed, 209 insertions, 0 deletions
diff --git a/NorthstarDLL/serverchathooks.cpp b/NorthstarDLL/serverchathooks.cpp new file mode 100644 index 00000000..2f5be5c1 --- /dev/null +++ b/NorthstarDLL/serverchathooks.cpp @@ -0,0 +1,209 @@ +#include "pch.h" +#include "serverchathooks.h" +#include <rapidjson/document.h> +#include <rapidjson/stringbuffer.h> +#include <rapidjson/writer.h> +#include "serverauthentication.h" +#include "squirrel.h" +#include "miscserverscript.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_ServerSquirrelManager->setupfunc("CServerGameDLL_ProcessMessageStartThread") != SQRESULT_ERROR) + { + g_ServerSquirrelManager->pusharg((int)senderPlayerId - 1); + g_ServerSquirrelManager->pusharg(text); + g_ServerSquirrelManager->pusharg(isTeam); + g_ServerSquirrelManager->call(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); +} + +SQRESULT SQ_SendMessage(void* sqvm) +{ + int playerIndex = ServerSq_getinteger(sqvm, 1); + const char* text = ServerSq_getstring(sqvm, 2); + bool isTeam = ServerSq_getbool(sqvm, 3); + + ChatSendMessage(playerIndex, text, isTeam); + + return SQRESULT_NULL; +} + +SQRESULT SQ_BroadcastMessage(void* sqvm) +{ + int fromPlayerIndex = ServerSq_getinteger(sqvm, 1); + int toPlayerIndex = ServerSq_getinteger(sqvm, 2); + const char* text = ServerSq_getstring(sqvm, 3); + bool isTeam = ServerSq_getbool(sqvm, 4); + bool isDead = ServerSq_getbool(sqvm, 5); + int messageType = ServerSq_getinteger(sqvm, 6); + + if (messageType < 1) + { + ServerSq_pusherror(sqvm, fmt::format("Invalid message type {}", messageType).c_str()); + return SQRESULT_ERROR; + } + + ChatBroadcastMessage(fromPlayerIndex, toPlayerIndex, text, isTeam, isDead, (CustomMessageType)messageType); + + return SQRESULT_NULL; +} + +void InitialiseServerChatHooks_Engine(HMODULE baseAddress) +{ + g_pServerGameDLL = (CServerGameDLL*)((char*)baseAddress + 0x13F0AA98); +} + +void InitialiseServerChatHooks_Server(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, + reinterpret_cast<void*>(CServerGameDLL__OnReceivedSayTextMessage), + &CServerGameDLL__OnReceivedSayTextMessageHook, + reinterpret_cast<LPVOID*>(&CServerGameDLL__OnReceivedSayTextMessageHookBase)); + + // Chat sending functions + g_ServerSquirrelManager->AddFuncRegistration("void", "NSSendMessage", "int playerIndex, string text, bool isTeam", "", SQ_SendMessage); + g_ServerSquirrelManager->AddFuncRegistration( + "void", + "NSBroadcastMessage", + "int fromPlayerIndex, int toPlayerIndex, string text, bool isTeam, bool isDead, int messageType", + "", + SQ_BroadcastMessage); +} |