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 dvander@547: #include 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@947: #include "MemoryUtils.h" dvander@947: #include "LibrarySys.h" dvander@949: #include "HandleSys.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@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@947: #define PSTATE_GAMEDEFS_SUPPORTED 6 dvander@947: #define PSTATE_GAMEDEFS_SIGNATURES 7 dvander@947: #define PSTATE_GAMEDEFS_SIGNATURES_SIG 8 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@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@947: 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@947: m_pSigs = 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@947: sm_trie_destroy(m_pSigs); 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@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@547: if ((strcmp(name, "*") == 0) dvander@947: || (strcmp(name, "#default") == 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@947: } dvander@947: else if (strcmp(name, "Keys") == 0) dvander@947: { dvander@548: m_ParseState = PSTATE_GAMEDEFS_KEYS; dvander@947: } dvander@947: else if ((strcmp(name, "#supported") == 0) dvander@947: && (strcmp(m_mod, "#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@947: } dvander@947: else if (strcmp(name, "Signatures") == 0) dvander@947: { dvander@947: m_ParseState = PSTATE_GAMEDEFS_SIGNATURES; dvander@947: } dvander@947: else dvander@947: { 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@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@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@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@947: } else if (m_ParseState == PSTATE_GAMEDEFS_SUPPORTED) { dvander@947: if (strcmp(key, "game") == 0 dvander@947: && strcmp(value, g_mod) == 0) dvander@947: { dvander@947: bShouldBeReadingDefault = true; 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@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@947: if (((strcmp(m_mod, "*") != 0) && strcmp(m_mod, "#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\")", 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@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; dvander@947: } dvander@947: m_ParseState = PSTATE_GAMEDEFS; dvander@947: break; dvander@947: } 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: { dvander@951: addrInBase = (void *)g_SMAPI->serverFactory(false); dvander@947: } else if (strcmp(s_TempSig.library, "engine") == 0) { dvander@951: addrInBase = (void *)g_SMAPI->engineFactory(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@947: m_pFile); 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@947: m_pFile); 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@947: m_pFile); dvander@947: } dvander@947: if (!final_addr) dvander@947: { dvander@947: g_Logger.LogError("[SM] Unable to find named symbol (symbol \"%s\") (library \"%s\") (gameconf \"%s\")", dvander@951: &s_TempSig.sig[1], dvander@947: s_TempSig.library, dvander@947: m_pFile); 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@947: char real_sig[255]; 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@947: for (size_t i=0; i= sizeof(real_sig)) dvander@947: { dvander@947: break; dvander@947: } dvander@947: real_sig[real_bytes++] = s_TempSig.sig[i]; dvander@947: if (s_TempSig.sig[i] == '\\' dvander@947: && s_TempSig.sig[i+1] == 'x') dvander@947: { dvander@947: if (i + 3 >= length) dvander@947: { dvander@947: continue; dvander@947: } dvander@947: /* Get the hex part */ dvander@947: char s_byte[3]; dvander@947: int r_byte; dvander@947: s_byte[0] = s_TempSig.sig[i+2]; dvander@947: s_byte[1] = s_TempSig.sig[i+3]; dvander@947: s_byte[2] = '\0'; dvander@947: /* Read it as an integer */ dvander@947: sscanf(s_byte, "%x", &r_byte); dvander@947: /* Save the value */ dvander@947: real_sig[real_bytes-1] = r_byte; dvander@947: /* Adjust index */ dvander@947: i += 3; dvander@947: } dvander@947: } dvander@947: dvander@947: if (real_bytes < 1) dvander@947: { dvander@969: g_Logger.LogError("[SM] Invalid signature (name \"%s\") (gameconf \"%s\")", dvander@947: m_offset, dvander@947: m_pFile); dvander@947: } else { dvander@947: final_addr = g_MemUtils.FindPattern(addrInBase, real_sig, real_bytes); dvander@947: if (!final_addr) dvander@947: { dvander@947: g_Logger.LogError("[SM] Unable to find signature (name \"%s\") (gameconf \"%s\")", dvander@947: m_offset, dvander@947: m_pFile); dvander@947: } dvander@947: } dvander@947: } dvander@947: dvander@947: #if defined PLATFORM_LINUX dvander@947: skip_find: dvander@947: #endif dvander@947: sm_trie_insert(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@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@947: g_SourceMod.BuildPath(Path_SM, path, sizeof(path), "gamedata/%s.txt", m_pFile); dvander@947: /* Backwards compatibility */ dvander@947: if (!g_LibSys.PathExists(path)) dvander@947: { dvander@947: g_SourceMod.BuildPath(Path_SM, path, sizeof(path), "configs/gamedata/%s.txt", m_pFile); dvander@947: } dvander@547: dvander@548: /* Initialize parse states */ dvander@547: m_IgnoreLevel = 0; dvander@947: bShouldBeReadingDefault = true; 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@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@729: strncopy(g_mod, g_SourceMod.GetGameFolderName(), 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@971: sm_trie_delete(m_pLookup, pConfig->m_pFile); 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: }