UNLOZE/sm-ext-TransmitManager
9
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix client disconnect unhook

Browse Source
This commit is contained in:
Kyle
2020-08-12 00:32:18 +08:00
parent c925c50667
commit df7ece96de
1 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extension.cpp
+13 -10
View File
@@ -6,8 +6,8 @@
constexpr auto MAX_EDICT = 2048; constexpr auto MAX_EDICT = 2048;
TransmitManager g_TransmitManager; TransmitManager g_Transmit;
SMEXT_LINK(&g_TransmitManager); SMEXT_LINK(&g_Transmit);
IGameConfig* g_pGameConf = nullptr; IGameConfig* g_pGameConf = nullptr;
ISDKHooks* g_pSDKHooks = nullptr; ISDKHooks* g_pSDKHooks = nullptr;
@@ -22,7 +22,7 @@ struct HookingEntity
bRemoveFlags = false; bRemoveFlags = false;
iOwnerEntity = -1; iOwnerEntity = -1;
iEntityIndex = gamehelpers->EntityToBCompatRef(pEntity); iEntityIndex = gamehelpers->EntityToBCompatRef(pEntity);
SourceHookId = SH_ADD_MANUALHOOK(SetTransmit, pEntity, SH_MEMBER(&g_TransmitManager, &TransmitManager::Hook_SetTransmit), false); SourceHookId = SH_ADD_MANUALHOOK(SetTransmit, pEntity, SH_MEMBER(&g_Transmit, &TransmitManager::Hook_SetTransmit), false);
for (auto i = 1; i < SM_MAXPLAYERS; i++) for (auto i = 1; i < SM_MAXPLAYERS; i++)
{ {
@@ -142,18 +142,21 @@ void TransmitManager::Hook_SetTransmit(CCheckTransmitInfo* pInfo, bool bAlways)
bool TransmitManager::SDK_OnLoad(char *error, size_t maxlen, bool late) bool TransmitManager::SDK_OnLoad(char *error, size_t maxlen, bool late)
{ {
sharesys->AddDependency(myself, "sdkhooks.ext", true, true); sharesys->AddDependency(myself, "sdkhooks.ext", true, true);
if (!sharesys->RequestInterface(SMINTERFACE_SDKHOOKS_NAME, SMINTERFACE_SDKHOOKS_VERSION, myself, reinterpret_cast<SMInterface**>(&g_pSDKHooks))) { if (!sharesys->RequestInterface(SMINTERFACE_SDKHOOKS_NAME, SMINTERFACE_SDKHOOKS_VERSION, myself, reinterpret_cast<SMInterface**>(&g_pSDKHooks)))
{
smutils->Format(error, maxlen, "Cannot get SDKHooks Interface"); smutils->Format(error, maxlen, "Cannot get SDKHooks Interface");
return false; return false;
} }
if (!gameconfs->LoadGameConfigFile("sdkhooks.games", &g_pGameConf, error, maxlen)) { if (!gameconfs->LoadGameConfigFile("sdkhooks.games", &g_pGameConf, error, maxlen))
{
smutils->Format(error, maxlen, "Failed to load SDKHooks gamedata."); smutils->Format(error, maxlen, "Failed to load SDKHooks gamedata.");
return false; return false;
} }
int offset = -1; auto offset = -1;
if (!g_pGameConf->GetOffset("SetTransmit", &offset)) { if (!g_pGameConf->GetOffset("SetTransmit", &offset))
{
smutils->Format(error, maxlen, "Failed to load 'SetTransmit' offset."); smutils->Format(error, maxlen, "Failed to load 'SetTransmit' offset.");
return false; return false;
} }
@@ -214,7 +217,7 @@ void TransmitManager::OnEntityDestroyed(CBaseEntity* pEntity)
void TransmitManager::OnClientPutInServer(int client) void TransmitManager::OnClientPutInServer(int client)
{ {
auto *pPlayer = playerhelpers->GetGamePlayer(client); auto *pPlayer = playerhelpers->GetGamePlayer(client);
if (!pPlayer || pPlayer->IsFakeClient()) if (!pPlayer || pPlayer->IsSourceTV() || pPlayer->IsReplay())
{ {
return; return;
} }
@@ -234,7 +237,7 @@ void TransmitManager::OnClientPutInServer(int client)
void TransmitManager::OnClientDisconnecting(int client) void TransmitManager::OnClientDisconnecting(int client)
{ {
auto* pPlayer = playerhelpers->GetGamePlayer(client); auto* pPlayer = playerhelpers->GetGamePlayer(client);
if (!pPlayer || pPlayer->IsFakeClient()) if (!pPlayer || pPlayer->IsSourceTV() || pPlayer->IsReplay() || !pPlayer->IsInGame())
{ {
// not in-game = no hook // not in-game = no hook
return; return;
@@ -339,7 +342,7 @@ static cell_t Native_AddEntityHooks(IPluginContext* pContext, const cell_t* para
return pContext->ThrowNativeError("Entity %d is invalid.", params[1]); return pContext->ThrowNativeError("Entity %d is invalid.", params[1]);
} }
g_TransmitManager.HookEntity(pEntity); g_Transmit.HookEntity(pEntity);
return 0; return 0;
} }