r5sdk/r5dev/common/netmessages.cpp

61 lines
1.8 KiB
C++

//===============================================================================//
//
// Purpose:
//
// $NoKeywords: $
//
//===============================================================================//
// netmessages.cpp: implementation of the CNetMessage types.
//
///////////////////////////////////////////////////////////////////////////////////
#include "core/stdafx.h"
#include "engine/net.h"
#include "common/netmessages.h"
bool SVC_Print::ProcessImpl()
{
if (this->m_szText)
{
DevMsg(eDLL_T::SERVER, "%s", this->m_szText);
}
return true; // Original just return true also.
}
bool SVC_UserMessage::ProcessImpl()
{
bf_read buf = m_DataIn;
int type = buf.ReadByte();
if (type == HUD_PRINTCONSOLE ||
type == HUD_PRINTCENTER)
{
char text[MAX_USER_MSG_DATA];
buf.ReadString(text, sizeof(text));
if (strnlen_s(text, sizeof(text)) >= NET_MIN_MESSAGE)
{
DevMsg(eDLL_T::SERVER, "%s", text);
}
}
return SVC_UserMessage_Process(this); // Need to return original.
}
void V_NetMessages::Attach() const
{
#if !defined(DEDICATED)
auto SVCPrint = &SVC_Print::ProcessImpl;
auto SVCUserMessage = &SVC_UserMessage::ProcessImpl;
CMemory::HookVirtualMethod((uintptr_t)g_pSVC_Print_VFTable, (LPVOID&)SVCPrint, 3, (LPVOID*)&SVC_Print_Process);
CMemory::HookVirtualMethod((uintptr_t)g_pSVC_UserMessage_VFTable, (LPVOID&)SVCUserMessage, 3, (LPVOID*)&SVC_UserMessage_Process);
#endif // DEDICATED
}
void V_NetMessages::Detach() const
{
#if !defined(DEDICATED)
void* hkRestore = nullptr;
CMemory::HookVirtualMethod((uintptr_t)g_pSVC_Print_VFTable, (LPVOID)SVC_Print_Process, 3, (LPVOID*)&hkRestore);
CMemory::HookVirtualMethod((uintptr_t)g_pSVC_UserMessage_VFTable, (LPVOID)SVC_UserMessage_Process, 3, (LPVOID*)&hkRestore);
#endif // DEDICATED
}