3 * =============================================================================
5 * Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved.
6 * =============================================================================
8 * This program is free software; you can redistribute it and/or modify it under
9 * the terms of the GNU General Public License, version 3.0, as published by the
10 * Free Software Foundation.
12 * This program is distributed in the hope that it will be useful, but WITHOUT
13 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
14 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
17 * You should have received a copy of the GNU General Public License along with
18 * this program. If not, see <http://www.gnu.org/licenses/>.
20 * As a special exception, AlliedModders LLC gives you permission to link the
21 * code of this program (as well as its derivative works) to "Half-Life 2," the
22 * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
23 * by the Valve Corporation. You must obey the GNU General Public License in
24 * all respects for all other code used. Additionally, AlliedModders LLC grants
25 * this exception to all derivative works. AlliedModders LLC defines further
26 * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
27 * or <http://www.sourcemod.net/license.php>.
33 #include "PluginSys.h"
35 #include "LibrarySys.h"
36 #include "HandleSys.h"
37 #include "ForwardSys.h"
38 #include "sourcemm_api.h"
39 #include "sourcemod.h"
41 #include "ExtensionSys.h"
42 #include "sm_srvcmds.h"
43 #include "sm_stringutil.h"
44 #include "ConCmdManager.h"
45 #include "PlayerManager.h"
46 #include "CoreConfig.h"
47 #include "logic_bridge.h"
49 CPluginManager g_PluginSys;
50 HandleType_t g_PluginType = 0;
51 IdentityType_t g_PluginIdent = 0;
53 CPlugin::CPlugin(const char *file)
55 static int MySerial = 0;
57 m_type = PluginType_Private;
58 m_status = Plugin_Uncompiled;
59 m_bSilentlyFailed = false;
60 m_serial = ++MySerial;
62 m_errormsg[sizeof(m_errormsg) - 1] = '\0';
63 UTIL_Format(m_filename, sizeof(m_filename), "%s", file);
66 m_pProps = sm_trie_create();
67 m_FakeNativesMissing = false;
68 m_LibraryMissing = false;
69 m_bGotAllLoaded = false;
70 m_pPhrases = translator->CreatePhraseCollection();
71 m_MaxClientsVar = NULL;
79 sec.pOwner = g_PluginSys.GetIdentity();
80 sec.pIdentity = sec.pOwner;
82 g_HandleSys.FreeHandle(m_handle, &sec);
83 g_ShareSys.DestroyIdentity(m_ident);
86 if (m_pRuntime != NULL)
94 sm_trie_destroy(m_pProps);
96 for (size_t i=0; i<m_configs.size(); i++)
101 if (m_pPhrases != NULL)
103 m_pPhrases->Destroy();
108 void CPlugin::InitIdentity()
112 m_ident = g_ShareSys.CreateIdentity(g_PluginIdent, this);
113 m_handle = g_HandleSys.CreateHandle(g_PluginType, this, g_PluginSys.GetIdentity(), g_PluginSys.GetIdentity(), NULL);
114 m_pRuntime->GetDefaultContext()->SetKey(1, m_ident);
115 m_pRuntime->GetDefaultContext()->SetKey(2, (IPlugin *)this);
119 unsigned int CPlugin::CalcMemUsage()
121 unsigned int base_size =
123 + sizeof(IdentityToken_t)
124 + (m_configs.size() * (sizeof(AutoConfig *) + sizeof(AutoConfig)))
125 + sm_trie_mem_usage(m_pProps);
127 for (unsigned int i = 0; i < m_configs.size(); i++)
129 base_size += m_configs[i]->autocfg.size();
130 base_size += m_configs[i]->folder.size();
133 for (List<String>::iterator i = m_Libraries.begin();
134 i != m_Libraries.end();
137 base_size += (*i).size();
140 for (List<String>::iterator i = m_RequiredLibs.begin();
141 i != m_RequiredLibs.end();
144 base_size += (*i).size();
150 Handle_t CPlugin::GetMyHandle()
155 CPlugin *CPlugin::CreatePlugin(const char *file, char *error, size_t maxlength)
157 char fullpath[PLATFORM_MAX_PATH];
158 g_SourceMod.BuildPath(Path_SM, fullpath, sizeof(fullpath), "plugins/%s", file);
159 FILE *fp = fopen(fullpath, "rb");
161 CPlugin *pPlugin = new CPlugin(file);
167 UTIL_Format(error, maxlength, "Unable to open file");
169 pPlugin->m_status = Plugin_BadLoad;
178 bool CPlugin::GetProperty(const char *prop, void **ptr, bool remove/* =false */)
180 bool exists = sm_trie_retrieve(m_pProps, prop, ptr);
182 if (exists && remove)
184 sm_trie_delete(m_pProps, prop);
190 bool CPlugin::SetProperty(const char *prop, void *ptr)
192 return sm_trie_insert(m_pProps, prop, ptr);
195 IPluginRuntime *CPlugin::GetRuntime()
200 void CPlugin::SetErrorState(PluginStatus status, const char *error_fmt, ...)
202 PluginStatus old_status = m_status;
205 if (old_status == Plugin_Running)
207 /* Tell everyone we're now paused */
208 g_PluginSys._SetPauseState(this, true);
212 va_start(ap, error_fmt);
213 UTIL_FormatArgs(m_errormsg, sizeof(m_errormsg), error_fmt, ap);
216 if (m_pRuntime != NULL)
218 m_pRuntime->SetPauseState(true);
222 bool CPlugin::UpdateInfo()
224 /* Now grab the info */
226 IPluginContext *base = GetBaseContext();
227 int err = base->FindPubvarByName("myinfo", &idx);
229 memset(&m_info, 0, sizeof(m_info));
231 if (err == SP_ERROR_NONE)
233 struct sm_plugininfo_c_t
241 sm_plugininfo_c_t *cinfo;
244 base->GetPubvarAddrs(idx, &local_addr, (cell_t **)&cinfo);
245 base->LocalToString(cinfo->name, (char **)&m_info.name);
246 base->LocalToString(cinfo->description, (char **)&m_info.description);
247 base->LocalToString(cinfo->author, (char **)&m_info.author);
248 base->LocalToString(cinfo->url, (char **)&m_info.url);
249 base->LocalToString(cinfo->version, (char **)&m_info.version);
252 m_info.author = m_info.author ? m_info.author : "";
253 m_info.description = m_info.description ? m_info.description : "";
254 m_info.name = m_info.name ? m_info.name : "";
255 m_info.url = m_info.url ? m_info.url : "";
256 m_info.version = m_info.version ? m_info.version : "";
258 if ((err = base->FindPubvarByName("__version", &idx)) == SP_ERROR_NONE)
260 struct __version_info
267 __version_info *info;
269 const char *pDate, *pTime, *pFileVers;
274 base->GetPubvarAddrs(idx, &local_addr, (cell_t **)&info);
275 m_FileVersion = info->version;
276 if (m_FileVersion >= 4)
278 base->LocalToString(info->date, (char **)&pDate);
279 base->LocalToString(info->time, (char **)&pTime);
280 UTIL_Format(m_DateTime, sizeof(m_DateTime), "%s %s", pDate, pTime);
282 if (m_FileVersion > 5)
284 base->LocalToString(info->filevers, (char **)&pFileVers);
285 SetErrorState(Plugin_Failed, "Newer SourceMod required (%s or higher)", pFileVers);
294 if ((err = base->FindPubvarByName("MaxClients", &idx)) == SP_ERROR_NONE)
296 base->GetPubvarByIndex(idx, &m_MaxClientsVar);
302 void CPlugin::SyncMaxClients(int max_clients)
304 if (m_MaxClientsVar == NULL)
309 *m_MaxClientsVar->offs = max_clients;
312 void CPlugin::Call_OnPluginStart()
314 if (m_status != Plugin_Loaded)
319 m_status = Plugin_Running;
321 SyncMaxClients(g_Players.MaxClients());
324 IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("OnPluginStart");
331 if ((err=pFunction->Execute(&result)) != SP_ERROR_NONE)
333 SetErrorState(Plugin_Error, "Error detected in plugin startup (see error logs)");
337 void CPlugin::Call_OnPluginEnd()
339 if (m_status > Plugin_Paused)
345 IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("OnPluginEnd");
351 pFunction->Execute(&result);
354 void CPlugin::Call_OnAllPluginsLoaded()
356 if (m_status > Plugin_Paused)
366 m_bGotAllLoaded = true;
369 IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("OnAllPluginsLoaded");
370 if (pFunction != NULL)
372 pFunction->Execute(&result);
375 if (g_SourceMod.IsMapRunning())
377 if ((pFunction = m_pRuntime->GetFunctionByName("OnMapStart")) != NULL)
379 pFunction->Execute(NULL);
383 if (SM_AreConfigsExecuted())
385 SM_ExecuteForPlugin(GetBaseContext());
389 APLRes CPlugin::Call_AskPluginLoad(char *error, size_t maxlength)
391 if (m_status != Plugin_Created)
393 return APLRes_Failure;
396 m_status = Plugin_Loaded;
400 bool haveNewAPL = false;
401 IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("AskPluginLoad2");
407 else if (!(pFunction = m_pRuntime->GetFunctionByName("AskPluginLoad")))
409 return APLRes_Success;
412 pFunction->PushCell(m_handle);
413 pFunction->PushCell(g_PluginSys.IsLateLoadTime() ? 1 : 0);
414 pFunction->PushStringEx(error, maxlength, 0, SM_PARAM_COPYBACK);
415 pFunction->PushCell(maxlength);
416 if ((err=pFunction->Execute(&result)) != SP_ERROR_NONE)
418 return APLRes_Failure;
423 return (APLRes)result;
427 return APLRes_Success;
431 return APLRes_Failure;
435 void *CPlugin::GetPluginStructure()
440 IPluginContext *CPlugin::GetBaseContext()
442 if (m_pRuntime == NULL)
447 return m_pRuntime->GetDefaultContext();
450 sp_context_t *CPlugin::GetContext()
455 const char *CPlugin::GetFilename()
460 PluginType CPlugin::GetType()
465 const sm_plugininfo_t *CPlugin::GetPublicInfo()
467 if (GetStatus() >= Plugin_Created)
474 unsigned int CPlugin::GetSerial()
479 PluginStatus CPlugin::GetStatus()
484 bool CPlugin::IsSilentlyFailed()
486 return m_bSilentlyFailed;
489 bool CPlugin::IsDebugging()
491 if (m_pRuntime == NULL)
499 void CPlugin::SetSilentlyFailed(bool sf)
501 m_bSilentlyFailed = sf;
504 void CPlugin::LibraryActions(bool dropping)
506 List<String>::iterator iter;
507 for (iter = m_Libraries.begin();
508 iter != m_Libraries.end();
511 g_PluginSys.OnLibraryAction((*iter).c_str(), true, dropping);
515 bool CPlugin::SetPauseState(bool paused)
517 if (paused && GetStatus() != Plugin_Running)
520 } else if (!paused && GetStatus() != Plugin_Paused) {
526 LibraryActions(true);
529 IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("OnPluginPauseChange");
533 pFunction->PushCell(paused ? 1 : 0);
534 pFunction->Execute(&result);
539 m_status = Plugin_Paused;
540 m_pRuntime->SetPauseState(true);
542 m_status = Plugin_Running;
543 m_pRuntime->SetPauseState(false);
546 g_PluginSys._SetPauseState(this, paused);
550 LibraryActions(false);
556 IdentityToken_t *CPlugin::GetIdentity()
561 bool CPlugin::IsRunnable()
563 return (m_status <= Plugin_Paused) ? true : false;
566 time_t CPlugin::GetFileTimeStamp()
568 char path[PLATFORM_MAX_PATH];
569 g_SourceMod.BuildPath(Path_SM, path, sizeof(path), "plugins/%s", m_filename);
570 #ifdef PLATFORM_WINDOWS
572 if (_stat(path, &s) != 0)
573 #elif defined PLATFORM_POSIX
575 if (stat(path, &s) != 0)
586 time_t CPlugin::GetTimeStamp()
591 void CPlugin::SetTimeStamp(time_t t)
596 IPhraseCollection *CPlugin::GetPhrases()
601 void CPlugin::DependencyDropped(CPlugin *pOwner)
608 List<String>::iterator reqlib_iter;
609 List<String>::iterator lib_iter;
610 for (lib_iter=pOwner->m_Libraries.begin(); lib_iter!=pOwner->m_Libraries.end(); lib_iter++)
612 for (reqlib_iter=m_RequiredLibs.begin(); reqlib_iter!=m_RequiredLibs.end(); reqlib_iter++)
614 if ((*reqlib_iter) == (*lib_iter))
616 m_LibraryMissing = true;
621 List<NativeEntry *>::iterator iter;
622 NativeEntry *pNative;
625 unsigned int unbound = 0;
627 for (iter = pOwner->m_Natives.begin();
628 iter != pOwner->m_Natives.end();
632 /* Find this native! */
633 if (m_pRuntime->FindNativeByName(pNative->name, &idx) != SP_ERROR_NONE)
638 m_pRuntime->GetNativeByIndex(idx, &native);
640 native->status = SP_NATIVE_UNBOUND;
646 m_FakeNativesMissing = true;
649 /* :IDEA: in the future, add native trapping? */
650 if (m_FakeNativesMissing || m_LibraryMissing)
652 SetErrorState(Plugin_Error, "Depends on plugin: %s", pOwner->GetFilename());
656 unsigned int CPlugin::GetConfigCount()
658 return (unsigned int)m_configs.size();
661 AutoConfig *CPlugin::GetConfig(unsigned int i)
663 if (i >= GetConfigCount())
671 void CPlugin::AddConfig(bool autoCreate, const char *cfg, const char *folder)
673 /* Do a check for duplicates to prevent double-execution */
674 for (size_t i = 0; i < m_configs.size(); i++)
676 if (m_configs[i]->autocfg.compare(cfg) == 0
677 && m_configs[i]->folder.compare(folder) == 0
678 && m_configs[i]->create == autoCreate)
684 AutoConfig *c = new AutoConfig;
688 c->create = autoCreate;
690 m_configs.push_back(c);
693 void CPlugin::DropEverything()
696 List<CPlugin *>::iterator iter;
697 List<WeakNative>::iterator wk_iter;
699 /* Tell everyone that depends on us that we're about to drop */
700 for (iter = m_Dependents.begin();
701 iter != m_Dependents.end();
705 pOther->DependencyDropped(this);
708 /* Note: we don't care about things we depend on.
709 * The reason is that extensions have their own cleanup
710 * code for plugins. Although the "right" design would be
711 * to centralize that here, i'm omitting it for now. Thus,
712 * the code below to walk the plugins list will suffice.
715 /* Other plugins could be holding weak references that were
716 * added by us. We need to clean all of those up now.
718 for (iter = g_PluginSys.m_plugins.begin();
719 iter != g_PluginSys.m_plugins.end();
722 (*iter)->DropRefsTo(this);
725 /* Proceed with the rest of the necessities. */
726 CNativeOwner::DropEverything();
729 bool CPlugin::AddFakeNative(IPluginFunction *pFunc, const char *name, SPVM_FAKENATIVE_FUNC func)
733 if ((pEntry = g_ShareSys.AddFakeNative(pFunc, name, func)) == NULL)
738 m_Natives.push_back(pEntry);
747 CPluginManager::CPluginIterator::CPluginIterator(List<CPlugin *> *_mylist)
753 IPlugin *CPluginManager::CPluginIterator::GetPlugin()
758 bool CPluginManager::CPluginIterator::MorePlugins()
760 return (current != mylist->end());
763 void CPluginManager::CPluginIterator::NextPlugin()
768 void CPluginManager::CPluginIterator::Release()
770 g_PluginSys.ReleaseIterator(this);
773 CPluginManager::CPluginIterator::~CPluginIterator()
777 void CPluginManager::CPluginIterator::Reset()
779 current = mylist->begin();
786 CPluginManager::CPluginManager()
788 m_LoadLookup = sm_trie_create();
789 m_AllPluginsLoaded = false;
791 m_LoadingLocked = false;
793 m_bBlockBadPlugins = true;
796 CPluginManager::~CPluginManager()
799 * Ignore the fact that there might be plugins in the cache.
800 * This usually means that Core is not being unloaded properly, and everything
801 * will crash anyway. YAY
803 sm_trie_destroy(m_LoadLookup);
805 CStack<CPluginManager::CPluginIterator *>::iterator iter;
806 for (iter=m_iters.begin(); iter!=m_iters.end(); iter++)
813 void CPluginManager::Shutdown()
815 List<CPlugin *>::iterator iter;
817 while ((iter = m_plugins.begin()) != m_plugins.end())
823 void CPluginManager::LoadAll_FirstPass(const char *config, const char *basedir)
825 /* First read in the database of plugin settings */
828 m_AllPluginsLoaded = false;
829 if ((err=textparsers->ParseFile_SMC(config, &m_PluginInfo, &states)) != SMCError_Okay)
831 g_Logger.LogError("[SM] Encountered fatal error parsing file \"%s\"", config);
832 const char *err_msg = textparsers->GetSMCErrorString(err);
835 g_Logger.LogError("[SM] Parse error encountered: \"%s\"", err_msg);
839 LoadPluginsFromDir(basedir, NULL);
842 void CPluginManager::LoadPluginsFromDir(const char *basedir, const char *localpath)
844 char base_path[PLATFORM_MAX_PATH];
846 /* Form the current path to start reading from */
847 if (localpath == NULL)
849 g_LibSys.PathFormat(base_path, sizeof(base_path), "%s", basedir);
851 g_LibSys.PathFormat(base_path, sizeof(base_path), "%s/%s", basedir, localpath);
854 IDirectory *dir = g_LibSys.OpenDirectory(base_path);
859 g_LibSys.GetPlatformError(error, sizeof(error));
860 g_Logger.LogError("[SM] Failure reading from plugins path: %s", localpath);
861 g_Logger.LogError("[SM] Platform returned error: %s", error);
865 while (dir->MoreFiles())
867 if (dir->IsEntryDirectory()
868 && (strcmp(dir->GetEntryName(), ".") != 0)
869 && (strcmp(dir->GetEntryName(), "..") != 0)
870 && (strcmp(dir->GetEntryName(), "disabled") != 0)
871 && (strcmp(dir->GetEntryName(), "optional") != 0))
873 char new_local[PLATFORM_MAX_PATH];
874 if (localpath == NULL)
876 /* If no path yet, don't add a former slash */
877 UTIL_Format(new_local, sizeof(new_local), "%s", dir->GetEntryName());
879 g_LibSys.PathFormat(new_local, sizeof(new_local), "%s/%s", localpath, dir->GetEntryName());
881 LoadPluginsFromDir(basedir, new_local);
882 } else if (dir->IsEntryFile()) {
883 const char *name = dir->GetEntryName();
884 size_t len = strlen(name);
886 && strcmp(&name[len-4], ".smx") == 0)
888 /* If the filename matches, load the plugin */
889 char plugin[PLATFORM_MAX_PATH];
890 if (localpath == NULL)
892 UTIL_Format(plugin, sizeof(plugin), "%s", name);
894 g_LibSys.PathFormat(plugin, sizeof(plugin), "%s/%s", localpath, name);
896 LoadAutoPlugin(plugin);
901 g_LibSys.CloseDirectory(dir);
904 LoadRes CPluginManager::_LoadPlugin(CPlugin **_plugin, const char *path, bool debug, PluginType type, char error[], size_t maxlength)
908 return LoadRes_NeverLoad;
912 bool no_load = false;
913 PluginSettings *pset;
914 unsigned int setcount = m_PluginInfo.GetSettingsNum();
915 for (unsigned int i=0; i<setcount; i++)
917 if ((pset = m_PluginInfo.GetSettingsIfMatch(i, path)) == NULL)
921 if (pset->blockload_val)
929 * Does this plugin already exist?
932 if (sm_trie_retrieve(m_LoadLookup, path, (void **)&pPlugin))
934 /* Check to see if we should try reloading it */
935 if (pPlugin->GetStatus() == Plugin_BadLoad
936 || pPlugin->GetStatus() == Plugin_Error
937 || pPlugin->GetStatus() == Plugin_Failed
940 UnloadPlugin(pPlugin);
948 return LoadRes_AlreadyLoaded;
954 return LoadRes_NeverLoad;
957 pPlugin = CPlugin::CreatePlugin(path, error, maxlength);
959 assert(pPlugin != NULL);
961 pPlugin->m_type = PluginType_MapUpdated;
963 ICompilation *co = NULL;
965 if (pPlugin->m_status == Plugin_Uncompiled)
967 co = g_pSourcePawn2->StartCompilation();
970 for (unsigned int i=0; i<setcount; i++)
972 if ((pset=m_PluginInfo.GetSettingsIfMatch(i, path)) == NULL)
976 pPlugin->m_type = pset->type_val;
979 for (unsigned int j=0; j<pset->opts_num; j++)
981 const char *key, *val;
982 m_PluginInfo.GetOptionsForPlugin(pset, j, &key, &val);
987 if ((err = co->SetOption(key, val)) == SP_ERROR_NONE)
991 UTIL_Format(error, maxlength, "Unable to set JIT option (key \"%s\") (value \"%s\")", key, val);
1001 /* Do the actual compiling */
1004 char fullpath[PLATFORM_MAX_PATH];
1005 g_SourceMod.BuildPath(Path_SM, fullpath, sizeof(fullpath), "plugins/%s", pPlugin->m_filename);
1007 pPlugin->m_pRuntime = g_pSourcePawn2->LoadPlugin(co, fullpath, &err);
1008 if (pPlugin->m_pRuntime == NULL)
1014 "Unable to load plugin (error %d: %s)",
1016 g_pSourcePawn2->GetErrorString(err));
1018 pPlugin->m_status = Plugin_BadLoad;
1022 if (pPlugin->UpdateInfo())
1024 pPlugin->m_status = Plugin_Created;
1030 UTIL_Format(error, maxlength, "%s", pPlugin->m_errormsg);
1036 if (pPlugin->GetStatus() == Plugin_Created)
1038 unsigned char *pCodeHash = pPlugin->m_pRuntime->GetCodeHash();
1040 char codeHashBuf[40];
1041 UTIL_Format(codeHashBuf, 40, "plugin_");
1042 for (int i = 0; i < 16; i++)
1043 UTIL_Format(codeHashBuf + 7 + (i * 2), 3, "%02x", pCodeHash[i]);
1045 const char *bulletinUrl = g_pGameConf->GetKeyValue(codeHashBuf);
1046 if (bulletinUrl != NULL)
1048 if (m_bBlockBadPlugins)
1052 if (bulletinUrl[0] != '\0')
1054 UTIL_Format(error, maxlength, "Known malware detected and blocked. See %s for more info", bulletinUrl);
1056 UTIL_Format(error, maxlength, "Possible malware or illegal plugin detected and blocked");
1059 pPlugin->m_status = Plugin_BadLoad;
1061 if (bulletinUrl[0] != '\0')
1063 g_Logger.LogMessage("%s: Known malware detected. See %s for more info, blocking disabled in core.cfg", pPlugin->GetFilename(), bulletinUrl);
1065 g_Logger.LogMessage("%s: Possible malware or illegal plugin detected, blocking disabled in core.cfg", pPlugin->GetFilename());
1071 LoadRes loadFailure = LoadRes_Failure;
1072 /* Get the status */
1073 if (pPlugin->GetStatus() == Plugin_Created)
1075 /* First native pass - add anything from Core */
1076 g_ShareSys.BindNativesToPlugin(pPlugin, true);
1077 pPlugin->InitIdentity();
1078 APLRes result = pPlugin->Call_AskPluginLoad(error, maxlength);
1081 case APLRes_Success:
1082 /* Autoload any modules */
1083 LoadOrRequireExtensions(pPlugin, 1, error, maxlength);
1085 case APLRes_Failure:
1086 pPlugin->SetErrorState(Plugin_Failed, "%s", error);
1087 loadFailure = LoadRes_Failure;
1089 case APLRes_SilentFailure:
1090 pPlugin->SetErrorState(Plugin_Failed, "%s", error);
1091 loadFailure = LoadRes_SilentFailure;
1092 pPlugin->SetSilentlyFailed(true);
1099 /* Save the time stamp */
1100 time_t t = pPlugin->GetFileTimeStamp();
1101 pPlugin->SetTimeStamp(t);
1108 return (pPlugin->GetStatus() == Plugin_Loaded) ? LoadRes_Successful : loadFailure;
1111 IPlugin *CPluginManager::LoadPlugin(const char *path, bool debug, PluginType type, char error[], size_t maxlength, bool *wasloaded)
1117 if ((res=_LoadPlugin(&pl, path, true, type, error, maxlength)) == LoadRes_Failure)
1123 if (res == LoadRes_AlreadyLoaded)
1129 if (res == LoadRes_NeverLoad)
1133 if (m_LoadingLocked)
1135 UTIL_Format(error, maxlength, "There is a global plugin loading lock in effect");
1139 UTIL_Format(error, maxlength, "This plugin is blocked from loading (see plugin_settings.cfg)");
1147 /* Run second pass if we need to */
1148 if (IsLateLoadTime() && pl->GetStatus() == Plugin_Loaded)
1150 if (!RunSecondPass(pl, error, maxlength))
1155 pl->Call_OnAllPluginsLoaded();
1161 void CPluginManager::LoadAutoPlugin(const char *plugin)
1165 char error[255] = "Unknown error";
1167 if ((res=_LoadPlugin(&pl, plugin, false, PluginType_MapUpdated, error, sizeof(error))) == LoadRes_Failure)
1169 g_Logger.LogError("[SM] Failed to load plugin \"%s\": %s.", plugin, error);
1171 pl->GetStatus() <= Plugin_Created ? Plugin_BadLoad : pl->GetStatus(),
1176 if (res == LoadRes_Successful || res == LoadRes_Failure || res == LoadRes_SilentFailure)
1182 void CPluginManager::AddPlugin(CPlugin *pPlugin)
1184 List<IPluginsListener *>::iterator iter;
1185 IPluginsListener *pListener;
1187 for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
1189 pListener = (*iter);
1190 pListener->OnPluginCreated(pPlugin);
1193 m_plugins.push_back(pPlugin);
1194 sm_trie_insert(m_LoadLookup, pPlugin->m_filename, pPlugin);
1197 void CPluginManager::LoadAll_SecondPass()
1199 List<CPlugin *>::iterator iter;
1203 for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
1206 if (pPlugin->GetStatus() == Plugin_Loaded)
1209 if (!RunSecondPass(pPlugin, error, sizeof(error)))
1211 g_Logger.LogError("[SM] Unable to load plugin \"%s\": %s", pPlugin->GetFilename(), error);
1212 pPlugin->SetErrorState(Plugin_Failed, "%s", error);
1217 m_AllPluginsLoaded = true;
1220 bool CPluginManager::FindOrRequirePluginDeps(CPlugin *pPlugin, char *error, size_t maxlength)
1229 IPluginContext *pBase = pPlugin->GetBaseContext();
1230 uint32_t num = pBase->GetPubVarsNum();
1231 sp_pubvar_t *pubvar;
1233 char pathfile[PLATFORM_MAX_PATH];
1235 for (uint32_t i=0; i<num; i++)
1237 if (pBase->GetPubvarByIndex(i, &pubvar) != SP_ERROR_NONE)
1241 if (strncmp(pubvar->name, "__pl_", 5) == 0)
1243 pl = (_pl *)pubvar->offs;
1244 if (pBase->LocalToString(pl->file, &file) != SP_ERROR_NONE)
1248 if (pBase->LocalToString(pl->name, &name) != SP_ERROR_NONE)
1252 g_LibSys.GetFileFromPath(pathfile, sizeof(pathfile), pPlugin->GetFilename());
1253 if (strcmp(pathfile, file) == 0)
1257 if (pl->required == false)
1259 IPluginFunction *pFunc;
1261 UTIL_Format(buffer, sizeof(buffer), "__pl_%s_SetNTVOptional", &pubvar->name[5]);
1262 if ((pFunc=pBase->GetFunctionByName(buffer)))
1265 pFunc->Execute(&res);
1266 if (pPlugin->GetBaseContext()->GetLastNativeError() != SP_ERROR_NONE)
1270 UTIL_Format(error, maxlength, "Fatal error during initializing plugin load");
1278 /* Check that we aren't registering the same library twice */
1279 if (pPlugin->m_RequiredLibs.find(name) == pPlugin->m_RequiredLibs.end())
1281 pPlugin->m_RequiredLibs.push_back(name);
1287 List<CPlugin *>::iterator iter;
1290 for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
1293 if (pl->m_Libraries.find(name) != pl->m_Libraries.end())
1303 UTIL_Format(error, maxlength, "Could not find required plugin \"%s\"", name);
1314 bool CPluginManager::LoadOrRequireExtensions(CPlugin *pPlugin, unsigned int pass, char *error, size_t maxlength)
1316 /* Find any extensions this plugin needs */
1325 IPluginContext *pBase = pPlugin->GetBaseContext();
1326 uint32_t num = pBase->GetPubVarsNum();
1327 sp_pubvar_t *pubvar;
1329 char path[PLATFORM_MAX_PATH];
1331 for (uint32_t i=0; i<num; i++)
1333 if (pBase->GetPubvarByIndex(i, &pubvar) != SP_ERROR_NONE)
1337 if (strncmp(pubvar->name, "__ext_", 6) == 0)
1339 ext = (_ext *)pubvar->offs;
1340 if (pBase->LocalToString(ext->file, &file) != SP_ERROR_NONE)
1344 if (pBase->LocalToString(ext->name, &name) != SP_ERROR_NONE)
1350 /* Attempt to auto-load if necessary */
1353 g_LibSys.PathFormat(path, PLATFORM_MAX_PATH, "%s", file);
1354 bool bErrorOnMissing = ext->required ? true : false;
1355 g_Extensions.LoadAutoExtension(path, bErrorOnMissing);
1360 /* Is this required? */
1363 g_LibSys.PathFormat(path, PLATFORM_MAX_PATH, "%s", file);
1364 if ((pExt = g_Extensions.FindExtensionByFile(path)) == NULL)
1366 pExt = g_Extensions.FindExtensionByName(name);
1368 /* :TODO: should we bind to unloaded extensions?
1369 * Currently the extension manager will ignore this.
1371 if (!pExt || !pExt->IsRunning(NULL, 0))
1375 UTIL_Format(error, maxlength, "Required extension \"%s\" file(\"%s\") not running", name, file);
1381 g_Extensions.BindChildPlugin(pExt, pPlugin);
1386 IPluginFunction *pFunc;
1388 UTIL_Format(buffer, sizeof(buffer), "__ext_%s_SetNTVOptional", &pubvar->name[6]);
1390 if ((pFunc = pBase->GetFunctionByName(buffer)) != NULL)
1393 pFunc->Execute(&res);
1394 if (pPlugin->GetBaseContext()->GetLastNativeError() != SP_ERROR_NONE)
1398 UTIL_Format(error, maxlength, "Fatal error during plugin initialization (ext req)");
1411 bool CPluginManager::RunSecondPass(CPlugin *pPlugin, char *error, size_t maxlength)
1413 /* Second pass for extension requirements */
1414 if (!LoadOrRequireExtensions(pPlugin, 2, error, maxlength))
1419 if (!FindOrRequirePluginDeps(pPlugin, error, maxlength))
1424 /* Run another binding pass */
1425 g_ShareSys.BindNativesToPlugin(pPlugin, false);
1427 /* Find any unbound natives. Right now, these are not allowed. */
1428 IPluginContext *pContext = pPlugin->GetBaseContext();
1429 uint32_t num = pContext->GetNativesNum();
1430 sp_native_t *native;
1431 for (unsigned int i=0; i<num; i++)
1433 if (pContext->GetNativeByIndex(i, &native) != SP_ERROR_NONE)
1437 if (native->status == SP_NATIVE_UNBOUND
1438 && native->name[0] != '@'
1439 && !(native->flags & SP_NTVFLAG_OPTIONAL))
1443 UTIL_Format(error, maxlength, "Native \"%s\" was not found", native->name);
1449 /* Finish by telling all listeners */
1450 List<IPluginsListener *>::iterator iter;
1451 IPluginsListener *pListener;
1452 for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
1454 pListener = (*iter);
1455 pListener->OnPluginLoaded(pPlugin);
1458 /* Tell this plugin to finish initializing itself */
1459 pPlugin->Call_OnPluginStart();
1461 /* Now, if we have fake natives, go through all plugins that might need rebinding */
1462 if (pPlugin->GetStatus() <= Plugin_Paused && pPlugin->m_Natives.size())
1464 List<CPlugin *>::iterator pl_iter;
1466 for (pl_iter = m_plugins.begin();
1467 pl_iter != m_plugins.end();
1470 pOther = (*pl_iter);
1471 if ((pOther->GetStatus() == Plugin_Error
1472 && (pOther->m_FakeNativesMissing || pOther->m_LibraryMissing))
1473 || pOther->m_FakeNativesMissing)
1475 TryRefreshDependencies(pOther);
1477 else if ((pOther->GetStatus() == Plugin_Running
1478 || pOther->GetStatus() == Plugin_Paused)
1479 && pOther != pPlugin)
1481 List<NativeEntry *>::iterator nv_iter;
1482 for (nv_iter = pPlugin->m_Natives.begin();
1483 nv_iter != pPlugin->m_Natives.end();
1486 g_ShareSys.BindNativeToPlugin(pOther, (*nv_iter));
1492 /* Go through our libraries and tell other plugins they're added */
1493 List<String>::iterator s_iter;
1494 for (s_iter = pPlugin->m_Libraries.begin();
1495 s_iter != pPlugin->m_Libraries.end();
1498 OnLibraryAction((*s_iter).c_str(), true, false);
1501 /* :TODO: optimize? does this even matter? */
1502 pPlugin->GetPhrases()->AddPhraseFile("core.phrases");
1507 void CPluginManager::TryRefreshDependencies(CPlugin *pPlugin)
1509 assert(pPlugin->GetBaseContext() != NULL);
1511 g_ShareSys.BindNativesToPlugin(pPlugin, false);
1513 List<String>::iterator lib_iter;
1514 List<String>::iterator req_iter;
1515 List<CPlugin *>::iterator pl_iter;
1517 for (req_iter=pPlugin->m_RequiredLibs.begin(); req_iter!=pPlugin->m_RequiredLibs.end(); req_iter++)
1520 for (pl_iter=m_plugins.begin(); pl_iter!=m_plugins.end(); pl_iter++)
1523 for (lib_iter=pl->m_Libraries.begin(); lib_iter!=pl->m_Libraries.end(); lib_iter++)
1525 if ((*req_iter) == (*lib_iter))
1533 pPlugin->SetErrorState(Plugin_Error, "Library not found: %s", (*req_iter).c_str());
1538 /* Find any unbound natives
1539 * Right now, these are not allowed
1541 IPluginContext *pContext = pPlugin->GetBaseContext();
1542 uint32_t num = pContext->GetNativesNum();
1543 sp_native_t *native;
1544 for (unsigned int i=0; i<num; i++)
1546 if (pContext->GetNativeByIndex(i, &native) != SP_ERROR_NONE)
1550 if (native->status == SP_NATIVE_UNBOUND
1551 && native->name[0] != '@'
1552 && !(native->flags & SP_NTVFLAG_OPTIONAL))
1554 pPlugin->SetErrorState(Plugin_Error, "Native not found: %s", native->name);
1559 if (pPlugin->GetStatus() == Plugin_Error)
1561 /* If we got here, all natives are okay again! */
1562 pPlugin->m_status = Plugin_Running;
1563 if (pPlugin->m_pRuntime->IsPaused())
1565 pPlugin->m_pRuntime->SetPauseState(false);
1566 _SetPauseState(pPlugin, false);
1571 bool CPluginManager::UnloadPlugin(IPlugin *plugin)
1573 CPlugin *pPlugin = (CPlugin *)plugin;
1575 /* This prevents removal during insertion or anything else weird */
1576 if (m_plugins.find(pPlugin) == m_plugins.end())
1581 IPluginContext *pContext = plugin->GetBaseContext();
1582 if (pContext != NULL && pContext->IsInExec())
1585 UTIL_Format(buffer, sizeof(buffer), "sm plugins unload %s\n", plugin->GetFilename());
1586 engine->ServerCommand(buffer);
1590 /* Remove us from the lookup table and linked list */
1591 m_plugins.remove(pPlugin);
1592 sm_trie_delete(m_LoadLookup, pPlugin->m_filename);
1594 /* Go through our libraries and tell other plugins they're gone */
1595 List<String>::iterator s_iter;
1596 for (s_iter = pPlugin->m_Libraries.begin();
1597 s_iter != pPlugin->m_Libraries.end();
1600 OnLibraryAction((*s_iter).c_str(), true, true);
1603 List<IPluginsListener *>::iterator iter;
1604 IPluginsListener *pListener;
1606 if (pPlugin->GetStatus() <= Plugin_Error)
1608 /* Notify listeners of unloading */
1609 for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
1611 pListener = (*iter);
1612 pListener->OnPluginUnloaded(pPlugin);
1615 pPlugin->Call_OnPluginEnd();
1618 pPlugin->DropEverything();
1620 for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
1622 /* Notify listeners of destruction */
1623 pListener = (*iter);
1624 pListener->OnPluginDestroyed(pPlugin);
1627 /* Tell the plugin to delete itself */
1633 IPlugin *CPluginManager::FindPluginByContext(const sp_context_t *ctx)
1636 IPluginContext *pContext;
1638 pContext = reinterpret_cast<IPluginContext *>(const_cast<sp_context_t *>(ctx));
1640 if (pContext->GetKey(2, (void **)&pPlugin))
1648 CPlugin *CPluginManager::GetPluginByCtx(const sp_context_t *ctx)
1650 return (CPlugin *)FindPluginByContext(ctx);
1653 unsigned int CPluginManager::GetPluginCount()
1655 return m_plugins.size();
1658 void CPluginManager::AddPluginsListener(IPluginsListener *listener)
1660 m_listeners.push_back(listener);
1663 void CPluginManager::RemovePluginsListener(IPluginsListener *listener)
1665 m_listeners.remove(listener);
1668 IPluginIterator *CPluginManager::GetPluginIterator()
1670 if (m_iters.empty())
1672 return new CPluginIterator(&m_plugins);
1674 CPluginIterator *iter = m_iters.front();
1681 void CPluginManager::ReleaseIterator(CPluginIterator *iter)
1686 bool CPluginManager::TestAliasMatch(const char *alias, const char *localpath)
1688 /* As an optimization, we do not call strlen, but compute the length in the first pass */
1689 size_t alias_len = 0;
1690 size_t local_len = 0;
1692 const char *ptr = alias;
1693 unsigned int alias_explicit_paths = 0;
1694 unsigned int alias_path_end = 0;
1695 while (*ptr != '\0')
1697 if (*ptr == '\\' || *ptr == '/')
1699 alias_explicit_paths++;
1700 alias_path_end = alias_len;
1706 if (alias_explicit_paths && alias_path_end == alias_len - 1)
1708 /* Trailing slash is totally invalid here */
1713 unsigned int local_explicit_paths = 0;
1714 unsigned int local_path_end = 0;
1715 while (*ptr != '\0')
1717 if (*ptr == '\\' || *ptr == '/')
1719 local_explicit_paths++;
1720 local_path_end = local_len;
1726 /* If the alias has more explicit paths than the real path,
1727 * no match will be possible.
1729 if (alias_explicit_paths > local_explicit_paths)
1734 if (alias_explicit_paths)
1736 /* We need to find if the paths match now. For example, these should all match:
1738 * csdm optional/csdm
1739 * csdm/ban optional/crab/csdm/ban
1741 const char *aliasptr = alias;
1742 const char *localptr = localpath;
1746 if (*aliasptr != *localptr)
1748 /* We have to knock one path off */
1749 local_explicit_paths--;
1750 if (alias_explicit_paths > local_explicit_paths)
1752 /* Skip out if we're gonna have an impossible match */
1755 /* Eat up localptr tokens until we get a result */
1756 while (((localptr - localpath) < (int)local_path_end)
1758 && *localptr != '\\')
1762 /* Check if we hit the end of our searchable area.
1763 * This probably isn't possible because of the path
1764 * count check, but it's a good idea anyway.
1766 if ((localptr - localpath) >= (int)local_path_end)
1770 /* Consume the slash token */
1773 /* Reset the alias pointer so we can continue consuming */
1779 * This is safe because if localptr terminates early, aliasptr will too
1783 /* We should never reach the end of the string because of this check. */
1784 bool aliasend = (aliasptr - alias) > (int)alias_path_end;
1785 bool localend = (localptr - localpath) > (int)local_path_end;
1786 if (aliasend || localend)
1788 if (aliasend && localend)
1790 /* we matched, and we can break out now */
1794 /* Otherwise, we've hit the end somehow and rest won't match up. Break out. */
1799 /* If we got here, it's safe to compare the next two tokens */
1800 if (*localptr != *aliasptr)
1811 /* If we got here, it's time to compare filenames */
1812 const char *aliasptr = alias;
1813 const char *localptr = localpath;
1815 if (alias_explicit_paths)
1817 aliasptr = &alias[alias_path_end + 1];
1820 if (local_explicit_paths)
1822 localptr = &localpath[local_path_end + 1];
1827 if (*aliasptr == '*')
1829 /* First, see if this is the last character */
1830 if ((unsigned)(aliasptr - alias) == alias_len - 1)
1832 /* If so, there's no need to match anything else */
1835 /* Otherwise, we need to search for an appropriate matching sequence in local.
1836 * Note that we only need to search up to the next asterisk.
1840 const char *local_orig = localptr;
1844 while (*aliasptr != '\0' && *aliasptr != '*')
1846 /* Since aliasptr is never '\0', localptr hitting the end will fail */
1847 if (*aliasptr != *localptr)
1857 /* If we didn't get a match, we need to advance the search stream.
1858 * This will let us skip tokens while still searching for another match.
1860 localptr = ++local_orig;
1861 /* Make sure we don't go out of bounds */
1862 if (*localptr == '\0')
1873 /* If we got a match, move on to the next token */
1876 } else if (*aliasptr == '\0') {
1877 if (*localptr == '\0'
1879 strcmp(localptr, ".smx") == 0)
1885 } else if (*aliasptr != *localptr) {
1895 bool CPluginManager::IsLateLoadTime() const
1897 return (m_AllPluginsLoaded || !g_SourceMod.IsMapLoading());
1900 void CPluginManager::OnSourceModAllInitialized()
1902 m_MyIdent = g_ShareSys.CreateCoreIdentity();
1905 g_HandleSys.InitAccessDefaults(NULL, &sec);
1907 sec.access[HandleAccess_Delete] = HANDLE_RESTRICT_IDENTITY;
1908 sec.access[HandleAccess_Clone] = HANDLE_RESTRICT_IDENTITY;
1910 g_PluginType = g_HandleSys.CreateType("Plugin", this, 0, NULL, &sec, m_MyIdent, NULL);
1911 g_PluginIdent = g_ShareSys.CreateIdentType("PLUGIN");
1913 g_RootMenu.AddRootConsoleCommand("plugins", "Manage Plugins", this);
1915 g_ShareSys.AddInterface(NULL, this);
1918 void CPluginManager::OnSourceModShutdown()
1920 g_RootMenu.RemoveRootConsoleCommand("plugins", this);
1924 g_HandleSys.RemoveType(g_PluginType, m_MyIdent);
1925 g_ShareSys.DestroyIdentType(g_PluginIdent);
1926 g_ShareSys.DestroyIdentity(m_MyIdent);
1929 ConfigResult CPluginManager::OnSourceModConfigChanged(const char *key,
1931 ConfigSource source,
1935 if (strcmp(key, "BlockBadPlugins") == 0) {
1936 if (strcasecmp(value, "yes") == 0)
1938 m_bBlockBadPlugins = true;
1939 } else if (strcasecmp(value, "no") == 0) {
1940 m_bBlockBadPlugins = false;
1942 UTIL_Format(error, maxlength, "Invalid value: must be \"yes\" or \"no\"");
1943 return ConfigResult_Reject;
1945 return ConfigResult_Accept;
1947 return ConfigResult_Ignore;
1950 void CPluginManager::OnHandleDestroy(HandleType_t type, void *object)
1952 /* We don't care about the internal object, actually */
1955 bool CPluginManager::GetHandleApproxSize(HandleType_t type, void *object, unsigned int *pSize)
1957 *pSize = ((CPlugin *)object)->CalcMemUsage();
1961 IPlugin *CPluginManager::PluginFromHandle(Handle_t handle, HandleError *err)
1969 sec.pIdentity = m_MyIdent;
1971 if ((_err=g_HandleSys.ReadHandle(handle, g_PluginType, &sec, (void **)&pPlugin)) != HandleError_None)
1984 CPlugin *CPluginManager::GetPluginByOrder(int num)
1986 if (num < 1 || num > (int)GetPluginCount())
1994 SourceHook::List<CPlugin *>::iterator iter;
1995 for (iter = m_plugins.begin(); iter != m_plugins.end() && id < num; iter++, id++) {}
2002 const char *CPluginManager::GetStatusText(PluginStatus st)
2006 case Plugin_Running:
2012 case Plugin_Uncompiled:
2013 return "Uncompiled";
2014 case Plugin_BadLoad:
2024 void CPluginManager::OnRootConsoleCommand(const char *cmdname, const CCommand &command)
2026 int argcount = command.ArgC();
2029 const char *cmd = command.Arg(2);
2030 if (strcmp(cmd, "list") == 0)
2033 unsigned int id = 1;
2034 int plnum = GetPluginCount();
2038 g_RootMenu.ConsolePrint("[SM] No plugins loaded");
2043 g_RootMenu.ConsolePrint("[SM] Listing %d plugin%s:", GetPluginCount(), (plnum > 1) ? "s" : "");
2047 SourceHook::List<CPlugin *>::iterator iter;
2048 SourceHook::List<CPlugin *> m_FailList;
2050 for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++, id++)
2053 assert(pl->GetStatus() != Plugin_Created);
2055 const sm_plugininfo_t *info = pl->GetPublicInfo();
2056 if (pl->GetStatus() != Plugin_Running && !pl->IsSilentlyFailed())
2058 len += UTIL_Format(buffer, sizeof(buffer), " %02d <%s>", id, GetStatusText(pl->GetStatus()));
2060 if (pl->GetStatus() <= Plugin_Error)
2062 /* Plugin has failed to load. */
2063 m_FailList.push_back(pl);
2068 len += UTIL_Format(buffer, sizeof(buffer), " %02d", id);
2070 if (pl->GetStatus() < Plugin_Created)
2072 if (pl->IsSilentlyFailed())
2073 len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " Disabled:");
2074 len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " \"%s\"", (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
2075 if (IS_STR_FILLED(info->version))
2077 len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " (%s)", info->version);
2079 if (IS_STR_FILLED(info->author))
2081 UTIL_Format(&buffer[len], sizeof(buffer)-len, " by %s", info->author);
2086 UTIL_Format(&buffer[len], sizeof(buffer)-len, " %s", pl->m_filename);
2088 g_RootMenu.ConsolePrint("%s", buffer);
2091 if (!m_FailList.empty())
2093 g_RootMenu.ConsolePrint("Load Errors:");
2095 SourceHook::List<CPlugin *>::iterator _iter;
2099 for (_iter=m_FailList.begin(); _iter!=m_FailList.end(); _iter++)
2101 pl = (CPlugin *)*_iter;
2102 g_RootMenu.ConsolePrint("%s: %s",(IS_STR_FILLED(pl->GetPublicInfo()->name)) ? pl->GetPublicInfo()->name : pl->GetFilename(), pl->m_errormsg);
2108 else if (strcmp(cmd, "load") == 0)
2112 g_RootMenu.ConsolePrint("[SM] Usage: sm plugins load <file>");
2118 const char *filename = command.Arg(3);
2120 char pluginfile[256];
2121 const char *ext = g_LibSys.GetFileExtension(filename) ? "" : ".smx";
2122 g_SourceMod.BuildPath(Path_None, pluginfile, sizeof(pluginfile), "%s%s", filename, ext);
2124 IPlugin *pl = LoadPlugin(pluginfile, false, PluginType_MapUpdated, error, sizeof(error), &wasloaded);
2128 g_RootMenu.ConsolePrint("[SM] Plugin %s is already loaded.", pluginfile);
2134 g_RootMenu.ConsolePrint("[SM] Loaded plugin %s successfully.", pluginfile);
2138 g_RootMenu.ConsolePrint("[SM] Plugin %s failed to load: %s.", pluginfile, error);
2143 else if (strcmp(cmd, "unload") == 0)
2147 g_RootMenu.ConsolePrint("[SM] Usage: sm plugins unload <#|file>");
2153 const char *arg = command.Arg(3);
2154 int id = strtol(arg, &end, 10);
2158 pl = GetPluginByOrder(id);
2161 g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
2167 char pluginfile[256];
2168 const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
2169 g_SourceMod.BuildPath(Path_None, pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
2171 if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
2173 g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
2178 char name[PLATFORM_MAX_PATH];
2180 if (pl->GetStatus() < Plugin_Created)
2182 const sm_plugininfo_t *info = pl->GetPublicInfo();
2183 UTIL_Format(name, sizeof(name), (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
2187 UTIL_Format(name, sizeof(name), "%s", pl->GetFilename());
2190 if (UnloadPlugin(pl))
2192 g_RootMenu.ConsolePrint("[SM] Plugin %s unloaded successfully.", name);
2196 g_RootMenu.ConsolePrint("[SM] Failed to unload plugin %s.", name);
2201 else if (strcmp(cmd, "unload_all") == 0)
2203 g_PluginSys.UnloadAll();
2204 g_RootMenu.ConsolePrint("[SM] All plugins have been unloaded.");
2207 else if (strcmp(cmd, "load_lock") == 0)
2209 if (m_LoadingLocked)
2211 g_RootMenu.ConsolePrint("[SM] There is already a loading lock in effect.");
2215 m_LoadingLocked = true;
2216 g_RootMenu.ConsolePrint("[SM] Loading is now locked; no plugins will be loaded or re-loaded.");
2220 else if (strcmp(cmd, "load_unlock") == 0)
2222 if (m_LoadingLocked)
2224 m_LoadingLocked = false;
2225 g_RootMenu.ConsolePrint("[SM] The loading lock is no longer in effect.");
2229 g_RootMenu.ConsolePrint("[SM] There was no loading lock in effect.");
2233 else if (strcmp(cmd, "info") == 0)
2237 g_RootMenu.ConsolePrint("[SM] Usage: sm plugins info <#>");
2243 const char *arg = command.Arg(3);
2244 int id = strtol(arg, &end, 10);
2248 pl = GetPluginByOrder(id);
2251 g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
2257 char pluginfile[256];
2258 const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
2259 g_SourceMod.BuildPath(Path_None, pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
2261 if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
2263 g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
2268 const sm_plugininfo_t *info = pl->GetPublicInfo();
2270 g_RootMenu.ConsolePrint(" Filename: %s", pl->GetFilename());
2271 if (pl->GetStatus() <= Plugin_Error)
2273 if (IS_STR_FILLED(info->name))
2275 if (IS_STR_FILLED(info->description))
2277 g_RootMenu.ConsolePrint(" Title: %s (%s)", info->name, info->description);
2279 g_RootMenu.ConsolePrint(" Title: %s", info->name);
2282 if (IS_STR_FILLED(info->author))
2284 g_RootMenu.ConsolePrint(" Author: %s", info->author);
2286 if (IS_STR_FILLED(info->version))
2288 g_RootMenu.ConsolePrint(" Version: %s", info->version);
2290 if (IS_STR_FILLED(info->url))
2292 g_RootMenu.ConsolePrint(" URL: %s", info->url);
2294 if (pl->GetStatus() == Plugin_Error)
2296 g_RootMenu.ConsolePrint(" Error: %s", pl->m_errormsg);
2300 if (pl->GetStatus() == Plugin_Running)
2302 g_RootMenu.ConsolePrint(" Status: running");
2306 g_RootMenu.ConsolePrint(" Status: not running");
2309 const char *typestr = "";
2310 switch (pl->GetType())
2312 case PluginType_MapUpdated:
2313 typestr = "Map Change if Updated";
2315 case PluginType_MapOnly:
2316 typestr = "Map Change";
2318 case PluginType_Private:
2319 case PluginType_Global:
2324 g_RootMenu.ConsolePrint(" Reloads: %s", typestr);
2326 if (pl->m_FileVersion >= 3)
2328 g_RootMenu.ConsolePrint(" Timestamp: %s", pl->m_DateTime);
2331 unsigned char *pCodeHash = pl->m_pRuntime->GetCodeHash();
2332 unsigned char *pDataHash = pl->m_pRuntime->GetDataHash();
2334 char combinedHash[33];
2335 for (int i = 0; i < 16; i++)
2336 UTIL_Format(combinedHash + (i * 2), 3, "%02x", pCodeHash[i] ^ pDataHash[i]);
2338 g_RootMenu.ConsolePrint(" Hash: %s", combinedHash);
2342 g_RootMenu.ConsolePrint(" Load error: %s", pl->m_errormsg);
2343 if (pl->GetStatus() < Plugin_Created)
2345 g_RootMenu.ConsolePrint(" File info: (title \"%s\") (version \"%s\")",
2346 info->name ? info->name : "<none>",
2347 info->version ? info->version : "<none>");
2348 if (IS_STR_FILLED(info->url))
2350 g_RootMenu.ConsolePrint(" File URL: %s", info->url);
2357 else if (strcmp(cmd, "refresh") == 0)
2359 g_SourceMod.DoGlobalPluginLoads();
2360 g_RootMenu.ConsolePrint("[SM] The plugin list has been refreshed and reloaded.");
2363 else if (strcmp(cmd, "reload") == 0)
2367 g_RootMenu.ConsolePrint("[SM] Usage: sm plugins reload <#|file>");
2373 const char *arg = command.Arg(3);
2374 int id = strtol(arg, &end, 10);
2378 pl = GetPluginByOrder(id);
2381 g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
2387 char pluginfile[256];
2388 const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
2389 g_SourceMod.BuildPath(Path_None, pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
2391 if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
2393 g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
2398 char name[PLATFORM_MAX_PATH];
2399 const sm_plugininfo_t *info = pl->GetPublicInfo();
2400 if (pl->GetStatus() <= Plugin_Paused)
2401 strcpy(name, (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
2403 strcpy(name, pl->GetFilename());
2405 if (ReloadPlugin(pl))
2407 g_RootMenu.ConsolePrint("[SM] Plugin %s reloaded successfully.", name);
2411 g_RootMenu.ConsolePrint("[SM] Failed to reload plugin %s.", name);
2418 /* Draw the main menu */
2419 g_RootMenu.ConsolePrint("SourceMod Plugins Menu:");
2420 g_RootMenu.DrawGenericOption("info", "Information about a plugin");
2421 g_RootMenu.DrawGenericOption("list", "Show loaded plugins");
2422 g_RootMenu.DrawGenericOption("load", "Load a plugin");
2423 g_RootMenu.DrawGenericOption("load_lock", "Prevents any more plugins from being loaded");
2424 g_RootMenu.DrawGenericOption("load_unlock", "Re-enables plugin loading");
2425 g_RootMenu.DrawGenericOption("refresh", "Reloads/refreshes all plugins in the plugins folder");
2426 g_RootMenu.DrawGenericOption("reload", "Reloads a plugin");
2427 g_RootMenu.DrawGenericOption("unload", "Unload a plugin");
2428 g_RootMenu.DrawGenericOption("unload_all", "Unloads all plugins");
2431 bool CPluginManager::ReloadPlugin(CPlugin *pl)
2433 List<CPlugin *>::iterator iter;
2434 char filename[PLATFORM_MAX_PATH];
2440 strcpy(filename, pl->m_filename);
2441 ptype = pl->GetType();
2443 for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++, id++)
2451 if (!UnloadPlugin(pl))
2455 if (!(newpl=LoadPlugin(filename, true, ptype, NULL, 0, &wasloaded)))
2460 for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
2462 if ((*iter) == (CPlugin *)newpl)
2464 m_plugins.erase(iter);
2470 for (i=1, iter=m_plugins.begin(); iter!=m_plugins.end() && i<id; iter++, i++)
2473 m_plugins.insert(iter, (CPlugin *)newpl);
2478 void CPluginManager::ReloadOrUnloadPlugins()
2480 /* If we're in a load lock, just skip this whole bit. */
2481 if (m_LoadingLocked)
2486 List<CPlugin *>::iterator iter;
2487 List<CPlugin *> tmp_list = m_plugins;
2491 for (iter=tmp_list.begin(); iter!=tmp_list.end(); iter++)
2494 if (pl->GetType() == PluginType_MapOnly)
2496 UnloadPlugin((IPlugin *)pl);
2498 else if (pl->GetType() == PluginType_MapUpdated)
2500 t = pl->GetFileTimeStamp();
2501 if (!t || t > pl->GetTimeStamp())
2503 pl->SetTimeStamp(t);
2504 UnloadPlugin((IPlugin *)pl);
2510 void CPluginManager::_SetPauseState(CPlugin *pl, bool paused)
2512 List<IPluginsListener *>::iterator iter;
2513 IPluginsListener *pListener;
2514 for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
2516 pListener = (*iter);
2517 pListener->OnPluginPauseChange(pl, paused);
2521 void CPluginManager::AddFunctionsToForward(const char *name, IChangeableForward *pForward)
2523 List<CPlugin *>::iterator iter;
2525 IPluginFunction *pFunction;
2527 for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++)
2531 if (pPlugin->GetStatus() <= Plugin_Paused)
2533 pFunction = pPlugin->GetBaseContext()->GetFunctionByName(name);
2537 pForward->AddFunction(pFunction);
2543 CPlugin *CPluginManager::GetPluginFromIdentity(IdentityToken_t *pToken)
2545 if (pToken->type != g_PluginIdent)
2550 return (CPlugin *)(pToken->ptr);
2553 void CPluginManager::OnLibraryAction(const char *lib, bool is_a_plugin, bool drop)
2555 List<CPlugin *>::iterator iter;
2572 const char *name = drop ? "OnLibraryRemoved" : "OnLibraryAdded";
2574 for (iter=m_plugins.begin();
2575 iter!=m_plugins.end();
2578 CPlugin *pl = (*iter);
2579 if (pl->GetStatus() != Plugin_Running)
2583 IPluginContext *pContext = pl->GetBaseContext();
2584 IPluginFunction *pf = pContext->GetFunctionByName(name);
2589 uint32_t num_vars = pContext->GetPubVarsNum();
2590 for (uint32_t i=0; i<num_vars; i++)
2592 sp_pubvar_t *pubvar;
2593 if (pContext->GetPubvarByIndex(i, &pubvar) != SP_ERROR_NONE)
2597 if (is_a_plugin && strncmp(pubvar->name, "__pl_", 5) == 0)
2599 plc = (_pl *)pubvar->offs;
2605 pContext->LocalToString(plc->name, &str);
2606 if (strcmp(str, lib) != 0)
2610 pf->PushString(lib);
2613 else if (!is_a_plugin && strncmp(pubvar->name, "__ext_", 6) == 0)
2615 ext = (_ext *)pubvar->offs;
2621 pContext->LocalToString(ext->name, &str);
2622 if (strcmp(str, lib) != 0)
2626 pf->PushString(lib);
2633 bool CPluginManager::LibraryExists(const char *lib)
2635 List<CPlugin *>::iterator iter;
2637 for (iter=m_plugins.begin();
2638 iter!=m_plugins.end();
2641 CPlugin *pl = (*iter);
2642 if (pl->GetStatus() != Plugin_Running)
2646 List<String>::iterator s_iter;
2647 for (s_iter = pl->m_Libraries.begin();
2648 s_iter != pl->m_Libraries.end();
2651 if ((*s_iter).compare(lib) == 0)
2661 void CPluginManager::AllPluginsLoaded()
2663 List<CPlugin *>::iterator iter;
2666 for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
2669 pl->Call_OnAllPluginsLoaded();
2673 void CPluginManager::UnloadAll()
2675 List<CPlugin *>::iterator iter;
2676 while ( (iter = m_plugins.begin()) != m_plugins.end() )
2678 UnloadPlugin((*iter));
2682 int CPluginManager::GetOrderOfPlugin(IPlugin *pl)
2685 List<CPlugin *>::iterator iter;
2687 for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++, id++)
2698 CPlugin *CPluginManager::FindPluginByConsoleArg(const char *arg)
2704 id = strtol(arg, &end, 10);
2708 pl = GetPluginByOrder(id);
2716 char pluginfile[256];
2717 const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
2718 UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
2720 if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
2729 void CPluginManager::OnSourceModMaxPlayersChanged(int newvalue)
2731 SyncMaxClients(newvalue);
2734 void CPluginManager::SyncMaxClients(int max_clients)
2736 List<CPlugin *>::iterator iter;
2738 for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++)
2740 (*iter)->SyncMaxClients(max_clients);
2744 void CPluginManager::ListPluginsToClient(CPlayer *player, const CCommand &args)
2747 unsigned int id = 0;
2748 int plnum = GetPluginCount();
2749 edict_t *e = player->GetEdict();
2750 unsigned int start = 0;
2754 ClientConsolePrint(e, "[SM] No plugins found.");
2758 if (args.ArgC() > 2)
2760 start = atoi(args.Arg(2));
2764 SourceHook::List<CPlugin *>::iterator iter;
2765 SourceHook::List<CPlugin *> m_FailList;
2767 for (iter = m_plugins.begin();
2768 iter != m_plugins.end();
2773 if (pl->GetStatus() != Plugin_Running)
2778 /* Count valid plugins */
2785 if (id - start > 10)
2791 const sm_plugininfo_t *info = pl->GetPublicInfo();
2792 len = UTIL_Format(buffer, sizeof(buffer), " \"%s\"", (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
2793 if (IS_STR_FILLED(info->version))
2795 len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " (%s)", info->version);
2797 if (IS_STR_FILLED(info->author))
2799 UTIL_Format(&buffer[len], sizeof(buffer)-len, " by %s", info->author);
2803 UTIL_Format(&buffer[len], sizeof(buffer)-len, " %s", pl->m_filename);
2805 ClientConsolePrint(e, "%s", buffer);
2808 /* See if we can get more plugins */
2809 while (iter != m_plugins.end())
2811 if ((*iter)->GetStatus() == Plugin_Running)
2817 /* Do we actually have more plugins? */
2818 if (iter != m_plugins.end())
2820 ClientConsolePrint(e, "To see more, type \"sm plugins %d\"", id);