Add ClientConnect and Disconnect forwards
Let plugins know when new spectators join or leave the SourceTV server.
This commit is contained in:
Peace-Maker
+269
-5
@@ -41,28 +41,292 @@ SH_DECL_HOOK1_void(IDemoRecorder, StopRecording, SH_NOATTRIB, 0, CGameInfo const
|
||||
SH_DECL_HOOK0_void(IDemoRecorder, StopRecording, SH_NOATTRIB, 0)
|
||||
#endif
|
||||
|
||||
#if SOURCE_ENGINE == SE_CSGO
|
||||
SH_DECL_MANUALHOOK13(CHLTVServer_ConnectClient, 0, 0, 0, IClient *, netadr_s &, int, int, int, const char *, const char *, const char *, int, CUtlVector<INetMessage *> &, bool, CrossPlayPlatform_t, const unsigned char *, int);
|
||||
SH_DECL_HOOK1_void(IClient, Disconnect, SH_NOATTRIB, 0, const char *);
|
||||
#else
|
||||
SH_DECL_MANUALHOOK9(CHLTVServer_ConnectClient, 0, 0, 0, IClient *, netadr_t &, int, int, int, int, const char *, const char *, const char *, int);
|
||||
SH_DECL_HOOK0_void_vafmt(IClient, Disconnect, SH_NOATTRIB, 0);
|
||||
#endif
|
||||
|
||||
void CForwardManager::Init()
|
||||
{
|
||||
m_StartRecordingFwd = forwards->CreateForward("SourceTV_OnStartRecording", ET_Ignore, 3, NULL, Param_Cell, Param_String);
|
||||
int offset = -1;
|
||||
if (!g_pGameConf->GetOffset("CHLTVServer::ConnectClient", &offset) || offset == -1)
|
||||
{
|
||||
smutils->LogError(myself, "Failed to get CHLTVServer::ConnectClient offset.");
|
||||
}
|
||||
else
|
||||
{
|
||||
SH_MANUALHOOK_RECONFIGURE(CHLTVServer_ConnectClient, offset, 0, 0);
|
||||
m_bHasClientConnectOffset = true;
|
||||
}
|
||||
m_StartRecordingFwd = forwards->CreateForward("SourceTV_OnStartRecording", ET_Ignore, 2, NULL, Param_Cell, Param_String);
|
||||
m_StopRecordingFwd = forwards->CreateForward("SourceTV_OnStopRecording", ET_Ignore, 3, NULL, Param_Cell, Param_String, Param_Cell);
|
||||
m_SpectatorPreConnectFwd = forwards->CreateForward("SourceTV_OnSpectatorPreConnect", ET_LowEvent, 4, NULL, Param_String, Param_String, Param_String, Param_String);
|
||||
m_SpectatorConnectedFwd = forwards->CreateForward("SourceTV_OnSpectatorConnected", ET_Ignore, 1, NULL, Param_Cell);
|
||||
m_SpectatorDisconnectFwd = forwards->CreateForward("SourceTV_OnSpectatorDisconnect", ET_Ignore, 2, NULL, Param_Cell, Param_String);
|
||||
m_SpectatorDisconnectedFwd = forwards->CreateForward("SourceTV_OnSpectatorDisconnected", ET_Ignore, 2, NULL, Param_Cell, Param_String);
|
||||
}
|
||||
|
||||
void CForwardManager::Shutdown()
|
||||
{
|
||||
forwards->ReleaseForward(m_StartRecordingFwd);
|
||||
forwards->ReleaseForward(m_StopRecordingFwd);
|
||||
forwards->ReleaseForward(m_SpectatorPreConnectFwd);
|
||||
forwards->ReleaseForward(m_SpectatorConnectedFwd);
|
||||
forwards->ReleaseForward(m_SpectatorDisconnectFwd);
|
||||
forwards->ReleaseForward(m_SpectatorDisconnectedFwd);
|
||||
}
|
||||
|
||||
void CForwardManager::HookRecorder(IDemoRecorder *recorder)
|
||||
{
|
||||
SH_ADD_HOOK(IDemoRecorder, StartRecording, recorder, SH_MEMBER(this, &CForwardManager::OnStartRecording_Post), false);
|
||||
SH_ADD_HOOK(IDemoRecorder, StopRecording, recorder, SH_MEMBER(this, &CForwardManager::OnStopRecording_Post), false);
|
||||
SH_ADD_HOOK(IDemoRecorder, StartRecording, recorder, SH_MEMBER(this, &CForwardManager::OnStartRecording_Post), true);
|
||||
SH_ADD_HOOK(IDemoRecorder, StopRecording, recorder, SH_MEMBER(this, &CForwardManager::OnStopRecording_Post), true);
|
||||
}
|
||||
|
||||
void CForwardManager::UnhookRecorder(IDemoRecorder *recorder)
|
||||
{
|
||||
SH_REMOVE_HOOK(IDemoRecorder, StartRecording, recorder, SH_MEMBER(this, &CForwardManager::OnStartRecording_Post), false);
|
||||
SH_REMOVE_HOOK(IDemoRecorder, StopRecording, recorder, SH_MEMBER(this, &CForwardManager::OnStopRecording_Post), false);
|
||||
SH_REMOVE_HOOK(IDemoRecorder, StartRecording, recorder, SH_MEMBER(this, &CForwardManager::OnStartRecording_Post), true);
|
||||
SH_REMOVE_HOOK(IDemoRecorder, StopRecording, recorder, SH_MEMBER(this, &CForwardManager::OnStopRecording_Post), true);
|
||||
}
|
||||
|
||||
void CForwardManager::HookServer(IServer *server)
|
||||
{
|
||||
if (!m_bHasClientConnectOffset)
|
||||
return;
|
||||
|
||||
SH_ADD_MANUALHOOK(CHLTVServer_ConnectClient, server, SH_MEMBER(this, &CForwardManager::OnSpectatorConnect), false);
|
||||
SH_ADD_MANUALHOOK(CHLTVServer_ConnectClient, server, SH_MEMBER(this, &CForwardManager::OnSpectatorConnect_Post), true);
|
||||
|
||||
// Hook all already connected clients as well for late loading
|
||||
for (int i = 0; i < server->GetClientCount(); i++)
|
||||
{
|
||||
IClient *client = server->GetClient(i);
|
||||
if (client->IsConnected())
|
||||
HookClient(client);
|
||||
}
|
||||
}
|
||||
|
||||
void CForwardManager::UnhookServer(IServer *server)
|
||||
{
|
||||
if (!m_bHasClientConnectOffset)
|
||||
return;
|
||||
|
||||
SH_REMOVE_MANUALHOOK(CHLTVServer_ConnectClient, server, SH_MEMBER(this, &CForwardManager::OnSpectatorConnect), false);
|
||||
SH_REMOVE_MANUALHOOK(CHLTVServer_ConnectClient, server, SH_MEMBER(this, &CForwardManager::OnSpectatorConnect_Post), true);
|
||||
|
||||
// Unhook all connected clients as well.
|
||||
for (int i = 0; i < server->GetClientCount(); i++)
|
||||
{
|
||||
IClient *client = server->GetClient(i);
|
||||
if (client->IsConnected())
|
||||
UnhookClient(client);
|
||||
}
|
||||
}
|
||||
|
||||
void CForwardManager::HookClient(IClient *client)
|
||||
{
|
||||
SH_ADD_HOOK(IClient, Disconnect, client, SH_MEMBER(this, &CForwardManager::OnSpectatorDisconnect), false);
|
||||
}
|
||||
|
||||
void CForwardManager::UnhookClient(IClient *client)
|
||||
{
|
||||
SH_REMOVE_HOOK(IClient, Disconnect, client, SH_MEMBER(this, &CForwardManager::OnSpectatorDisconnect), false);
|
||||
}
|
||||
|
||||
#if SOURCE_ENGINE == SE_CSGO
|
||||
// CBaseServer::RejectConnection(ns_address const&, char const*, ...)
|
||||
static void RejectConnection(IServer *server, netadr_t &address, char *pchReason)
|
||||
{
|
||||
static ICallWrapper *pRejectConnection = nullptr;
|
||||
|
||||
if (!pRejectConnection)
|
||||
{
|
||||
int offset = -1;
|
||||
if (!g_pGameConf->GetOffset("CHLTVServer::RejectConnection", &offset) || offset == -1)
|
||||
{
|
||||
smutils->LogError(myself, "Failed to get CHLTVServer::RejectConnection offset.");
|
||||
return;
|
||||
}
|
||||
|
||||
PassInfo pass[4];
|
||||
pass[0].flags = PASSFLAG_BYVAL;
|
||||
pass[0].type = PassType_Basic;
|
||||
pass[0].size = sizeof(void *);
|
||||
pass[1].flags = PASSFLAG_BYVAL;
|
||||
pass[1].type = PassType_Basic;
|
||||
pass[1].size = sizeof(void *);
|
||||
pass[2].flags = PASSFLAG_BYVAL;
|
||||
pass[2].type = PassType_Basic;
|
||||
pass[2].size = sizeof(char *);
|
||||
pass[3].flags = PASSFLAG_BYVAL;
|
||||
pass[3].type = PassType_Basic;
|
||||
pass[3].size = sizeof(char *);
|
||||
|
||||
void **vtable = *(void ***)server;
|
||||
void *func = vtable[offset];
|
||||
|
||||
pRejectConnection = bintools->CreateCall(func, CallConv_Cdecl, NULL, pass, 4);
|
||||
}
|
||||
|
||||
static char fmt[] = "%s";
|
||||
|
||||
if (pRejectConnection)
|
||||
{
|
||||
unsigned char vstk[sizeof(void *) * 2 + sizeof(char *) * 2];
|
||||
unsigned char *vptr = vstk;
|
||||
|
||||
*(void **)vptr = (void *)server;
|
||||
vptr += sizeof(void *);
|
||||
*(void **)vptr = (void *)&address;
|
||||
vptr += sizeof(void *);
|
||||
*(char **)vptr = fmt;
|
||||
vptr += sizeof(char *);
|
||||
*(char **)vptr = pchReason;
|
||||
|
||||
pRejectConnection->Execute(vstk, NULL);
|
||||
}
|
||||
}
|
||||
#else
|
||||
static void RejectConnection(IServer *server, netadr_t &address, int iClientChallenge, char *pchReason)
|
||||
{
|
||||
static ICallWrapper *pRejectConnection = nullptr;
|
||||
|
||||
if (!pRejectConnection)
|
||||
{
|
||||
int offset = -1;
|
||||
if (!g_pGameConf->GetOffset("CHLTVServer::RejectConnection", &offset) || offset == -1)
|
||||
{
|
||||
smutils->LogError(myself, "Failed to get CHLTVServer::RejectConnection offset.");
|
||||
return;
|
||||
}
|
||||
|
||||
PassInfo pass[3];
|
||||
pass[0].flags = PASSFLAG_BYVAL;
|
||||
pass[0].type = PassType_Basic;
|
||||
pass[0].size = sizeof(netadr_t *);
|
||||
pass[1].flags = PASSFLAG_BYVAL;
|
||||
pass[1].type = PassType_Basic;
|
||||
pass[1].size = sizeof(int);
|
||||
pass[2].flags = PASSFLAG_BYVAL;
|
||||
pass[2].type = PassType_Basic;
|
||||
pass[2].size = sizeof(char *);
|
||||
|
||||
pRejectConnection = bintools->CreateVCall(offset, 0, 0, NULL, pass, 3);
|
||||
}
|
||||
|
||||
if (pRejectConnection)
|
||||
{
|
||||
unsigned char vstk[sizeof(void *) + sizeof(netadr_t *) + sizeof(int) + sizeof(char *)];
|
||||
unsigned char *vptr = vstk;
|
||||
|
||||
*(void **)vptr = (void *)server;
|
||||
vptr += sizeof(void *);
|
||||
*(netadr_t **)vptr = &address;
|
||||
vptr += sizeof(netadr_t *);
|
||||
*(int *)vptr = iClientChallenge;
|
||||
vptr += sizeof(int);
|
||||
*(char **)vptr = pchReason;
|
||||
|
||||
pRejectConnection->Execute(vstk, NULL);
|
||||
}
|
||||
}
|
||||
#endif
|
||||
|
||||
// Mimic Connect extension https://forums.alliedmods.net/showthread.php?t=162489
|
||||
// Thanks asherkin!
|
||||
char passwordBuffer[255];
|
||||
#if SOURCE_ENGINE == SE_CSGO
|
||||
// CHLTVServer::ConnectClient(ns_address const&, int, int, int, char const*, char const*, char const*, int, CUtlVector<CNetMessagePB<16, CCLCMsg_SplitPlayerConnect, 0, true> *, CUtlMemory<CNetMessagePB<16, CCLCMsg_SplitPlayerConnect, 0, true> *, int>> &, bool, CrossPlayPlatform_t, unsigned char const*, int)
|
||||
IClient *CForwardManager::OnSpectatorConnect(netadr_s & address, int nProtocol, int iChallenge, int nAuthProtocol, const char *pchName, const char *pchPassword, const char *pCookie, int cbCookie, CUtlVector<INetMessage *> &pSplitPlayerConnectVector, bool bUnknown, CrossPlayPlatform_t platform, const unsigned char *pUnknown, int iUnknown)
|
||||
#else
|
||||
IClient *CForwardManager::OnSpectatorConnect(netadr_t & address, int nProtocol, int iChallenge, int iClientChallenge, int nAuthProtocol, const char *pchName, const char *pchPassword, const char *pCookie, int cbCookie)
|
||||
#endif
|
||||
{
|
||||
if (!pCookie || cbCookie < sizeof(uint64))
|
||||
RETURN_META_VALUE(MRES_IGNORED, nullptr);
|
||||
|
||||
char ipString[16];
|
||||
V_snprintf(ipString, sizeof(ipString), "%u.%u.%u.%u", address.ip[0], address.ip[1], address.ip[2], address.ip[3]);
|
||||
V_strncpy(passwordBuffer, pchPassword, 255);
|
||||
|
||||
// SourceTV doesn't validate steamids?!
|
||||
|
||||
char rejectReason[255];
|
||||
|
||||
m_SpectatorPreConnectFwd->PushString(pchName);
|
||||
m_SpectatorPreConnectFwd->PushStringEx(passwordBuffer, 255, SM_PARAM_STRING_UTF8 | SM_PARAM_STRING_COPY, SM_PARAM_COPYBACK);
|
||||
m_SpectatorPreConnectFwd->PushString(ipString);
|
||||
m_SpectatorPreConnectFwd->PushStringEx(rejectReason, 255, SM_PARAM_STRING_UTF8 | SM_PARAM_STRING_COPY, SM_PARAM_COPYBACK);
|
||||
|
||||
cell_t retVal = 1;
|
||||
m_SpectatorPreConnectFwd->Execute(&retVal);
|
||||
|
||||
if (retVal == 0)
|
||||
{
|
||||
IServer *server = META_IFACEPTR(IServer);
|
||||
#if SOURCE_ENGINE == SE_CSGO
|
||||
RejectConnection(server, address, rejectReason);
|
||||
#else
|
||||
RejectConnection(server, address, iClientChallenge, rejectReason);
|
||||
#endif
|
||||
RETURN_META_VALUE(MRES_SUPERCEDE, nullptr);
|
||||
}
|
||||
|
||||
pchPassword = passwordBuffer;
|
||||
#if SOURCE_ENGINE == SE_CSGO
|
||||
RETURN_META_VALUE_MNEWPARAMS(MRES_IGNORED, nullptr, CHLTVServer_ConnectClient, (address, nProtocol, iChallenge, nAuthProtocol, pchName, pchPassword, pCookie, cbCookie, pSplitPlayerConnectVector, bUnknown, platform, pUnknown, iUnknown));
|
||||
#else
|
||||
RETURN_META_VALUE_MNEWPARAMS(MRES_IGNORED, nullptr, CHLTVServer_ConnectClient, (address, nProtocol, iChallenge, iClientChallenge, nAuthProtocol, pchName, pchPassword, pCookie, cbCookie));
|
||||
#endif
|
||||
}
|
||||
|
||||
#if SOURCE_ENGINE == SE_CSGO
|
||||
IClient *CForwardManager::OnSpectatorConnect_Post(netadr_s & address, int nProtocol, int iChallenge, int nAuthProtocol, const char *pchName, const char *pchPassword, const char *pCookie, int cbCookie, CUtlVector<INetMessage *> &pSplitPlayerConnectVector, bool bUnknown, CrossPlayPlatform_t platform, const unsigned char * pUnknown, int iUnknown)
|
||||
#else
|
||||
IClient *CForwardManager::OnSpectatorConnect_Post(netadr_t & address, int nProtocol, int iChallenge, int iClientChallenge, int nAuthProtocol, const char *pchName, const char *pchPassword, const char *pCookie, int cbCookie)
|
||||
#endif
|
||||
{
|
||||
IClient *client = META_RESULT_ORIG_RET(IClient *);
|
||||
if (!client)
|
||||
RETURN_META_VALUE(MRES_IGNORED, nullptr);
|
||||
|
||||
HookClient(client);
|
||||
|
||||
m_SpectatorConnectedFwd->PushCell(client->GetPlayerSlot()+1);
|
||||
m_SpectatorConnectedFwd->Execute();
|
||||
|
||||
RETURN_META_VALUE(MRES_IGNORED, nullptr);
|
||||
}
|
||||
|
||||
void CForwardManager::OnSpectatorDisconnect(const char *reason)
|
||||
{
|
||||
IClient *client = META_IFACEPTR(IClient);
|
||||
if (!client)
|
||||
RETURN_META(MRES_IGNORED);
|
||||
|
||||
UnhookClient(client);
|
||||
|
||||
char disconnectReason[255];
|
||||
V_strncpy(disconnectReason, reason, 255);
|
||||
int clientIndex = client->GetPlayerSlot() + 1;
|
||||
|
||||
m_SpectatorDisconnectFwd->PushCell(clientIndex);
|
||||
m_SpectatorDisconnectFwd->PushStringEx(disconnectReason, 255, SM_PARAM_STRING_UTF8 | SM_PARAM_STRING_COPY, SM_PARAM_COPYBACK);
|
||||
m_SpectatorDisconnectFwd->Execute();
|
||||
|
||||
#if SOURCE_ENGINE == SE_CSGO
|
||||
SH_CALL(client, &IClient::Disconnect)(disconnectReason);
|
||||
#else
|
||||
SH_CALL(client, &IClient::Disconnect)("%s", disconnectReason);
|
||||
#endif
|
||||
|
||||
m_SpectatorDisconnectedFwd->PushCell(clientIndex);
|
||||
m_SpectatorDisconnectedFwd->PushString(disconnectReason);
|
||||
m_SpectatorDisconnectedFwd->Execute();
|
||||
|
||||
RETURN_META(MRES_SUPERCEDE);
|
||||
}
|
||||
|
||||
void CForwardManager::OnStartRecording_Post(const char *filename, bool bContinuously)
|
||||
|
||||
Reference in New Issue
Block a user