ds@580: /** ds@2825: * vim: set ts=4 sw=4 tw=99 noet : ds@1330: * ============================================================================= dvander@1231: * SourceMod ds@2825: * Copyright (C) 2004-2009 AlliedModders LLC. All rights reserved. ds@1330: * ============================================================================= ds@653: * 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 . ds@653: */ ds@580: ds@580: #include "ConVarManager.h" ds@696: #include "HalfLife2.h" ds@580: #include "ForwardSys.h" ds@580: #include "sm_srvcmds.h" ds@580: #include "sm_stringutil.h" ds@580: #include dvander@1582: #include dvander@3872: #include "logic_bridge.h" ds@580: ds@580: ConVarManager g_ConVarManager; ds@580: ds@2400: #if SOURCE_ENGINE >= SE_ORANGEBOX dvander@1770: SH_DECL_HOOK3_void(ICvar, CallGlobalChangeCallbacks, SH_NOATTRIB, false, ConVar *, const char *, float); dvander@1770: #else skamonkey@3820: SH_DECL_HOOK2_void(ICvar, CallGlobalChangeCallback, SH_NOATTRIB, false, ConVar *, const char *); dvander@1770: #endif dvander@1770: skamonkey@3942: #if SOURCE_ENGINE == SE_DOTA skamonkey@3942: SH_DECL_HOOK5_void(IServerGameDLL, OnQueryCvarValueFinished, SH_NOATTRIB, 0, QueryCvarCookie_t, CEntityIndex, EQueryCvarValueStatus, const char *, const char *); skamonkey@3942: SH_DECL_HOOK5_void(IServerPluginCallbacks, OnQueryCvarValueFinished, SH_NOATTRIB, 0, QueryCvarCookie_t, CEntityIndex, EQueryCvarValueStatus, const char *, const char *); skamonkey@3942: #elif SOURCE_ENGINE != SE_DARKMESSIAH ds@696: SH_DECL_HOOK5_void(IServerGameDLL, OnQueryCvarValueFinished, SH_NOATTRIB, 0, QueryCvarCookie_t, edict_t *, EQueryCvarValueStatus, const char *, const char *); ds@696: SH_DECL_HOOK5_void(IServerPluginCallbacks, OnQueryCvarValueFinished, SH_NOATTRIB, 0, QueryCvarCookie_t, edict_t *, EQueryCvarValueStatus, const char *, const char *); ds@2583: #endif ds@696: ds@582: const ParamType CONVARCHANGE_PARAMS[] = {Param_Cell, Param_String, Param_String}; ds@582: typedef List ConVarList; dvander@1582: KTrie convar_cache; ds@582: dvander@2890: class ConVarReentrancyGuard dvander@2890: { dvander@2890: ConVar *cvar; dvander@2890: ConVarReentrancyGuard *up; dvander@2890: public: dvander@2890: static ConVarReentrancyGuard *chain; dvander@2890: dvander@2890: ConVarReentrancyGuard(ConVar *cvar) dvander@2890: : cvar(cvar), up(chain) dvander@2890: { dvander@2890: chain = this; dvander@2890: } dvander@2890: dvander@2890: ~ConVarReentrancyGuard() dvander@2890: { dvander@2890: assert(chain == this); dvander@2890: chain = up; dvander@2890: } dvander@2890: dvander@2890: static bool IsCvarInChain(ConVar *cvar) dvander@2890: { dvander@2890: ConVarReentrancyGuard *guard = chain; dvander@2890: while (guard != NULL) dvander@2890: { dvander@2890: if (guard->cvar == cvar) dvander@2890: return true; dvander@2890: guard = guard->up; dvander@2890: } dvander@2890: return false; dvander@2890: } dvander@2890: }; dvander@2890: dvander@2890: ConVarReentrancyGuard *ConVarReentrancyGuard::chain = NULL; dvander@2890: dvander@1581: ConVarManager::ConVarManager() : m_ConVarType(0), m_bIsDLLQueryHooked(false), m_bIsVSPQueryHooked(false) ds@580: { ds@580: } ds@580: ds@580: ConVarManager::~ConVarManager() ds@580: { ds@580: } ds@580: dvander@1770: void ConVarManager::OnSourceModStartup(bool late) dvander@1770: { dvander@1770: HandleAccess sec; dvander@1770: dvander@1770: /* Set up access rights for the 'ConVar' handle type */ dvander@1770: sec.access[HandleAccess_Read] = 0; dvander@1770: sec.access[HandleAccess_Delete] = HANDLE_RESTRICT_IDENTITY | HANDLE_RESTRICT_OWNER; dvander@1770: sec.access[HandleAccess_Clone] = HANDLE_RESTRICT_IDENTITY | HANDLE_RESTRICT_OWNER; dvander@1770: dvander@1770: /* Create the 'ConVar' handle type */ dvander@3872: m_ConVarType = handlesys->CreateType("ConVar", this, 0, NULL, &sec, g_pCoreIdent, NULL); dvander@1770: } dvander@1770: ds@580: void ConVarManager::OnSourceModAllInitialized() ds@580: { dvander@1581: /** dvander@1581: * Episode 2 has this function by default, but the older versions do not. dvander@1581: */ ds@2400: #if SOURCE_ENGINE == SE_EPISODEONE ds@696: if (g_SMAPI->GetGameDLLVersion() >= 6) ds@696: { kyle@3819: SH_ADD_HOOK(IServerGameDLL, OnQueryCvarValueFinished, gamedll, SH_MEMBER(this, &ConVarManager::OnQueryCvarValueFinished), false); dvander@1581: m_bIsDLLQueryHooked = true; ds@696: } dvander@1759: #endif ds@696: skamonkey@3820: #if SOURCE_ENGINE >= SE_ORANGEBOX kyle@3819: SH_ADD_HOOK(ICvar, CallGlobalChangeCallbacks, icvar, SH_STATIC(OnConVarChanged), false); skamonkey@3820: #else skamonkey@3820: SH_ADD_HOOK(ICvar, CallGlobalChangeCallback, icvar, SH_STATIC(OnConVarChanged), false); skamonkey@3820: #endif ds@580: dvander@3872: scripts->AddPluginsListener(this); ds@2825: ds@582: /* Add the 'convars' option to the 'sm' console command */ ds@990: g_RootMenu.AddRootConsoleCommand("cvars", "View convars created by a plugin", this); ds@580: } ds@580: ds@580: void ConVarManager::OnSourceModShutdown() ds@580: { dvander@1582: List::iterator iter = m_ConVars.begin(); dvander@1582: HandleSecurity sec(NULL, g_pCoreIdent); dvander@1582: dvander@1582: /* Iterate list of ConVarInfo structures, remove every one of them */ dvander@1582: while (iter != m_ConVars.end()) dvander@1582: { dvander@1582: ConVarInfo *pInfo = (*iter); dvander@1582: dvander@1582: iter = m_ConVars.erase(iter); dvander@1582: dvander@3872: handlesys->FreeHandle(pInfo->handle, &sec); dvander@1582: if (pInfo->pChangeForward != NULL) dvander@1582: { dvander@1582: g_Forwards.ReleaseForward(pInfo->pChangeForward); dvander@1582: } dvander@1582: if (pInfo->sourceMod) dvander@1582: { dvander@1582: /* If we created it, we won't be tracking it, therefore it is dvander@1582: * safe to remove everything in one go. dvander@1582: */ dvander@1582: META_UNREGCVAR(pInfo->pVar); dvander@1582: delete [] pInfo->pVar->GetName(); dvander@1582: delete [] pInfo->pVar->GetHelpText(); dvander@1582: delete [] pInfo->pVar->GetDefault(); dvander@1582: delete pInfo->pVar; dvander@1582: } dvander@1582: else dvander@1582: { dvander@1582: /* If we didn't create it, we might be tracking it. Also, dvander@1582: * it could be unreadable. dvander@1582: */ dvander@1582: UntrackConCommandBase(pInfo->pVar, this); dvander@1582: } dvander@1582: dvander@1582: /* It's not safe to read the name here, so we simply delete the dvander@1582: * the info struct and clear the lookup cache at the end. dvander@1582: */ dvander@1582: delete pInfo; dvander@1582: } dvander@1582: convar_cache.clear(); dvander@1582: ds@2583: #if SOURCE_ENGINE != SE_DARKMESSIAH dvander@1582: /* Unhook things */ dvander@1581: if (m_bIsDLLQueryHooked) ds@696: { kyle@3819: SH_REMOVE_HOOK(IServerGameDLL, OnQueryCvarValueFinished, gamedll, SH_MEMBER(this, &ConVarManager::OnQueryCvarValueFinished), false); dvander@1581: m_bIsDLLQueryHooked = false; dvander@1581: } dvander@1581: else if (m_bIsVSPQueryHooked) dvander@1581: { skamonkey@3942: #if SOURCE_ENGINE != SE_DOTA kyle@3819: SH_REMOVE_HOOK(IServerPluginCallbacks, OnQueryCvarValueFinished, vsp_interface, SH_MEMBER(this, &ConVarManager::OnQueryCvarValueFinished), false); skamonkey@3942: #endif dvander@1581: m_bIsVSPQueryHooked = false; ds@696: } ds@2583: #endif ds@696: skamonkey@3820: #if SOURCE_ENGINE >= SE_ORANGEBOX kyle@3819: SH_REMOVE_HOOK(ICvar, CallGlobalChangeCallbacks, icvar, SH_STATIC(OnConVarChanged), false); skamonkey@3820: #else skamonkey@3820: SH_REMOVE_HOOK(ICvar, CallGlobalChangeCallback, icvar, SH_STATIC(OnConVarChanged), false); skamonkey@3820: #endif dvander@1770: ds@582: /* Remove the 'convars' option from the 'sm' console command */ ds@990: g_RootMenu.RemoveRootConsoleCommand("cvars", this); ds@580: dvander@3872: scripts->RemovePluginsListener(this); ds@2825: ds@582: /* Remove the 'ConVar' handle type */ dvander@3872: handlesys->RemoveType(m_ConVarType, g_pCoreIdent); ds@580: } ds@580: dvander@1581: /** dvander@1581: * Orange Box will never use this. dvander@1581: */ dvander@1581: void ConVarManager::OnSourceModVSPReceived() ds@696: { dvander@1581: /** dvander@1581: * Don't bother if the DLL is already hooked. ds@696: */ dvander@1581: if (m_bIsDLLQueryHooked) ds@696: { ds@696: return; ds@696: } ds@696: dvander@1581: /* For later MM:S versions, use the updated API, since it's cleaner. */ dvander@2846: #if defined METAMOD_PLAPI_VERSION || PLAPI_VERSION >= 11 dvander@1581: int engine = g_SMAPI->GetSourceEngineBuild(); dvander@1581: if (engine == SOURCE_ENGINE_ORIGINAL || vsp_version < 2) ds@696: { dvander@1581: return; dvander@1581: } dvander@1581: #else dvander@1581: if (g_HL2.IsOriginalEngine() || vsp_version < 2) dvander@1581: { dvander@1581: return; dvander@1581: } dvander@1581: #endif ds@696: skamonkey@3942: #if SOURCE_ENGINE != SE_DARKMESSIAH && SOURCE_ENGINE != SE_DOTA kyle@3819: SH_ADD_HOOK(IServerPluginCallbacks, OnQueryCvarValueFinished, vsp_interface, SH_MEMBER(this, &ConVarManager::OnQueryCvarValueFinished), false); dvander@1581: m_bIsVSPQueryHooked = true; ds@2583: #endif ds@696: } ds@696: dvander@1582: bool convar_cache_lookup(const char *name, ConVarInfo **pVar) ds@1145: { dvander@1582: ConVarInfo **pLookup = convar_cache.retrieve(name); dvander@1582: if (pLookup != NULL) ds@1145: { dvander@1582: *pVar = *pLookup; dvander@1582: return true; dvander@1582: } dvander@1582: else dvander@1582: { dvander@1582: return false; ds@1145: } ds@1145: } ds@1145: dvander@1582: void ConVarManager::OnUnlinkConCommandBase(ConCommandBase *pBase, const char *name, bool is_read_safe) dvander@1582: { dvander@1582: /* Only check convars that have not been created by SourceMod's core */ dvander@1582: ConVarInfo *pInfo; dvander@1582: if (!convar_cache_lookup(name, &pInfo)) dvander@1582: { dvander@1582: return; dvander@1582: } ds@1145: ds@1145: HandleSecurity sec(NULL, g_pCoreIdent); ds@1145: dvander@1582: /* Remove it from our cache */ dvander@1582: m_ConVars.remove(pInfo); dvander@1582: convar_cache.remove(name); dvander@1582: dvander@1582: /* Now make sure no plugins are referring to this pointer */ dvander@3872: IPluginIterator *pl_iter = scripts->GetPluginIterator(); dvander@1582: while (pl_iter->MorePlugins()) ds@1145: { dvander@1582: IPlugin *pl = pl_iter->GetPlugin(); ds@1145: dvander@1582: ConVarList *pConVarList; dvander@1582: if (pl->GetProperty("ConVarList", (void **)&pConVarList, true) dvander@1582: && pConVarList != NULL) ds@1145: { dvander@1582: pConVarList->remove(pInfo->pVar); dvander@1582: } ds@1145: dvander@1582: pl_iter->NextPlugin(); ds@1145: } dvander@1582: dvander@1582: /* Free resources */ dvander@3872: handlesys->FreeHandle(pInfo->handle, &sec); dvander@1582: delete pInfo; ds@1145: } ds@1145: ds@582: void ConVarManager::OnPluginUnloaded(IPlugin *plugin) ds@580: { ds@582: ConVarList *pConVarList; ds@2825: List::iterator iter; ds@580: ds@582: /* If plugin has a convar list, free its memory */ ds@582: if (plugin->GetProperty("ConVarList", (void **)&pConVarList, true)) ds@580: { ds@582: delete pConVarList; ds@580: } ds@2825: kyle@3995: const IPluginRuntime * pRuntime = plugin->GetRuntime(); kyle@3995: ds@2825: /* Remove convar queries for this plugin that haven't returned results yet */ kyle@3995: for (iter = m_ConVarQueries.begin(); iter != m_ConVarQueries.end();) ds@2825: { ds@2825: ConVarQuery &query = (*iter); kyle@3995: if (query.pCallback->GetParentRuntime() == pRuntime) ds@2825: { kyle@3995: iter = m_ConVarQueries.erase(iter); kyle@3995: continue; ds@2825: } kyle@3995: kyle@3995: ++iter; ds@2825: } ds@580: } ds@580: ds@580: void ConVarManager::OnHandleDestroy(HandleType_t type, void *object) ds@580: { ds@580: } ds@580: dvander@1766: bool ConVarManager::GetHandleApproxSize(HandleType_t type, void *object, unsigned int *pSize) dvander@1766: { dvander@1770: *pSize = sizeof(ConVar) + sizeof(ConVarInfo); dvander@1766: return true; dvander@1766: } dvander@1766: faluco@1541: void ConVarManager::OnRootConsoleCommand(const char *cmdname, const CCommand &command) ds@580: { faluco@1541: int argcount = command.ArgC(); ds@580: if (argcount >= 3) ds@580: { asherkin@3367: bool wantReset = false; asherkin@3367: ds@582: /* Get plugin index that was passed */ dvander@1902: const char *arg = command.Arg(2); asherkin@3367: if (argcount >= 4 && strcmp(arg, "reset") == 0) asherkin@3367: { asherkin@3367: wantReset = true; asherkin@3367: arg = command.Arg(3); asherkin@3367: } ds@580: ds@582: /* Get plugin object */ dvander@3872: IPlugin *plugin = scripts->FindPluginByConsoleArg(arg); ds@582: ds@582: if (!plugin) ds@580: { dvander@1902: g_RootMenu.ConsolePrint("[SM] Plugin \"%s\" was not found.", arg); ds@580: return; ds@580: } ds@580: ds@582: /* Get plugin name */ ds@582: const sm_plugininfo_t *plinfo = plugin->GetPublicInfo(); ds@582: const char *plname = IS_STR_FILLED(plinfo->name) ? plinfo->name : plugin->GetFilename(); ds@580: ds@582: ConVarList *pConVarList; ds@582: ConVarList::iterator iter; ds@580: ds@582: /* If no convar list... */ ds@582: if (!plugin->GetProperty("ConVarList", (void **)&pConVarList)) ds@580: { ds@582: g_RootMenu.ConsolePrint("[SM] No convars found for: %s", plname); ds@580: return; ds@580: } ds@580: asherkin@3367: if (!wantReset) asherkin@3367: { asherkin@3367: g_RootMenu.ConsolePrint("[SM] Listing %d convars for: %s", pConVarList->size(), plname); asherkin@3367: g_RootMenu.ConsolePrint(" %-32.31s %s", "[Name]", "[Value]"); asherkin@3367: } asherkin@3367: asherkin@3367: /* Iterate convar list and display/reset each one */ ds@582: for (iter = pConVarList->begin(); iter != pConVarList->end(); iter++) ds@580: { asherkin@3367: /*const */ConVar *pConVar = const_cast(*iter); asherkin@3367: if (!wantReset) asherkin@3367: { asherkin@3367: g_RootMenu.ConsolePrint(" %-32.31s %s", pConVar->GetName(), pConVar->GetString()); asherkin@3367: } else { asherkin@3367: pConVar->Revert(); asherkin@3367: } asherkin@3367: } asherkin@3367: asherkin@3367: if (wantReset) asherkin@3367: { asherkin@3367: g_RootMenu.ConsolePrint("[SM] Reset %d convars for: %s", pConVarList->size(), plname); ds@580: } ds@580: ds@580: return; ds@580: } ds@580: ds@582: /* Display usage of subcommand */ asherkin@3367: g_RootMenu.ConsolePrint("[SM] Usage: sm cvars [reset] "); ds@580: } ds@580: ds@582: Handle_t ConVarManager::CreateConVar(IPluginContext *pContext, const char *name, const char *defaultVal, const char *description, int flags, bool hasMin, float min, bool hasMax, float max) ds@580: { ds@580: ConVar *pConVar = NULL; ds@582: ConVarInfo *pInfo = NULL; ds@580: Handle_t hndl = 0; ds@580: ds@582: /* Find out if the convar exists already */ ds@580: pConVar = icvar->FindVar(name); ds@580: ds@582: /* If the convar already exists... */ ds@582: if (pConVar) ds@580: { ds@759: /* Add convar to plugin's list */ ds@759: AddConVarToPluginList(pContext, pConVar); ds@759: ds@582: /* First find out if we already have a handle to it */ dvander@1582: if (convar_cache_lookup(name, &pInfo)) ds@580: { ds@582: return pInfo->handle; dvander@1581: } dvander@1581: else dvander@1581: { ds@582: /* Create and initialize ConVarInfo structure */ ds@582: pInfo = new ConVarInfo(); ds@582: pInfo->sourceMod = false; ds@696: pInfo->pChangeForward = NULL; dvander@1582: pInfo->pVar = pConVar; dvander@1582: dvander@1582: /* If we don't, then create a new handle from the convar */ dvander@3872: hndl = handlesys->CreateHandle(m_ConVarType, pInfo, NULL, g_pCoreIdent, NULL); dvander@1582: if (hndl == BAD_HANDLE) dvander@1582: { dvander@1582: delete pInfo; dvander@1582: return BAD_HANDLE; dvander@1582: } dvander@1582: dvander@1582: pInfo->handle = hndl; ds@580: ds@582: /* Insert struct into caches */ ds@582: m_ConVars.push_back(pInfo); dvander@1582: convar_cache.insert(name, pInfo); dvander@1582: TrackConCommandBase(pConVar, this); ds@580: ds@580: return hndl; ds@580: } ds@580: } ds@580: ds@2400: /* Prevent creating a convar that has the same name as a console command */ ds@2400: if (FindCommand(name)) ds@580: { ds@2400: return BAD_HANDLE; ds@580: } ds@580: ds@582: /* Create and initialize ConVarInfo structure */ ds@582: pInfo = new ConVarInfo(); ds@582: pInfo->handle = hndl; ds@582: pInfo->sourceMod = true; ds@696: pInfo->pChangeForward = NULL; dvander@1582: dvander@1582: /* Create a handle from the new convar */ dvander@3872: hndl = handlesys->CreateHandle(m_ConVarType, pInfo, NULL, g_pCoreIdent, NULL); dvander@1582: if (hndl == BAD_HANDLE) dvander@1582: { dvander@1582: delete pInfo; dvander@1582: return BAD_HANDLE; dvander@1582: } dvander@1582: dvander@1582: pInfo->handle = hndl; dvander@1582: dvander@1582: /* Since an existing convar (or concmd with the same name) was not found , now we can finally create it */ dvander@1582: pConVar = new ConVar(sm_strdup(name), sm_strdup(defaultVal), flags, sm_strdup(description), hasMin, min, hasMax, max); dvander@1582: pInfo->pVar = pConVar; dvander@1582: dvander@1582: /* Add convar to plugin's list */ dvander@1582: AddConVarToPluginList(pContext, pConVar); ds@580: ds@582: /* Insert struct into caches */ ds@582: m_ConVars.push_back(pInfo); dvander@1582: convar_cache.insert(name, pInfo); ds@580: ds@580: return hndl; ds@580: } ds@580: ds@580: Handle_t ConVarManager::FindConVar(const char *name) ds@580: { ds@580: ConVar *pConVar = NULL; ds@582: ConVarInfo *pInfo; ds@582: Handle_t hndl; ds@580: ds@582: /* Search for convar */ ds@580: pConVar = icvar->FindVar(name); ds@580: ds@582: /* If it doesn't exist, then return an invalid handle */ ds@582: if (!pConVar) ds@580: { ds@580: return BAD_HANDLE; ds@580: } ds@580: ds@582: /* At this point, the convar exists. So, find out if we already have a handle */ dvander@1582: if (convar_cache_lookup(name, &pInfo)) ds@580: { ds@582: return pInfo->handle; ds@580: } ds@580: ds@729: /* Create and initialize ConVarInfo structure */ ds@582: pInfo = new ConVarInfo(); ds@582: pInfo->sourceMod = false; ds@696: pInfo->pChangeForward = NULL; dvander@1582: pInfo->pVar = pConVar; dvander@1582: dvander@1582: /* If we don't have a handle, then create a new one */ dvander@3872: hndl = handlesys->CreateHandle(m_ConVarType, pInfo, NULL, g_pCoreIdent, NULL); dvander@1582: if (hndl == BAD_HANDLE) dvander@1582: { dvander@1582: delete pInfo; dvander@1582: return BAD_HANDLE; dvander@1582: } dvander@1582: dvander@1582: pInfo->handle = hndl; ds@580: ds@582: /* Insert struct into our caches */ ds@582: m_ConVars.push_back(pInfo); dvander@1582: convar_cache.insert(name, pInfo); dvander@1582: TrackConCommandBase(pConVar, this); ds@580: ds@580: return hndl; ds@580: } ds@580: dvander@1770: void ConVarManager::AddConVarChangeListener(const char *name, IConVarChangeListener *pListener) dvander@1770: { dvander@1770: ConVarInfo *pInfo; dvander@1770: dvander@1770: if (FindConVar(name) == BAD_HANDLE) dvander@1770: { dvander@1770: return; dvander@1770: } dvander@1770: dvander@1770: /* Find the convar in the lookup trie */ dvander@1770: if (convar_cache_lookup(name, &pInfo)) dvander@1770: { dvander@1770: pInfo->changeListeners.push_back(pListener); dvander@1770: } dvander@1770: } dvander@1770: dvander@1770: void ConVarManager::RemoveConVarChangeListener(const char *name, IConVarChangeListener *pListener) dvander@1770: { dvander@1770: ConVarInfo *pInfo; dvander@1770: dvander@1770: /* Find the convar in the lookup trie */ dvander@1770: if (convar_cache_lookup(name, &pInfo)) dvander@1770: { dvander@1770: pInfo->changeListeners.remove(pListener); dvander@1770: } dvander@1770: } dvander@1770: ds@582: void ConVarManager::HookConVarChange(ConVar *pConVar, IPluginFunction *pFunction) ds@580: { ds@582: ConVarInfo *pInfo; ds@582: IChangeableForward *pForward; ds@580: ds@582: /* Find the convar in the lookup trie */ dvander@1582: if (convar_cache_lookup(pConVar->GetName(), &pInfo)) ds@580: { ds@582: /* Get the forward */ ds@696: pForward = pInfo->pChangeForward; ds@580: ds@582: /* If forward does not exist, create it */ ds@582: if (!pForward) ds@582: { ds@582: pForward = g_Forwards.CreateForwardEx(NULL, ET_Ignore, 3, CONVARCHANGE_PARAMS); ds@696: pInfo->pChangeForward = pForward; ds@580: } ds@582: ds@582: /* Add function to forward's list */ ds@582: pForward->AddFunction(pFunction); ds@580: } ds@580: } ds@580: ds@582: void ConVarManager::UnhookConVarChange(ConVar *pConVar, IPluginFunction *pFunction) ds@580: { ds@582: ConVarInfo *pInfo; ds@582: IChangeableForward *pForward; ds@582: IPluginContext *pContext = pFunction->GetParentContext(); ds@580: ds@582: /* Find the convar in the lookup trie */ dvander@1582: if (convar_cache_lookup(pConVar->GetName(), &pInfo)) ds@580: { ds@582: /* Get the forward */ ds@696: pForward = pInfo->pChangeForward; ds@580: ds@582: /* If the forward doesn't exist, we can't unhook anything */ ds@582: if (!pForward) ds@580: { ds@580: pContext->ThrowNativeError("Convar \"%s\" has no active hook", pConVar->GetName()); ds@580: return; ds@580: } ds@580: ds@582: /* Remove the function from the forward's list */ ds@582: if (!pForward->RemoveFunction(pFunction)) ds@580: { ds@580: pContext->ThrowNativeError("Invalid hook callback specified for convar \"%s\"", pConVar->GetName()); ds@580: return; ds@580: } ds@580: ds@582: /* If the forward now has 0 functions in it... */ dvander@2890: if (pForward->GetFunctionCount() == 0 && dvander@2890: !ConVarReentrancyGuard::IsCvarInChain(pConVar)) ds@580: { ds@582: /* Free this forward */ ds@582: g_Forwards.ReleaseForward(pForward); ds@696: pInfo->pChangeForward = NULL; ds@582: } ds@582: } ds@582: } ds@582: ds@696: QueryCvarCookie_t ConVarManager::QueryClientConVar(edict_t *pPlayer, const char *name, IPluginFunction *pCallback, Handle_t hndl) ds@696: { ds@2583: QueryCvarCookie_t cookie = 0; ds@696: ds@2583: #if SOURCE_ENGINE != SE_DARKMESSIAH ds@696: /* Call StartQueryCvarValue() in either the IVEngineServer or IServerPluginHelpers depending on situation */ dvander@1581: if (m_bIsDLLQueryHooked) ds@696: { skamonkey@3913: #if SOURCE_ENGINE == SE_DOTA skamonkey@3942: cookie = engine->StartQueryCvarValue(CEntityIndex(IndexOfEdict(pPlayer)), name); skamonkey@3913: #else ds@696: cookie = engine->StartQueryCvarValue(pPlayer, name); skamonkey@3913: #endif dvander@1581: } skamonkey@3854: #if SOURCE_ENGINE != SE_DOTA dvander@1581: else if (m_bIsVSPQueryHooked) dvander@1581: { ds@696: cookie = serverpluginhelpers->StartQueryCvarValue(pPlayer, name); ds@696: } skamonkey@3854: #endif dvander@1581: else dvander@1581: { dvander@1581: return InvalidQueryCvarCookie; dvander@1581: } ds@696: ds@696: ConVarQuery query = {cookie, pCallback, hndl}; ds@696: m_ConVarQueries.push_back(query); ds@2583: #endif ds@696: ds@696: return cookie; ds@696: } ds@696: ds@582: void ConVarManager::AddConVarToPluginList(IPluginContext *pContext, const ConVar *pConVar) ds@582: { ds@582: ConVarList *pConVarList; ds@582: ConVarList::iterator iter; ds@582: bool inserted = false; ds@582: const char *orig = pConVar->GetName(); ds@582: dvander@3872: IPlugin *plugin = scripts->FindPluginByContext(pContext->GetContext()); ds@580: ds@582: /* Check plugin for an existing convar list */ ds@582: if (!plugin->GetProperty("ConVarList", (void **)&pConVarList)) ds@582: { ds@582: pConVarList = new ConVarList(); ds@582: plugin->SetProperty("ConVarList", pConVarList); dvander@1581: } dvander@1581: else if (pConVarList->find(pConVar) != pConVarList->end()) dvander@1581: { ds@867: /* If convar is already in list, then don't add it */ ds@867: return; ds@582: } ds@582: ds@582: /* Insert convar into list which is sorted alphabetically */ ds@582: for (iter = pConVarList->begin(); iter != pConVarList->end(); iter++) ds@582: { ds@582: if (strcmp(orig, (*iter)->GetName()) < 0) ds@582: { ds@582: pConVarList->insert(iter, pConVar); ds@582: inserted = true; ds@582: break; ds@580: } ds@580: } ds@582: ds@582: if (!inserted) ds@582: { ds@582: pConVarList->push_back(pConVar); ds@582: } ds@580: } ds@580: ds@2400: #if SOURCE_ENGINE >= SE_ORANGEBOX dvander@1770: void ConVarManager::OnConVarChanged(ConVar *pConVar, const char *oldValue, float flOldValue) faluco@1541: #else ds@580: void ConVarManager::OnConVarChanged(ConVar *pConVar, const char *oldValue) faluco@1541: #endif ds@580: { ds@582: /* If the values are the same, exit early in order to not trigger callbacks */ dvander@1544: if (strcmp(pConVar->GetString(), oldValue) == 0) ds@580: { ds@580: return; ds@580: } ds@580: ds@582: ConVarInfo *pInfo; ds@580: ds@582: /* Find the convar in the lookup trie */ dvander@1582: if (!convar_cache_lookup(pConVar->GetName(), &pInfo)) dvander@1582: { dvander@1582: return; dvander@1582: } ds@580: ds@696: IChangeableForward *pForward = pInfo->pChangeForward; ds@580: dvander@1770: if (pInfo->changeListeners.size() != 0) ds@580: { dvander@1770: for (List::iterator i = pInfo->changeListeners.begin(); dvander@1770: i != pInfo->changeListeners.end(); dvander@1770: i++) dvander@1770: { ds@2400: #if SOURCE_ENGINE >= SE_ORANGEBOX dvander@1770: (*i)->OnConVarChanged(pConVar, oldValue, flOldValue); faluco@1541: #else dvander@1770: (*i)->OnConVarChanged(pConVar, oldValue, atof(oldValue)); faluco@1541: #endif dvander@1770: } ds@580: } ds@580: dvander@1770: if (pForward != NULL) dvander@1770: { dvander@2890: ConVarReentrancyGuard guard(pConVar); dvander@2890: dvander@1770: /* Now call forwards in plugins that have hooked this */ dvander@1770: pForward->PushCell(pInfo->handle); dvander@1770: pForward->PushString(oldValue); dvander@1770: pForward->PushString(pConVar->GetString()); dvander@1770: pForward->Execute(NULL); dvander@1770: } ds@580: } ds@627: dvander@1581: bool ConVarManager::IsQueryingSupported() dvander@1581: { dvander@1581: return (m_bIsDLLQueryHooked || m_bIsVSPQueryHooked); dvander@1581: } dvander@1581: ds@2583: #if SOURCE_ENGINE != SE_DARKMESSIAH skamonkey@3942: #if SOURCE_ENGINE == SE_DOTA skamonkey@3942: void ConVarManager::OnQueryCvarValueFinished(QueryCvarCookie_t cookie, CEntityIndex player, EQueryCvarValueStatus result, const char *cvarName, const char *cvarValue) skamonkey@3942: #else ds@696: void ConVarManager::OnQueryCvarValueFinished(QueryCvarCookie_t cookie, edict_t *pPlayer, EQueryCvarValueStatus result, const char *cvarName, const char *cvarValue) skamonkey@3942: #endif // SE_DOTA ds@696: { ds@696: IPluginFunction *pCallback = NULL; ds@696: cell_t value = 0; ds@696: List::iterator iter; ds@696: ds@696: for (iter = m_ConVarQueries.begin(); iter != m_ConVarQueries.end(); iter++) ds@696: { ds@696: ConVarQuery &query = (*iter); ds@696: if (query.cookie == cookie) ds@696: { ds@696: pCallback = query.pCallback; ds@696: value = query.value; ds@696: break; ds@696: } ds@696: } ds@696: ds@696: if (pCallback) ds@696: { ds@696: cell_t ret; ds@696: ds@696: pCallback->PushCell(cookie); skamonkey@3942: #if SOURCE_ENGINE == SE_DOTA skamonkey@3942: pCallback->PushCell(player.Get()); skamonkey@3942: #else ds@2400: pCallback->PushCell(IndexOfEdict(pPlayer)); skamonkey@3942: #endif ds@696: pCallback->PushCell(result); ds@696: pCallback->PushString(cvarName); ds@696: ds@696: if (result == eQueryCvarValueStatus_ValueIntact) ds@696: { ds@696: pCallback->PushString(cvarValue); dvander@1581: } dvander@1581: else dvander@1581: { ds@696: pCallback->PushString("\0"); ds@696: } ds@696: ds@696: pCallback->PushCell(value); ds@696: pCallback->Execute(&ret); ds@696: ds@696: m_ConVarQueries.erase(iter); ds@696: } ds@696: } ds@2583: #endif ds@696: dvander@1582: HandleError ConVarManager::ReadConVarHandle(Handle_t hndl, ConVar **pVar) dvander@1582: { dvander@1582: ConVarInfo *pInfo; dvander@1582: HandleError error; dvander@1582: dvander@3872: if ((error = handlesys->ReadHandle(hndl, m_ConVarType, NULL, (void **)&pInfo)) != HandleError_None) dvander@1582: { dvander@1582: return error; dvander@1582: } dvander@1582: dvander@1582: if (pVar) dvander@1582: { dvander@1582: *pVar = pInfo->pVar; dvander@1582: } dvander@1582: dvander@1582: return error; dvander@1582: }