/**

 * vim: set ts=4 :

 * =============================================================================

 * SourceMod

 * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.

 * =============================================================================

 *

 * This program is free software; you can redistribute it and/or modify it under

 * the terms of the GNU General Public License, version 3.0, as published by the

 * Free Software Foundation.

 * 

 * This program is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

 * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more

 * details.

 *

 * You should have received a copy of the GNU General Public License along with

 * this program.  If not, see <http://www.gnu.org/licenses/>.

 *

 * As a special exception, AlliedModders LLC gives you permission to link the

 * code of this program (as well as its derivative works) to "Half-Life 2," the

 * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software

 * by the Valve Corporation.  You must obey the GNU General Public License in

 * all respects for all other code used.  Additionally, AlliedModders LLC grants

 * this exception to all derivative works.  AlliedModders LLC defines further

 * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),

 * or <http://www.sourcemod.net/license.php>.

 *

 * Version: $Id$

 */

#include "HalfLife2.h"

#include "sourcemod.h"

#include "sourcemm_api.h"

#include "UserMessages.h"

#include "PlayerManager.h"

#include "sm_stringutil.h"

#include <IGameConfigs.h>

#include <compat_wrappers.h>

#include <Logger.h>

#include "LibrarySys.h"

#include "logic_bridge.h"

#include <tier0/mem.h>

#if SOURCE_ENGINE == SE_DOTA

#include <game/shared/protobuf/usermessages.pb.h>

#elif SOURCE_ENGINE == SE_CSGO

#include <cstrike15_usermessages.pb.h>

#endif

typedef ICommandLine *(*FakeGetCommandLine)();

#if defined _WIN32

#define TIER0_NAME			"tier0.dll"

#define VSTDLIB_NAME		"vstdlib.dll"

#elif defined __APPLE__

#define TIER0_NAME			"libtier0.dylib"

#define VSTDLIB_NAME		"libvstdlib.dylib"

#elif defined __linux__

#if SOURCE_ENGINE == SE_HL2DM || SOURCE_ENGINE == SE_DODS || SOURCE_ENGINE == SE_CSS || SOURCE_ENGINE == SE_TF2 || SOURCE_ENGINE == SE_LEFT4DEAD2

#define TIER0_NAME			"libtier0_srv.so"

#define VSTDLIB_NAME		"libvstdlib_srv.so"

#elif SOURCE_ENGINE >= SE_LEFT4DEAD

#define TIER0_NAME			"libtier0.so"

#define VSTDLIB_NAME		"libvstdlib.so"

#else

#define TIER0_NAME			"tier0_i486.so"

#define VSTDLIB_NAME		"vstdlib_i486.so"

#endif

#endif

CHalfLife2 g_HL2;

ConVar *sv_lan = NULL;

static void *g_EntList = NULL;

static int entInfoOffset = -1;

namespace SourceHook

{

	template<>

	int HashFunction<datamap_t *>(datamap_t * const &k)

	{

		return reinterpret_cast<int>(k);

	}

	template<>

	int Compare<datamap_t *>(datamap_t * const &k1, datamap_t * const &k2)

	{

		return (k1-k2);

	}

}

CHalfLife2::CHalfLife2()

{

	m_pClasses = sm_trie_create();

}

CHalfLife2::~CHalfLife2()

{

	sm_trie_destroy(m_pClasses);

	List<DataTableInfo *>::iterator iter;

	DataTableInfo *pInfo;

	for (iter=m_Tables.begin(); iter!=m_Tables.end(); iter++)

	{

		pInfo = (*iter);

		delete pInfo;

	}

	m_Tables.clear();

	THash<datamap_t *, DataMapTrie>::iterator h_iter;

	for (h_iter=m_Maps.begin(); h_iter!=m_Maps.end(); h_iter++)

	{

		if (h_iter->val.trie)

		{

			delete h_iter->val.trie;

			h_iter->val.trie = NULL;

		}

	}

	m_Maps.clear();

}

#if SOURCE_ENGINE != SE_DARKMESSIAH

CSharedEdictChangeInfo *g_pSharedChangeInfo = NULL;

#endif

#if !defined METAMOD_PLAPI_VERSION || PLAPI_VERSION < 11

bool is_original_engine = false;

#endif

void CHalfLife2::OnSourceModStartup(bool late)

{

#if SOURCE_ENGINE != SE_DARKMESSIAH

	/* The Ship currently is the only known game to use an older version of the engine */

#if defined METAMOD_PLAPI_VERSION || PLAPI_VERSION >= 11

	if (g_SMAPI->GetSourceEngineBuild() == SOURCE_ENGINE_ORIGINAL)

#else

	if (strcasecmp(g_SourceMod.GetGameFolderName(), "ship") == 0)

#endif

	{

		is_original_engine = true;

	}

	else if (g_pSharedChangeInfo == NULL)

	{

		g_pSharedChangeInfo = engine->GetSharedEdictChangeInfo();

	}

#endif

}

void CHalfLife2::OnSourceModAllInitialized()

{

	m_MsgTextMsg = g_UserMsgs.GetMessageIndex("TextMsg");

	m_HinTextMsg = g_UserMsgs.GetMessageIndex("HintText");

	m_SayTextMsg = g_UserMsgs.GetMessageIndex("SayText");

	m_VGUIMenu = g_UserMsgs.GetMessageIndex("VGUIMenu");

	sharesys->AddInterface(NULL, this);

}

void CHalfLife2::OnSourceModAllInitialized_Post()

{

	InitLogicalEntData();

	InitCommandLine();

}

void CHalfLife2::InitLogicalEntData()

{

	char *addr = NULL;

	/*

	 * gEntList and/or g_pEntityList

	 *

	 * First try to lookup pointer directly for platforms with symbols.

	 * If symbols aren't present (Windows or stripped Linux/Mac), 

	 * attempt find via LevelShutdown + offset

	 */

	if (g_pGameConf->GetMemSig("gEntList", (void **)&addr))

	{

#if !defined PLATFORM_WINDOWS

		if (!addr)

		{

			// Key exists so notify if lookup fails, but try other method.

			g_Logger.LogError("Failed lookup of gEntList directly - Reverting to lookup via LevelShutdown");

		}

		else

		{

#endif

			g_EntList = reinterpret_cast<void *>(addr);

#if !defined PLATFORM_WINDOWS

		}

#endif

	}

	if (!g_EntList)

	{

		if (!g_pGameConf->GetMemSig("LevelShutdown", (void **)&addr))

		{

			g_Logger.LogError("Logical Entities not supported by this mod (LevelShutdown) - Reverting to networkable entities only");

			return;

		}

		if (!addr)

		{

			g_Logger.LogError("Failed lookup of LevelShutdown - Reverting to networkable entities only");

			return;

		}

		int offset;

		if (!g_pGameConf->GetOffset("gEntList", &offset))

		{

			g_Logger.LogError("Logical Entities not supported by this mod (gEntList) - Reverting to networkable entities only");

			return;

		}

		g_EntList = *reinterpret_cast<void **>(addr + offset);

	}

	if (!g_EntList)

	{

		g_Logger.LogError("Failed lookup of gEntList - Reverting to networkable entities only");

		return;

	}

	if (!g_pGameConf->GetOffset("EntInfo", &entInfoOffset))

	{

		g_Logger.LogError("Logical Entities not supported by this mod (EntInfo) - Reverting to networkable entities only");

		return;

	}

}

void CHalfLife2::InitCommandLine()

{

	char path[PLATFORM_MAX_PATH];

	char error[256];

	g_SourceMod.BuildPath(Path_Game, path, sizeof(path), "../bin/" TIER0_NAME);

	if (!g_LibSys.IsPathFile(path))

	{

		g_Logger.LogError("Could not find path for: " TIER0_NAME);

		return;

	}

	ILibrary *lib = g_LibSys.OpenLibrary(path, error, sizeof(error));

	m_pGetCommandLine = lib->GetSymbolAddress("CommandLine_Tier0");

	/* '_Tier0' dropped on Alien Swarm version */

	if (m_pGetCommandLine == NULL)

	{

		m_pGetCommandLine = lib->GetSymbolAddress("CommandLine");

	}

	if (m_pGetCommandLine == NULL)

	{

		/* We probably have a Ship engine. */

		lib->CloseLibrary();

		g_SourceMod.BuildPath(Path_Game, path, sizeof(path), "../bin/" VSTDLIB_NAME);

		if (!g_LibSys.IsPathFile(path))

		{

			g_Logger.LogError("Could not find path for: " VSTDLIB_NAME);

			return;

		}

		if ((lib = g_LibSys.OpenLibrary(path, error, sizeof(error))) == NULL)

		{

			g_Logger.LogError("Could not load %s: %s", path, error);

			return;

		}

		m_pGetCommandLine = lib->GetSymbolAddress("CommandLine");

		if (m_pGetCommandLine == NULL)

		{

			g_Logger.LogError("Could not locate any command line functionality");

		}

		lib->CloseLibrary();

	}

}

ICommandLine *CHalfLife2::GetValveCommandLine()

{

	if (!m_pGetCommandLine)

		return NULL;

	return ((FakeGetCommandLine)((FakeGetCommandLine *)m_pGetCommandLine))();

}

#if !defined METAMOD_PLAPI_VERSION || PLAPI_VERSION < 11

bool CHalfLife2::IsOriginalEngine()

{

	return is_original_engine;

}

#endif

#if SOURCE_ENGINE != SE_DARKMESSIAH

IChangeInfoAccessor *CBaseEdict::GetChangeAccessor()

{

#if SOURCE_ENGINE == SE_DOTA

	return engine->GetChangeAccessor( IndexOfEdict((const edict_t *)this) );

#else

	return engine->GetChangeAccessor( (const edict_t *)this );

#endif

}

#endif

bool UTIL_FindInSendTable(SendTable *pTable, 

						  const char *name,

						  sm_sendprop_info_t *info,

						  unsigned int offset)

{

	const char *pname;

	int props = pTable->GetNumProps();

	SendProp *prop;

	for (int i=0; i<props; i++)

	{

		prop = pTable->GetProp(i);

		pname = prop->GetName();

		if (pname && strcmp(name, pname) == 0)

		{

			info->prop = prop;

			info->actual_offset = offset + info->prop->GetOffset();

			return true;

		}

		if (prop->GetDataTable())

		{

			if (UTIL_FindInSendTable(prop->GetDataTable(), 

				name,

				info,

				offset + prop->GetOffset())

				)

			{

				return true;

			}

		}

	}

	return false;

}

bool UTIL_FindDataMapInfo(datamap_t *pMap, const char *name, sm_datatable_info_t *pDataTable)

{

	while (pMap)

	{

		for (int i = 0; i < pMap->dataNumFields; ++i)

		{

			if (pMap->dataDesc[i].fieldName == NULL)

			{

				continue;

			}

			if (strcmp(name, pMap->dataDesc[i].fieldName) == 0)

			{

				pDataTable->prop = &(pMap->dataDesc[i]);

				pDataTable->actual_offset = GetTypeDescOffs(pDataTable->prop);

				return true;

			}

			if (pMap->dataDesc[i].td == NULL || !UTIL_FindDataMapInfo(pMap->dataDesc[i].td, name, pDataTable))

			{

				continue;

			}

			pDataTable->actual_offset += GetTypeDescOffs(&(pMap->dataDesc[i]));

			return true;

		}

		pMap = pMap->baseMap;

	}

	return false; 

}

ServerClass *CHalfLife2::FindServerClass(const char *classname)

{

	DataTableInfo *pInfo = _FindServerClass(classname);

	if (!pInfo)

	{

		return NULL;

	}

	return pInfo->sc;

}

DataTableInfo *CHalfLife2::_FindServerClass(const char *classname)

{

	DataTableInfo *pInfo = NULL;

	if (!sm_trie_retrieve(m_pClasses, classname, (void **)&pInfo))

	{

		ServerClass *sc = gamedll->GetAllServerClasses();

		while (sc)

		{

			if (strcmp(classname, sc->GetName()) == 0)

			{

				pInfo = new DataTableInfo;

				pInfo->sc = sc;

				sm_trie_insert(m_pClasses, classname, pInfo);

				m_Tables.push_back(pInfo);

				break;

			}

			sc = sc->m_pNext;

		}

		if (!pInfo)

		{

			return NULL;

		}

	}

	return pInfo;

}

bool CHalfLife2::FindSendPropInfo(const char *classname, const char *offset, sm_sendprop_info_t *info)

{

	DataTableInfo *pInfo;

	sm_sendprop_info_t *prop;

	if ((pInfo = _FindServerClass(classname)) == NULL)

	{

		return false;

	}

	if ((prop = pInfo->lookup.retrieve(offset)) == NULL)

	{

		sm_sendprop_info_t temp_info;

		if (!UTIL_FindInSendTable(pInfo->sc->m_pTable, offset, &temp_info, 0))

		{

			return false;

		}

		pInfo->lookup.insert(offset, temp_info);

		*info = temp_info;

	}

	else

	{

		*info = *prop;

	}

	return true;

}

SendProp *CHalfLife2::FindInSendTable(const char *classname, const char *offset)

{

	sm_sendprop_info_t info;

	if (!FindSendPropInfo(classname, offset, &info))

	{

		return NULL;

	}

	return info.prop;

}

typedescription_t *CHalfLife2::FindInDataMap(datamap_t *pMap, const char *offset)

{

	sm_datatable_info_t dt_info;

	if (!(this->FindDataMapInfo(pMap, offset, &dt_info)))

	{

		return NULL;

	}

	return dt_info.prop;

}

bool CHalfLife2::FindDataMapInfo(datamap_t *pMap, const char *offset, sm_datatable_info_t *pDataTable)

{

	DataMapTrie &val = m_Maps[pMap];

	if (!val.trie)

	{

		val.trie = new KTrie<sm_datatable_info_t>;

	}

	sm_datatable_info_t * pNewTable = val.trie->retrieve(offset);

	if (!pNewTable)

	{

		if (UTIL_FindDataMapInfo(pMap, offset, pDataTable))

		{

			val.trie->insert(offset, *pDataTable);

		}

		else

		{

			pDataTable->prop = NULL;

		}

	}

	else

	{

		*pDataTable = *pNewTable;

	}

	return (pDataTable->prop != NULL);

}

void CHalfLife2::SetEdictStateChanged(edict_t *pEdict, unsigned short offset)

{

#if SOURCE_ENGINE != SE_DARKMESSIAH

	if (g_pSharedChangeInfo != NULL)

	{

		if (offset)

		{

			pEdict->StateChanged(offset);

		}

		else

		{

			pEdict->StateChanged();

		}

	}

	else

#endif

	{

		pEdict->m_fStateFlags |= FL_EDICT_CHANGED;

	}

}

bool CHalfLife2::TextMsg(int client, int dest, const char *msg)

{

#ifndef USE_PROTOBUF_USERMESSAGES

	bf_write *pBitBuf = NULL;

#endif

	cell_t players[] = {client};

	if (dest == HUD_PRINTTALK)

	{

		const char *chat_saytext = g_pGameConf->GetKeyValue("ChatSayText");

		/* Use SayText user message instead */

		if (chat_saytext != NULL && strcmp(chat_saytext, "yes") == 0)

		{

			char buffer[192];

			UTIL_Format(buffer, sizeof(buffer), "%s\1\n", msg);

#if SOURCE_ENGINE == SE_DOTA

			CUserMsg_SayText *pMsg;

			if ((pMsg = (CUserMsg_SayText *)g_UserMsgs.StartProtobufMessage(m_SayTextMsg, players, 1, USERMSG_RELIABLE)) == NULL)

			{

				return false;

			}

			pMsg->set_client(0);

			pMsg->set_text(buffer);

			pMsg->set_chat(false);

#elif SOURCE_ENGINE == SE_CSGO

			CCSUsrMsg_SayText *pMsg;

			if ((pMsg = (CCSUsrMsg_SayText *)g_UserMsgs.StartProtobufMessage(m_SayTextMsg, players, 1, USERMSG_RELIABLE)) == NULL)

			{

				return false;

			}

			pMsg->set_ent_idx(0);

			pMsg->set_text(buffer);

			pMsg->set_chat(false);

#else

			if ((pBitBuf = g_UserMsgs.StartBitBufMessage(m_SayTextMsg, players, 1, USERMSG_RELIABLE)) == NULL)

			{

				return false;

			}

			pBitBuf->WriteByte(0);

			pBitBuf->WriteString(buffer);

			pBitBuf->WriteByte(1);

#endif

			g_UserMsgs.EndMessage();

			return true;

		}

	}

#if SOURCE_ENGINE == SE_DOTA

	CUserMsg_TextMsg *pMsg;

	if ((pMsg = (CUserMsg_TextMsg *)g_UserMsgs.StartProtobufMessage(m_MsgTextMsg, players, 1, USERMSG_RELIABLE)) == NULL)

	{

		return false;

	}

	pMsg->set_dest(dest);

	pMsg->add_param(msg);

	pMsg->add_param("");

	pMsg->add_param("");

	pMsg->add_param("");

	pMsg->add_param("");

#elif SOURCE_ENGINE == SE_CSGO

	CCSUsrMsg_TextMsg *pMsg;

	if ((pMsg = (CCSUsrMsg_TextMsg *)g_UserMsgs.StartProtobufMessage(m_MsgTextMsg, players, 1, USERMSG_RELIABLE)) == NULL)

	{

		return false;

	}

	// Client tries to read all 5 'params' and will crash if less

	pMsg->set_msg_dst(dest);

	pMsg->add_params(msg);

	pMsg->add_params("");

	pMsg->add_params("");

	pMsg->add_params("");

	pMsg->add_params("");

#else

	if ((pBitBuf = g_UserMsgs.StartBitBufMessage(m_MsgTextMsg, players, 1, USERMSG_RELIABLE)) == NULL)

	{

		return false;

	}

	pBitBuf->WriteByte(dest);

	pBitBuf->WriteString(msg);

#endif

	g_UserMsgs.EndMessage();

	return true;

}

bool CHalfLife2::HintTextMsg(int client, const char *msg)

{

	cell_t players[] = {client};