dvander@592: /**
ds@653:  * vim: set ts=4 :
ds@1330:  * =============================================================================
dvander@1231:  * SourceMod
dvander@1231:  * Copyright (C) 2004-2007 AlliedModders LLC.  All rights reserved.
ds@1330:  * =============================================================================
dvander@592:  *
ds@1330:  * This program is free software; you can redistribute it and/or modify it under
ds@1330:  * the terms of the GNU General Public License, version 3.0, as published by the
ds@1330:  * Free Software Foundation.
dvander@1231:  * 
ds@1330:  * This program is distributed in the hope that it will be useful, but WITHOUT
ds@1330:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
ds@1330:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
ds@1330:  * details.
dvander@1231:  *
ds@1330:  * You should have received a copy of the GNU General Public License along with
ds@1330:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@1231:  *
ds@1330:  * As a special exception, AlliedModders LLC gives you permission to link the
ds@1330:  * code of this program (as well as its derivative works) to "Half-Life 2," the
ds@1330:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
ds@1330:  * by the Valve Corporation.  You must obey the GNU General Public License in
ds@1330:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
ds@1330:  * this exception to all derivative works.  AlliedModders LLC defines further
ds@1330:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
ds@1330:  * or <http://www.sourcemod.net/license.php>.
dvander@592:  *
dvander@592:  * Version: $Id$
dvander@592:  */
dvander@592: 
dvander@592: #include "HalfLife2.h"
ds@696: #include "sourcemod.h"
dvander@592: #include "sourcemm_api.h"
ds@873: #include "UserMessages.h"
dvander@1338: #include "PlayerManager.h"
dvander@592: 
dvander@592: CHalfLife2 g_HL2;
dvander@685: bool g_IsOriginalEngine = false;
dvander@592: 
faluco@682: namespace SourceHook
faluco@682: {
faluco@682: 	template<>
faluco@682: 	int HashFunction<datamap_t *>(datamap_t * const &k)
faluco@682: 	{
faluco@682: 		return reinterpret_cast<int>(k);
faluco@682: 	}
faluco@682: 
faluco@682: 	template<>
faluco@682: 	int Compare<datamap_t *>(datamap_t * const &k1, datamap_t * const &k2)
faluco@682: 	{
faluco@682: 		return (k1-k2);
faluco@682: 	}
faluco@682: }
faluco@682: 
dvander@592: CHalfLife2::CHalfLife2()
dvander@592: {
dvander@592: 	m_pClasses = sm_trie_create();
dvander@592: }
dvander@592: 
dvander@592: CHalfLife2::~CHalfLife2()
dvander@592: {
dvander@592: 	sm_trie_destroy(m_pClasses);
dvander@592: 
dvander@592: 	List<DataTableInfo *>::iterator iter;
dvander@592: 	DataTableInfo *pInfo;
dvander@592: 	for (iter=m_Tables.begin(); iter!=m_Tables.end(); iter++)
dvander@592: 	{
dvander@592: 		pInfo = (*iter);
dvander@592: 		sm_trie_destroy(pInfo->lookup);
dvander@592: 		delete pInfo;
dvander@592: 	}
dvander@592: 
dvander@592: 	m_Tables.clear();
faluco@682: 
faluco@682: 	THash<datamap_t *, DataMapTrie>::iterator h_iter;
faluco@682: 	for (h_iter=m_Maps.begin(); h_iter!=m_Maps.end(); h_iter++)
faluco@682: 	{
faluco@682: 		if (h_iter->val.trie)
faluco@682: 		{
faluco@682: 			sm_trie_destroy(h_iter->val.trie);
faluco@682: 			h_iter->val.trie = NULL;
faluco@682: 		}
faluco@682: 	}
faluco@682: 
faluco@682: 	m_Maps.clear();
dvander@592: }
dvander@592: 
dvander@677: CSharedEdictChangeInfo *g_pSharedChangeInfo = NULL;
dvander@677: 
dvander@592: void CHalfLife2::OnSourceModStartup(bool late)
dvander@592: {
ds@696: 	/* The Ship currently is the only known game to use an older version of the engine */
ds@729: 	if (strcasecmp(g_SourceMod.GetGameFolderName(), "ship") == 0)
dvander@677: 	{
ds@696: 		/* :TODO: Better engine versioning - perhaps something added to SourceMM? */
ds@696: 		g_IsOriginalEngine = true;
ds@696: 	} else if (!g_pSharedChangeInfo) {
dvander@677: 		g_pSharedChangeInfo = engine->GetSharedEdictChangeInfo();
dvander@677: 	}
dvander@592: }
dvander@592: 
ds@873: void CHalfLife2::OnSourceModAllInitialized()
ds@873: {
ds@873: 	m_MsgTextMsg = g_UserMsgs.GetMessageIndex("TextMsg");
faluco@1218: 	m_HinTextMsg = g_UserMsgs.GetMessageIndex("HintText");
faluco@1218: 	m_VGUIMenu = g_UserMsgs.GetMessageIndex("VGUIMenu");
dvander@965: 	g_ShareSys.AddInterface(NULL, this);
ds@873: }
ds@873: 
dvander@678: IChangeInfoAccessor *CBaseEdict::GetChangeAccessor()
dvander@678: {
dvander@678: 	return engine->GetChangeAccessor( (const edict_t *)this );
dvander@678: }
dvander@678: 
dvander@592: SendProp *UTIL_FindInSendTable(SendTable *pTable, const char *name)
dvander@592: {
dvander@592: 	const char *pname;
dvander@592: 	int props = pTable->GetNumProps();
dvander@592: 	SendProp *prop;
dvander@592: 
dvander@592: 	for (int i=0; i<props; i++)
dvander@592: 	{
dvander@592: 		prop = pTable->GetProp(i);
dvander@592: 		pname = prop->GetName();
dvander@592: 		if (pname && strcmp(name, pname) == 0)
dvander@592: 		{
dvander@592: 			return prop;
dvander@592: 		}
dvander@592: 		if (prop->GetDataTable())
dvander@592: 		{
dvander@592: 			if ((prop=UTIL_FindInSendTable(prop->GetDataTable(), name)) != NULL)
dvander@592: 			{
dvander@592: 				return prop;
dvander@592: 			}
dvander@592: 		}
dvander@592: 	}
dvander@592: 
dvander@592: 	return NULL;
dvander@592: }
dvander@592: 
faluco@682: typedescription_t *UTIL_FindInDataMap(datamap_t *pMap, const char *name)
faluco@682: {
faluco@682: 	while (pMap)
faluco@682: 	{
faluco@682: 		for (int i=0; i<pMap->dataNumFields; i++)
faluco@682: 		{
dvander@709: 			if (pMap->dataDesc[i].fieldName == NULL)
dvander@709: 			{
dvander@709: 				continue;
dvander@709: 			}
faluco@682: 			if (strcmp(name, pMap->dataDesc[i].fieldName) == 0)
faluco@682: 			{
faluco@682: 				return &(pMap->dataDesc[i]);
faluco@682: 			}
faluco@682: 			if (pMap->dataDesc[i].td)
faluco@682: 			{
faluco@682: 				typedescription_t *_td;
faluco@682: 				if ((_td=UTIL_FindInDataMap(pMap->dataDesc[i].td, name)) != NULL)
faluco@682: 				{
faluco@682: 					return _td;
faluco@682: 				}
faluco@682: 			}
faluco@682: 		}
faluco@682: 		pMap = pMap->baseMap;
faluco@682: 	}
faluco@682: 
faluco@682: 	return NULL; 
faluco@682: }
faluco@682: 
dvander@592: ServerClass *CHalfLife2::FindServerClass(const char *classname)
dvander@592: {
dvander@592: 	DataTableInfo *pInfo = _FindServerClass(classname);
dvander@592: 	if (!pInfo)
dvander@592: 	{
dvander@592: 		return NULL;
dvander@592: 	}
dvander@592: 
dvander@592: 	return pInfo->sc;
dvander@592: }
dvander@592: 
dvander@592: DataTableInfo *CHalfLife2::_FindServerClass(const char *classname)
dvander@592: {
dvander@592: 	DataTableInfo *pInfo = NULL;
dvander@592: 
dvander@592: 	if (!sm_trie_retrieve(m_pClasses, classname, (void **)&pInfo))
dvander@592: 	{
dvander@592: 		ServerClass *sc = gamedll->GetAllServerClasses();
dvander@592: 		while (sc)
dvander@592: 		{
dvander@592: 			if (strcmp(classname, sc->GetName()) == 0)
dvander@592: 			{
dvander@592: 				pInfo = new DataTableInfo;
dvander@592: 				pInfo->lookup = sm_trie_create();
dvander@592: 				pInfo->sc = sc;
dvander@592: 				sm_trie_insert(m_pClasses, classname, pInfo);
faluco@682: 				m_Tables.push_back(pInfo);
dvander@592: 				break;
dvander@592: 			}
dvander@592: 			sc = sc->m_pNext;
dvander@592: 		}
dvander@592: 		if (!pInfo)
dvander@592: 		{
dvander@592: 			return NULL;
dvander@592: 		}
dvander@592: 	}
dvander@592: 
dvander@592: 	return pInfo;
dvander@592: }
dvander@592: 
dvander@592: SendProp *CHalfLife2::FindInSendTable(const char *classname, const char *offset)
dvander@592: {
dvander@592: 	DataTableInfo *pInfo = _FindServerClass(classname);
dvander@592: 
dvander@592: 	if (!pInfo)
dvander@592: 	{
dvander@592: 		return NULL;
dvander@592: 	}
dvander@592: 
faluco@682: 	SendProp *pProp = NULL;
dvander@592: 	if (!sm_trie_retrieve(pInfo->lookup, offset, (void **)&pProp))
dvander@592: 	{
dvander@592: 		if ((pProp = UTIL_FindInSendTable(pInfo->sc->m_pTable, offset)) != NULL)
dvander@592: 		{
dvander@592: 			sm_trie_insert(pInfo->lookup, offset, pProp);
dvander@592: 		}
dvander@592: 	}
dvander@592: 
dvander@592: 	return pProp;
dvander@592: }
faluco@682: 
faluco@682: typedescription_t *CHalfLife2::FindInDataMap(datamap_t *pMap, const char *offset)
faluco@682: {
faluco@682: 	typedescription_t *td = NULL;
faluco@682: 	DataMapTrie &val = m_Maps[pMap];
faluco@682: 
faluco@682: 	if (!val.trie)
faluco@682: 	{
faluco@682: 		val.trie = sm_trie_create();
faluco@682: 	}
faluco@682: 	if (!sm_trie_retrieve(val.trie, offset, (void **)&td))
faluco@682: 	{
faluco@682: 		if ((td = UTIL_FindInDataMap(pMap, offset)) != NULL)
faluco@682: 		{
faluco@682: 			sm_trie_insert(val.trie, offset, td);
faluco@682: 		}
faluco@682: 	}
faluco@682: 
faluco@682: 	return td;
faluco@682: }
dvander@685: 
dvander@685: void CHalfLife2::SetEdictStateChanged(edict_t *pEdict, unsigned short offset)
dvander@685: {
dvander@685: 	if (!g_IsOriginalEngine)
dvander@685: 	{
dvander@685: 		if (offset)
dvander@685: 		{
dvander@685: 			pEdict->StateChanged(offset);
dvander@685: 		} else {
dvander@685: 			pEdict->StateChanged();
dvander@685: 		}
dvander@685: 	} else {
dvander@685: 		pEdict->m_fStateFlags |= FL_EDICT_CHANGED;
dvander@685: 	}
dvander@685: }
ds@873: 
dvander@1132: bool CHalfLife2::TextMsg(int client, int dest, const char *msg)
ds@873: {
ds@873: 	bf_write *pBitBuf = NULL;
ds@873: 	cell_t players[] = {client};
ds@873: 
dvander@1132: 	if ((pBitBuf = g_UserMsgs.StartMessage(m_MsgTextMsg, players, 1, USERMSG_RELIABLE)) == NULL)
dvander@1132: 	{
dvander@1132: 		return false;
dvander@1132: 	}
dvander@1132: 
ds@873: 	pBitBuf->WriteByte(dest);
ds@873: 	pBitBuf->WriteString(msg);
ds@873: 	g_UserMsgs.EndMessage();
dvander@1132: 
dvander@1132: 	return true;
ds@873: }
faluco@1218: 
teame06@1264: bool CHalfLife2::HintTextMsg(int client, const char *msg)
faluco@1218: {
faluco@1218: 	bf_write *pBitBuf = NULL;
faluco@1218: 	cell_t players[] = {client};
faluco@1218: 
faluco@1218: 	if ((pBitBuf = g_UserMsgs.StartMessage(m_HinTextMsg, players, 1, USERMSG_RELIABLE)) == NULL)
faluco@1218: 	{
faluco@1218: 		return false;
faluco@1218: 	}
faluco@1218: 
faluco@1218: 	pBitBuf->WriteByte(1);
faluco@1218: 	pBitBuf->WriteString(msg);
faluco@1218: 	g_UserMsgs.EndMessage();
faluco@1218: 
faluco@1218: 	return true;
faluco@1218: }
faluco@1218: 
faluco@1218: bool CHalfLife2::ShowVGUIMenu(int client, const char *name, KeyValues *data, bool show)
faluco@1218: {
faluco@1218: 	bf_write *pBitBuf = NULL;
faluco@1218: 	KeyValues *SubKey = NULL;
faluco@1218: 	int count = 0;
faluco@1218: 	cell_t players[] = {client};
faluco@1218: 
faluco@1218: 	if ((pBitBuf = g_UserMsgs.StartMessage(m_VGUIMenu, players, 1, USERMSG_RELIABLE)) == NULL)
faluco@1218: 	{
faluco@1218: 		return false;
faluco@1218: 	}
faluco@1218: 
faluco@1218: 	if (data)
faluco@1218: 	{
faluco@1218: 		SubKey = data->GetFirstSubKey();
faluco@1218: 		while (SubKey)
faluco@1218: 		{
faluco@1218: 			count++;
faluco@1218: 			SubKey = SubKey->GetNextKey();
faluco@1218: 		}
faluco@1218: 		SubKey = data->GetFirstSubKey();
faluco@1218: 	}
faluco@1218: 
faluco@1218: 	pBitBuf->WriteString(name);
faluco@1218: 	pBitBuf->WriteByte((show) ? 1 : 0);
faluco@1218: 	pBitBuf->WriteByte(count);
faluco@1218: 	while (SubKey)
faluco@1218: 	{
faluco@1218: 		pBitBuf->WriteString(SubKey->GetName());
faluco@1218: 		pBitBuf->WriteString(SubKey->GetString());
faluco@1218: 		SubKey = SubKey->GetNextKey();
faluco@1218: 	}
faluco@1218: 	g_UserMsgs.EndMessage();
faluco@1218: 
faluco@1218: 	return true;
faluco@1218: }
dvander@1338: 
dvander@1338: void CHalfLife2::AddToFakeCliCmdQueue(int client, int userid, const char *cmd)
dvander@1338: {
dvander@1338: 	DelayedFakeCliCmd *pFake;
dvander@1338: 
dvander@1338: 	if (m_FreeCmds.empty())
dvander@1338: 	{
dvander@1338: 		pFake = new DelayedFakeCliCmd;
dvander@1338: 	} else {
dvander@1338: 		pFake = m_FreeCmds.front();
dvander@1338: 		m_FreeCmds.pop();
dvander@1338: 	}
dvander@1338: 
dvander@1338: 	pFake->client = client;
dvander@1338: 	pFake->userid = userid;
dvander@1338: 	pFake->cmd.assign(cmd);
dvander@1338: 
dvander@1338: 	m_CmdQueue.push(pFake);
dvander@1338: }
dvander@1338: 
dvander@1338: void CHalfLife2::ProcessFakeCliCmdQueue()
dvander@1338: {
dvander@1338: 	while (!m_CmdQueue.empty())
dvander@1338: 	{
dvander@1338: 		DelayedFakeCliCmd *pFake = m_CmdQueue.first();
dvander@1338: 
dvander@1338: 		if (g_Players.GetClientOfUserId(pFake->userid) == pFake->client)
dvander@1338: 		{
dvander@1338: 			CPlayer *pPlayer = g_Players.GetPlayerByIndex(pFake->client);
dvander@1338: 			serverpluginhelpers->ClientCommand(pPlayer->GetEdict(), pFake->cmd.c_str());
dvander@1338: 		}
dvander@1338: 
dvander@1338: 		m_CmdQueue.pop();
dvander@1338: 	}
dvander@1338: }