dvander@547: /**
ds@653: * vim: set ts=4 :
ds@1330: * =============================================================================
dvander@1231: * SourceMod
ds@1947: * Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved.
ds@1330: * =============================================================================
dvander@547: *
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 .
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 .
dvander@547: *
dvander@547: * Version: $Id$
dvander@547: */
dvander@547:
dvander@547: #include
dvander@547: #include
dvander@592: #include "GameConfigs.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@947: #include "MemoryUtils.h"
dvander@947: #include "LibrarySys.h"
dvander@949: #include "HandleSys.h"
dvander@1307: #include "sm_crc32.h"
dvander@947:
dvander@947: #if defined PLATFORM_LINUX
dvander@947: #include
dvander@947: #endif
dvander@547:
dvander@616: GameConfigManager g_GameConfigs;
dvander@547: IGameConfig *g_pGameConf = NULL;
dvander@2628: static char g_Game[256];
dvander@2628: static char g_GameDesc[256] = {'!', '\0'};
dvander@2628: static char g_GameName[256] = {'$', '\0'};
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@947: #define PSTATE_GAMEDEFS_SUPPORTED 6
dvander@947: #define PSTATE_GAMEDEFS_SIGNATURES 7
dvander@947: #define PSTATE_GAMEDEFS_SIGNATURES_SIG 8
dvander@1307: #define PSTATE_GAMEDEFS_CRC 9
dvander@1307: #define PSTATE_GAMEDEFS_CRC_BINARY 10
pred@2217: #define PSTATE_GAMEDEFS_CUSTOM 11
dvander@547:
dvander@547: #if defined PLATFORM_WINDOWS
dvander@1307: #define PLATFORM_NAME "windows"
dvander@1307: #define PLATFORM_SERVER_BINARY "server.dll"
dvander@547: #elif defined PLATFORM_LINUX
dvander@1307: #define PLATFORM_NAME "linux"
dvander@1307: #define PLATFORM_SERVER_BINARY "server_i486.so"
dvander@2471: #elif defined PLATFORM_APPLE
dvander@2471: #define PLATFORM_NAME "undef"
dvander@2471: #define PLATFORM_SERVER_BINARY "undef.dylib"
dvander@547: #endif
dvander@547:
dvander@947: struct TempSigInfo
dvander@947: {
dvander@947: void Reset()
dvander@947: {
dvander@947: library[0] = '\0';
dvander@947: sig[0] = '\0';
dvander@947: }
dvander@947: char sig[512];
dvander@947: char library[64];
dvander@947: } s_TempSig;
dvander@1307: unsigned int s_ServerBinCRC;
dvander@1307: bool s_ServerBinCRC_Ok = false;
dvander@947:
dvander@2524: static bool DoesGameMatch(const char *value)
dvander@2524: {
dvander@2828: #if defined PLATFORM_WINDOWS
dvander@2828: if (strcasecmp(value, g_Game) == 0 ||
dvander@2828: #else
dvander@2524: if (strcmp(value, g_Game) == 0 ||
dvander@2828: #endif
dvander@2524: strcmp(value, g_GameDesc) == 0 ||
dvander@2524: strcmp(value, g_GameName) == 0)
dvander@2524: {
dvander@2524: return true;
dvander@2524: }
dvander@2524: return false;
dvander@2524: }
dvander@2524:
dvander@2524: static bool DoesEngineMatch(const char *value)
dvander@2524: {
dvander@2524: #if SOURCE_ENGINE == SE_EPISODEONE
dvander@2524: if (strcmp(value, "original") == 0)
ds@2583: #elif SOURCE_ENGINE == SE_DARKMESSIAH
ds@2583: if (strcmp(value, "darkmessiah") == 0)
dvander@2524: #elif SOURCE_ENGINE == SE_ORANGEBOX
dvander@2524: if (strcmp(value, "orangebox") == 0)
pred@2768: #elif SOURCE_ENGINE == SE_ORANGEBOXVALVE
pred@2768: if (strcmp(value, "orangebox_valve") == 0)
dvander@2524: #elif SOURCE_ENGINE == SE_LEFT4DEAD
dvander@2524: if (strcmp(value, "left4dead") == 0)
dvander@2524: #else
dvander@2524: #error "Unknown engine type"
dvander@2524: #endif
dvander@2524: {
dvander@2524: return true;
dvander@2524: }
dvander@2524: return false;
dvander@2524: }
dvander@2524:
dvander@547: CGameConfig::CGameConfig(const char *file)
dvander@547: {
dvander@2524: strncopy(m_File, file, sizeof(m_File));
dvander@547: m_pOffsets = sm_trie_create();
dvander@547: m_pProps = sm_trie_create();
dvander@548: m_pKeys = sm_trie_create();
dvander@947: m_pSigs = sm_trie_create();
dvander@548: m_pStrings = new BaseStringTable(512);
faluco@641: m_RefCount = 0;
pred@2217:
pred@2217: m_CustomLevel = 0;
pred@2217: m_CustomHandler = NULL;
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@947: sm_trie_destroy(m_pSigs);
dvander@548: delete m_pStrings;
dvander@547: }
dvander@547:
dvander@1656: SMCResult CGameConfig::ReadSMC_NewSection(const SMCStates *states, const char *name)
dvander@547: {
dvander@547: if (m_IgnoreLevel)
dvander@547: {
dvander@547: m_IgnoreLevel++;
dvander@1656: return SMCResult_Continue;
dvander@547: }
dvander@947:
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@2524: if (strcmp(name, "*") == 0 ||
dvander@2524: strcmp(name, "#default") == 0 ||
dvander@2524: DoesGameMatch(name))
dvander@547: {
dvander@1307: bShouldBeReadingDefault = true;
dvander@547: m_ParseState = PSTATE_GAMEDEFS;
ds@1064: strncopy(m_Game, name, sizeof(m_Game));
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@947: }
dvander@947: else if (strcmp(name, "Keys") == 0)
dvander@947: {
dvander@548: m_ParseState = PSTATE_GAMEDEFS_KEYS;
dvander@947: }
dvander@2524: else if ((strcmp(name, "#supported") == 0) && (strcmp(m_Game, "#default") == 0))
dvander@947: {
dvander@947: m_ParseState = PSTATE_GAMEDEFS_SUPPORTED;
dvander@947: /* Ignore this section unless we get a game. */
dvander@947: bShouldBeReadingDefault = false;
dvander@2524: had_game = false;
dvander@2524: matched_game = false;
dvander@2524: had_engine = false;
dvander@2524: matched_engine = false;
dvander@947: }
dvander@947: else if (strcmp(name, "Signatures") == 0)
dvander@947: {
dvander@947: m_ParseState = PSTATE_GAMEDEFS_SIGNATURES;
dvander@947: }
dvander@1307: else if (strcmp(name, "CRC") == 0)
dvander@1307: {
dvander@1307: m_ParseState = PSTATE_GAMEDEFS_CRC;
dvander@1307: bShouldBeReadingDefault = false;
dvander@1307: }
dvander@947: else
dvander@947: {
pred@2217: ITextListener_SMC **pListen = g_GameConfigs.m_customHandlers.retrieve(name);
pred@2217:
pred@2217: if (pListen != NULL)
pred@2217: {
pred@2217: m_CustomLevel = 0;
pred@2217: m_ParseState = PSTATE_GAMEDEFS_CUSTOM;
pred@2217: m_CustomHandler = *pListen;
pred@2217: m_CustomHandler->ReadSMC_ParseStart();
pred@2217: break;
pred@2217: }
pred@2217:
dvander@547: m_IgnoreLevel++;
dvander@547: }
dvander@547: break;
dvander@547: }
dvander@547: case PSTATE_GAMEDEFS_OFFSETS:
dvander@547: {
ds@1064: m_Prop[0] = '\0';
ds@1064: 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@947: case PSTATE_GAMEDEFS_SIGNATURES:
dvander@947: {
dvander@947: strncopy(m_offset, name, sizeof(m_offset));
dvander@947: s_TempSig.Reset();
dvander@947: m_ParseState = PSTATE_GAMEDEFS_SIGNATURES_SIG;
dvander@947: break;
dvander@947: }
dvander@1307: case PSTATE_GAMEDEFS_CRC:
dvander@1307: {
dvander@1307: char error[255];
dvander@1307: error[0] = '\0';
dvander@1307: if (strcmp(name, "server") != 0)
dvander@1307: {
dvander@1307: UTIL_Format(error, sizeof(error), "Unrecognized library \"%s\"", name);
dvander@1307: }
dvander@1307: else if (!s_ServerBinCRC_Ok)
dvander@1307: {
dvander@1307: FILE *fp;
dvander@1307: char path[PLATFORM_MAX_PATH];
dvander@1307:
dvander@1307: g_SourceMod.BuildPath(Path_Game, path, sizeof(path), "bin/" PLATFORM_SERVER_BINARY);
dvander@1307: if ((fp = fopen(path, "rb")) == NULL)
dvander@1307: {
dvander@1307: UTIL_Format(error, sizeof(error), "Could not open binary: %s", path);
dvander@1307: } else {
dvander@1307: size_t size;
dvander@1307: void *buffer;
dvander@1307:
dvander@1307: fseek(fp, 0, SEEK_END);
dvander@1307: size = ftell(fp);
dvander@1307: fseek(fp, 0, SEEK_SET);
dvander@1307:
dvander@1307: buffer = malloc(size);
dvander@1307: fread(buffer, size, 1, fp);
dvander@1307: s_ServerBinCRC = UTIL_CRC32(buffer, size);
dvander@1307: free(buffer);
dvander@1307: s_ServerBinCRC_Ok = true;
pred@2219: fclose(fp);
dvander@1307: }
dvander@1307: }
dvander@1307: if (error[0] != '\0')
dvander@1307: {
dvander@1307: m_IgnoreLevel = 1;
dvander@2524: g_Logger.LogError("[SM] Error while parsing CRC section for \"%s\" (%s):", m_Game, m_CurFile);
dvander@1307: g_Logger.LogError("[SM] %s", error);
dvander@1307: } else {
dvander@1307: m_ParseState = PSTATE_GAMEDEFS_CRC_BINARY;
dvander@1307: }
dvander@1307: break;
dvander@1307: }
pred@2217: case PSTATE_GAMEDEFS_CUSTOM:
pred@2217: {
pred@2217: m_CustomLevel++;
pred@2217: return m_CustomHandler->ReadSMC_NewSection(states, name);
pred@2217: break;
pred@2217: }
dvander@547: /* No sub-sections allowed:
dvander@547: case PSTATE_GAMEDEFS_OFFSETS_OFFSET:
dvander@548: case PSTATE_GAMEDEFS_KEYS:
dvander@947: case PSTATE_GAMEDEFS_SUPPORTED:
dvander@947: case PSTATE_GAMEDEFS_SIGNATURES_SIG:
dvander@1307: case PSTATE_GAMEDEFS_CRC_BINARY:
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@1656: return SMCResult_Continue;
dvander@547: }
dvander@547:
dvander@1656: SMCResult CGameConfig::ReadSMC_KeyValue(const SMCStates *states, const char *key, const char *value)
dvander@547: {
dvander@547: if (m_IgnoreLevel)
dvander@547: {
dvander@1656: return SMCResult_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: {
ds@1064: strncopy(m_Class, value, sizeof(m_Class));
dvander@547: } else if (strcmp(key, "prop") == 0) {
ds@1064: strncopy(m_Prop, value, sizeof(m_Prop));
dvander@547: } else if (strcmp(key, PLATFORM_NAME) == 0) {
dvander@547: int val = atoi(value);
dvander@1057: sm_trie_replace(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@1057: sm_trie_replace(m_pKeys, key, (void *)id);
dvander@947: } else if (m_ParseState == PSTATE_GAMEDEFS_SUPPORTED) {
dvander@2524: if (strcmp(key, "game") == 0)
dvander@947: {
dvander@2524: had_game = true;
dvander@2524: if (DoesGameMatch(value))
dvander@2524: {
dvander@2524: matched_game = true;
dvander@2524: }
dvander@2524: if ((!had_engine && matched_game) || (matched_engine && matched_game))
dvander@2524: {
dvander@2524: bShouldBeReadingDefault = true;
dvander@2524: }
dvander@2524: }
dvander@2524: else if (strcmp(key, "engine") == 0)
dvander@2524: {
dvander@2524: had_engine = true;
dvander@2524: if (DoesEngineMatch(value))
dvander@2524: {
dvander@2524: matched_engine = true;
dvander@2524: }
dvander@2524: if ((!had_game && matched_engine) || (matched_game && matched_engine))
dvander@2524: {
dvander@2524: bShouldBeReadingDefault = true;
dvander@2524: }
dvander@947: }
dvander@947: } else if (m_ParseState == PSTATE_GAMEDEFS_SIGNATURES_SIG) {
dvander@947: if (strcmp(key, PLATFORM_NAME) == 0)
dvander@947: {
dvander@947: strncopy(s_TempSig.sig, value, sizeof(s_TempSig.sig));
dvander@947: } else if (strcmp(key, "library") == 0) {
dvander@947: strncopy(s_TempSig.library, value, sizeof(s_TempSig.library));
dvander@947: }
dvander@1307: } else if (m_ParseState == PSTATE_GAMEDEFS_CRC_BINARY) {
dvander@1307: if (strcmp(key, PLATFORM_NAME) == 0
dvander@1307: && s_ServerBinCRC_Ok
dvander@1307: && !bShouldBeReadingDefault)
dvander@1307: {
dvander@1307: unsigned int crc = 0;
dvander@1307: sscanf(value, "%08X", &crc);
dvander@1307: if (s_ServerBinCRC == crc)
dvander@1307: {
dvander@1307: bShouldBeReadingDefault = true;
dvander@1307: }
dvander@1307: }
pred@2217: } else if (m_ParseState == PSTATE_GAMEDEFS_CUSTOM) {
pred@2217: return m_CustomHandler->ReadSMC_KeyValue(states, key, value);
dvander@547: }
dvander@547:
dvander@1656: return SMCResult_Continue;
dvander@547: }
dvander@547:
dvander@1656: SMCResult CGameConfig::ReadSMC_LeavingSection(const SMCStates *states)
dvander@547: {
dvander@547: if (m_IgnoreLevel)
dvander@547: {
dvander@547: m_IgnoreLevel--;
dvander@1656: return SMCResult_Continue;
dvander@547: }
dvander@547:
pred@2217: if (m_CustomLevel)
pred@2217: {
pred@2217: m_CustomLevel--;
pred@2217: m_CustomHandler->ReadSMC_LeavingSection(states);
pred@2217: return SMCResult_Continue;
pred@2217: }
pred@2217:
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: }
pred@2217: case PSTATE_GAMEDEFS_CUSTOM:
pred@2217: {
pred@2217: m_ParseState = PSTATE_GAMEDEFS;
pred@2217: m_CustomHandler->ReadSMC_ParseEnd(false, false);
pred@2217: break;
pred@2217: }
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... */
ds@1064: if (m_Class[0] != '\0'
ds@1064: && m_Prop[0] != '\0')
dvander@547: {
ds@1064: SendProp *pProp = g_HL2.FindInSendTable(m_Class, m_Prop);
dvander@547: if (pProp)
dvander@547: {
dvander@547: int val = pProp->GetOffset();
dvander@1057: sm_trie_replace(m_pOffsets, m_offset, (void *)val);
dvander@1057: sm_trie_replace(m_pProps, m_offset, pProp);
dvander@547: } else {
dvander@547: /* Check if it's a non-default game and no offsets exist */
ds@1064: if (((strcmp(m_Game, "*") != 0) && strcmp(m_Game, "#default") != 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\")",
ds@1064: m_Class,
ds@1064: m_Prop,
dvander@2524: m_CurFile,
ds@1064: m_Game);
dvander@547: }
dvander@547: }
dvander@547: }
dvander@547: m_ParseState = PSTATE_GAMEDEFS_OFFSETS;
dvander@547: break;
dvander@547: }
dvander@1307: case PSTATE_GAMEDEFS_CRC:
dvander@947: case PSTATE_GAMEDEFS_SUPPORTED:
dvander@947: {
dvander@947: if (!bShouldBeReadingDefault)
dvander@947: {
dvander@947: /* If we shouldn't read the rest of this section, set the ignore level. */
dvander@947: m_IgnoreLevel = 1;
ds@1043: m_ParseState = PSTATE_GAMES;
ds@1043: } else {
ds@1043: m_ParseState = PSTATE_GAMEDEFS;
dvander@947: }
dvander@947: break;
dvander@947: }
dvander@1307: case PSTATE_GAMEDEFS_CRC_BINARY:
dvander@1307: {
dvander@1307: m_ParseState = PSTATE_GAMEDEFS_CRC;
dvander@1307: break;
dvander@1307: }
dvander@969: case PSTATE_GAMEDEFS_SIGNATURES:
dvander@969: {
dvander@969: m_ParseState = PSTATE_GAMEDEFS;
dvander@969: break;
dvander@969: }
dvander@947: case PSTATE_GAMEDEFS_SIGNATURES_SIG:
dvander@947: {
dvander@947: if (s_TempSig.library[0] == '\0')
dvander@947: {
dvander@947: /* assume server */
dvander@947: strncopy(s_TempSig.library, "server", sizeof(s_TempSig.library));
dvander@947: }
dvander@947: void *addrInBase = NULL;
dvander@947: if (strcmp(s_TempSig.library, "server") == 0)
dvander@947: {
faluco@1541: addrInBase = (void *)g_SMAPI->GetServerFactory(false);
dvander@947: } else if (strcmp(s_TempSig.library, "engine") == 0) {
faluco@1541: addrInBase = (void *)g_SMAPI->GetEngineFactory(false);
dvander@947: }
dvander@947: void *final_addr = NULL;
dvander@947: if (addrInBase == NULL)
dvander@947: {
dvander@947: g_Logger.LogError("[SM] Unrecognized library \"%s\" (gameconf \"%s\")",
dvander@947: s_TempSig.library,
dvander@2524: m_CurFile);
dvander@947: } else {
dvander@947: #if defined PLATFORM_LINUX
dvander@951: if (s_TempSig.sig[0] == '@')
dvander@947: {
dvander@947: Dl_info info;
dvander@947: /* GNU only: returns 0 on error, inconsistent! >:[ */
dvander@973: if (dladdr(addrInBase, &info) != 0)
dvander@947: {
dvander@947: void *handle = dlopen(info.dli_fname, RTLD_NOW);
dvander@947: if (handle)
dvander@947: {
dvander@951: final_addr = dlsym(handle, &s_TempSig.sig[1]);
dvander@947: dlclose(handle);
dvander@947: } else {
dvander@947: g_Logger.LogError("[SM] Unable to load library \"%s\" (gameconf \"%s\")",
dvander@947: s_TempSig.library,
dvander@2524: m_File);
dvander@947: }
dvander@947: } else {
dvander@947: g_Logger.LogError("[SM] Unable to find library \"%s\" in memory (gameconf \"%s\")",
dvander@947: s_TempSig.library,
dvander@2524: m_File);
dvander@947: }
dvander@947: }
dvander@947: if (final_addr)
dvander@947: {
dvander@947: goto skip_find;
dvander@947: }
dvander@947: #endif
dvander@947: /* First, preprocess the signature */
dvander@2806: unsigned char real_sig[511];
dvander@947: size_t real_bytes;
dvander@947: size_t length;
dvander@947:
dvander@947: real_bytes = 0;
dvander@947: length = strlen(s_TempSig.sig);
dvander@947:
dvander@2806: real_bytes = UTIL_DecodeHexString(real_sig, sizeof(real_sig), s_TempSig.sig);
dvander@947:
dvander@1061: if (real_bytes >= 1)
dvander@947: {
dvander@2806: final_addr = g_MemUtils.FindPattern(addrInBase, (char*)real_sig, real_bytes);
dvander@947: }
dvander@947: }
dvander@947:
dvander@947: #if defined PLATFORM_LINUX
dvander@947: skip_find:
dvander@947: #endif
dvander@1057: sm_trie_replace(m_pSigs, m_offset, final_addr);
dvander@964: m_ParseState = PSTATE_GAMEDEFS_SIGNATURES;
dvander@947:
dvander@947: break;
dvander@947: }
dvander@547: }
dvander@547:
dvander@1656: return SMCResult_Continue;
dvander@547: }
dvander@547:
dvander@2524: #define MSTATE_NONE 0
dvander@2524: #define MSTATE_MAIN 1
dvander@2524: #define MSTATE_FILE 2
dvander@2524:
dvander@2524: class MasterReader : public ITextListener_SMC
dvander@2524: {
dvander@2524: public:
dvander@2524: virtual void ReadSMC_ParseStart()
dvander@2524: {
dvander@2524: state = MSTATE_NONE;
dvander@2524: ignoreLevel = 0;
dvander@2524: }
dvander@2524:
dvander@2524: virtual SMCResult ReadSMC_NewSection(const SMCStates *states, const char *name)
dvander@2524: {
dvander@2524: if (ignoreLevel)
dvander@2524: {
dvander@2524: return SMCResult_Continue;
dvander@2524: }
dvander@2524:
dvander@2524: if (state == MSTATE_NONE)
dvander@2524: {
dvander@2524: if (strcmp(name, "Game Master") == 0)
dvander@2524: {
dvander@2524: state = MSTATE_MAIN;
dvander@2524: }
dvander@2524: else
dvander@2524: {
dvander@2524: ignoreLevel++;
dvander@2524: }
dvander@2524: }
dvander@2524: else if (state == MSTATE_MAIN)
dvander@2524: {
dvander@2524: strncopy(cur_file, name, sizeof(cur_file));
dvander@2524: had_engine = false;
dvander@2524: matched_engine = false;
dvander@2524: had_game = false;
dvander@2524: matched_game = false;
dvander@2524: state = MSTATE_FILE;
dvander@2524: }
dvander@2524: else if (state == MSTATE_FILE)
dvander@2524: {
dvander@2524: ignoreLevel++;
dvander@2524: }
dvander@2524:
dvander@2524: return SMCResult_Continue;
dvander@2524: }
dvander@2524:
dvander@2524: virtual SMCResult ReadSMC_KeyValue(const SMCStates *states, const char *key, const char *value)
dvander@2524: {
dvander@2524: if (ignoreLevel || state != MSTATE_FILE)
dvander@2524: {
dvander@2524: return SMCResult_Continue;
dvander@2524: }
dvander@2524:
dvander@2524: if (strcmp(key, "engine") == 0)
dvander@2524: {
dvander@2524: had_engine = true;
dvander@2524: if (DoesEngineMatch(value))
dvander@2524: {
dvander@2524: matched_engine = true;
dvander@2524: }
dvander@2524: }
dvander@2524: else if (strcmp(key, "game") == 0)
dvander@2524: {
dvander@2524: had_game = true;
dvander@2524: if (DoesGameMatch(value))
dvander@2524: {
dvander@2524: matched_game = true;
dvander@2524: }
dvander@2524: }
dvander@2524:
dvander@2524: return SMCResult_Continue;
dvander@2524: }
dvander@2524:
dvander@2524: virtual SMCResult ReadSMC_LeavingSection(const SMCStates *states)
dvander@2524: {
dvander@2524: if (ignoreLevel)
dvander@2524: {
dvander@2524: ignoreLevel--;
dvander@2524: return SMCResult_Continue;
dvander@2524: }
dvander@2524:
dvander@2524: if (state == MSTATE_FILE)
dvander@2524: {
dvander@2524: /* The four success conditions:
dvander@2524: * 1. Needed nothing.
dvander@2524: * 2. Needed game only.
dvander@2524: * 3. Needed engine only.
dvander@2524: * 4. Needed both engine and game.
dvander@2524: * Final result is minimized via k-map.
dvander@2524: */
dvander@2524: #if 0
dvander@2524: if ((!had_engine && !had_game) ||
dvander@2524: (!had_engine && (had_game && matched_game)) ||
dvander@2524: (!had_game && (had_engine && matched_engine)) ||
dvander@2524: ((had_game && had_engine) && (matched_game && matched_engine)))
dvander@2524: #endif
dvander@2524: if ((!had_engine && !had_game) ||
dvander@2524: (!had_engine && matched_game) ||
dvander@2524: (!had_game && matched_engine) ||
dvander@2524: (matched_engine && matched_game))
dvander@2524: {
dvander@2524: fileList->push_back(cur_file);
dvander@2524: }
dvander@2524: state = MSTATE_MAIN;
dvander@2524: }
dvander@2524: else if (state == MSTATE_MAIN)
dvander@2524: {
dvander@2524: state = MSTATE_NONE;
dvander@2524: }
dvander@2524:
dvander@2524: return SMCResult_Continue;
dvander@2524: }
dvander@2524: public:
dvander@2524: List *fileList;
dvander@2524: unsigned int state;
dvander@2524: unsigned int ignoreLevel;
dvander@2524: char cur_file[PLATFORM_MAX_PATH];
dvander@2524: bool had_engine;
dvander@2524: bool matched_engine;
dvander@2524: bool had_game;
dvander@2524: bool matched_game;
dvander@2524: };
dvander@2524:
dvander@2524: static MasterReader master_reader;
dvander@2524:
dvander@547: bool CGameConfig::Reparse(char *error, size_t maxlength)
dvander@547: {
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@2524: char path[PLATFORM_MAX_PATH];
dvander@2524:
dvander@2524: /* See if we can use the extended gamedata format. */
dvander@2524: g_SourceMod.BuildPath(Path_SM, path, sizeof(path), "gamedata/%s/master.games.txt", m_File);
dvander@2524: if (!g_LibSys.PathExists(path))
dvander@2524: {
dvander@2524: /* Nope, use the old mechanism. */
dvander@2524: UTIL_Format(path, sizeof(path), "%s.txt", m_File);
dvander@2524: return EnterFile(path, error, maxlength);
dvander@2524: }
dvander@2524:
dvander@2524: /* Otherwise, it's time to parse the master. */
dvander@2524: SMCError err;
dvander@2524: SMCStates state = {0, 0};
dvander@2524: List fileList;
dvander@2524: master_reader.fileList = &fileList;
dvander@2524:
dvander@2524: err = textparsers->ParseSMCFile(path, &master_reader, &state, error, maxlength);
dvander@2524: if (err != SMCError_Okay)
dvander@2524: {
dvander@2524: const char *msg = textparsers->GetSMCErrorString(err);
dvander@2524:
dvander@2524: g_Logger.LogError("[SM] Error parsing master gameconf file \"%s\":", path);
dvander@2524: g_Logger.LogError("[SM] Error %d on line %d, col %d: %s",
dvander@2524: err,
dvander@2524: state.line,
dvander@2524: state.col,
dvander@2524: msg ? msg : "Unknown error");
dvander@2524: return false;
dvander@2524: }
dvander@2524:
dvander@2524: /* Go through each file we found and parse it. */
dvander@2524: List::iterator iter;
dvander@2524: for (iter = fileList.begin(); iter != fileList.end(); iter++)
dvander@2524: {
dvander@2524: UTIL_Format(path, sizeof(path), "%s/%s", m_File, (*iter).c_str());
dvander@2524: if (!EnterFile(path, error, maxlength))
dvander@2524: {
dvander@2524: return false;
dvander@2524: }
dvander@2524: }
dvander@2524:
pred@2578: /* Parse the contents of the 'custom' directory */
pred@2582: g_SourceMod.BuildPath(Path_SM, path, sizeof(path), "gamedata/%s/custom", m_File);
pred@2578: IDirectory *customDir = g_LibSys.OpenDirectory(path);
pred@2578:
pred@2578: if (!customDir)
pred@2578: {
pred@2578: return true;
pred@2578: }
pred@2578:
pred@2578: while (customDir->MoreFiles())
pred@2578: {
pred@2578: if (!customDir->IsEntryFile())
pred@2578: {
pred@2578: customDir->NextEntry();
pred@2578: continue;
pred@2578: }
pred@2578:
pred@2578: const char *curFile = customDir->GetEntryName();
pred@2578:
pred@2578: /* Only allow .txt files */
pred@2578: int len = strlen(curFile);
pred@2579: if (len > 4 && strcmp(&curFile[len-4], ".txt") != 0)
pred@2578: {
pred@2578: customDir->NextEntry();
pred@2578: continue;
pred@2578: }
pred@2578:
pred@2578: UTIL_Format(path, sizeof(path), "%s/custom/%s", m_File, curFile);
pred@2578: if (!EnterFile(path, error, maxlength))
pred@2578: {
pred@2579: g_LibSys.CloseDirectory(customDir);
pred@2578: return false;
pred@2578: }
pred@2578:
pred@2578: customDir->NextEntry();
pred@2578: }
pred@2578:
pred@2579: g_LibSys.CloseDirectory(customDir);
dvander@2524: return true;
dvander@2524: }
dvander@2524:
dvander@2524: bool CGameConfig::EnterFile(const char *file, char *error, size_t maxlength)
dvander@2524: {
dvander@2524: SMCError err;
dvander@2524: SMCStates state = {0, 0};
dvander@2524:
dvander@2524: g_SourceMod.BuildPath(Path_SM, m_CurFile, sizeof(m_CurFile), "gamedata/%s", file);
dvander@2524:
dvander@2524: /* Initialize parse states */
dvander@2524: m_IgnoreLevel = 0;
dvander@2524: bShouldBeReadingDefault = true;
dvander@2524: m_ParseState = PSTATE_NONE;
dvander@2524:
dvander@2524: if ((err=textparsers->ParseSMCFile(m_CurFile, this, &state, error, maxlength))
dvander@1656: != SMCError_Okay)
dvander@547: {
dvander@1885: const char *msg;
dvander@1885:
dvander@1885: msg = textparsers->GetSMCErrorString(err);
dvander@1885:
dvander@2524: g_Logger.LogError("[SM] Error parsing gameconfig file \"%s\":", m_CurFile);
dvander@1885: g_Logger.LogError("[SM] Error %d on line %d, col %d: %s",
dvander@1885: err,
dvander@1885: state.line,
dvander@1885: state.col,
dvander@1885: msg ? msg : "Unknown error");
pred@2217:
pred@2217: if (m_ParseState == PSTATE_GAMEDEFS_CUSTOM)
pred@2217: {
pred@2217: //error occurred while parsing a custom section
pred@2217: m_CustomHandler->ReadSMC_ParseEnd(true, true);
pred@2217: m_CustomHandler = NULL;
pred@2217: m_CustomLevel = 0;
pred@2217: }
pred@2217:
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@947: bool CGameConfig::GetMemSig(const char *key, void **addr)
dvander@947: {
dvander@947: return sm_trie_retrieve(m_pSigs, key, addr);
dvander@947: }
dvander@947:
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@1064: strncopy(g_Game, g_SourceMod.GetGameFolderName(), sizeof(g_Game));
ds@1064: strncopy(g_GameDesc + 1, SERVER_CALL(GetGameDescription)(), sizeof(g_GameDesc) - 1);
dvander@1762:
dvander@1762: KeyValues *pGameInfo = new KeyValues("GameInfo");
ds@1944: if (g_HL2.KVLoadFromFile(pGameInfo, basefilesystem, "gameinfo.txt"))
dvander@1762: {
dvander@1762: const char *str;
dvander@1762: if ((str = pGameInfo->GetString("game", NULL)) != NULL)
dvander@1762: {
dvander@1762: strncopy(g_GameName + 1, str, sizeof(g_GameName) - 1);
dvander@1762: }
dvander@1762: }
dvander@1762: pGameInfo->deleteThis();
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@2517: #if 0
pred@2142: /* A crash was detected during last load - We block the gamedata loading so it hopefully won't happen again */
pred@2142: if (g_blockGameDataLoad)
pred@2142: {
pred@2142: UTIL_Format(error, maxlength, "GameData loaded blocked due to detected crash");
pred@2142: return false;
pred@2142: }
dvander@2517: #endif
pred@2142:
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@2524: sm_trie_delete(m_pLookup, pConfig->m_File);
dvander@547: delete pConfig;
dvander@547: }
dvander@547: }
dvander@949:
dvander@949: extern HandleType_t g_GameConfigsType;
dvander@949:
dvander@949: IGameConfig *GameConfigManager::ReadHandle(Handle_t hndl, IdentityToken_t *ident, HandleError *err)
dvander@949: {
dvander@949: HandleSecurity sec(ident, g_pCoreIdent);
dvander@949: IGameConfig *conf = NULL;
dvander@949: HandleError _err = g_HandleSys.ReadHandle(hndl, g_GameConfigsType, &sec, (void **)&conf);
dvander@949:
dvander@949: if (err)
dvander@949: {
dvander@949: *err = _err;
dvander@949: }
dvander@949:
dvander@949: return conf;
dvander@949: }
pred@2217:
dvander@2562: void GameConfigManager::AddUserConfigHook(const char *sectionname, ITextListener_SMC *listener)
pred@2217: {
pred@2217: m_customHandlers.insert(sectionname, listener);
pred@2217: }
pred@2217:
dvander@2562: void GameConfigManager::RemoveUserConfigHook(const char *sectionname, ITextListener_SMC *listener)
pred@2217: {
pred@2217: ITextListener_SMC **pListener = m_customHandlers.retrieve(sectionname);
pred@2217:
pred@2217: if (pListener == NULL)
pred@2217: {
pred@2217: return;
pred@2217: }
pred@2217:
pred@2217: if (*pListener != listener)
pred@2217: {
pred@2217: return;
pred@2217: }
pred@2217:
pred@2217: m_customHandlers.remove(sectionname);
pred@2217:
pred@2217: return;
pred@2217: }
dvander@2562:
dvander@2562: void GameConfigManager::AcquireLock()
dvander@2562: {
dvander@2562: }
dvander@2562:
dvander@2562: void GameConfigManager::ReleaseLock()
dvander@2562: {
dvander@2562: }