sourcemod-central: core/GameConfigs.cpp@130639d897c2 (annotated)

dvander@547	1	/**
ds@2863	2	* vim: set ts=4 sw=4 tw=99 noet :
ds@1330	3	* =============================================================================
dvander@1231	4	* SourceMod
ds@2863	5	* Copyright (C) 2004-2009 AlliedModders LLC. All rights reserved.
ds@1330	6	* =============================================================================
dvander@547	7	*
ds@1330	8	* This program is free software; you can redistribute it and/or modify it under
ds@1330	9	* the terms of the GNU General Public License, version 3.0, as published by the
ds@1330	10	* Free Software Foundation.
dvander@1231	11	*
ds@1330	12	* This program is distributed in the hope that it will be useful, but WITHOUT
ds@1330	13	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
ds@1330	14	* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
ds@1330	15	* details.
dvander@1231	16	*
ds@1330	17	* You should have received a copy of the GNU General Public License along with
ds@1330	18	* this program. If not, see <http://www.gnu.org/licenses/>.
dvander@1231	19	*
ds@1330	20	* As a special exception, AlliedModders LLC gives you permission to link the
ds@1330	21	* code of this program (as well as its derivative works) to "Half-Life 2," the
ds@1330	22	* "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
ds@1330	23	* by the Valve Corporation. You must obey the GNU General Public License in
ds@1330	24	* all respects for all other code used. Additionally, AlliedModders LLC grants
ds@1330	25	* this exception to all derivative works. AlliedModders LLC defines further
ds@1330	26	* exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
ds@1330	27	* or <http://www.sourcemod.net/license.php>.
dvander@547	28	*/
dvander@547	29
dvander@547	30	#include <string.h>
dvander@547	31	#include <stdlib.h>
dvander@592	32	#include "GameConfigs.h"
dvander@547	33	#include "sm_stringutil.h"
dvander@547	34	#include "sourcemod.h"
dvander@547	35	#include "sourcemm_api.h"
dvander@592	36	#include "HalfLife2.h"
dvander@592	37	#include "Logger.h"
dvander@547	38	#include "ShareSys.h"
dvander@947	39	#include "MemoryUtils.h"
dvander@947	40	#include "LibrarySys.h"
dvander@949	41	#include "HandleSys.h"
dvander@2906	42	#include "logic_bridge.h"
dvander@947	43
dvander@947	44	#if defined PLATFORM_LINUX
dvander@947	45	#include <dlfcn.h>
dvander@947	46	#endif
dvander@547	47
dvander@616	48	GameConfigManager g_GameConfigs;
dvander@547	49	IGameConfig *g_pGameConf = NULL;
dvander@2628	50	static char g_Game[256];
dvander@2628	51	static char g_GameDesc[256] = {'!', '\0'};
dvander@2628	52	static char g_GameName[256] = {'$', '\0'};
dvander@547	53
dvander@547	54	#define PSTATE_NONE 0
dvander@547	55	#define PSTATE_GAMES 1
dvander@547	56	#define PSTATE_GAMEDEFS 2
dvander@547	57	#define PSTATE_GAMEDEFS_OFFSETS 3
dvander@547	58	#define PSTATE_GAMEDEFS_OFFSETS_OFFSET 4
dvander@548	59	#define PSTATE_GAMEDEFS_KEYS 5
dvander@947	60	#define PSTATE_GAMEDEFS_SUPPORTED 6
dvander@947	61	#define PSTATE_GAMEDEFS_SIGNATURES 7
dvander@947	62	#define PSTATE_GAMEDEFS_SIGNATURES_SIG 8
dvander@1307	63	#define PSTATE_GAMEDEFS_CRC 9
dvander@1307	64	#define PSTATE_GAMEDEFS_CRC_BINARY 10
pred@2217	65	#define PSTATE_GAMEDEFS_CUSTOM 11
downtown1@2901	66	#define PSTATE_GAMEDEFS_ADDRESSES 12
downtown1@2901	67	#define PSTATE_GAMEDEFS_ADDRESSES_ADDRESS 13
downtown1@2901	68	#define PSTATE_GAMEDEFS_ADDRESSES_ADDRESS_READ 14
dvander@547	69
dvander@547	70	#if defined PLATFORM_WINDOWS
dvander@1307	71	#define PLATFORM_NAME "windows"
dvander@1307	72	#define PLATFORM_SERVER_BINARY "server.dll"
dvander@547	73	#elif defined PLATFORM_LINUX
dvander@1307	74	#define PLATFORM_NAME "linux"
dvander@1307	75	#define PLATFORM_SERVER_BINARY "server_i486.so"
dvander@2471	76	#elif defined PLATFORM_APPLE
dvander@2471	77	#define PLATFORM_NAME "undef"
dvander@2471	78	#define PLATFORM_SERVER_BINARY "undef.dylib"
dvander@547	79	#endif
dvander@547	80
dvander@947	81	struct TempSigInfo
dvander@947	82	{
dvander@947	83	void Reset()
dvander@947	84	{
dvander@947	85	library[0] = '\0';
dvander@947	86	sig[0] = '\0';
dvander@947	87	}
dvander@947	88	char sig[512];
dvander@947	89	char library[64];
dvander@947	90	} s_TempSig;
dvander@1307	91	unsigned int s_ServerBinCRC;
dvander@1307	92	bool s_ServerBinCRC_Ok = false;
dvander@947	93
dvander@2524	94	static bool DoesGameMatch(const char *value)
dvander@2524	95	{
dvander@2828	96	#if defined PLATFORM_WINDOWS
dvander@2828	97	if (strcasecmp(value, g_Game) == 0 \|\|
dvander@2828	98	#else
dvander@2524	99	if (strcmp(value, g_Game) == 0 \|\|
dvander@2828	100	#endif
dvander@2524	101	strcmp(value, g_GameDesc) == 0 \|\|
dvander@2524	102	strcmp(value, g_GameName) == 0)
dvander@2524	103	{
dvander@2524	104	return true;
dvander@2524	105	}
dvander@2524	106	return false;
dvander@2524	107	}
dvander@2524	108
dvander@2524	109	static bool DoesEngineMatch(const char *value)
dvander@2524	110	{
dvander@2524	111	#if SOURCE_ENGINE == SE_EPISODEONE
dvander@2524	112	if (strcmp(value, "original") == 0)
ds@2583	113	#elif SOURCE_ENGINE == SE_DARKMESSIAH
ds@2583	114	if (strcmp(value, "darkmessiah") == 0)
dvander@2524	115	#elif SOURCE_ENGINE == SE_ORANGEBOX
dvander@2524	116	if (strcmp(value, "orangebox") == 0)
pred@2768	117	#elif SOURCE_ENGINE == SE_ORANGEBOXVALVE
pred@2768	118	if (strcmp(value, "orangebox_valve") == 0)
dvander@2524	119	#elif SOURCE_ENGINE == SE_LEFT4DEAD
dvander@2524	120	if (strcmp(value, "left4dead") == 0)
fyrenmoo@2847	121	#elif SOURCE_ENGINE == SE_LEFT4DEAD2
fyrenmoo@2847	122	if (strcmp(value, "left4dead2") == 0)
dvander@2524	123	#else
dvander@2524	124	#error "Unknown engine type"
dvander@2524	125	#endif
dvander@2524	126	{
dvander@2524	127	return true;
dvander@2524	128	}
dvander@2524	129	return false;
dvander@2524	130	}
dvander@2524	131
dvander@547	132	CGameConfig::CGameConfig(const char *file)
dvander@547	133	{
dvander@2524	134	strncopy(m_File, file, sizeof(m_File));
dvander@547	135	m_pOffsets = sm_trie_create();
dvander@547	136	m_pProps = sm_trie_create();
dvander@548	137	m_pKeys = sm_trie_create();
downtown1@2901	138	m_pAddresses = new KTrie<AddressConf>();
dvander@947	139	m_pSigs = sm_trie_create();
dvander@548	140	m_pStrings = new BaseStringTable(512);
faluco@641	141	m_RefCount = 0;
pred@2217	142
pred@2217	143	m_CustomLevel = 0;
pred@2217	144	m_CustomHandler = NULL;
dvander@547	145	}
dvander@547	146
dvander@547	147	CGameConfig::~CGameConfig()
dvander@547	148	{
dvander@547	149	sm_trie_destroy(m_pOffsets);
dvander@547	150	sm_trie_destroy(m_pProps);
dvander@548	151	sm_trie_destroy(m_pKeys);
downtown1@2901	152	delete m_pAddresses;
dvander@947	153	sm_trie_destroy(m_pSigs);
dvander@548	154	delete m_pStrings;
dvander@547	155	}
dvander@547	156
dvander@1656	157	SMCResult CGameConfig::ReadSMC_NewSection(const SMCStates states, const char name)
dvander@547	158	{
dvander@547	159	if (m_IgnoreLevel)
dvander@547	160	{
dvander@547	161	m_IgnoreLevel++;
dvander@1656	162	return SMCResult_Continue;
dvander@547	163	}
dvander@947	164
dvander@547	165	switch (m_ParseState)
dvander@547	166	{
dvander@547	167	case PSTATE_NONE:
dvander@547	168	{
dvander@547	169	if (strcmp(name, "Games") == 0)
dvander@547	170	{
dvander@547	171	m_ParseState = PSTATE_GAMES;
dvander@547	172	} else {
dvander@547	173	m_IgnoreLevel++;
dvander@547	174	}
dvander@547	175	break;
dvander@547	176	}
dvander@547	177	case PSTATE_GAMES:
dvander@547	178	{
dvander@2524	179	if (strcmp(name, "*") == 0 \|\|
dvander@2524	180	strcmp(name, "#default") == 0 \|\|
dvander@2524	181	DoesGameMatch(name))
dvander@547	182	{
dvander@1307	183	bShouldBeReadingDefault = true;
dvander@547	184	m_ParseState = PSTATE_GAMEDEFS;
ds@1064	185	strncopy(m_Game, name, sizeof(m_Game));
dvander@547	186	} else {
dvander@547	187	m_IgnoreLevel++;
dvander@547	188	}
dvander@547	189	break;
dvander@547	190	}
dvander@547	191	case PSTATE_GAMEDEFS:
dvander@547	192	{
dvander@547	193	if (strcmp(name, "Offsets") == 0)
dvander@547	194	{
dvander@547	195	m_ParseState = PSTATE_GAMEDEFS_OFFSETS;
dvander@947	196	}
dvander@947	197	else if (strcmp(name, "Keys") == 0)
dvander@947	198	{
dvander@548	199	m_ParseState = PSTATE_GAMEDEFS_KEYS;
dvander@947	200	}
dvander@2524	201	else if ((strcmp(name, "#supported") == 0) && (strcmp(m_Game, "#default") == 0))
dvander@947	202	{
dvander@947	203	m_ParseState = PSTATE_GAMEDEFS_SUPPORTED;
dvander@947	204	/* Ignore this section unless we get a game. */
dvander@947	205	bShouldBeReadingDefault = false;
dvander@2524	206	had_game = false;
dvander@2524	207	matched_game = false;
dvander@2524	208	had_engine = false;
dvander@2524	209	matched_engine = false;
dvander@947	210	}
dvander@947	211	else if (strcmp(name, "Signatures") == 0)
dvander@947	212	{
dvander@947	213	m_ParseState = PSTATE_GAMEDEFS_SIGNATURES;
dvander@947	214	}
dvander@1307	215	else if (strcmp(name, "CRC") == 0)
dvander@1307	216	{
dvander@1307	217	m_ParseState = PSTATE_GAMEDEFS_CRC;
dvander@1307	218	bShouldBeReadingDefault = false;
dvander@1307	219	}
downtown1@2901	220	else if (strcmp(name, "Addresses") == 0)
downtown1@2901	221	{
downtown1@2901	222	m_ParseState = PSTATE_GAMEDEFS_ADDRESSES;
downtown1@2901	223	}
dvander@947	224	else
dvander@947	225	{
pred@2217	226	ITextListener_SMC **pListen = g_GameConfigs.m_customHandlers.retrieve(name);
pred@2217	227
pred@2217	228	if (pListen != NULL)
pred@2217	229	{
pred@2217	230	m_CustomLevel = 0;
pred@2217	231	m_ParseState = PSTATE_GAMEDEFS_CUSTOM;
pred@2217	232	m_CustomHandler = *pListen;
pred@2217	233	m_CustomHandler->ReadSMC_ParseStart();
pred@2217	234	break;
pred@2217	235	}
pred@2217	236
dvander@547	237	m_IgnoreLevel++;
dvander@547	238	}
dvander@547	239	break;
dvander@547	240	}
dvander@547	241	case PSTATE_GAMEDEFS_OFFSETS:
dvander@547	242	{
ds@1064	243	m_Prop[0] = '\0';
ds@1064	244	m_Class[0] = '\0';
dvander@547	245	strncopy(m_offset, name, sizeof(m_offset));
dvander@547	246	m_ParseState = PSTATE_GAMEDEFS_OFFSETS_OFFSET;
dvander@547	247	break;
dvander@547	248	}
dvander@947	249	case PSTATE_GAMEDEFS_SIGNATURES:
dvander@947	250	{
dvander@947	251	strncopy(m_offset, name, sizeof(m_offset));
dvander@947	252	s_TempSig.Reset();
dvander@947	253	m_ParseState = PSTATE_GAMEDEFS_SIGNATURES_SIG;
dvander@947	254	break;
dvander@947	255	}
dvander@1307	256	case PSTATE_GAMEDEFS_CRC:
dvander@1307	257	{
dvander@1307	258	char error[255];
dvander@1307	259	error[0] = '\0';
dvander@1307	260	if (strcmp(name, "server") != 0)
dvander@1307	261	{
dvander@1307	262	UTIL_Format(error, sizeof(error), "Unrecognized library \"%s\"", name);
dvander@1307	263	}
dvander@1307	264	else if (!s_ServerBinCRC_Ok)
dvander@1307	265	{
dvander@1307	266	FILE *fp;
dvander@1307	267	char path[PLATFORM_MAX_PATH];
dvander@1307	268
dvander@1307	269	g_SourceMod.BuildPath(Path_Game, path, sizeof(path), "bin/" PLATFORM_SERVER_BINARY);
dvander@1307	270	if ((fp = fopen(path, "rb")) == NULL)
dvander@1307	271	{
dvander@1307	272	UTIL_Format(error, sizeof(error), "Could not open binary: %s", path);
dvander@1307	273	} else {
dvander@1307	274	size_t size;
dvander@1307	275	void *buffer;
dvander@1307	276
dvander@1307	277	fseek(fp, 0, SEEK_END);
dvander@1307	278	size = ftell(fp);
dvander@1307	279	fseek(fp, 0, SEEK_SET);
dvander@1307	280
dvander@1307	281	buffer = malloc(size);
dvander@1307	282	fread(buffer, size, 1, fp);
dvander@1307	283	s_ServerBinCRC = UTIL_CRC32(buffer, size);
dvander@1307	284	free(buffer);
dvander@1307	285	s_ServerBinCRC_Ok = true;
pred@2219	286	fclose(fp);
dvander@1307	287	}
dvander@1307	288	}
dvander@1307	289	if (error[0] != '\0')
dvander@1307	290	{
dvander@1307	291	m_IgnoreLevel = 1;
dvander@2524	292	g_Logger.LogError("[SM] Error while parsing CRC section for \"%s\" (%s):", m_Game, m_CurFile);
dvander@1307	293	g_Logger.LogError("[SM] %s", error);
dvander@1307	294	} else {
dvander@1307	295	m_ParseState = PSTATE_GAMEDEFS_CRC_BINARY;
dvander@1307	296	}
dvander@1307	297	break;
dvander@1307	298	}
pred@2217	299	case PSTATE_GAMEDEFS_CUSTOM:
pred@2217	300	{
pred@2217	301	m_CustomLevel++;
pred@2217	302	return m_CustomHandler->ReadSMC_NewSection(states, name);
pred@2217	303	break;
pred@2217	304	}
downtown1@2901	305	case PSTATE_GAMEDEFS_ADDRESSES:
downtown1@2901	306	{
downtown1@2901	307	m_Address[0] = '\0';
downtown1@2901	308	m_AddressSignature[0] = '\0';
downtown1@2901	309	m_AddressReadCount = 0;
downtown1@2901	310
downtown1@2901	311	strncopy(m_Address, name, sizeof(m_Address));
downtown1@2901	312	m_ParseState = PSTATE_GAMEDEFS_ADDRESSES_ADDRESS;
downtown1@2901	313
downtown1@2901	314	break;
downtown1@2901	315	}
downtown1@2901	316	case PSTATE_GAMEDEFS_ADDRESSES_ADDRESS:
downtown1@2901	317	{
downtown1@2901	318	if (strcmp(name, PLATFORM_NAME) == 0)
downtown1@2901	319	{
downtown1@2901	320	m_ParseState = PSTATE_GAMEDEFS_ADDRESSES_ADDRESS_READ;
downtown1@2901	321	}
downtown1@2901	322	else
downtown1@2901	323	{
downtown1@2901	324	if (strcmp(name, "linux") != 0 && strcmp(name, "windows") != 0)
downtown1@2901	325	{
downtown1@2901	326	g_Logger.LogError("[SM] Error while parsing Address section for \"%s\" (%s):", m_Address, m_CurFile);
downtown1@2901	327	g_Logger.LogError("[SM] Unrecognized platform \"%s\"", name);
downtown1@2901	328	}
downtown1@2901	329	m_IgnoreLevel = 1;
downtown1@2901	330	}
downtown1@2901	331	break;
downtown1@2901	332	}
dvander@547	333	/* No sub-sections allowed:
dvander@547	334	case PSTATE_GAMEDEFS_OFFSETS_OFFSET:
dvander@548	335	case PSTATE_GAMEDEFS_KEYS:
dvander@947	336	case PSTATE_GAMEDEFS_SUPPORTED:
dvander@947	337	case PSTATE_GAMEDEFS_SIGNATURES_SIG:
dvander@1307	338	case PSTATE_GAMEDEFS_CRC_BINARY:
downtown1@2901	339	case PSTATE_GAMEDEFS_ADDRESSES_ADDRESS_READ:
dvander@547	340	*/
dvander@547	341	default:
dvander@547	342	{
dvander@547	343	/* If we don't know what we got, start ignoring */
dvander@547	344	m_IgnoreLevel++;
dvander@547	345	break;
dvander@547	346	}
dvander@547	347	}
dvander@547	348
dvander@1656	349	return SMCResult_Continue;
dvander@547	350	}
dvander@547	351
dvander@1656	352	SMCResult CGameConfig::ReadSMC_KeyValue(const SMCStates states, const char key, const char *value)
dvander@547	353	{
dvander@547	354	if (m_IgnoreLevel)
dvander@547	355	{
dvander@1656	356	return SMCResult_Continue;
dvander@547	357	}
dvander@547	358
dvander@547	359	if (m_ParseState == PSTATE_GAMEDEFS_OFFSETS_OFFSET)
dvander@547	360	{
dvander@547	361	if (strcmp(key, "class") == 0)
dvander@547	362	{
ds@1064	363	strncopy(m_Class, value, sizeof(m_Class));
dvander@547	364	} else if (strcmp(key, "prop") == 0) {
ds@1064	365	strncopy(m_Prop, value, sizeof(m_Prop));
dvander@547	366	} else if (strcmp(key, PLATFORM_NAME) == 0) {
dvander@547	367	int val = atoi(value);
dvander@1057	368	sm_trie_replace(m_pOffsets, m_offset, (void *)val);
dvander@547	369	}
dvander@548	370	} else if (m_ParseState == PSTATE_GAMEDEFS_KEYS) {
dvander@548	371	int id = m_pStrings->AddString(value);
dvander@1057	372	sm_trie_replace(m_pKeys, key, (void *)id);
dvander@947	373	} else if (m_ParseState == PSTATE_GAMEDEFS_SUPPORTED) {
dvander@2524	374	if (strcmp(key, "game") == 0)
dvander@947	375	{
dvander@2524	376	had_game = true;
dvander@2524	377	if (DoesGameMatch(value))
dvander@2524	378	{
dvander@2524	379	matched_game = true;
dvander@2524	380	}
dvander@2524	381	if ((!had_engine && matched_game) \|\| (matched_engine && matched_game))
dvander@2524	382	{
dvander@2524	383	bShouldBeReadingDefault = true;
dvander@2524	384	}
dvander@2524	385	}
dvander@2524	386	else if (strcmp(key, "engine") == 0)
dvander@2524	387	{
dvander@2524	388	had_engine = true;
dvander@2524	389	if (DoesEngineMatch(value))
dvander@2524	390	{
dvander@2524	391	matched_engine = true;
dvander@2524	392	}
dvander@2524	393	if ((!had_game && matched_engine) \|\| (matched_game && matched_engine))
dvander@2524	394	{
dvander@2524	395	bShouldBeReadingDefault = true;
dvander@2524	396	}
dvander@947	397	}
dvander@947	398	} else if (m_ParseState == PSTATE_GAMEDEFS_SIGNATURES_SIG) {
dvander@947	399	if (strcmp(key, PLATFORM_NAME) == 0)
dvander@947	400	{
dvander@947	401	strncopy(s_TempSig.sig, value, sizeof(s_TempSig.sig));
dvander@947	402	} else if (strcmp(key, "library") == 0) {
dvander@947	403	strncopy(s_TempSig.library, value, sizeof(s_TempSig.library));
dvander@947	404	}
dvander@1307	405	} else if (m_ParseState == PSTATE_GAMEDEFS_CRC_BINARY) {
dvander@1307	406	if (strcmp(key, PLATFORM_NAME) == 0
dvander@1307	407	&& s_ServerBinCRC_Ok
dvander@1307	408	&& !bShouldBeReadingDefault)
dvander@1307	409	{
dvander@1307	410	unsigned int crc = 0;
dvander@1307	411	sscanf(value, "%08X", &crc);
dvander@1307	412	if (s_ServerBinCRC == crc)
dvander@1307	413	{
dvander@1307	414	bShouldBeReadingDefault = true;
dvander@1307	415	}
dvander@1307	416	}
downtown1@2901	417	} else if (m_ParseState == PSTATE_GAMEDEFS_ADDRESSES_ADDRESS \|\| m_ParseState == PSTATE_GAMEDEFS_ADDRESSES_ADDRESS_READ) {
downtown1@2901	418	if (strcmp(key, "read") == 0) {
downtown1@2901	419	int limit = sizeof(m_AddressRead)/sizeof(m_AddressRead[0]);
downtown1@2901	420	if (m_AddressReadCount < limit)
downtown1@2901	421	{
downtown1@2901	422	m_AddressRead[m_AddressReadCount] = atoi(value);
downtown1@2901	423	m_AddressReadCount++;
downtown1@2901	424	}
downtown1@2901	425	else
downtown1@2901	426	{
downtown1@2901	427	g_Logger.LogError("[SM] Error parsing Address \"%s\", does not support more than %d read offsets (gameconf \"%s\")", m_Address, limit, m_CurFile);
downtown1@2901	428	}
downtown1@2901	429	} else if (strcmp(key, "signature") == 0) {
downtown1@2901	430	strncopy(m_AddressSignature, value, sizeof(m_AddressSignature));
downtown1@2901	431	}
pred@2217	432	} else if (m_ParseState == PSTATE_GAMEDEFS_CUSTOM) {
pred@2217	433	return m_CustomHandler->ReadSMC_KeyValue(states, key, value);
dvander@547	434	}
dvander@547	435
dvander@1656	436	return SMCResult_Continue;
dvander@547	437	}
dvander@547	438
dvander@1656	439	SMCResult CGameConfig::ReadSMC_LeavingSection(const SMCStates *states)
dvander@547	440	{
dvander@547	441	if (m_IgnoreLevel)
dvander@547	442	{
dvander@547	443	m_IgnoreLevel--;
dvander@1656	444	return SMCResult_Continue;
dvander@547	445	}
dvander@547	446
pred@2217	447	if (m_CustomLevel)
pred@2217	448	{
pred@2217	449	m_CustomLevel--;
pred@2217	450	m_CustomHandler->ReadSMC_LeavingSection(states);
pred@2217	451	return SMCResult_Continue;
pred@2217	452	}
pred@2217	453
dvander@547	454	switch (m_ParseState)
dvander@547	455	{
dvander@547	456	case PSTATE_GAMES:
dvander@547	457	{
dvander@547	458	m_ParseState = PSTATE_NONE;
dvander@547	459	break;
dvander@547	460	}
dvander@547	461	case PSTATE_GAMEDEFS:
dvander@547	462	{
dvander@547	463	m_ParseState = PSTATE_GAMES;
dvander@547	464	break;
dvander@547	465	}
pred@2217	466	case PSTATE_GAMEDEFS_CUSTOM:
pred@2217	467	{
pred@2217	468	m_ParseState = PSTATE_GAMEDEFS;
pred@2217	469	m_CustomHandler->ReadSMC_ParseEnd(false, false);
pred@2217	470	break;
pred@2217	471	}
dvander@548	472	case PSTATE_GAMEDEFS_KEYS:
dvander@547	473	case PSTATE_GAMEDEFS_OFFSETS:
dvander@547	474	{
dvander@547	475	m_ParseState = PSTATE_GAMEDEFS;
dvander@547	476	break;
dvander@547	477	}
dvander@547	478	case PSTATE_GAMEDEFS_OFFSETS_OFFSET:
dvander@547	479	{
dvander@547	480	/* Parse the offset... */
ds@1064	481	if (m_Class[0] != '\0'
ds@1064	482	&& m_Prop[0] != '\0')
dvander@547	483	{
ds@1064	484	SendProp *pProp = g_HL2.FindInSendTable(m_Class, m_Prop);
dvander@547	485	if (pProp)
dvander@547	486	{
dvander@547	487	int val = pProp->GetOffset();
dvander@1057	488	sm_trie_replace(m_pOffsets, m_offset, (void *)val);
dvander@1057	489	sm_trie_replace(m_pProps, m_offset, pProp);
dvander@547	490	} else {
dvander@547	491	/* Check if it's a non-default game and no offsets exist */
ds@1064	492	if (((strcmp(m_Game, "*") != 0) && strcmp(m_Game, "#default") != 0)
dvander@547	493	&& (!sm_trie_retrieve(m_pOffsets, m_offset, NULL)))
dvander@547	494	{
dvander@547	495	g_Logger.LogError("[SM] Unable to find property %s.%s (file \"%s\") (mod \"%s\")",
ds@1064	496	m_Class,
ds@1064	497	m_Prop,
dvander@2524	498	m_CurFile,
ds@1064	499	m_Game);
dvander@547	500	}
dvander@547	501	}
dvander@547	502	}
dvander@547	503	m_ParseState = PSTATE_GAMEDEFS_OFFSETS;
dvander@547	504	break;
dvander@547	505	}
dvander@1307	506	case PSTATE_GAMEDEFS_CRC:
dvander@947	507	case PSTATE_GAMEDEFS_SUPPORTED:
dvander@947	508	{
dvander@947	509	if (!bShouldBeReadingDefault)
dvander@947	510	{
dvander@947	511	/* If we shouldn't read the rest of this section, set the ignore level. */
dvander@947	512	m_IgnoreLevel = 1;
ds@1043	513	m_ParseState = PSTATE_GAMES;
ds@1043	514	} else {
ds@1043	515	m_ParseState = PSTATE_GAMEDEFS;
dvander@947	516	}
dvander@947	517	break;
dvander@947	518	}
dvander@1307	519	case PSTATE_GAMEDEFS_CRC_BINARY:
dvander@1307	520	{
dvander@1307	521	m_ParseState = PSTATE_GAMEDEFS_CRC;
dvander@1307	522	break;
dvander@1307	523	}
dvander@969	524	case PSTATE_GAMEDEFS_SIGNATURES:
dvander@969	525	{
dvander@969	526	m_ParseState = PSTATE_GAMEDEFS;
dvander@969	527	break;
dvander@969	528	}
dvander@947	529	case PSTATE_GAMEDEFS_SIGNATURES_SIG:
dvander@947	530	{
dvander@947	531	if (s_TempSig.library[0] == '\0')
dvander@947	532	{
dvander@947	533	/* assume server */
dvander@947	534	strncopy(s_TempSig.library, "server", sizeof(s_TempSig.library));
dvander@947	535	}
dvander@947	536	void *addrInBase = NULL;
dvander@947	537	if (strcmp(s_TempSig.library, "server") == 0)
dvander@947	538	{
faluco@1541	539	addrInBase = (void *)g_SMAPI->GetServerFactory(false);
dvander@947	540	} else if (strcmp(s_TempSig.library, "engine") == 0) {
faluco@1541	541	addrInBase = (void *)g_SMAPI->GetEngineFactory(false);
dvander@947	542	}
dvander@947	543	void *final_addr = NULL;
dvander@947	544	if (addrInBase == NULL)
dvander@947	545	{
dvander@947	546	g_Logger.LogError("[SM] Unrecognized library \"%s\" (gameconf \"%s\")",
dvander@947	547	s_TempSig.library,
dvander@2524	548	m_CurFile);
dvander@947	549	} else {
dvander@947	550	#if defined PLATFORM_LINUX
dvander@951	551	if (s_TempSig.sig[0] == '@')
dvander@947	552	{
dvander@947	553	Dl_info info;
dvander@947	554	/* GNU only: returns 0 on error, inconsistent! >:[ */
dvander@973	555	if (dladdr(addrInBase, &info) != 0)
dvander@947	556	{
dvander@947	557	void *handle = dlopen(info.dli_fname, RTLD_NOW);
dvander@947	558	if (handle)
dvander@947	559	{
fyrenmoo@2962	560	#if (SOURCE_ENGINE == SE_ORANGEBOXVALVE) \|\| (SOURCE_ENGINE == SE_LEFT4DEAD2)
ds@2863	561	final_addr = g_MemUtils.ResolveSymbol(handle, &s_TempSig.sig[1]);
ds@2863	562	#else
dvander@951	563	final_addr = dlsym(handle, &s_TempSig.sig[1]);
ds@2863	564	#endif
dvander@947	565	dlclose(handle);
dvander@947	566	} else {
dvander@947	567	g_Logger.LogError("[SM] Unable to load library \"%s\" (gameconf \"%s\")",
dvander@947	568	s_TempSig.library,
dvander@2524	569	m_File);
dvander@947	570	}
dvander@947	571	} else {
dvander@947	572	g_Logger.LogError("[SM] Unable to find library \"%s\" in memory (gameconf \"%s\")",
dvander@947	573	s_TempSig.library,
dvander@2524	574	m_File);
dvander@947	575	}
dvander@947	576	}
dvander@947	577	if (final_addr)
dvander@947	578	{
dvander@947	579	goto skip_find;
dvander@947	580	}
dvander@947	581	#endif
dvander@947	582	/* First, preprocess the signature */
dvander@2806	583	unsigned char real_sig[511];
dvander@947	584	size_t real_bytes;
dvander@947	585	size_t length;
dvander@947	586
dvander@947	587	real_bytes = 0;
dvander@947	588	length = strlen(s_TempSig.sig);
dvander@947	589
dvander@2806	590	real_bytes = UTIL_DecodeHexString(real_sig, sizeof(real_sig), s_TempSig.sig);
dvander@947	591
dvander@1061	592	if (real_bytes >= 1)
dvander@947	593	{
dvander@2806	594	final_addr = g_MemUtils.FindPattern(addrInBase, (char*)real_sig, real_bytes);
dvander@947	595	}
dvander@947	596	}
dvander@947	597
dvander@947	598	#if defined PLATFORM_LINUX
dvander@947	599	skip_find:
dvander@947	600	#endif
dvander@1057	601	sm_trie_replace(m_pSigs, m_offset, final_addr);
dvander@964	602	m_ParseState = PSTATE_GAMEDEFS_SIGNATURES;
dvander@947	603
dvander@947	604	break;
dvander@947	605	}
downtown1@2901	606	case PSTATE_GAMEDEFS_ADDRESSES:
downtown1@2901	607	{
downtown1@2901	608	m_ParseState = PSTATE_GAMEDEFS;
downtown1@2901	609	break;
downtown1@2901	610	}
downtown1@2901	611	case PSTATE_GAMEDEFS_ADDRESSES_ADDRESS:
downtown1@2901	612	{
downtown1@2901	613	m_ParseState = PSTATE_GAMEDEFS_ADDRESSES;
downtown1@2901	614
downtown1@2901	615	if (m_Address[0] == '\0')
downtown1@2901	616	{
downtown1@2901	617	g_Logger.LogError("[SM] Address sections must have names (gameconf \"%s\")", m_CurFile);
downtown1@2901	618	break;
downtown1@2901	619	}
downtown1@2901	620	if (m_AddressSignature[0] == '\0')
downtown1@2901	621	{
downtown1@2901	622	g_Logger.LogError("[SM] Address section for \"%s\" did not specify a signature (gameconf \"%s\")", m_Address, m_CurFile);
downtown1@2901	623	break;
downtown1@2901	624	}
downtown1@2901	625
downtown1@2901	626	AddressConf addrConf(m_AddressSignature, sizeof(m_AddressSignature), m_AddressReadCount, m_AddressRead);
downtown1@2901	627	m_pAddresses->replace(m_Address, addrConf);
downtown1@2901	628
downtown1@2901	629	break;
downtown1@2901	630	}
downtown1@2901	631	case PSTATE_GAMEDEFS_ADDRESSES_ADDRESS_READ:
downtown1@2901	632	{
downtown1@2901	633	m_ParseState = PSTATE_GAMEDEFS_ADDRESSES_ADDRESS;
downtown1@2901	634	break;
downtown1@2901	635	}
dvander@547	636	}
dvander@547	637
dvander@1656	638	return SMCResult_Continue;
dvander@547	639	}
dvander@547	640
dvander@2524	641	#define MSTATE_NONE 0
dvander@2524	642	#define MSTATE_MAIN 1
dvander@2524	643	#define MSTATE_FILE 2
dvander@2524	644
dvander@2524	645	class MasterReader : public ITextListener_SMC
dvander@2524	646	{
dvander@2524	647	public:
dvander@2524	648	virtual void ReadSMC_ParseStart()
dvander@2524	649	{
dvander@2524	650	state = MSTATE_NONE;
dvander@2524	651	ignoreLevel = 0;
dvander@2524	652	}
dvander@2524	653
dvander@2524	654	virtual SMCResult ReadSMC_NewSection(const SMCStates states, const char name)
dvander@2524	655	{
dvander@2524	656	if (ignoreLevel)
dvander@2524	657	{
dvander@2524	658	return SMCResult_Continue;
dvander@2524	659	}
dvander@2524	660
dvander@2524	661	if (state == MSTATE_NONE)
dvander@2524	662	{
dvander@2524	663	if (strcmp(name, "Game Master") == 0)
dvander@2524	664	{
dvander@2524	665	state = MSTATE_MAIN;
dvander@2524	666	}
dvander@2524	667	else
dvander@2524	668	{
dvander@2524	669	ignoreLevel++;
dvander@2524	670	}
dvander@2524	671	}
dvander@2524	672	else if (state == MSTATE_MAIN)
dvander@2524	673	{
dvander@2524	674	strncopy(cur_file, name, sizeof(cur_file));
dvander@2524	675	had_engine = false;
dvander@2524	676	matched_engine = false;
dvander@2524	677	had_game = false;
dvander@2524	678	matched_game = false;
dvander@2524	679	state = MSTATE_FILE;
dvander@2524	680	}
dvander@2524	681	else if (state == MSTATE_FILE)
dvander@2524	682	{
dvander@2524	683	ignoreLevel++;
dvander@2524	684	}
dvander@2524	685
dvander@2524	686	return SMCResult_Continue;
dvander@2524	687	}
dvander@2524	688
dvander@2524	689	virtual SMCResult ReadSMC_KeyValue(const SMCStates states, const char key, const char *value)
dvander@2524	690	{
dvander@2524	691	if (ignoreLevel \|\| state != MSTATE_FILE)
dvander@2524	692	{
dvander@2524	693	return SMCResult_Continue;
dvander@2524	694	}
dvander@2524	695
dvander@2524	696	if (strcmp(key, "engine") == 0)
dvander@2524	697	{
dvander@2524	698	had_engine = true;
dvander@2524	699	if (DoesEngineMatch(value))
dvander@2524	700	{
dvander@2524	701	matched_engine = true;
dvander@2524	702	}
dvander@2524	703	}
dvander@2524	704	else if (strcmp(key, "game") == 0)
dvander@2524	705	{
dvander@2524	706	had_game = true;
dvander@2524	707	if (DoesGameMatch(value))
dvander@2524	708	{
dvander@2524	709	matched_game = true;
dvander@2524	710	}
dvander@2524	711	}
dvander@2524	712
dvander@2524	713	return SMCResult_Continue;
dvander@2524	714	}
dvander@2524	715
dvander@2524	716	virtual SMCResult ReadSMC_LeavingSection(const SMCStates *states)
dvander@2524	717	{
dvander@2524	718	if (ignoreLevel)
dvander@2524	719	{
dvander@2524	720	ignoreLevel--;
dvander@2524	721	return SMCResult_Continue;
dvander@2524	722	}
dvander@2524	723
dvander@2524	724	if (state == MSTATE_FILE)
dvander@2524	725	{
dvander@2524	726	/* The four success conditions:
dvander@2524	727	* 1. Needed nothing.
dvander@2524	728	* 2. Needed game only.
dvander@2524	729	* 3. Needed engine only.
dvander@2524	730	* 4. Needed both engine and game.
dvander@2524	731	* Final result is minimized via k-map.
dvander@2524	732	*/
dvander@2524	733	#if 0
dvander@2524	734	if ((!had_engine && !had_game) \|\|
dvander@2524	735	(!had_engine && (had_game && matched_game)) \|\|
dvander@2524	736	(!had_game && (had_engine && matched_engine)) \|\|
dvander@2524	737	((had_game && had_engine) && (matched_game && matched_engine)))
dvander@2524	738	#endif
dvander@2524	739	if ((!had_engine && !had_game) \|\|
dvander@2524	740	(!had_engine && matched_game) \|\|
dvander@2524	741	(!had_game && matched_engine) \|\|
dvander@2524	742	(matched_engine && matched_game))
dvander@2524	743	{
dvander@2524	744	fileList->push_back(cur_file);
dvander@2524	745	}
dvander@2524	746	state = MSTATE_MAIN;
dvander@2524	747	}
dvander@2524	748	else if (state == MSTATE_MAIN)
dvander@2524	749	{
dvander@2524	750	state = MSTATE_NONE;
dvander@2524	751	}
dvander@2524	752
dvander@2524	753	return SMCResult_Continue;
dvander@2524	754	}
dvander@2524	755	public:
dvander@2524	756	List<String> *fileList;
dvander@2524	757	unsigned int state;
dvander@2524	758	unsigned int ignoreLevel;
dvander@2524	759	char cur_file[PLATFORM_MAX_PATH];
dvander@2524	760	bool had_engine;
dvander@2524	761	bool matched_engine;
dvander@2524	762	bool had_game;
dvander@2524	763	bool matched_game;
dvander@2524	764	};
dvander@2524	765
dvander@2524	766	static MasterReader master_reader;
dvander@2524	767
dvander@547	768	bool CGameConfig::Reparse(char *error, size_t maxlength)
dvander@547	769	{
dvander@548	770	/* Reset cached data */
dvander@548	771	m_pStrings->Reset();
dvander@547	772	sm_trie_clear(m_pOffsets);
dvander@548	773	sm_trie_clear(m_pProps);
dvander@548	774	sm_trie_clear(m_pKeys);
downtown1@2901	775	m_pAddresses->clear();
dvander@547	776
dvander@2524	777	char path[PLATFORM_MAX_PATH];
dvander@2524	778
dvander@2524	779	/* See if we can use the extended gamedata format. */
dvander@2524	780	g_SourceMod.BuildPath(Path_SM, path, sizeof(path), "gamedata/%s/master.games.txt", m_File);
dvander@2524	781	if (!g_LibSys.PathExists(path))
dvander@2524	782	{
dvander@2524	783	/* Nope, use the old mechanism. */
dvander@2524	784	UTIL_Format(path, sizeof(path), "%s.txt", m_File);
dvander@2524	785	return EnterFile(path, error, maxlength);
dvander@2524	786	}
dvander@2524	787
dvander@2524	788	/* Otherwise, it's time to parse the master. */
dvander@2524	789	SMCError err;
dvander@2524	790	SMCStates state = {0, 0};
dvander@2524	791	List<String> fileList;
dvander@2524	792	master_reader.fileList = &fileList;
dvander@2524	793
dvander@2524	794	err = textparsers->ParseSMCFile(path, &master_reader, &state, error, maxlength);
dvander@2524	795	if (err != SMCError_Okay)
dvander@2524	796	{
dvander@2524	797	const char *msg = textparsers->GetSMCErrorString(err);
dvander@2524	798
dvander@2524	799	g_Logger.LogError("[SM] Error parsing master gameconf file \"%s\":", path);
dvander@2524	800	g_Logger.LogError("[SM] Error %d on line %d, col %d: %s",
dvander@2524	801	err,
dvander@2524	802	state.line,
dvander@2524	803	state.col,
dvander@2524	804	msg ? msg : "Unknown error");
dvander@2524	805	return false;
dvander@2524	806	}
dvander@2524	807
dvander@2524	808	/* Go through each file we found and parse it. */
dvander@2524	809	List<String>::iterator iter;
dvander@2524	810	for (iter = fileList.begin(); iter != fileList.end(); iter++)
dvander@2524	811	{
dvander@2524	812	UTIL_Format(path, sizeof(path), "%s/%s", m_File, (*iter).c_str());
dvander@2524	813	if (!EnterFile(path, error, maxlength))
dvander@2524	814	{
dvander@2524	815	return false;
dvander@2524	816	}
dvander@2524	817	}
dvander@2524	818
pred@2578	819	/* Parse the contents of the 'custom' directory */
pred@2582	820	g_SourceMod.BuildPath(Path_SM, path, sizeof(path), "gamedata/%s/custom", m_File);
pred@2578	821	IDirectory *customDir = g_LibSys.OpenDirectory(path);
pred@2578	822
pred@2578	823	if (!customDir)
pred@2578	824	{
pred@2578	825	return true;
pred@2578	826	}
pred@2578	827
pred@2578	828	while (customDir->MoreFiles())
pred@2578	829	{
pred@2578	830	if (!customDir->IsEntryFile())
pred@2578	831	{
pred@2578	832	customDir->NextEntry();
pred@2578	833	continue;
pred@2578	834	}
pred@2578	835
pred@2578	836	const char *curFile = customDir->GetEntryName();
pred@2578	837
pred@2578	838	/* Only allow .txt files */
pred@2578	839	int len = strlen(curFile);
pred@2579	840	if (len > 4 && strcmp(&curFile[len-4], ".txt") != 0)
pred@2578	841	{
pred@2578	842	customDir->NextEntry();
pred@2578	843	continue;
pred@2578	844	}
pred@2578	845
pred@2578	846	UTIL_Format(path, sizeof(path), "%s/custom/%s", m_File, curFile);
pred@2578	847	if (!EnterFile(path, error, maxlength))
pred@2578	848	{
pred@2579	849	g_LibSys.CloseDirectory(customDir);
pred@2578	850	return false;
pred@2578	851	}
pred@2578	852
pred@2578	853	customDir->NextEntry();
pred@2578	854	}
pred@2578	855
pred@2579	856	g_LibSys.CloseDirectory(customDir);
dvander@2524	857	return true;
dvander@2524	858	}
dvander@2524	859
dvander@2524	860	bool CGameConfig::EnterFile(const char file, char error, size_t maxlength)
dvander@2524	861	{
dvander@2524	862	SMCError err;
dvander@2524	863	SMCStates state = {0, 0};
dvander@2524	864
dvander@2524	865	g_SourceMod.BuildPath(Path_SM, m_CurFile, sizeof(m_CurFile), "gamedata/%s", file);
dvander@2524	866
dvander@2524	867	/* Initialize parse states */
dvander@2524	868	m_IgnoreLevel = 0;
dvander@2524	869	bShouldBeReadingDefault = true;
dvander@2524	870	m_ParseState = PSTATE_NONE;
dvander@2524	871
dvander@2524	872	if ((err=textparsers->ParseSMCFile(m_CurFile, this, &state, error, maxlength))
dvander@1656	873	!= SMCError_Okay)
dvander@547	874	{
dvander@1885	875	const char *msg;
dvander@1885	876
dvander@1885	877	msg = textparsers->GetSMCErrorString(err);
dvander@1885	878
dvander@2524	879	g_Logger.LogError("[SM] Error parsing gameconfig file \"%s\":", m_CurFile);
dvander@1885	880	g_Logger.LogError("[SM] Error %d on line %d, col %d: %s",
dvander@1885	881	err,
dvander@1885	882	state.line,
dvander@1885	883	state.col,
dvander@1885	884	msg ? msg : "Unknown error");
pred@2217	885
pred@2217	886	if (m_ParseState == PSTATE_GAMEDEFS_CUSTOM)
pred@2217	887	{
pred@2217	888	//error occurred while parsing a custom section
pred@2217	889	m_CustomHandler->ReadSMC_ParseEnd(true, true);
pred@2217	890	m_CustomHandler = NULL;
pred@2217	891	m_CustomLevel = 0;
pred@2217	892	}
pred@2217	893
dvander@547	894	return false;
dvander@547	895	}
dvander@547	896
dvander@547	897	return true;
dvander@547	898	}
dvander@547	899
dvander@547	900	bool CGameConfig::GetOffset(const char key, int value)
dvander@547	901	{
dvander@547	902	void *obj;
dvander@547	903
dvander@547	904	if (!sm_trie_retrieve(m_pOffsets, key, &obj))
dvander@547	905	{
dvander@547	906	return false;
dvander@547	907	}
dvander@547	908
dvander@547	909	*value = (int)obj;
dvander@547	910
dvander@547	911	return true;
dvander@547	912	}
dvander@547	913
dvander@547	914	const char CGameConfig::GetKeyValue(const char key)
dvander@547	915	{
dvander@548	916	void *obj;
dvander@548	917	if (!sm_trie_retrieve(m_pKeys, key, &obj))
dvander@548	918	{
dvander@548	919	return NULL;
dvander@548	920	}
dvander@548	921	return m_pStrings->GetString((int)obj);
dvander@547	922	}
dvander@547	923
downtown1@2901	924	//memory addresses below 0x10000 are automatically considered invalid for dereferencing
downtown1@2901	925	#define VALID_MINIMUM_MEMORY_ADDRESS 0x10000
downtown1@2901	926
downtown1@2901	927	bool CGameConfig::GetAddress(const char key, void *retaddr)
downtown1@2901	928	{
downtown1@2901	929	AddressConf *addrConf;
downtown1@2901	930
downtown1@2901	931	addrConf = m_pAddresses->retrieve(key);
downtown1@2901	932	if (!addrConf)
downtown1@2901	933	{
downtown1@2901	934	*retaddr = NULL;
downtown1@2901	935	return false;
downtown1@2901	936	}
downtown1@2901	937
downtown1@2901	938	void *addr;
downtown1@2901	939	if (!GetMemSig(addrConf->signatureName, &addr))
downtown1@2901	940	{
downtown1@2901	941	*retaddr = NULL;
downtown1@2901	942	return false;
downtown1@2901	943	}
downtown1@2901	944
downtown1@2901	945	for (int i = 0; i < addrConf->readCount; ++i)
downtown1@2901	946	{
downtown1@2901	947	int offset = addrConf->read[i];
downtown1@2901	948
downtown1@2901	949	//NULLs in the middle of an indirection chain are bad, end NULL is ok
downtown1@2901	950	if (addr == NULL \|\| reinterpret_cast<uintptr_t>(addr) < VALID_MINIMUM_MEMORY_ADDRESS)
downtown1@2901	951	{
downtown1@2901	952	*retaddr = NULL;
downtown1@2901	953	return false;
downtown1@2901	954	}
downtown1@2901	955	addr = (reinterpret_cast<void>(reinterpret_cast<uint8_t>(addr) + offset));
downtown1@2901	956	}
downtown1@2901	957
downtown1@2901	958	*retaddr = addr;
downtown1@2901	959	return true;
downtown1@2901	960	}
downtown1@2901	961
downtown1@2901	962	CGameConfig::AddressConf::AddressConf(char sigName, unsigned sigLength, unsigned readCount, int read)
downtown1@2901	963	{
downtown1@2901	964	unsigned readLimit = min(readCount, sizeof(this->read) / sizeof(this->read[0]));
downtown1@2901	965
downtown1@2901	966	strncopy(signatureName, sigName, sizeof(signatureName) / sizeof(signatureName[0]));
downtown1@2901	967	this->readCount = readLimit;
downtown1@2901	968	memcpy(&this->read[0], read, sizeof(this->read[0])*readLimit);
downtown1@2901	969	}
downtown1@2901	970
dvander@547	971	SendProp CGameConfig::GetSendProp(const char key)
dvander@547	972	{
dvander@547	973	SendProp *pProp;
dvander@547	974
dvander@547	975	if (!sm_trie_retrieve(m_pProps, key, (void **)&pProp))
dvander@547	976	{
dvander@547	977	return NULL;
dvander@547	978	}
dvander@547	979
dvander@547	980	return pProp;
dvander@547	981	}
dvander@547	982
dvander@947	983	bool CGameConfig::GetMemSig(const char key, void *addr)
dvander@947	984	{
dvander@947	985	return sm_trie_retrieve(m_pSigs, key, addr);
dvander@947	986	}
dvander@947	987
dvander@547	988	void CGameConfig::IncRefCount()
dvander@547	989	{
dvander@547	990	m_RefCount++;
dvander@547	991	}
dvander@547	992
dvander@547	993	unsigned int CGameConfig::DecRefCount()
dvander@547	994	{
dvander@547	995	m_RefCount--;
dvander@547	996	return m_RefCount;
dvander@547	997	}
dvander@547	998
dvander@616	999	GameConfigManager::GameConfigManager()
dvander@547	1000	{
dvander@547	1001	m_pLookup = sm_trie_create();
dvander@547	1002	}
dvander@547	1003
dvander@616	1004	GameConfigManager::~GameConfigManager()
dvander@547	1005	{
dvander@547	1006	sm_trie_destroy(m_pLookup);
dvander@547	1007	}
dvander@547	1008
dvander@616	1009	void GameConfigManager::OnSourceModStartup(bool late)
dvander@547	1010	{
dvander@547	1011	LoadGameConfigFile("core.games", &g_pGameConf, NULL, 0);
dvander@547	1012
ds@1064	1013	strncopy(g_Game, g_SourceMod.GetGameFolderName(), sizeof(g_Game));
ds@1064	1014	strncopy(g_GameDesc + 1, SERVER_CALL(GetGameDescription)(), sizeof(g_GameDesc) - 1);
dvander@1762	1015
dvander@1762	1016	KeyValues *pGameInfo = new KeyValues("GameInfo");
ds@1944	1017	if (g_HL2.KVLoadFromFile(pGameInfo, basefilesystem, "gameinfo.txt"))
dvander@1762	1018	{
dvander@1762	1019	const char *str;
dvander@1762	1020	if ((str = pGameInfo->GetString("game", NULL)) != NULL)
dvander@1762	1021	{
dvander@1762	1022	strncopy(g_GameName + 1, str, sizeof(g_GameName) - 1);
dvander@1762	1023	}
dvander@1762	1024	}
dvander@1762	1025	pGameInfo->deleteThis();
dvander@547	1026	}
dvander@547	1027
dvander@616	1028	void GameConfigManager::OnSourceModAllInitialized()
dvander@547	1029	{
dvander@547	1030	/* NOW initialize the game file */
dvander@547	1031	CGameConfig pGameConf = (CGameConfig )g_pGameConf;
dvander@547	1032
dvander@547	1033	char error[255];
dvander@547	1034	if (!pGameConf->Reparse(error, sizeof(error)))
dvander@547	1035	{
dvander@547	1036	/* :TODO: log */
dvander@547	1037	}
dvander@547	1038
dvander@547	1039	g_ShareSys.AddInterface(NULL, this);
dvander@547	1040	}
dvander@547	1041
dvander@616	1042	void GameConfigManager::OnSourceModAllShutdown()
dvander@547	1043	{
dvander@547	1044	CloseGameConfigFile(g_pGameConf);
dvander@547	1045	}
dvander@547	1046
dvander@616	1047	bool GameConfigManager::LoadGameConfigFile(const char file, IGameConfig _pConfig, char error, size_t maxlength)
dvander@547	1048	{
dvander@2517	1049	#if 0
pred@2142	1050	/* A crash was detected during last load - We block the gamedata loading so it hopefully won't happen again */
pred@2142	1051	if (g_blockGameDataLoad)
pred@2142	1052	{
pred@2142	1053	UTIL_Format(error, maxlength, "GameData loaded blocked due to detected crash");
pred@2142	1054	return false;
pred@2142	1055	}
dvander@2517	1056	#endif
pred@2142	1057
dvander@547	1058	CGameConfig *pConfig;
dvander@547	1059
dvander@547	1060	if (sm_trie_retrieve(m_pLookup, file, (void **)&pConfig))
dvander@547	1061	{
dvander@547	1062	pConfig->IncRefCount();
dvander@547	1063	*_pConfig = pConfig;
dvander@547	1064	return true;
dvander@547	1065	}
dvander@547	1066
dvander@547	1067	pConfig = new CGameConfig(file);
dvander@547	1068
dvander@547	1069	/* :HACKHACK: Don't parse the main config file */
dvander@547	1070	bool retval = true;
dvander@547	1071	if (_pConfig != &g_pGameConf)
dvander@547	1072	{
dvander@547	1073	retval = pConfig->Reparse(error, maxlength);
dvander@547	1074	}
dvander@547	1075
dvander@547	1076	m_cfgs.push_back(pConfig);
dvander@547	1077	sm_trie_insert(m_pLookup, file, pConfig);
dvander@547	1078
dvander@547	1079	pConfig->IncRefCount();
dvander@547	1080
dvander@547	1081	*_pConfig = pConfig;
dvander@547	1082
dvander@547	1083	return retval;
dvander@547	1084	}
dvander@547	1085
dvander@616	1086	void GameConfigManager::CloseGameConfigFile(IGameConfig *cfg)
dvander@547	1087	{
dvander@547	1088	CGameConfig pConfig = (CGameConfig )cfg;
dvander@547	1089
dvander@547	1090	if (pConfig->DecRefCount() == 0)
dvander@547	1091	{
dvander@2524	1092	sm_trie_delete(m_pLookup, pConfig->m_File);
dvander@547	1093	delete pConfig;
dvander@547	1094	}
dvander@547	1095	}
dvander@949	1096
dvander@949	1097	extern HandleType_t g_GameConfigsType;
dvander@949	1098
dvander@949	1099	IGameConfig GameConfigManager::ReadHandle(Handle_t hndl, IdentityToken_t ident, HandleError *err)
dvander@949	1100	{
dvander@949	1101	HandleSecurity sec(ident, g_pCoreIdent);
dvander@949	1102	IGameConfig *conf = NULL;
dvander@949	1103	HandleError _err = g_HandleSys.ReadHandle(hndl, g_GameConfigsType, &sec, (void **)&conf);
dvander@949	1104
dvander@949	1105	if (err)
dvander@949	1106	{
dvander@949	1107	*err = _err;
dvander@949	1108	}
dvander@949	1109
dvander@949	1110	return conf;
dvander@949	1111	}
pred@2217	1112
dvander@2562	1113	void GameConfigManager::AddUserConfigHook(const char sectionname, ITextListener_SMC listener)
pred@2217	1114	{
pred@2217	1115	m_customHandlers.insert(sectionname, listener);
pred@2217	1116	}
pred@2217	1117
dvander@2562	1118	void GameConfigManager::RemoveUserConfigHook(const char sectionname, ITextListener_SMC listener)
pred@2217	1119	{
pred@2217	1120	ITextListener_SMC **pListener = m_customHandlers.retrieve(sectionname);
pred@2217	1121
pred@2217	1122	if (pListener == NULL)
pred@2217	1123	{
pred@2217	1124	return;
pred@2217	1125	}
pred@2217	1126
pred@2217	1127	if (*pListener != listener)
pred@2217	1128	{
pred@2217	1129	return;
pred@2217	1130	}
pred@2217	1131
pred@2217	1132	m_customHandlers.remove(sectionname);
pred@2217	1133
pred@2217	1134	return;
pred@2217	1135	}
dvander@2562	1136
dvander@2562	1137	void GameConfigManager::AcquireLock()
dvander@2562	1138	{
dvander@2562	1139	}
dvander@2562	1140
dvander@2562	1141	void GameConfigManager::ReleaseLock()
dvander@2562	1142	{
dvander@2562	1143	}

author	Fyren <fyrenmoo@gmail.com>
	Thu Apr 29 00:08:08 2010 -0700 (2010-04-29)
changeset 2962	130639d897c2
parent 2906	33cc9854bd97
child 2999	53dfd73380c3
permissions	-rw-r--r--