dvander@547: /**
ds@653:  * vim: set ts=4 :
dvander@547:  * ===============================================================
dvander@547:  * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.
dvander@547:  * ===============================================================
dvander@547:  *
dvander@547:  * This file is not open source and may not be copied without explicit
dvander@547:  * written permission of AlliedModders LLC.  This file may not be redistributed 
dvander@547:  * in whole or significant part.
dvander@547:  * For information, see LICENSE.txt or http://www.sourcemod.net/license.php
dvander@547:  *
dvander@547:  * Version: $Id$
dvander@547:  */
dvander@547: 
dvander@547: #include <string.h>
dvander@547: #include <stdlib.h>
dvander@592: #include "GameConfigs.h"
dvander@591: #include "TextParsers.h"
dvander@547: #include "sm_stringutil.h"
dvander@547: #include "sourcemod.h"
dvander@547: #include "sourcemm_api.h"
dvander@592: #include "HalfLife2.h"
dvander@592: #include "Logger.h"
dvander@547: #include "ShareSys.h"
dvander@547: 
dvander@616: GameConfigManager g_GameConfigs;
dvander@547: IGameConfig *g_pGameConf = NULL;
dvander@547: char g_mod[255];
dvander@547: 
dvander@547: #define PSTATE_NONE						0
dvander@547: #define PSTATE_GAMES					1
dvander@547: #define PSTATE_GAMEDEFS					2
dvander@547: #define PSTATE_GAMEDEFS_OFFSETS			3
dvander@547: #define PSTATE_GAMEDEFS_OFFSETS_OFFSET	4
dvander@548: #define PSTATE_GAMEDEFS_KEYS			5
dvander@547: 
dvander@547: #if defined PLATFORM_WINDOWS
dvander@547: #define PLATFORM_NAME	"windows"
dvander@547: #elif defined PLATFORM_LINUX
dvander@547: #define PLATFORM_NAME	"linux"
dvander@547: #endif
dvander@547: 
dvander@547: CGameConfig::CGameConfig(const char *file)
dvander@547: {
dvander@547: 	m_pFile = sm_strdup(file);
dvander@547: 	m_pOffsets = sm_trie_create();
dvander@547: 	m_pProps = sm_trie_create();
dvander@548: 	m_pKeys = sm_trie_create();
dvander@548: 	m_pStrings = new BaseStringTable(512);
faluco@641: 	m_RefCount = 0;
dvander@547: }
dvander@547: 
dvander@547: CGameConfig::~CGameConfig()
dvander@547: {
dvander@547: 	sm_trie_destroy(m_pOffsets);
dvander@547: 	sm_trie_destroy(m_pProps);
dvander@548: 	sm_trie_destroy(m_pKeys);
dvander@547: 	delete [] m_pFile;
dvander@548: 	delete m_pStrings;
dvander@547: }
dvander@547: 
dvander@547: SMCParseResult CGameConfig::ReadSMC_NewSection(const char *name, bool opt_quotes)
dvander@547: {
dvander@547: 	if (m_IgnoreLevel)
dvander@547: 	{
dvander@547: 		m_IgnoreLevel++;
dvander@547: 		return SMCParse_Continue;
dvander@547: 	}
dvander@547: 	switch (m_ParseState)
dvander@547: 	{
dvander@547: 	case PSTATE_NONE:
dvander@547: 		{
dvander@547: 			if (strcmp(name, "Games") == 0)
dvander@547: 			{
dvander@547: 				m_ParseState = PSTATE_GAMES;
dvander@547: 			} else {
dvander@547: 				m_IgnoreLevel++;
dvander@547: 			}
dvander@547: 			break;
dvander@547: 		}
dvander@547: 	case PSTATE_GAMES:
dvander@547: 		{
dvander@547: 			if ((strcmp(name, "*") == 0)
dvander@547: 				|| (strcmp(name, g_mod) == 0))
dvander@547: 			{
dvander@547: 				m_ParseState = PSTATE_GAMEDEFS;
dvander@547: 				strncopy(m_mod, name, sizeof(m_mod));
dvander@547: 			} else {
dvander@547: 				m_IgnoreLevel++;
dvander@547: 			}
dvander@547: 			break;
dvander@547: 		}
dvander@547: 	case PSTATE_GAMEDEFS:
dvander@547: 		{
dvander@547: 			if (strcmp(name, "Offsets") == 0)
dvander@547: 			{
dvander@547: 				m_ParseState = PSTATE_GAMEDEFS_OFFSETS;
dvander@548: 			} else if (strcmp(name, "Keys") == 0) {
dvander@548: 				m_ParseState = PSTATE_GAMEDEFS_KEYS;
dvander@547: 			} else {
dvander@547: 				m_IgnoreLevel++;
dvander@547: 			}
dvander@547: 			break;
dvander@547: 		}
dvander@547: 	case PSTATE_GAMEDEFS_OFFSETS:
dvander@547: 		{
dvander@547: 			m_prop[0] = '\0';
dvander@547: 			m_class[0] = '\0';
dvander@547: 			strncopy(m_offset, name, sizeof(m_offset));
dvander@547: 			m_ParseState = PSTATE_GAMEDEFS_OFFSETS_OFFSET;
dvander@547: 			break;
dvander@547: 		}
dvander@547: 	/* No sub-sections allowed:
dvander@547: 	 case PSTATE_GAMEDEFS_OFFSETS_OFFSET:
dvander@548: 	 case PSTATE_GAMEDEFS_KEYS:
dvander@547: 	 */
dvander@547: 	default:
dvander@547: 		{
dvander@547: 			/* If we don't know what we got, start ignoring */
dvander@547: 			m_IgnoreLevel++;
dvander@547: 			break;
dvander@547: 		}
dvander@547: 	}
dvander@547: 
dvander@547: 	return SMCParse_Continue;
dvander@547: }
dvander@547: 
dvander@547: SMCParseResult CGameConfig::ReadSMC_KeyValue(const char *key, const char *value, bool key_quotes, bool value_quotes)
dvander@547: {
dvander@547: 	if (m_IgnoreLevel)
dvander@547: 	{
dvander@547: 		return SMCParse_Continue;
dvander@547: 	}
dvander@547: 
dvander@547: 	if (m_ParseState == PSTATE_GAMEDEFS_OFFSETS_OFFSET)
dvander@547: 	{
dvander@547: 		if (strcmp(key, "class") == 0)
dvander@547: 		{
dvander@547: 			strncopy(m_class, value, sizeof(m_class));
dvander@547: 		} else if (strcmp(key, "prop") == 0) {
dvander@547: 			strncopy(m_prop, value, sizeof(m_prop));
dvander@547: 		} else if (strcmp(key, PLATFORM_NAME) == 0) {
dvander@547: 			int val = atoi(value);
dvander@547: 			sm_trie_insert(m_pOffsets, m_offset, (void *)val);
dvander@547: 		}
dvander@548: 	} else if (m_ParseState == PSTATE_GAMEDEFS_KEYS) {
dvander@548: 		int id = m_pStrings->AddString(value);
dvander@548: 		sm_trie_insert(m_pKeys, key, (void *)id);
dvander@547: 	}
dvander@547: 
dvander@547: 	return SMCParse_Continue;
dvander@547: }
dvander@547: 
dvander@547: SMCParseResult CGameConfig::ReadSMC_LeavingSection()
dvander@547: {
dvander@547: 	if (m_IgnoreLevel)
dvander@547: 	{
dvander@547: 		m_IgnoreLevel--;
dvander@547: 		return SMCParse_Continue;
dvander@547: 	}
dvander@547: 
dvander@547: 	switch (m_ParseState)
dvander@547: 	{
dvander@547: 	case PSTATE_GAMES:
dvander@547: 		{
dvander@547: 			m_ParseState = PSTATE_NONE;
dvander@547: 			break;
dvander@547: 		}
dvander@547: 	case PSTATE_GAMEDEFS:
dvander@547: 		{
dvander@547: 			m_ParseState = PSTATE_GAMES;
dvander@547: 			break;
dvander@547: 		}
dvander@548: 	case PSTATE_GAMEDEFS_KEYS:
dvander@547: 	case PSTATE_GAMEDEFS_OFFSETS:
dvander@547: 		{
dvander@547: 			m_ParseState = PSTATE_GAMEDEFS;
dvander@547: 			break;
dvander@547: 		}
dvander@547: 	case PSTATE_GAMEDEFS_OFFSETS_OFFSET:
dvander@547: 		{
dvander@547: 			/* Parse the offset... */
dvander@547: 			if (m_class[0] != '\0'
dvander@547: 				&& m_prop[0] != '\0')
dvander@547: 			{
dvander@547: 				SendProp *pProp = g_HL2.FindInSendTable(m_class, m_prop);
dvander@547: 				if (pProp)
dvander@547: 				{
dvander@547: 					int val = pProp->GetOffset();
dvander@547: 					sm_trie_insert(m_pOffsets, m_offset, (void *)val);
dvander@547: 					sm_trie_insert(m_pProps, m_offset, pProp);
dvander@547: 				} else {
dvander@547: 					/* Check if it's a non-default game and no offsets exist */
dvander@547: 					if ((strcmp(m_mod, "*") != 0)
dvander@547: 						&& (!sm_trie_retrieve(m_pOffsets, m_offset, NULL)))
dvander@547: 					{
dvander@547: 						g_Logger.LogError("[SM] Unable to find property %s.%s (file \"%s\") (mod \"%s\")", 
dvander@547: 							m_class,
dvander@547: 							m_prop,
dvander@547: 							m_pFile,
dvander@547: 							m_mod);
dvander@547: 					}
dvander@547: 				}
dvander@547: 			}
dvander@547: 			m_ParseState = PSTATE_GAMEDEFS_OFFSETS;
dvander@547: 			break;
dvander@547: 		}
dvander@547: 	}
dvander@547: 
dvander@547: 	return SMCParse_Continue;
dvander@547: }
dvander@547: 
dvander@547: bool CGameConfig::Reparse(char *error, size_t maxlength)
dvander@547: {
dvander@547: 	SMCParseError err;
dvander@547: 
dvander@547: 	char path[PLATFORM_MAX_PATH];
dvander@547: 	g_SourceMod.BuildPath(Path_SM, path, sizeof(path), "configs/gamedata/%s.txt", m_pFile);
dvander@547: 
dvander@548: 	/* Initialize parse states */
dvander@547: 	m_IgnoreLevel = 0;
dvander@547: 	m_ParseState = PSTATE_NONE;
dvander@548: 	/* Reset cached data */
dvander@548: 	m_pStrings->Reset();
dvander@547: 	sm_trie_clear(m_pOffsets);
dvander@548: 	sm_trie_clear(m_pProps);
dvander@548: 	sm_trie_clear(m_pKeys);
dvander@547: 
dvander@547: 	if ((err=g_TextParser.ParseFile_SMC(path, this, NULL, NULL))
dvander@547: 		!= SMCParse_Okay)
dvander@547: 	{
dvander@547: 		if (error && (err != SMCParse_Custom))
dvander@547: 		{
dvander@547: 			const char *str = g_TextParser.GetSMCErrorString(err);
dvander@551: 			snprintf(error, maxlength, "%s", str);
dvander@547: 		}
dvander@547: 		return false;
dvander@547: 	}
dvander@547: 
dvander@547: 	return true;
dvander@547: }
dvander@547: 
dvander@547: bool CGameConfig::GetOffset(const char *key, int *value)
dvander@547: {
dvander@547: 	void *obj;
dvander@547: 
dvander@547: 	if (!sm_trie_retrieve(m_pOffsets, key, &obj))
dvander@547: 	{
dvander@547: 		return false;
dvander@547: 	}
dvander@547: 
dvander@547: 	*value = (int)obj;
dvander@547: 	
dvander@547: 	return true;
dvander@547: }
dvander@547: 
dvander@547: const char *CGameConfig::GetKeyValue(const char *key)
dvander@547: {
dvander@548: 	void *obj;
dvander@548: 	if (!sm_trie_retrieve(m_pKeys, key, &obj))
dvander@548: 	{
dvander@548: 		return NULL;
dvander@548: 	}
dvander@548: 	return m_pStrings->GetString((int)obj);
dvander@547: }
dvander@547: 
dvander@547: SendProp *CGameConfig::GetSendProp(const char *key)
dvander@547: {
dvander@547: 	SendProp *pProp;
dvander@547: 
dvander@547: 	if (!sm_trie_retrieve(m_pProps, key, (void **)&pProp))
dvander@547: 	{
dvander@547: 		return NULL;
dvander@547: 	}
dvander@547: 
dvander@547: 	return pProp;
dvander@547: }
dvander@547: 
dvander@547: void CGameConfig::IncRefCount()
dvander@547: {
dvander@547: 	m_RefCount++;
dvander@547: }
dvander@547: 
dvander@547: unsigned int CGameConfig::DecRefCount()
dvander@547: {
dvander@547: 	m_RefCount--;
dvander@547: 	return m_RefCount;
dvander@547: }
dvander@547: 
dvander@616: GameConfigManager::GameConfigManager()
dvander@547: {
dvander@547: 	m_pLookup = sm_trie_create();
dvander@547: }
dvander@547: 
dvander@616: GameConfigManager::~GameConfigManager()
dvander@547: {
dvander@547: 	sm_trie_destroy(m_pLookup);
dvander@547: }
dvander@547: 
dvander@616: void GameConfigManager::OnSourceModStartup(bool late)
dvander@547: {
dvander@547: 	LoadGameConfigFile("core.games", &g_pGameConf, NULL, 0);
dvander@547: 
ds@696: 	strncopy(g_mod, g_SourceMod.GetModFolderName(), sizeof(g_mod));
dvander@547: }
dvander@547: 
dvander@616: void GameConfigManager::OnSourceModAllInitialized()
dvander@547: {
dvander@547: 	/* NOW initialize the game file */
dvander@547: 	CGameConfig *pGameConf = (CGameConfig *)g_pGameConf;
dvander@547: 	
dvander@547: 	char error[255];
dvander@547: 	if (!pGameConf->Reparse(error, sizeof(error)))
dvander@547: 	{
dvander@547: 		/* :TODO: log */
dvander@547: 	}
dvander@547: 
dvander@547: 	g_ShareSys.AddInterface(NULL, this);
dvander@547: }
dvander@547: 
dvander@616: void GameConfigManager::OnSourceModAllShutdown()
dvander@547: {
dvander@547: 	CloseGameConfigFile(g_pGameConf);
dvander@547: }
dvander@547: 
dvander@616: bool GameConfigManager::LoadGameConfigFile(const char *file, IGameConfig **_pConfig, char *error, size_t maxlength)
dvander@547: {
dvander@547: 	CGameConfig *pConfig;
dvander@547: 
dvander@547: 	if (sm_trie_retrieve(m_pLookup, file, (void **)&pConfig))
dvander@547: 	{
dvander@547: 		pConfig->IncRefCount();
dvander@547: 		*_pConfig = pConfig;
dvander@547: 		return true;
dvander@547: 	}
dvander@547: 
dvander@547: 	pConfig = new CGameConfig(file);
dvander@547: 
dvander@547: 	/* :HACKHACK: Don't parse the main config file */
dvander@547: 	bool retval = true;
dvander@547: 	if (_pConfig != &g_pGameConf)
dvander@547: 	{
dvander@547: 		retval = pConfig->Reparse(error, maxlength);
dvander@547: 	}
dvander@547: 
dvander@547: 	m_cfgs.push_back(pConfig);
dvander@547: 	sm_trie_insert(m_pLookup, file, pConfig);
dvander@547: 
dvander@547: 	pConfig->IncRefCount();
dvander@547: 
dvander@547: 	*_pConfig = pConfig;
dvander@547: 
dvander@547: 	return retval;
dvander@547: }
dvander@547: 
dvander@616: void GameConfigManager::CloseGameConfigFile(IGameConfig *cfg)
dvander@547: {
dvander@547: 	CGameConfig *pConfig = (CGameConfig *)cfg;
dvander@547: 
dvander@547: 	if (pConfig->DecRefCount() == 0)
dvander@547: 	{
dvander@547: 		delete pConfig;
dvander@547: 	}
dvander@547: }