dvander@1945: /**
dvander@1945:  * vim: set ts=4 :
dvander@1945:  * =============================================================================
pred@1963:  * SourceMod Team Fortress 2 Extension
ds@1947:  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
dvander@1945:  * =============================================================================
dvander@1945:  *
dvander@1945:  * This program is free software; you can redistribute it and/or modify it under
dvander@1945:  * the terms of the GNU General Public License, version 3.0, as published by the
dvander@1945:  * Free Software Foundation.
dvander@1945:  * 
dvander@1945:  * This program is distributed in the hope that it will be useful, but WITHOUT
dvander@1945:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
dvander@1945:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
dvander@1945:  * details.
dvander@1945:  *
dvander@1945:  * You should have received a copy of the GNU General Public License along with
dvander@1945:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@1945:  *
dvander@1945:  * As a special exception, AlliedModders LLC gives you permission to link the
dvander@1945:  * code of this program (as well as its derivative works) to "Half-Life 2," the
dvander@1945:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
dvander@1945:  * by the Valve Corporation.  You must obey the GNU General Public License in
dvander@1945:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
dvander@1945:  * this exception to all derivative works.  AlliedModders LLC defines further
dvander@1945:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
dvander@1945:  * or <http://www.sourcemod.net/license.php>.
dvander@1945:  *
dvander@1945:  * Version: $Id$
dvander@1945:  */
dvander@1945: 
dvander@2783: #include <sourcemod_version.h>
dvander@1945: #include "extension.h"
pred@1963: #include "util.h"
dvander@1945: #include "RegNatives.h"
dvander@1945: #include "iplayerinfo.h"
pred@1963: #include "sm_trie_tpl.h"
pred@1979: #include "criticals.h"
skamonkey@3056: #include "holiday.h"
crimson@3151: #include "gameplayrules.h"
pred@2136: #include "CDetour/detours.h"
dvander@1945: 
dvander@1945: /**
dvander@1945:  * @file extension.cpp
dvander@1945:  * @brief Implement extension code here.
dvander@1945:  */
dvander@1945: 
dvander@1945: 
dvander@1945: TF2Tools g_TF2Tools;
dvander@1945: IGameConfig *g_pGameConf = NULL;
dvander@1945: 
dvander@1945: IBinTools *g_pBinTools = NULL;
dvander@1945: 
dvander@1945: SMEXT_LINK(&g_TF2Tools);
dvander@1945: 
pred@1979: ICvar *icvar = NULL;
pred@1979: 
pred@1963: CGlobalVars *gpGlobals = NULL;
pred@1963: 
pred@1963: sm_sendprop_info_t *playerSharedOffset;
dvander@1945: 
dvander@1945: extern sp_nativeinfo_t g_TFNatives[];
dvander@1945: 
pred@1963: int g_resourceEntity;
pred@1963: 
pred@1963: SH_DECL_HOOK3_void(IServerGameDLL, ServerActivate, SH_NOATTRIB, 0, edict_t *, int , int);
pred@1963: 
dvander@1945: bool TF2Tools::SDK_OnLoad(char *error, size_t maxlength, bool late)
dvander@1945: {
pred@2069: 	if (strcmp(g_pSM->GetGameFolderName(), "tf") != 0)
pred@2069: 	{
pred@2123: 		UTIL_Format(error, maxlength, "Cannot Load TF2 Extension on mods other than TF2");
pred@2069: 		return false;
pred@2069: 	}
pred@2069: 
pred@1963: 	ServerClass *sc = UTIL_FindServerClass("CTFPlayer");
pred@1963: 	if (sc == NULL)
pred@1963: 	{
pred@2123: 		UTIL_Format(error, maxlength, "Could not find CTFPlayer server class");
pred@1963: 		return false;
pred@1963: 	}
pred@1963: 
pred@1963: 	playerSharedOffset = new sm_sendprop_info_t;
pred@1963: 
pred@1963: 	if (!UTIL_FindDataTable(sc->m_pTable, "DT_TFPlayerShared", playerSharedOffset, 0))
pred@1963: 	{
pred@2123: 		UTIL_Format(error, maxlength, "Could not find DT_TFPlayerShared data table");
pred@1963: 		return false;
pred@1963: 	}
pred@1963: 
dvander@1945: 	sharesys->AddDependency(myself, "bintools.ext", true, true);
dvander@1945: 
pred@2123: 	char conf_error[255] = "";
dvander@1945: 	if (!gameconfs->LoadGameConfigFile("sm-tf2.games", &g_pGameConf, conf_error, sizeof(conf_error)))
dvander@1945: 	{
pred@2600: 		if (conf_error[0])
dvander@1945: 		{
pred@2123: 			UTIL_Format(error, maxlength, "Could not read sm-tf2.games.txt: %s", conf_error);
dvander@1945: 		}
dvander@1945: 		return false;
dvander@1945: 	}
dvander@1945: 
pred@2136: 	CDetourManager::Init(g_pSM->GetScriptingEngine(), g_pGameConf);
pred@2136: 
dvander@1945: 	sharesys->AddNatives(myself, g_TFNatives);
dvander@1945: 	sharesys->RegisterLibrary(myself, "tf2");
dvander@1945: 
pred@2619: 	plsys->AddPluginsListener(this);
pred@2619: 
ds@1952: 	playerhelpers->RegisterCommandTargetProcessor(this);
dvander@1945: 
pred@2136: 	g_critForward = forwards->CreateForward("TF2_CalcIsAttackCritical", ET_Hook, 4, NULL, Param_Cell, Param_Cell, Param_String, Param_CellByRef);
skamonkey@3056: 	g_getHolidayForward = forwards->CreateForward("TF2_OnGetHoliday", ET_Event, 1, NULL, Param_CellByRef);
crimson@3151: 	g_waitingPlayersStartForward = forwards->CreateForward("TF2_OnWaitingForPlayersStart", ET_Ignore, 0, NULL);
crimson@3151: 	g_waitingPlayersEndForward = forwards->CreateForward("TF2_OnWaitingForPlayersEnd", ET_Ignore, 0, NULL);
pred@1979: 
pred@1979: 	g_pCVar = icvar;
dvander@1945: 
skamonkey@3056: 	m_CritDetoursEnabled = false;
skamonkey@3056: 	m_GetHolidayDetourEnabled = false;
crimson@3151: 	m_RulesDetoursEnabled = false;
pred@2619: 
dvander@1945: 	return true;
dvander@1945: }
dvander@1945: 
dvander@2790: const char *TF2Tools::GetExtensionVerString()
dvander@2783: {
dvander@2783: 	return SM_FULL_VERSION;
dvander@2783: }
dvander@2783: 
dvander@2790: const char *TF2Tools::GetExtensionDateString()
dvander@2783: {
dvander@2783: 	return SM_BUILD_TIMESTAMP;
dvander@2783: }
dvander@2783: 
dvander@1945: bool TF2Tools::SDK_OnMetamodLoad(ISmmAPI *ismm, char *error, size_t maxlen, bool late)
dvander@1945: {
dvander@1945: 	GET_V_IFACE_CURRENT(GetEngineFactory, engine, IVEngineServer, INTERFACEVERSION_VENGINESERVER);
dvander@1945: 
pred@1963: 	gpGlobals = ismm->GetCGlobals();
pred@1963: 
pred@1963: 	SH_ADD_HOOK(IServerGameDLL, ServerActivate, gamedll, SH_STATIC(OnServerActivate), true);
pred@1963: 
pred@1979: 	GET_V_IFACE_CURRENT(GetEngineFactory, icvar, ICvar, CVAR_INTERFACE_VERSION);
pred@1979: 
pred@1979: 	GET_V_IFACE_CURRENT(GetServerFactory, gameents, IServerGameEnts, INTERFACEVERSION_SERVERGAMEENTS);
pred@1979: 
pred@2075: 	GET_V_IFACE_CURRENT(GetEngineFactory, m_GameEventManager, IGameEventManager2, INTERFACEVERSION_GAMEEVENTSMANAGER2);
pred@2075: 	m_GameEventManager->AddListener(this, "teamplay_restart_round", true);
pred@2075: 
dvander@1945: 	return true;
dvander@1945: }
dvander@1945: 
dvander@1945: void TF2Tools::SDK_OnUnload()
dvander@1945: {
pred@1963: 	SH_REMOVE_HOOK(IServerGameDLL, ServerActivate, gamedll, SH_STATIC(OnServerActivate), true);
pred@1963: 
dvander@1945: 	g_RegNatives.UnregisterAll();
dvander@1945: 	gameconfs->CloseGameConfigFile(g_pGameConf);
pred@1963: 	playerhelpers->UnregisterCommandTargetProcessor(this);
pred@2136: 
pred@2619: 	plsys->RemovePluginsListener(this);
pred@2136: 
pred@2619: 	forwards->ReleaseForward(g_critForward);
skamonkey@3056: 	forwards->ReleaseForward(g_getHolidayForward);
crimson@3151: 	forwards->ReleaseForward(g_waitingPlayersStartForward);
crimson@3151: 	forwards->ReleaseForward(g_waitingPlayersEndForward);
dvander@1945: }
dvander@1945: 
dvander@1945: void TF2Tools::SDK_OnAllLoaded()
dvander@1945: {
dvander@1945: 	SM_GET_LATE_IFACE(BINTOOLS, g_pBinTools);
ds@1952: }
ds@1952: 
pred@1979: bool TF2Tools::RegisterConCommandBase(ConCommandBase *pVar)
pred@1979: {
pred@1979: 	return g_SMAPI->RegisterConCommandBase(g_PLAPI, pVar);
pred@1979: }
pred@1979: 
pred@2075: void TF2Tools::FireGameEvent( IGameEvent *event )
pred@2075: {
pred@2075: 	timersys->NotifyOfGameStart();
pred@2075: 	timersys->MapTimeLeftChanged();
dvander@1945: }
dvander@1945: 
dvander@1945: bool TF2Tools::QueryRunning(char *error, size_t maxlength)
dvander@1945: {
dvander@1945: 	SM_CHECK_IFACE(BINTOOLS, g_pBinTools);
dvander@1945: 
dvander@1945: 	return true;
dvander@1945: }
dvander@1945: 
dvander@1945: bool TF2Tools::QueryInterfaceDrop(SMInterface *pInterface)
dvander@1945: {
dvander@1945: 	if (pInterface == g_pBinTools)
dvander@1945: 	{
dvander@1945: 		return false;
dvander@1945: 	}
dvander@1945: 
dvander@1945: 	return IExtensionInterface::QueryInterfaceDrop(pInterface);
dvander@1945: }
dvander@1945: 
dvander@1945: void TF2Tools::NotifyInterfaceDrop(SMInterface *pInterface)
dvander@1945: {
dvander@1945: 	g_RegNatives.UnregisterAll();
dvander@1945: }
dvander@1945: 
pred@1963: void OnServerActivate(edict_t *pEdictList, int edictCount, int clientMax)
dvander@1945: {
pred@1963: 	g_resourceEntity = FindResourceEntity();
pred@1963: }
dvander@1945: 
dvander@1945: bool TF2Tools::ProcessCommandTarget(cmd_target_info_t *info)
dvander@1945: {
dvander@1945: 	int max_clients;
dvander@1945: 	IPlayerInfo *pInfo;
dvander@1945: 	unsigned int team_index = 0;
dvander@1945: 	IGamePlayer *pPlayer, *pAdmin;
dvander@1945: 
dvander@1945: 	if ((info->flags & COMMAND_FILTER_NO_MULTI) == COMMAND_FILTER_NO_MULTI)
dvander@1945: 	{
dvander@1945: 		return false;
dvander@1945: 	}
dvander@1945: 
dvander@1945: 	if (info->admin)
dvander@1945: 	{
dvander@1945: 		if ((pAdmin = playerhelpers->GetGamePlayer(info->admin)) == NULL)
dvander@1945: 		{
dvander@1945: 			return false;
dvander@1945: 		}
dvander@1945: 		if (!pAdmin->IsInGame())
dvander@1945: 		{
dvander@1945: 			return false;
dvander@1945: 		}
dvander@1945: 	}
dvander@1945: 	else
dvander@1945: 	{
dvander@1945: 		pAdmin = NULL;
dvander@1945: 	}
dvander@1945: 
dvander@1945: 	if (strcmp(info->pattern, "@red") == 0 )
dvander@1945: 	{
dvander@1945: 		team_index = 2;
dvander@1945: 	}
dvander@1945: 	else if (strcmp(info->pattern, "@blue") == 0)
dvander@1945: 	{
dvander@1945: 		team_index = 3;
dvander@1945: 	}
dvander@1945: 	else
dvander@1945: 	{
dvander@1945: 		return false;
dvander@1945: 	}
dvander@1945: 
dvander@1945: 	info->num_targets = 0;
dvander@1945: 
dvander@1945: 	max_clients = playerhelpers->GetMaxClients();
dvander@1945: 	for (int i = 1; 
dvander@1945: 		 i <= max_clients && (cell_t)info->num_targets < info->max_targets; 
dvander@1945: 		 i++)
dvander@1945: 	{
dvander@1945: 		if ((pPlayer = playerhelpers->GetGamePlayer(i)) == NULL)
dvander@1945: 		{
dvander@1945: 			continue;
dvander@1945: 		}
dvander@1945: 		if (!pPlayer->IsInGame())
dvander@1945: 		{
dvander@1945: 			continue;
dvander@1945: 		}
dvander@1945: 		if ((pInfo = pPlayer->GetPlayerInfo()) == NULL)
dvander@1945: 		{
dvander@1945: 			continue;
dvander@1945: 		}
dvander@1945: 		if (pInfo->GetTeamIndex() != (int)team_index)
dvander@1945: 		{
dvander@1945: 			continue;
dvander@1945: 		}
dvander@1945: 		if (playerhelpers->FilterCommandTarget(pAdmin, pPlayer, info->flags) 
dvander@1945: 			!= COMMAND_TARGET_VALID)
dvander@1945: 		{
dvander@1945: 			continue;
dvander@1945: 		}
dvander@1945: 		info->targets[info->num_targets] = i;
dvander@1945: 		info->num_targets++;
dvander@1945: 	}
dvander@1945: 
dvander@1945: 	if (info->num_targets == 0)
dvander@1945: 	{
dvander@1945: 		info->reason = COMMAND_TARGET_EMPTY_FILTER;
dvander@1945: 	}
dvander@1945: 	else
dvander@1945: 	{
dvander@1945: 		info->reason = COMMAND_TARGET_VALID;
dvander@1945: 	}
dvander@1945: 
dvander@1945: 	info->target_name_style = COMMAND_TARGETNAME_RAW;
dvander@1945: 	if (team_index == 2)
dvander@1945: 	{
dvander@1945: 		UTIL_Format(info->target_name, info->target_name_maxlength, "Red Team");
dvander@1945: 	}
dvander@1945: 	else if (team_index == 3)
dvander@1945: 	{
dvander@1945: 		UTIL_Format(info->target_name, info->target_name_maxlength, "Blue Team");
dvander@1945: 	}
dvander@1945: 
dvander@1945: 	return true;
dvander@1945: }
dvander@1945: 
pred@2619: void TF2Tools::OnPluginLoaded(IPlugin *plugin)
pred@2619: {
skamonkey@3056: 	if (!m_CritDetoursEnabled && g_critForward->GetFunctionCount())
pred@2619: 	{
skamonkey@3056: 		InitialiseCritDetours();
skamonkey@3056: 		m_CritDetoursEnabled = true;
skamonkey@3056: 	}
skamonkey@3056: 	if (!m_GetHolidayDetourEnabled && g_getHolidayForward->GetFunctionCount())
skamonkey@3056: 	{
skamonkey@3056: 		InitialiseGetHolidayDetour();
skamonkey@3056: 		m_GetHolidayDetourEnabled = true;
pred@2619: 	}
crimson@3151: 	if (!m_RulesDetoursEnabled)
crimson@3151: 	{
crimson@3151: 		if(g_waitingPlayersStartForward->GetFunctionCount() || g_waitingPlayersEndForward->GetFunctionCount())
crimson@3151: 		{
crimson@3151: 			InitialiseRulesDetours();
crimson@3151: 			m_RulesDetoursEnabled = true;
crimson@3151: 		}
crimson@3151: 	}
pred@2619: }
pred@2619: 
pred@2619: void TF2Tools::OnPluginUnloaded(IPlugin *plugin)
pred@2619: {
skamonkey@3056: 	if (m_CritDetoursEnabled && !g_critForward->GetFunctionCount())
pred@2619: 	{
skamonkey@3056: 		RemoveCritDetours();
skamonkey@3056: 		m_CritDetoursEnabled = false;
skamonkey@3056: 	}
skamonkey@3056: 	if (m_GetHolidayDetourEnabled && !g_getHolidayForward->GetFunctionCount())
skamonkey@3056: 	{
skamonkey@3056: 		RemoveGetHolidayDetour();
skamonkey@3056: 		m_GetHolidayDetourEnabled = false;
pred@2619: 	}
crimson@3151: 	if (m_RulesDetoursEnabled)
crimson@3151: 	{
crimson@3151: 		if(!g_waitingPlayersStartForward->GetFunctionCount() || !g_waitingPlayersEndForward->GetFunctionCount())
crimson@3151: 		{
crimson@3151: 			RemoveRulesDetours();
crimson@3151: 			m_RulesDetoursEnabled = false;
crimson@3151: 		}
crimson@3151: 	}
pred@2619: }
pred@1963: int FindResourceEntity()
pred@1963: {
pred@1963: 	return FindEntityByNetClass(-1, "CTFPlayerResource");
pred@1963: }
pred@1963: 
pred@1963: 
pred@1963: int FindEntityByNetClass(int start, const char *classname)
pred@1963: {
pred@1963: 	edict_t *current;
pred@1963: 
pred@1963: 	for (int i = ((start != -1) ? start : 0); i < gpGlobals->maxEntities; i++)
pred@1963: 	{
pred@1963: 		current = engine->PEntityOfEntIndex(i);
pred@1963: 		if (current == NULL)
pred@1963: 		{
pred@1963: 			continue;
pred@1963: 		}
pred@1963: 
pred@1963: 		IServerNetworkable *network = current->GetNetworkable();
pred@1963: 
pred@1963: 		if (network == NULL)
pred@1963: 		{
pred@1963: 			continue;
pred@1963: 		}
pred@1963: 
pred@1963: 		ServerClass *sClass = network->GetServerClass();
pred@1963: 		const char *name = sClass->GetName();
pred@1963: 		
pred@1963: 
pred@1963: 		if (strcmp(name, classname) == 0)
pred@1963: 		{
pred@1963: 			return i;
pred@1963: 		}
pred@1963: 	}
pred@1963: 
pred@1963: 	return -1;
pred@1963: }
pred@1963: 
pred@1963: TFClassType ClassnameToType(const char *classname)
pred@1963: {
pred@1963: 	static KTrie<TFClassType> trie;
pred@1963: 	static bool filled = false;
pred@1963: 	
pred@1963: 	if (!filled)
pred@1963: 	{
pred@1963: 		trie.insert("scout", TFClass_Scout);
pred@1963: 		trie.insert("sniper", TFClass_Sniper);
pred@1963: 		trie.insert("soldier", TFClass_Soldier);
pred@1963: 		trie.insert("demoman", TFClass_DemoMan);
pred@1963: 		trie.insert("demo", TFClass_DemoMan);
pred@1963: 		trie.insert("medic", TFClass_Medic);
pred@1963: 		trie.insert("heavy", TFClass_Heavy);
pred@1963: 		trie.insert("hwg", TFClass_Heavy);
pred@1963: 		trie.insert("pyro", TFClass_Pyro);
pred@1963: 		trie.insert("spy", TFClass_Spy);
pred@1963: 		trie.insert("engineer", TFClass_Engineer);
pred@1963: 		trie.insert("engy", TFClass_Engineer);
pred@1963: 
pred@1963: 		filled = true;	
pred@1963: 	}
pred@1963: 	
pred@1963: 	TFClassType *value;
pred@1963: 
pred@1963: 	if (!(value = trie.retrieve(classname)))
pred@1963: 	{
pred@1963: 		return TFClass_Unknown;	
pred@1963: 	}
pred@1963: 
pred@1963: 	return *value;
pred@1963: }
pred@1963: 
pred@1963: 
pred@1963: 
pred@1963: /**
pred@1963:  * A picture of a blue crab given to me as a gift and stored here for safe keeping
pred@1963:  *
pred@1963:  * http://www.democracycellproject.net/blog/archives/Clown%20car.jpg
pred@1963:  */
pred@1963: