1.1 --- a/core/systems/PluginSys.cpp Fri Jul 11 06:51:21 2008 +0000
1.2 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000
1.3 @@ -1,2692 +0,0 @@
1.4 -/**
1.5 - * vim: set ts=4 :
1.6 - * =============================================================================
1.7 - * SourceMod
1.8 - * Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved.
1.9 - * =============================================================================
1.10 - *
1.11 - * This program is free software; you can redistribute it and/or modify it under
1.12 - * the terms of the GNU General Public License, version 3.0, as published by the
1.13 - * Free Software Foundation.
1.14 - *
1.15 - * This program is distributed in the hope that it will be useful, but WITHOUT
1.16 - * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
1.17 - * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
1.18 - * details.
1.19 - *
1.20 - * You should have received a copy of the GNU General Public License along with
1.21 - * this program. If not, see <http://www.gnu.org/licenses/>.
1.22 - *
1.23 - * As a special exception, AlliedModders LLC gives you permission to link the
1.24 - * code of this program (as well as its derivative works) to "Half-Life 2," the
1.25 - * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
1.26 - * by the Valve Corporation. You must obey the GNU General Public License in
1.27 - * all respects for all other code used. Additionally, AlliedModders LLC grants
1.28 - * this exception to all derivative works. AlliedModders LLC defines further
1.29 - * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
1.30 - * or <http://www.sourcemod.net/license.php>.
1.31 - *
1.32 - * Version: $Id$
1.33 - */
1.34 -
1.35 -#include <stdio.h>
1.36 -#include "PluginSys.h"
1.37 -#include "ShareSys.h"
1.38 -#include "LibrarySys.h"
1.39 -#include "HandleSys.h"
1.40 -#include "ForwardSys.h"
1.41 -#include "sourcemm_api.h"
1.42 -#include "sourcemod.h"
1.43 -#include "TextParsers.h"
1.44 -#include "Logger.h"
1.45 -#include "ExtensionSys.h"
1.46 -#include "sm_srvcmds.h"
1.47 -#include "sm_stringutil.h"
1.48 -#include "ConCmdManager.h"
1.49 -#include "PlayerManager.h"
1.50 -#include "CoreConfig.h"
1.51 -#include "Translator.h"
1.52 -
1.53 -CPluginManager g_PluginSys;
1.54 -HandleType_t g_PluginType = 0;
1.55 -IdentityType_t g_PluginIdent = 0;
1.56 -
1.57 -CPlugin::CPlugin(const char *file)
1.58 -{
1.59 - static int MySerial = 0;
1.60 -
1.61 - m_type = PluginType_Private;
1.62 - m_status = Plugin_Uncompiled;
1.63 - m_serial = ++MySerial;
1.64 - m_pRuntime = NULL;
1.65 - m_errormsg[256] = '\0';
1.66 - snprintf(m_filename, sizeof(m_filename), "%s", file);
1.67 - m_handle = 0;
1.68 - m_ident = NULL;
1.69 - m_pProps = sm_trie_create();
1.70 - m_FakeNativesMissing = false;
1.71 - m_LibraryMissing = false;
1.72 - m_bGotAllLoaded = false;
1.73 - m_pPhrases = g_Translator.CreatePhraseCollection();
1.74 -}
1.75 -
1.76 -CPlugin::~CPlugin()
1.77 -{
1.78 - if (m_handle)
1.79 - {
1.80 - HandleSecurity sec;
1.81 - sec.pOwner = g_PluginSys.GetIdentity();
1.82 - sec.pIdentity = sec.pOwner;
1.83 -
1.84 - g_HandleSys.FreeHandle(m_handle, &sec);
1.85 - g_ShareSys.DestroyIdentity(m_ident);
1.86 - }
1.87 -
1.88 - if (m_pRuntime != NULL)
1.89 - {
1.90 - delete m_pRuntime;
1.91 - m_pRuntime = NULL;
1.92 - }
1.93 -
1.94 - if (m_pProps)
1.95 - {
1.96 - sm_trie_destroy(m_pProps);
1.97 - }
1.98 - for (size_t i=0; i<m_configs.size(); i++)
1.99 - {
1.100 - delete m_configs[i];
1.101 - }
1.102 - m_configs.clear();
1.103 - if (m_pPhrases != NULL)
1.104 - {
1.105 - m_pPhrases->Destroy();
1.106 - m_pPhrases = NULL;
1.107 - }
1.108 -}
1.109 -
1.110 -void CPlugin::InitIdentity()
1.111 -{
1.112 - if (!m_handle)
1.113 - {
1.114 - m_ident = g_ShareSys.CreateIdentity(g_PluginIdent, this);
1.115 - m_handle = g_HandleSys.CreateHandle(g_PluginType, this, g_PluginSys.GetIdentity(), g_PluginSys.GetIdentity(), NULL);
1.116 - m_pRuntime->GetDefaultContext()->SetKey(1, m_ident);
1.117 - m_pRuntime->GetDefaultContext()->SetKey(2, (IPlugin *)this);
1.118 - }
1.119 -}
1.120 -
1.121 -unsigned int CPlugin::CalcMemUsage()
1.122 -{
1.123 - unsigned int base_size =
1.124 - sizeof(CPlugin)
1.125 - + sizeof(IdentityToken_t)
1.126 - + (m_configs.size() * (sizeof(AutoConfig *) + sizeof(AutoConfig)))
1.127 - + sm_trie_mem_usage(m_pProps);
1.128 -
1.129 - for (unsigned int i = 0; i < m_configs.size(); i++)
1.130 - {
1.131 - base_size += m_configs[i]->autocfg.size();
1.132 - base_size += m_configs[i]->folder.size();
1.133 - }
1.134 -
1.135 - for (List<String>::iterator i = m_Libraries.begin();
1.136 - i != m_Libraries.end();
1.137 - i++)
1.138 - {
1.139 - base_size += (*i).size();
1.140 - }
1.141 -
1.142 - for (List<String>::iterator i = m_RequiredLibs.begin();
1.143 - i != m_RequiredLibs.end();
1.144 - i++)
1.145 - {
1.146 - base_size += (*i).size();
1.147 - }
1.148 -
1.149 - return base_size;
1.150 -}
1.151 -
1.152 -Handle_t CPlugin::GetMyHandle()
1.153 -{
1.154 - return m_handle;
1.155 -}
1.156 -
1.157 -CPlugin *CPlugin::CreatePlugin(const char *file, char *error, size_t maxlength)
1.158 -{
1.159 - char fullpath[PLATFORM_MAX_PATH];
1.160 - g_SourceMod.BuildPath(Path_SM, fullpath, sizeof(fullpath), "plugins/%s", file);
1.161 - FILE *fp = fopen(fullpath, "rb");
1.162 -
1.163 - CPlugin *pPlugin = new CPlugin(file);
1.164 -
1.165 - if (!fp)
1.166 - {
1.167 - if (error)
1.168 - {
1.169 - snprintf(error, maxlength, "Unable to open file");
1.170 - }
1.171 - pPlugin->m_status = Plugin_BadLoad;
1.172 - return pPlugin;
1.173 - }
1.174 -
1.175 - return pPlugin;
1.176 -}
1.177 -
1.178 -bool CPlugin::GetProperty(const char *prop, void **ptr, bool remove/* =false */)
1.179 -{
1.180 - bool exists = sm_trie_retrieve(m_pProps, prop, ptr);
1.181 -
1.182 - if (exists && remove)
1.183 - {
1.184 - sm_trie_delete(m_pProps, prop);
1.185 - }
1.186 -
1.187 - return exists;
1.188 -}
1.189 -
1.190 -bool CPlugin::SetProperty(const char *prop, void *ptr)
1.191 -{
1.192 - return sm_trie_insert(m_pProps, prop, ptr);
1.193 -}
1.194 -
1.195 -IPluginRuntime *CPlugin::GetRuntime()
1.196 -{
1.197 - return m_pRuntime;
1.198 -}
1.199 -
1.200 -void CPlugin::SetErrorState(PluginStatus status, const char *error_fmt, ...)
1.201 -{
1.202 - PluginStatus old_status = m_status;
1.203 - m_status = status;
1.204 -
1.205 - if (old_status == Plugin_Running)
1.206 - {
1.207 - /* Tell everyone we're now paused */
1.208 - g_PluginSys._SetPauseState(this, true);
1.209 - }
1.210 -
1.211 - va_list ap;
1.212 - va_start(ap, error_fmt);
1.213 - vsnprintf(m_errormsg, sizeof(m_errormsg), error_fmt, ap);
1.214 - va_end(ap);
1.215 -
1.216 - if (m_pRuntime != NULL)
1.217 - {
1.218 - m_pRuntime->SetPauseState(true);
1.219 - }
1.220 -}
1.221 -
1.222 -void CPlugin::UpdateInfo()
1.223 -{
1.224 - /* Now grab the info */
1.225 - uint32_t idx;
1.226 - IPluginContext *base = GetBaseContext();
1.227 - int err = base->FindPubvarByName("myinfo", &idx);
1.228 -
1.229 - memset(&m_info, 0, sizeof(m_info));
1.230 -
1.231 - if (err == SP_ERROR_NONE)
1.232 - {
1.233 - struct sm_plugininfo_c_t
1.234 - {
1.235 - cell_t name;
1.236 - cell_t description;
1.237 - cell_t author;
1.238 - cell_t version;
1.239 - cell_t url;
1.240 - };
1.241 - sm_plugininfo_c_t *cinfo;
1.242 - cell_t local_addr;
1.243 -
1.244 - base->GetPubvarAddrs(idx, &local_addr, (cell_t **)&cinfo);
1.245 - base->LocalToString(cinfo->name, (char **)&m_info.name);
1.246 - base->LocalToString(cinfo->description, (char **)&m_info.description);
1.247 - base->LocalToString(cinfo->author, (char **)&m_info.author);
1.248 - base->LocalToString(cinfo->url, (char **)&m_info.url);
1.249 - base->LocalToString(cinfo->version, (char **)&m_info.version);
1.250 - }
1.251 -
1.252 - m_info.author = m_info.author ? m_info.author : "";
1.253 - m_info.description = m_info.description ? m_info.description : "";
1.254 - m_info.name = m_info.name ? m_info.name : "";
1.255 - m_info.url = m_info.url ? m_info.url : "";
1.256 - m_info.version = m_info.version ? m_info.version : "";
1.257 -
1.258 - if ((err = base->FindPubvarByName("__version", &idx)) == SP_ERROR_NONE)
1.259 - {
1.260 - struct __version_info
1.261 - {
1.262 - cell_t version;
1.263 - cell_t filevers;
1.264 - cell_t date;
1.265 - cell_t time;
1.266 - };
1.267 - __version_info *info;
1.268 - cell_t local_addr;
1.269 - const char *pDate, *pTime;
1.270 -
1.271 - pDate = "";
1.272 - pTime = "";
1.273 -
1.274 - base->GetPubvarAddrs(idx, &local_addr, (cell_t **)&info);
1.275 - m_FileVersion = info->version;
1.276 - if (m_FileVersion >= 3)
1.277 - {
1.278 - base->LocalToString(info->date, (char **)&pDate);
1.279 - base->LocalToString(info->time, (char **)&pTime);
1.280 - UTIL_Format(m_DateTime, sizeof(m_DateTime), "%s %s", pDate, pTime);
1.281 - }
1.282 - }
1.283 - else
1.284 - {
1.285 - m_FileVersion = 0;
1.286 - }
1.287 -}
1.288 -
1.289 -void CPlugin::Call_OnPluginStart()
1.290 -{
1.291 - if (m_status != Plugin_Loaded)
1.292 - {
1.293 - return;
1.294 - }
1.295 -
1.296 - m_status = Plugin_Running;
1.297 -
1.298 - cell_t result;
1.299 - IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("OnPluginStart");
1.300 - if (!pFunction)
1.301 - {
1.302 - return;
1.303 - }
1.304 -
1.305 - int err;
1.306 - if ((err=pFunction->Execute(&result)) != SP_ERROR_NONE)
1.307 - {
1.308 - SetErrorState(Plugin_Error, "Error detected in plugin startup (see error logs)");
1.309 - }
1.310 -}
1.311 -
1.312 -void CPlugin::Call_OnPluginEnd()
1.313 -{
1.314 - if (m_status > Plugin_Paused)
1.315 - {
1.316 - return;
1.317 - }
1.318 -
1.319 - cell_t result;
1.320 - IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("OnPluginEnd");
1.321 - if (!pFunction)
1.322 - {
1.323 - return;
1.324 - }
1.325 -
1.326 - pFunction->Execute(&result);
1.327 -}
1.328 -
1.329 -void CPlugin::Call_OnAllPluginsLoaded()
1.330 -{
1.331 - if (m_status > Plugin_Paused)
1.332 - {
1.333 - return;
1.334 - }
1.335 -
1.336 - if (m_bGotAllLoaded)
1.337 - {
1.338 - return;
1.339 - }
1.340 -
1.341 - m_bGotAllLoaded = true;
1.342 -
1.343 - cell_t result;
1.344 - IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("OnAllPluginsLoaded");
1.345 - if (pFunction != NULL)
1.346 - {
1.347 - pFunction->Execute(&result);
1.348 - }
1.349 -
1.350 - if (g_OnMapStarted)
1.351 - {
1.352 - if ((pFunction = m_pRuntime->GetFunctionByName("OnMapStart")) != NULL)
1.353 - {
1.354 - pFunction->Execute(NULL);
1.355 - }
1.356 - }
1.357 -
1.358 - if (SM_AreConfigsExecuted())
1.359 - {
1.360 - SM_ExecuteForPlugin(GetBaseContext());
1.361 - }
1.362 -}
1.363 -
1.364 -bool CPlugin::Call_AskPluginLoad(char *error, size_t maxlength)
1.365 -{
1.366 - if (m_status != Plugin_Created)
1.367 - {
1.368 - return false;
1.369 - }
1.370 -
1.371 - m_status = Plugin_Loaded;
1.372 -
1.373 - int err;
1.374 - cell_t result;
1.375 - IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("AskPluginLoad");
1.376 -
1.377 - if (!pFunction)
1.378 - {
1.379 - return true;
1.380 - }
1.381 -
1.382 - pFunction->PushCell(m_handle);
1.383 - pFunction->PushCell(g_PluginSys.IsLateLoadTime() ? 1 : 0);
1.384 - pFunction->PushStringEx(error, maxlength, 0, SM_PARAM_COPYBACK);
1.385 - pFunction->PushCell(maxlength);
1.386 - if ((err=pFunction->Execute(&result)) != SP_ERROR_NONE)
1.387 - {
1.388 - return false;
1.389 - }
1.390 -
1.391 - if (!result || m_status != Plugin_Loaded)
1.392 - {
1.393 - return false;
1.394 - }
1.395 -
1.396 - return true;
1.397 -}
1.398 -
1.399 -void *CPlugin::GetPluginStructure()
1.400 -{
1.401 - return NULL;
1.402 -}
1.403 -
1.404 -IPluginContext *CPlugin::GetBaseContext()
1.405 -{
1.406 - if (m_pRuntime == NULL)
1.407 - {
1.408 - return NULL;
1.409 - }
1.410 -
1.411 - return m_pRuntime->GetDefaultContext();
1.412 -}
1.413 -
1.414 -sp_context_t *CPlugin::GetContext()
1.415 -{
1.416 - return NULL;
1.417 -}
1.418 -
1.419 -const char *CPlugin::GetFilename()
1.420 -{
1.421 - return m_filename;
1.422 -}
1.423 -
1.424 -PluginType CPlugin::GetType()
1.425 -{
1.426 - return m_type;
1.427 -}
1.428 -
1.429 -const sm_plugininfo_t *CPlugin::GetPublicInfo()
1.430 -{
1.431 - return &m_info;
1.432 -}
1.433 -
1.434 -unsigned int CPlugin::GetSerial()
1.435 -{
1.436 - return m_serial;
1.437 -}
1.438 -
1.439 -PluginStatus CPlugin::GetStatus()
1.440 -{
1.441 - return m_status;
1.442 -}
1.443 -
1.444 -bool CPlugin::IsDebugging()
1.445 -{
1.446 - if (m_pRuntime == NULL)
1.447 - {
1.448 - return false;
1.449 - }
1.450 -
1.451 - return m_pRuntime->IsDebugging();
1.452 -}
1.453 -
1.454 -void CPlugin::LibraryActions(bool dropping)
1.455 -{
1.456 - List<String>::iterator iter;
1.457 - for (iter = m_Libraries.begin();
1.458 - iter != m_Libraries.end();
1.459 - iter++)
1.460 - {
1.461 - g_PluginSys.OnLibraryAction((*iter).c_str(), true, dropping);
1.462 - }
1.463 -}
1.464 -
1.465 -bool CPlugin::SetPauseState(bool paused)
1.466 -{
1.467 - if (paused && GetStatus() != Plugin_Running)
1.468 - {
1.469 - return false;
1.470 - } else if (!paused && GetStatus() != Plugin_Paused) {
1.471 - return false;
1.472 - }
1.473 -
1.474 - if (paused)
1.475 - {
1.476 - LibraryActions(true);
1.477 - }
1.478 -
1.479 - IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("OnPluginPauseChange");
1.480 - if (pFunction)
1.481 - {
1.482 - cell_t result;
1.483 - pFunction->PushCell(paused ? 1 : 0);
1.484 - pFunction->Execute(&result);
1.485 - }
1.486 -
1.487 - if (paused)
1.488 - {
1.489 - m_status = Plugin_Paused;
1.490 - m_pRuntime->SetPauseState(true);
1.491 - } else {
1.492 - m_status = Plugin_Running;
1.493 - m_pRuntime->SetPauseState(false);
1.494 - }
1.495 -
1.496 - g_PluginSys._SetPauseState(this, paused);
1.497 -
1.498 - if (!paused)
1.499 - {
1.500 - LibraryActions(false);
1.501 - }
1.502 -
1.503 - return true;
1.504 -}
1.505 -
1.506 -IdentityToken_t *CPlugin::GetIdentity()
1.507 -{
1.508 - return m_ident;
1.509 -}
1.510 -
1.511 -bool CPlugin::ToggleDebugMode(bool debug, char *error, size_t maxlength)
1.512 -{
1.513 - int err;
1.514 -
1.515 - if (!IsRunnable())
1.516 - {
1.517 - if (error)
1.518 - {
1.519 - snprintf(error, maxlength, "Plugin is not runnable.");
1.520 - }
1.521 - return false;
1.522 - }
1.523 -
1.524 - if (debug && IsDebugging())
1.525 - {
1.526 - if (error)
1.527 - {
1.528 - snprintf(error, maxlength, "Plugin is already in debug mode.");
1.529 - }
1.530 - return false;
1.531 - }
1.532 - else if (!debug && !IsDebugging())
1.533 - {
1.534 - if (error)
1.535 - {
1.536 - snprintf(error, maxlength, "Plugins is already in production mode.");
1.537 - }
1.538 - return false;
1.539 - }
1.540 -
1.541 - ICompilation *co = g_pSourcePawn2->StartCompilation();
1.542 -
1.543 - if (!co->SetOption("debug", (debug) ? "1" : "0"))
1.544 - {
1.545 - if (error)
1.546 - {
1.547 - snprintf(error, maxlength, "Failed to change plugin mode (JIT failure).");
1.548 - }
1.549 - return false;
1.550 - }
1.551 -
1.552 - if ((err = m_pRuntime->ApplyCompilationOptions(co)) != SP_ERROR_NONE)
1.553 - {
1.554 - if (error)
1.555 - {
1.556 - snprintf(error, maxlength, "Failed to recompile plugin (JIT error %d).", err);
1.557 - }
1.558 - return false;
1.559 - }
1.560 -
1.561 - UpdateInfo();
1.562 -
1.563 - return true;
1.564 -}
1.565 -
1.566 -bool CPlugin::IsRunnable()
1.567 -{
1.568 - return (m_status <= Plugin_Paused) ? true : false;
1.569 -}
1.570 -
1.571 -time_t CPlugin::GetFileTimeStamp()
1.572 -{
1.573 - char path[PLATFORM_MAX_PATH];
1.574 - g_SourceMod.BuildPath(Path_SM, path, sizeof(path), "plugins/%s", m_filename);
1.575 -#ifdef PLATFORM_WINDOWS
1.576 - struct _stat s;
1.577 - if (_stat(path, &s) != 0)
1.578 -#elif defined PLATFORM_POSIX
1.579 - struct stat s;
1.580 - if (stat(path, &s) != 0)
1.581 -#endif
1.582 - {
1.583 - return 0;
1.584 - }
1.585 - else
1.586 - {
1.587 - return s.st_mtime;
1.588 - }
1.589 -}
1.590 -
1.591 -time_t CPlugin::GetTimeStamp()
1.592 -{
1.593 - return m_LastAccess;
1.594 -}
1.595 -
1.596 -void CPlugin::SetTimeStamp(time_t t)
1.597 -{
1.598 - m_LastAccess = t;
1.599 -}
1.600 -
1.601 -IPhraseCollection *CPlugin::GetPhrases()
1.602 -{
1.603 - return m_pPhrases;
1.604 -}
1.605 -
1.606 -void CPlugin::DependencyDropped(CPlugin *pOwner)
1.607 -{
1.608 - if (!m_pRuntime)
1.609 - {
1.610 - return;
1.611 - }
1.612 -
1.613 - List<String>::iterator reqlib_iter;
1.614 - List<String>::iterator lib_iter;
1.615 - for (lib_iter=pOwner->m_Libraries.begin(); lib_iter!=pOwner->m_Libraries.end(); lib_iter++)
1.616 - {
1.617 - for (reqlib_iter=m_RequiredLibs.begin(); reqlib_iter!=m_RequiredLibs.end(); reqlib_iter++)
1.618 - {
1.619 - if ((*reqlib_iter) == (*lib_iter))
1.620 - {
1.621 - m_LibraryMissing = true;
1.622 - }
1.623 - }
1.624 - }
1.625 -
1.626 - List<NativeEntry *>::iterator iter;
1.627 - NativeEntry *pNative;
1.628 - sp_native_t *native;
1.629 - uint32_t idx;
1.630 - unsigned int unbound = 0;
1.631 -
1.632 - for (iter = pOwner->m_Natives.begin();
1.633 - iter != pOwner->m_Natives.end();
1.634 - iter++)
1.635 - {
1.636 - pNative = (*iter);
1.637 - /* Find this native! */
1.638 - if (m_pRuntime->FindNativeByName(pNative->name, &idx) != SP_ERROR_NONE)
1.639 - {
1.640 - continue;
1.641 - }
1.642 - /* Unbind it */
1.643 - m_pRuntime->GetNativeByIndex(idx, &native);
1.644 - native->pfn = NULL;
1.645 - native->status = SP_NATIVE_UNBOUND;
1.646 - unbound++;
1.647 - }
1.648 -
1.649 - if (unbound)
1.650 - {
1.651 - m_FakeNativesMissing = true;
1.652 - }
1.653 -
1.654 - /* :IDEA: in the future, add native trapping? */
1.655 - if (m_FakeNativesMissing || m_LibraryMissing)
1.656 - {
1.657 - SetErrorState(Plugin_Error, "Depends on plugin: %s", pOwner->GetFilename());
1.658 - }
1.659 -}
1.660 -
1.661 -unsigned int CPlugin::GetConfigCount()
1.662 -{
1.663 - return (unsigned int)m_configs.size();
1.664 -}
1.665 -
1.666 -AutoConfig *CPlugin::GetConfig(unsigned int i)
1.667 -{
1.668 - if (i >= GetConfigCount())
1.669 - {
1.670 - return NULL;
1.671 - }
1.672 -
1.673 - return m_configs[i];
1.674 -}
1.675 -
1.676 -void CPlugin::AddConfig(bool autoCreate, const char *cfg, const char *folder)
1.677 -{
1.678 - /* Do a check for duplicates to prevent double-execution */
1.679 - for (size_t i = 0; i < m_configs.size(); i++)
1.680 - {
1.681 - if (m_configs[i]->autocfg.compare(cfg) == 0
1.682 - && m_configs[i]->folder.compare(folder) == 0
1.683 - && m_configs[i]->create == autoCreate)
1.684 - {
1.685 - return;
1.686 - }
1.687 - }
1.688 -
1.689 - AutoConfig *c = new AutoConfig;
1.690 -
1.691 - c->autocfg = cfg;
1.692 - c->folder = folder;
1.693 - c->create = autoCreate;
1.694 -
1.695 - m_configs.push_back(c);
1.696 -}
1.697 -
1.698 -void CPlugin::DropEverything()
1.699 -{
1.700 - CPlugin *pOther;
1.701 - List<CPlugin *>::iterator iter;
1.702 - List<WeakNative>::iterator wk_iter;
1.703 -
1.704 - /* Tell everyone that depends on us that we're about to drop */
1.705 - for (iter = m_Dependents.begin();
1.706 - iter != m_Dependents.end();
1.707 - iter++)
1.708 - {
1.709 - pOther = (*iter);
1.710 - pOther->DependencyDropped(this);
1.711 - }
1.712 -
1.713 - /* Note: we don't care about things we depend on.
1.714 - * The reason is that extensions have their own cleanup
1.715 - * code for plugins. Although the "right" design would be
1.716 - * to centralize that here, i'm omitting it for now. Thus,
1.717 - * the code below to walk the plugins list will suffice.
1.718 - */
1.719 -
1.720 - /* Other plugins could be holding weak references that were
1.721 - * added by us. We need to clean all of those up now.
1.722 - */
1.723 - for (iter = g_PluginSys.m_plugins.begin();
1.724 - iter != g_PluginSys.m_plugins.end();
1.725 - iter++)
1.726 - {
1.727 - (*iter)->DropRefsTo(this);
1.728 - }
1.729 -
1.730 - /* Proceed with the rest of the necessities. */
1.731 - CNativeOwner::DropEverything();
1.732 -}
1.733 -
1.734 -bool CPlugin::AddFakeNative(IPluginFunction *pFunc, const char *name, SPVM_FAKENATIVE_FUNC func)
1.735 -{
1.736 - NativeEntry *pEntry;
1.737 -
1.738 - if ((pEntry = g_ShareSys.AddFakeNative(pFunc, name, func)) == NULL)
1.739 - {
1.740 - return false;
1.741 - }
1.742 -
1.743 - m_Natives.push_back(pEntry);
1.744 -
1.745 - return true;
1.746 -}
1.747 -
1.748 -/*******************
1.749 - * PLUGIN ITERATOR *
1.750 - *******************/
1.751 -
1.752 -CPluginManager::CPluginIterator::CPluginIterator(List<CPlugin *> *_mylist)
1.753 -{
1.754 - mylist = _mylist;
1.755 - Reset();
1.756 -}
1.757 -
1.758 -IPlugin *CPluginManager::CPluginIterator::GetPlugin()
1.759 -{
1.760 - return (*current);
1.761 -}
1.762 -
1.763 -bool CPluginManager::CPluginIterator::MorePlugins()
1.764 -{
1.765 - return (current != mylist->end());
1.766 -}
1.767 -
1.768 -void CPluginManager::CPluginIterator::NextPlugin()
1.769 -{
1.770 - current++;
1.771 -}
1.772 -
1.773 -void CPluginManager::CPluginIterator::Release()
1.774 -{
1.775 - g_PluginSys.ReleaseIterator(this);
1.776 -}
1.777 -
1.778 -CPluginManager::CPluginIterator::~CPluginIterator()
1.779 -{
1.780 -}
1.781 -
1.782 -void CPluginManager::CPluginIterator::Reset()
1.783 -{
1.784 - current = mylist->begin();
1.785 -}
1.786 -
1.787 -/******************
1.788 - * PLUGIN MANAGER *
1.789 - ******************/
1.790 -
1.791 -CPluginManager::CPluginManager()
1.792 -{
1.793 - m_LoadLookup = sm_trie_create();
1.794 - m_AllPluginsLoaded = false;
1.795 - m_MyIdent = NULL;
1.796 - m_LoadingLocked = false;
1.797 -}
1.798 -
1.799 -CPluginManager::~CPluginManager()
1.800 -{
1.801 - /* :NOTICE:
1.802 - * Ignore the fact that there might be plugins in the cache.
1.803 - * This usually means that Core is not being unloaded properly, and everything
1.804 - * will crash anyway. YAY
1.805 - */
1.806 - sm_trie_destroy(m_LoadLookup);
1.807 -
1.808 - CStack<CPluginManager::CPluginIterator *>::iterator iter;
1.809 - for (iter=m_iters.begin(); iter!=m_iters.end(); iter++)
1.810 - {
1.811 - delete (*iter);
1.812 - }
1.813 - m_iters.popall();
1.814 -}
1.815 -
1.816 -void CPluginManager::Shutdown()
1.817 -{
1.818 - List<CPlugin *>::iterator iter;
1.819 -
1.820 - while ((iter = m_plugins.begin()) != m_plugins.end())
1.821 - {
1.822 - UnloadPlugin(*iter);
1.823 - }
1.824 -}
1.825 -
1.826 -void CPluginManager::LoadAll_FirstPass(const char *config, const char *basedir)
1.827 -{
1.828 - /* First read in the database of plugin settings */
1.829 - SMCError err;
1.830 - SMCStates states;
1.831 - m_AllPluginsLoaded = false;
1.832 - if ((err=textparsers->ParseFile_SMC(config, &m_PluginInfo, &states)) != SMCError_Okay)
1.833 - {
1.834 - g_Logger.LogError("[SM] Encountered fatal error parsing file \"%s\"", config);
1.835 - const char *err_msg = textparsers->GetSMCErrorString(err);
1.836 - if (err_msg)
1.837 - {
1.838 - g_Logger.LogError("[SM] Parse error encountered: \"%s\"", err_msg);
1.839 - }
1.840 - }
1.841 -
1.842 - LoadPluginsFromDir(basedir, NULL);
1.843 -}
1.844 -
1.845 -void CPluginManager::LoadPluginsFromDir(const char *basedir, const char *localpath)
1.846 -{
1.847 - char base_path[PLATFORM_MAX_PATH];
1.848 -
1.849 - /* Form the current path to start reading from */
1.850 - if (localpath == NULL)
1.851 - {
1.852 - g_LibSys.PathFormat(base_path, sizeof(base_path), "%s", basedir);
1.853 - } else {
1.854 - g_LibSys.PathFormat(base_path, sizeof(base_path), "%s/%s", basedir, localpath);
1.855 - }
1.856 -
1.857 - IDirectory *dir = g_LibSys.OpenDirectory(base_path);
1.858 -
1.859 - if (!dir)
1.860 - {
1.861 - char error[256];
1.862 - g_LibSys.GetPlatformError(error, sizeof(error));
1.863 - g_Logger.LogError("[SM] Failure reading from plugins path: %s", localpath);
1.864 - g_Logger.LogError("[SM] Platform returned error: %s", error);
1.865 - return;
1.866 - }
1.867 -
1.868 - while (dir->MoreFiles())
1.869 - {
1.870 - if (dir->IsEntryDirectory()
1.871 - && (strcmp(dir->GetEntryName(), ".") != 0)
1.872 - && (strcmp(dir->GetEntryName(), "..") != 0)
1.873 - && (strcmp(dir->GetEntryName(), "disabled") != 0)
1.874 - && (strcmp(dir->GetEntryName(), "optional") != 0))
1.875 - {
1.876 - char new_local[PLATFORM_MAX_PATH];
1.877 - if (localpath == NULL)
1.878 - {
1.879 - /* If no path yet, don't add a former slash */
1.880 - snprintf(new_local, sizeof(new_local), "%s", dir->GetEntryName());
1.881 - } else {
1.882 - g_LibSys.PathFormat(new_local, sizeof(new_local), "%s/%s", localpath, dir->GetEntryName());
1.883 - }
1.884 - LoadPluginsFromDir(basedir, new_local);
1.885 - } else if (dir->IsEntryFile()) {
1.886 - const char *name = dir->GetEntryName();
1.887 - size_t len = strlen(name);
1.888 - if (len >= 4
1.889 - && strcmp(&name[len-4], ".smx") == 0)
1.890 - {
1.891 - /* If the filename matches, load the plugin */
1.892 - char plugin[PLATFORM_MAX_PATH];
1.893 - if (localpath == NULL)
1.894 - {
1.895 - snprintf(plugin, sizeof(plugin), "%s", name);
1.896 - } else {
1.897 - g_LibSys.PathFormat(plugin, sizeof(plugin), "%s/%s", localpath, name);
1.898 - }
1.899 - LoadAutoPlugin(plugin);
1.900 - }
1.901 - }
1.902 - dir->NextEntry();
1.903 - }
1.904 - g_LibSys.CloseDirectory(dir);
1.905 -}
1.906 -
1.907 -LoadRes CPluginManager::_LoadPlugin(CPlugin **_plugin, const char *path, bool debug, PluginType type, char error[], size_t maxlength)
1.908 -{
1.909 - if (m_LoadingLocked)
1.910 - {
1.911 - return LoadRes_NeverLoad;
1.912 - }
1.913 -
1.914 - int err;
1.915 - bool no_load = false;
1.916 - PluginSettings *pset;
1.917 - unsigned int setcount = m_PluginInfo.GetSettingsNum();
1.918 - for (unsigned int i=0; i<setcount; i++)
1.919 - {
1.920 - if ((pset = m_PluginInfo.GetSettingsIfMatch(i, path)) == NULL)
1.921 - {
1.922 - continue;
1.923 - }
1.924 - if (pset->blockload_val)
1.925 - {
1.926 - no_load = true;
1.927 - break;
1.928 - }
1.929 - }
1.930 -
1.931 - /**
1.932 - * Does this plugin already exist?
1.933 - */
1.934 - CPlugin *pPlugin;
1.935 - if (sm_trie_retrieve(m_LoadLookup, path, (void **)&pPlugin))
1.936 - {
1.937 - /* Check to see if we should try reloading it */
1.938 - if (pPlugin->GetStatus() == Plugin_BadLoad
1.939 - || pPlugin->GetStatus() == Plugin_Error
1.940 - || pPlugin->GetStatus() == Plugin_Failed
1.941 - || no_load)
1.942 - {
1.943 - UnloadPlugin(pPlugin);
1.944 - }
1.945 - else
1.946 - {
1.947 - if (_plugin)
1.948 - {
1.949 - *_plugin = pPlugin;
1.950 - }
1.951 - return LoadRes_AlreadyLoaded;
1.952 - }
1.953 - }
1.954 -
1.955 - if (no_load)
1.956 - {
1.957 - return LoadRes_NeverLoad;
1.958 - }
1.959 -
1.960 - pPlugin = CPlugin::CreatePlugin(path, error, maxlength);
1.961 -
1.962 - assert(pPlugin != NULL);
1.963 -
1.964 - pPlugin->m_type = PluginType_MapUpdated;
1.965 -
1.966 - ICompilation *co = NULL;
1.967 -
1.968 - if (pPlugin->m_status == Plugin_Uncompiled)
1.969 - {
1.970 - co = g_pSourcePawn2->StartCompilation();
1.971 - }
1.972 -
1.973 - for (unsigned int i=0; i<setcount; i++)
1.974 - {
1.975 - if ((pset=m_PluginInfo.GetSettingsIfMatch(i, path)) == NULL)
1.976 - {
1.977 - continue;
1.978 - }
1.979 - pPlugin->m_type = pset->type_val;
1.980 - if (co)
1.981 - {
1.982 - for (unsigned int j=0; j<pset->opts_num; j++)
1.983 - {
1.984 - const char *key, *val;
1.985 - m_PluginInfo.GetOptionsForPlugin(pset, j, &key, &val);
1.986 - if (!key || !val)
1.987 - {
1.988 - continue;
1.989 - }
1.990 - if ((err = co->SetOption(key, val)) == SP_ERROR_NONE)
1.991 - {
1.992 - if (error)
1.993 - {
1.994 - snprintf(error, maxlength, "Unable to set JIT option (key \"%s\") (value \"%s\")", key, val);
1.995 - }
1.996 - co->Abort();
1.997 - co = NULL;
1.998 - break;
1.999 - }
1.1000 - }
1.1001 - }
1.1002 - }
1.1003 -
1.1004 - /* Do the actual compiling */
1.1005 - if (co != NULL)
1.1006 - {
1.1007 - char fullpath[PLATFORM_MAX_PATH];
1.1008 - g_SourceMod.BuildPath(Path_SM, fullpath, sizeof(fullpath), "plugins/%s", pPlugin->m_filename);
1.1009 -
1.1010 - pPlugin->m_pRuntime = g_pSourcePawn2->LoadPlugin(co, fullpath, &err);
1.1011 - if (pPlugin->m_pRuntime == NULL)
1.1012 - {
1.1013 - snprintf(error,
1.1014 - maxlength,
1.1015 - "Unable to load plugin (error %d: %s)",
1.1016 - err,
1.1017 - g_pSourcePawn2->GetErrorString(err));
1.1018 - }
1.1019 - else
1.1020 - {
1.1021 - pPlugin->UpdateInfo();
1.1022 - pPlugin->m_status = Plugin_Created;
1.1023 - }
1.1024 - }
1.1025 -
1.1026 - /* Get the status */
1.1027 - if (pPlugin->GetStatus() == Plugin_Created)
1.1028 - {
1.1029 - /* First native pass - add anything from Core */
1.1030 - g_ShareSys.BindNativesToPlugin(pPlugin, true);
1.1031 - pPlugin->InitIdentity();
1.1032 - if (pPlugin->Call_AskPluginLoad(error, maxlength))
1.1033 - {
1.1034 - /* Autoload any modules */
1.1035 - LoadOrRequireExtensions(pPlugin, 1, error, maxlength);
1.1036 - }
1.1037 - else
1.1038 - {
1.1039 - pPlugin->SetErrorState(Plugin_Failed, "%s", error);
1.1040 - }
1.1041 - }
1.1042 -
1.1043 - /* Save the time stamp */
1.1044 - time_t t = pPlugin->GetFileTimeStamp();
1.1045 - pPlugin->SetTimeStamp(t);
1.1046 -
1.1047 - if (_plugin)
1.1048 - {
1.1049 - *_plugin = pPlugin;
1.1050 - }
1.1051 -
1.1052 - return (pPlugin->GetStatus() == Plugin_Loaded) ? LoadRes_Successful : LoadRes_Failure;
1.1053 -}
1.1054 -
1.1055 -IPlugin *CPluginManager::LoadPlugin(const char *path, bool debug, PluginType type, char error[], size_t maxlength, bool *wasloaded)
1.1056 -{
1.1057 - CPlugin *pl;
1.1058 - LoadRes res;
1.1059 -
1.1060 - *wasloaded = false;
1.1061 - if ((res=_LoadPlugin(&pl, path, debug, type, error, maxlength)) == LoadRes_Failure)
1.1062 - {
1.1063 - delete pl;
1.1064 - return NULL;
1.1065 - }
1.1066 -
1.1067 - if (res == LoadRes_AlreadyLoaded)
1.1068 - {
1.1069 - *wasloaded = true;
1.1070 - return pl;
1.1071 - }
1.1072 -
1.1073 - if (res == LoadRes_NeverLoad)
1.1074 - {
1.1075 - if (m_LoadingLocked)
1.1076 - {
1.1077 - UTIL_Format(error, maxlength, "There is a global plugin loading lock in effect");
1.1078 - }
1.1079 - else
1.1080 - {
1.1081 - UTIL_Format(error, maxlength, "This plugin is blocked from loading (see plugin_settings.cfg)");
1.1082 - }
1.1083 - return NULL;
1.1084 - }
1.1085 -
1.1086 - AddPlugin(pl);
1.1087 -
1.1088 - /* Run second pass if we need to */
1.1089 - if (IsLateLoadTime() && pl->GetStatus() == Plugin_Loaded)
1.1090 - {
1.1091 - if (!RunSecondPass(pl, error, maxlength))
1.1092 - {
1.1093 - UnloadPlugin(pl);
1.1094 - return NULL;
1.1095 - }
1.1096 - pl->Call_OnAllPluginsLoaded();
1.1097 - }
1.1098 -
1.1099 - return pl;
1.1100 -}
1.1101 -
1.1102 -void CPluginManager::LoadAutoPlugin(const char *plugin)
1.1103 -{
1.1104 - CPlugin *pl = NULL;
1.1105 - LoadRes res;
1.1106 - char error[255] = "Unknown error";
1.1107 -
1.1108 - if ((res=_LoadPlugin(&pl, plugin, false, PluginType_MapUpdated, error, sizeof(error))) == LoadRes_Failure)
1.1109 - {
1.1110 - g_Logger.LogError("[SM] Failed to load plugin \"%s\": %s", plugin, error);
1.1111 - pl->SetErrorState(Plugin_Failed, "%s", error);
1.1112 - }
1.1113 -
1.1114 - if (res == LoadRes_Successful)
1.1115 - {
1.1116 - AddPlugin(pl);
1.1117 - }
1.1118 -}
1.1119 -
1.1120 -void CPluginManager::AddPlugin(CPlugin *pPlugin)
1.1121 -{
1.1122 - List<IPluginsListener *>::iterator iter;
1.1123 - IPluginsListener *pListener;
1.1124 -
1.1125 - for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
1.1126 - {
1.1127 - pListener = (*iter);
1.1128 - pListener->OnPluginCreated(pPlugin);
1.1129 - }
1.1130 -
1.1131 - m_plugins.push_back(pPlugin);
1.1132 - sm_trie_insert(m_LoadLookup, pPlugin->m_filename, pPlugin);
1.1133 -}
1.1134 -
1.1135 -void CPluginManager::LoadAll_SecondPass()
1.1136 -{
1.1137 - List<CPlugin *>::iterator iter;
1.1138 - CPlugin *pPlugin;
1.1139 -
1.1140 - char error[256];
1.1141 - for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
1.1142 - {
1.1143 - pPlugin = (*iter);
1.1144 - if (pPlugin->GetStatus() == Plugin_Loaded)
1.1145 - {
1.1146 - error[0] = '\0';
1.1147 - if (!RunSecondPass(pPlugin, error, sizeof(error)))
1.1148 - {
1.1149 - g_Logger.LogError("[SM] Unable to load plugin \"%s\": %s", pPlugin->GetFilename(), error);
1.1150 - pPlugin->SetErrorState(Plugin_Failed, "%s", error);
1.1151 - }
1.1152 - }
1.1153 - }
1.1154 -
1.1155 - m_AllPluginsLoaded = true;
1.1156 -}
1.1157 -
1.1158 -bool CPluginManager::FindOrRequirePluginDeps(CPlugin *pPlugin, char *error, size_t maxlength)
1.1159 -{
1.1160 - struct _pl
1.1161 - {
1.1162 - cell_t name;
1.1163 - cell_t file;
1.1164 - cell_t required;
1.1165 - } *pl;
1.1166 -
1.1167 - IPluginContext *pBase = pPlugin->GetBaseContext();
1.1168 - uint32_t num = pBase->GetPubVarsNum();
1.1169 - sp_pubvar_t *pubvar;
1.1170 - char *name, *file;
1.1171 - char pathfile[PLATFORM_MAX_PATH];
1.1172 -
1.1173 - for (uint32_t i=0; i<num; i++)
1.1174 - {
1.1175 - if (pBase->GetPubvarByIndex(i, &pubvar) != SP_ERROR_NONE)
1.1176 - {
1.1177 - continue;
1.1178 - }
1.1179 - if (strncmp(pubvar->name, "__pl_", 5) == 0)
1.1180 - {
1.1181 - pl = (_pl *)pubvar->offs;
1.1182 - if (pBase->LocalToString(pl->file, &file) != SP_ERROR_NONE)
1.1183 - {
1.1184 - continue;
1.1185 - }
1.1186 - if (pBase->LocalToString(pl->name, &name) != SP_ERROR_NONE)
1.1187 - {
1.1188 - continue;
1.1189 - }
1.1190 - g_LibSys.GetFileFromPath(pathfile, sizeof(pathfile), pPlugin->GetFilename());
1.1191 - if (strcmp(pathfile, file) == 0)
1.1192 - {
1.1193 - continue;
1.1194 - }
1.1195 - if (pl->required == false)
1.1196 - {
1.1197 - IPluginFunction *pFunc;
1.1198 - char buffer[64];
1.1199 - UTIL_Format(buffer, sizeof(buffer), "__pl_%s_SetNTVOptional", &pubvar->name[5]);
1.1200 - if ((pFunc=pBase->GetFunctionByName(buffer)))
1.1201 - {
1.1202 - cell_t res;
1.1203 - pFunc->Execute(&res);
1.1204 - if (pPlugin->GetBaseContext()->GetLastNativeError() != SP_ERROR_NONE)
1.1205 - {
1.1206 - if (error)
1.1207 - {
1.1208 - snprintf(error, maxlength, "Fatal error during initializing plugin load");
1.1209 - }
1.1210 - return false;
1.1211 - }
1.1212 - }
1.1213 - }
1.1214 - else
1.1215 - {
1.1216 - /* Check that we aren't registering the same library twice */
1.1217 - if (pPlugin->m_RequiredLibs.find(name) == pPlugin->m_RequiredLibs.end())
1.1218 - {
1.1219 - pPlugin->m_RequiredLibs.push_back(name);
1.1220 - }
1.1221 - else
1.1222 - {
1.1223 - continue;
1.1224 - }
1.1225 - List<CPlugin *>::iterator iter;
1.1226 - CPlugin *pl;
1.1227 - bool found = false;
1.1228 - for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
1.1229 - {
1.1230 - pl = (*iter);
1.1231 - if (pl->m_Libraries.find(name) != pl->m_Libraries.end())
1.1232 - {
1.1233 - found = true;
1.1234 - break;
1.1235 - }
1.1236 - }
1.1237 - if (!found)
1.1238 - {
1.1239 - if (error)
1.1240 - {
1.1241 - snprintf(error, maxlength, "Could not find required plugin \"%s\"", name);
1.1242 - }
1.1243 - return false;
1.1244 - }
1.1245 - }
1.1246 - }
1.1247 - }
1.1248 -
1.1249 - return true;
1.1250 -}
1.1251 -
1.1252 -bool CPluginManager::LoadOrRequireExtensions(CPlugin *pPlugin, unsigned int pass, char *error, size_t maxlength)
1.1253 -{
1.1254 - /* Find any extensions this plugin needs */
1.1255 - struct _ext
1.1256 - {
1.1257 - cell_t name;
1.1258 - cell_t file;
1.1259 - cell_t autoload;
1.1260 - cell_t required;
1.1261 - } *ext;
1.1262 -
1.1263 - IPluginContext *pBase = pPlugin->GetBaseContext();
1.1264 - uint32_t num = pBase->GetPubVarsNum();
1.1265 - sp_pubvar_t *pubvar;
1.1266 - IExtension *pExt;
1.1267 - char path[PLATFORM_MAX_PATH];
1.1268 - char *file, *name;
1.1269 - for (uint32_t i=0; i<num; i++)
1.1270 - {
1.1271 - if (pBase->GetPubvarByIndex(i, &pubvar) != SP_ERROR_NONE)
1.1272 - {
1.1273 - continue;
1.1274 - }
1.1275 - if (strncmp(pubvar->name, "__ext_", 6) == 0)
1.1276 - {
1.1277 - ext = (_ext *)pubvar->offs;
1.1278 - if (pBase->LocalToString(ext->file, &file) != SP_ERROR_NONE)
1.1279 - {
1.1280 - continue;
1.1281 - }
1.1282 - if (pBase->LocalToString(ext->name, &name) != SP_ERROR_NONE)
1.1283 - {
1.1284 - continue;
1.1285 - }
1.1286 - if (pass == 1)
1.1287 - {
1.1288 - /* Attempt to auto-load if necessary */
1.1289 - if (ext->autoload)
1.1290 - {
1.1291 - g_LibSys.PathFormat(path, PLATFORM_MAX_PATH, "%s", file);
1.1292 - g_Extensions.LoadAutoExtension(path);
1.1293 - }
1.1294 - }
1.1295 - else if (pass == 2)
1.1296 - {
1.1297 - /* Is this required? */
1.1298 - if (ext->required)
1.1299 - {
1.1300 - g_LibSys.PathFormat(path, PLATFORM_MAX_PATH, "%s", file);
1.1301 - if ((pExt = g_Extensions.FindExtensionByFile(path)) == NULL)
1.1302 - {
1.1303 - pExt = g_Extensions.FindExtensionByName(name);
1.1304 - }
1.1305 - /* :TODO: should we bind to unloaded extensions?
1.1306 - * Currently the extension manager will ignore this.
1.1307 - */
1.1308 - if (!pExt || !pExt->IsRunning(NULL, 0))
1.1309 - {
1.1310 - if (error)
1.1311 - {
1.1312 - snprintf(error, maxlength, "Required extension \"%s\" file(\"%s\") not running", name, file);
1.1313 - }
1.1314 - return false;
1.1315 - }
1.1316 - else
1.1317 - {
1.1318 - g_Extensions.BindChildPlugin(pExt, pPlugin);
1.1319 - }
1.1320 - }
1.1321 - else
1.1322 - {
1.1323 - IPluginFunction *pFunc;
1.1324 - char buffer[64];
1.1325 - UTIL_Format(buffer, sizeof(buffer), "__ext_%s_SetNTVOptional", &pubvar->name[6]);
1.1326 -
1.1327 - if ((pFunc = pBase->GetFunctionByName(buffer)) != NULL)
1.1328 - {
1.1329 - cell_t res;
1.1330 - pFunc->Execute(&res);
1.1331 - if (pPlugin->GetBaseContext()->GetLastNativeError() != SP_ERROR_NONE)
1.1332 - {
1.1333 - if (error)
1.1334 - {
1.1335 - UTIL_Format(error, maxlength, "Fatal error during plugin initialization (ext req)");
1.1336 - }
1.1337 - return false;
1.1338 - }
1.1339 - }
1.1340 - }
1.1341 - }
1.1342 - }
1.1343 - }
1.1344 -
1.1345 - return true;
1.1346 -}
1.1347 -
1.1348 -bool CPluginManager::RunSecondPass(CPlugin *pPlugin, char *error, size_t maxlength)
1.1349 -{
1.1350 - /* Second pass for extension requirements */
1.1351 - if (!LoadOrRequireExtensions(pPlugin, 2, error, maxlength))
1.1352 - {
1.1353 - return false;
1.1354 - }
1.1355 -
1.1356 - if (!FindOrRequirePluginDeps(pPlugin, error, maxlength))
1.1357 - {
1.1358 - return false;
1.1359 - }
1.1360 -
1.1361 - /* Run another binding pass */
1.1362 - g_ShareSys.BindNativesToPlugin(pPlugin, false);
1.1363 -
1.1364 - /* Find any unbound natives. Right now, these are not allowed. */
1.1365 - IPluginContext *pContext = pPlugin->GetBaseContext();
1.1366 - uint32_t num = pContext->GetNativesNum();
1.1367 - sp_native_t *native;
1.1368 - for (unsigned int i=0; i<num; i++)
1.1369 - {
1.1370 - if (pContext->GetNativeByIndex(i, &native) != SP_ERROR_NONE)
1.1371 - {
1.1372 - break;
1.1373 - }
1.1374 - if (native->status == SP_NATIVE_UNBOUND
1.1375 - && native->name[0] != '@'
1.1376 - && !(native->flags & SP_NTVFLAG_OPTIONAL))
1.1377 - {
1.1378 - if (error)
1.1379 - {
1.1380 - snprintf(error, maxlength, "Native \"%s\" was not found.", native->name);
1.1381 - }
1.1382 - return false;
1.1383 - }
1.1384 - }
1.1385 -
1.1386 - /* Finish by telling all listeners */
1.1387 - List<IPluginsListener *>::iterator iter;
1.1388 - IPluginsListener *pListener;
1.1389 - for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
1.1390 - {
1.1391 - pListener = (*iter);
1.1392 - pListener->OnPluginLoaded(pPlugin);
1.1393 - }
1.1394 -
1.1395 - /* Tell this plugin to finish initializing itself */
1.1396 - pPlugin->Call_OnPluginStart();
1.1397 -
1.1398 - /* Now, if we have fake natives, go through all plugins that might need rebinding */
1.1399 - if (pPlugin->GetStatus() <= Plugin_Paused && pPlugin->m_Natives.size())
1.1400 - {
1.1401 - List<CPlugin *>::iterator pl_iter;
1.1402 - CPlugin *pOther;
1.1403 - for (pl_iter = m_plugins.begin();
1.1404 - pl_iter != m_plugins.end();
1.1405 - pl_iter++)
1.1406 - {
1.1407 - pOther = (*pl_iter);
1.1408 - if ((pOther->GetStatus() == Plugin_Error
1.1409 - && (pOther->m_FakeNativesMissing || pOther->m_LibraryMissing))
1.1410 - || pOther->m_FakeNativesMissing)
1.1411 - {
1.1412 - TryRefreshDependencies(pOther);
1.1413 - }
1.1414 - else if ((pOther->GetStatus() == Plugin_Running
1.1415 - || pOther->GetStatus() == Plugin_Paused)
1.1416 - && pOther != pPlugin)
1.1417 - {
1.1418 - List<NativeEntry *>::iterator nv_iter;
1.1419 - for (nv_iter = pPlugin->m_Natives.begin();
1.1420 - nv_iter != pPlugin->m_Natives.end();
1.1421 - nv_iter++)
1.1422 - {
1.1423 - g_ShareSys.BindNativeToPlugin(pOther, (*nv_iter));
1.1424 - }
1.1425 - }
1.1426 - }
1.1427 - }
1.1428 -
1.1429 - /* Go through our libraries and tell other plugins they're added */
1.1430 - List<String>::iterator s_iter;
1.1431 - for (s_iter = pPlugin->m_Libraries.begin();
1.1432 - s_iter != pPlugin->m_Libraries.end();
1.1433 - s_iter++)
1.1434 - {
1.1435 - OnLibraryAction((*s_iter).c_str(), true, false);
1.1436 - }
1.1437 -
1.1438 - /* :TODO: optimize? does this even matter? */
1.1439 - pPlugin->GetPhrases()->AddPhraseFile("core.phrases");
1.1440 -
1.1441 - return true;
1.1442 -}
1.1443 -
1.1444 -void CPluginManager::TryRefreshDependencies(CPlugin *pPlugin)
1.1445 -{
1.1446 - assert(pPlugin->GetBaseContext() != NULL);
1.1447 -
1.1448 - g_ShareSys.BindNativesToPlugin(pPlugin, false);
1.1449 -
1.1450 - List<String>::iterator lib_iter;
1.1451 - List<String>::iterator req_iter;
1.1452 - List<CPlugin *>::iterator pl_iter;
1.1453 - CPlugin *pl;
1.1454 - for (req_iter=pPlugin->m_RequiredLibs.begin(); req_iter!=pPlugin->m_RequiredLibs.end(); req_iter++)
1.1455 - {
1.1456 - bool found = false;
1.1457 - for (pl_iter=m_plugins.begin(); pl_iter!=m_plugins.end(); pl_iter++)
1.1458 - {
1.1459 - pl = (*pl_iter);
1.1460 - for (lib_iter=pl->m_Libraries.begin(); lib_iter!=pl->m_Libraries.end(); lib_iter++)
1.1461 - {
1.1462 - if ((*req_iter) == (*lib_iter))
1.1463 - {
1.1464 - found = true;
1.1465 - }
1.1466 - }
1.1467 - }
1.1468 - if (!found)
1.1469 - {
1.1470 - pPlugin->SetErrorState(Plugin_Error, "Library not found: %s", (*req_iter).c_str());
1.1471 - return;
1.1472 - }
1.1473 - }
1.1474 -
1.1475 - /* Find any unbound natives
1.1476 - * Right now, these are not allowed
1.1477 - */
1.1478 - IPluginContext *pContext = pPlugin->GetBaseContext();
1.1479 - uint32_t num = pContext->GetNativesNum();
1.1480 - sp_native_t *native;
1.1481 - for (unsigned int i=0; i<num; i++)
1.1482 - {
1.1483 - if (pContext->GetNativeByIndex(i, &native) != SP_ERROR_NONE)
1.1484 - {
1.1485 - break;
1.1486 - }
1.1487 - if (native->status == SP_NATIVE_UNBOUND && !(native->flags & SP_NTVFLAG_OPTIONAL))
1.1488 - {
1.1489 - pPlugin->SetErrorState(Plugin_Error, "Native not found: %s", native->name);
1.1490 - return;
1.1491 - }
1.1492 - }
1.1493 -
1.1494 - if (pPlugin->GetStatus() == Plugin_Error)
1.1495 - {
1.1496 - /* If we got here, all natives are okay again! */
1.1497 - pPlugin->m_status = Plugin_Running;
1.1498 - if (pPlugin->m_pRuntime->IsPaused())
1.1499 - {
1.1500 - pPlugin->m_pRuntime->SetPauseState(false);
1.1501 - _SetPauseState(pPlugin, false);
1.1502 - }
1.1503 - }
1.1504 -}
1.1505 -
1.1506 -bool CPluginManager::UnloadPlugin(IPlugin *plugin)
1.1507 -{
1.1508 - CPlugin *pPlugin = (CPlugin *)plugin;
1.1509 -
1.1510 - /* This prevents removal during insertion or anything else weird */
1.1511 - if (m_plugins.find(pPlugin) == m_plugins.end())
1.1512 - {
1.1513 - return false;
1.1514 - }
1.1515 -
1.1516 - IPluginContext *pContext = plugin->GetBaseContext();
1.1517 - if (pContext->IsInExec())
1.1518 - {
1.1519 - char buffer[255];
1.1520 - UTIL_Format(buffer, sizeof(buffer), "sm plugins unload %s\n", plugin->GetFilename());
1.1521 - engine->ServerCommand(buffer);
1.1522 - return false;
1.1523 - }
1.1524 -
1.1525 - /* Remove us from the lookup table and linked list */
1.1526 - m_plugins.remove(pPlugin);
1.1527 - sm_trie_delete(m_LoadLookup, pPlugin->m_filename);
1.1528 -
1.1529 - /* Go through our libraries and tell other plugins they're gone */
1.1530 - List<String>::iterator s_iter;
1.1531 - for (s_iter = pPlugin->m_Libraries.begin();
1.1532 - s_iter != pPlugin->m_Libraries.end();
1.1533 - s_iter++)
1.1534 - {
1.1535 - OnLibraryAction((*s_iter).c_str(), true, true);
1.1536 - }
1.1537 -
1.1538 - List<IPluginsListener *>::iterator iter;
1.1539 - IPluginsListener *pListener;
1.1540 -
1.1541 - if (pPlugin->GetStatus() <= Plugin_Error)
1.1542 - {
1.1543 - /* Notify listeners of unloading */
1.1544 - for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
1.1545 - {
1.1546 - pListener = (*iter);
1.1547 - pListener->OnPluginUnloaded(pPlugin);
1.1548 - }
1.1549 - /* Notify plugin */
1.1550 - pPlugin->Call_OnPluginEnd();
1.1551 - }
1.1552 -
1.1553 - pPlugin->DropEverything();
1.1554 -
1.1555 - for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
1.1556 - {
1.1557 - /* Notify listeners of destruction */
1.1558 - pListener = (*iter);
1.1559 - pListener->OnPluginDestroyed(pPlugin);
1.1560 - }
1.1561 -
1.1562 - /* Tell the plugin to delete itself */
1.1563 - delete pPlugin;
1.1564 -
1.1565 - return true;
1.1566 -}
1.1567 -
1.1568 -IPlugin *CPluginManager::FindPluginByContext(const sp_context_t *ctx)
1.1569 -{
1.1570 - IPlugin *pPlugin;
1.1571 - IPluginContext *pContext;
1.1572 -
1.1573 - pContext = reinterpret_cast<IPluginContext *>(const_cast<sp_context_t *>(ctx));
1.1574 -
1.1575 - if (pContext->GetKey(2, (void **)&pPlugin))
1.1576 - {
1.1577 - return pPlugin;
1.1578 - }
1.1579 -
1.1580 - return NULL;
1.1581 -}
1.1582 -
1.1583 -CPlugin *CPluginManager::GetPluginByCtx(const sp_context_t *ctx)
1.1584 -{
1.1585 - return (CPlugin *)FindPluginByContext(ctx);
1.1586 -}
1.1587 -
1.1588 -unsigned int CPluginManager::GetPluginCount()
1.1589 -{
1.1590 - return m_plugins.size();
1.1591 -}
1.1592 -
1.1593 -void CPluginManager::AddPluginsListener(IPluginsListener *listener)
1.1594 -{
1.1595 - m_listeners.push_back(listener);
1.1596 -}
1.1597 -
1.1598 -void CPluginManager::RemovePluginsListener(IPluginsListener *listener)
1.1599 -{
1.1600 - m_listeners.remove(listener);
1.1601 -}
1.1602 -
1.1603 -IPluginIterator *CPluginManager::GetPluginIterator()
1.1604 -{
1.1605 - if (m_iters.empty())
1.1606 - {
1.1607 - return new CPluginIterator(&m_plugins);
1.1608 - } else {
1.1609 - CPluginIterator *iter = m_iters.front();
1.1610 - m_iters.pop();
1.1611 - iter->Reset();
1.1612 - return iter;
1.1613 - }
1.1614 -}
1.1615 -
1.1616 -void CPluginManager::ReleaseIterator(CPluginIterator *iter)
1.1617 -{
1.1618 - m_iters.push(iter);
1.1619 -}
1.1620 -
1.1621 -bool CPluginManager::TestAliasMatch(const char *alias, const char *localpath)
1.1622 -{
1.1623 - /* As an optimization, we do not call strlen, but compute the length in the first pass */
1.1624 - size_t alias_len = 0;
1.1625 - size_t local_len = 0;
1.1626 -
1.1627 - const char *ptr = alias;
1.1628 - unsigned int alias_explicit_paths = 0;
1.1629 - unsigned int alias_path_end = 0;
1.1630 - while (*ptr != '\0')
1.1631 - {
1.1632 - if (*ptr == '\\' || *ptr == '/')
1.1633 - {
1.1634 - alias_explicit_paths++;
1.1635 - alias_path_end = alias_len;
1.1636 - }
1.1637 - alias_len++;
1.1638 - ptr++;
1.1639 - }
1.1640 -
1.1641 - if (alias_explicit_paths && alias_path_end == alias_len - 1)
1.1642 - {
1.1643 - /* Trailing slash is totally invalid here */
1.1644 - return false;
1.1645 - }
1.1646 -
1.1647 - ptr = localpath;
1.1648 - unsigned int local_explicit_paths = 0;
1.1649 - unsigned int local_path_end = 0;
1.1650 - while (*ptr != '\0')
1.1651 - {
1.1652 - if (*ptr == '\\' || *ptr == '/')
1.1653 - {
1.1654 - local_explicit_paths++;
1.1655 - local_path_end = local_len;
1.1656 - }
1.1657 - local_len++;
1.1658 - ptr++;
1.1659 - }
1.1660 -
1.1661 - /* If the alias has more explicit paths than the real path,
1.1662 - * no match will be possible.
1.1663 - */
1.1664 - if (alias_explicit_paths > local_explicit_paths)
1.1665 - {
1.1666 - return false;
1.1667 - }
1.1668 -
1.1669 - if (alias_explicit_paths)
1.1670 - {
1.1671 - /* We need to find if the paths match now. For example, these should all match:
1.1672 - * csdm csdm
1.1673 - * csdm optional/csdm
1.1674 - * csdm/ban optional/crab/csdm/ban
1.1675 - */
1.1676 - const char *aliasptr = alias;
1.1677 - const char *localptr = localpath;
1.1678 - bool match = true;
1.1679 - do
1.1680 - {
1.1681 - if (*aliasptr != *localptr)
1.1682 - {
1.1683 - /* We have to knock one path off */
1.1684 - local_explicit_paths--;
1.1685 - if (alias_explicit_paths > local_explicit_paths)
1.1686 - {
1.1687 - /* Skip out if we're gonna have an impossible match */
1.1688 - return false;
1.1689 - }
1.1690 - /* Eat up localptr tokens until we get a result */
1.1691 - while (((localptr - localpath) < (int)local_path_end)
1.1692 - && *localptr != '/'
1.1693 - && *localptr != '\\')
1.1694 - {
1.1695 - localptr++;
1.1696 - }
1.1697 - /* Check if we hit the end of our searchable area.
1.1698 - * This probably isn't possible because of the path
1.1699 - * count check, but it's a good idea anyway.
1.1700 - */
1.1701 - if ((localptr - localpath) >= (int)local_path_end)
1.1702 - {
1.1703 - return false;
1.1704 - } else {
1.1705 - /* Consume the slash token */
1.1706 - localptr++;
1.1707 - }
1.1708 - /* Reset the alias pointer so we can continue consuming */
1.1709 - aliasptr = alias;
1.1710 - match = false;
1.1711 - continue;
1.1712 - }
1.1713 - /* Note:
1.1714 - * This is safe because if localptr terminates early, aliasptr will too
1.1715 - */
1.1716 - do
1.1717 - {
1.1718 - /* We should never reach the end of the string because of this check. */
1.1719 - bool aliasend = (aliasptr - alias) > (int)alias_path_end;
1.1720 - bool localend = (localptr - localpath) > (int)local_path_end;
1.1721 - if (aliasend || localend)
1.1722 - {
1.1723 - if (aliasend && localend)
1.1724 - {
1.1725 - /* we matched, and we can break out now */
1.1726 - match = true;
1.1727 - break;
1.1728 - }
1.1729 - /* Otherwise, we've hit the end somehow and rest won't match up. Break out. */
1.1730 - match = false;
1.1731 - break;
1.1732 - }
1.1733 -
1.1734 - /* If we got here, it's safe to compare the next two tokens */
1.1735 - if (*localptr != *aliasptr)
1.1736 - {
1.1737 - match = false;
1.1738 - break;
1.1739 - }
1.1740 - localptr++;
1.1741 - aliasptr++;
1.1742 - } while (true);
1.1743 - } while (!match);
1.1744 - }
1.1745 -
1.1746 - /* If we got here, it's time to compare filenames */
1.1747 - const char *aliasptr = alias;
1.1748 - const char *localptr = localpath;
1.1749 -
1.1750 - if (alias_explicit_paths)
1.1751 - {
1.1752 - aliasptr = &alias[alias_path_end + 1];
1.1753 - }
1.1754 -
1.1755 - if (local_explicit_paths)
1.1756 - {
1.1757 - localptr = &localpath[local_path_end + 1];
1.1758 - }
1.1759 -
1.1760 - while (true)
1.1761 - {
1.1762 - if (*aliasptr == '*')
1.1763 - {
1.1764 - /* First, see if this is the last character */
1.1765 - if ((unsigned)(aliasptr - alias) == alias_len - 1)
1.1766 - {
1.1767 - /* If so, there's no need to match anything else */
1.1768 - return true;
1.1769 - }
1.1770 - /* Otherwise, we need to search for an appropriate matching sequence in local.
1.1771 - * Note that we only need to search up to the next asterisk.
1.1772 - */
1.1773 - aliasptr++;
1.1774 - bool match = true;
1.1775 - const char *local_orig = localptr;
1.1776 - do
1.1777 - {
1.1778 - match = true;
1.1779 - while (*aliasptr != '\0' && *aliasptr != '*')
1.1780 - {
1.1781 - /* Since aliasptr is never '\0', localptr hitting the end will fail */
1.1782 - if (*aliasptr != *localptr)
1.1783 - {
1.1784 - match = false;
1.1785 - break;
1.1786 - }
1.1787 - aliasptr++;
1.1788 - localptr++;
1.1789 - }
1.1790 - if (!match)
1.1791 - {
1.1792 - /* If we didn't get a match, we need to advance the search stream.
1.1793 - * This will let us skip tokens while still searching for another match.
1.1794 - */
1.1795 - localptr = ++local_orig;
1.1796 - /* Make sure we don't go out of bounds */
1.1797 - if (*localptr == '\0')
1.1798 - {
1.1799 - break;
1.1800 - }
1.1801 - }
1.1802 - } while (!match);
1.1803 -
1.1804 - if (!match)
1.1805 - {
1.1806 - return false;
1.1807 - } else {
1.1808 - /* If we got a match, move on to the next token */
1.1809 - continue;
1.1810 - }
1.1811 - } else if (*aliasptr == '\0') {
1.1812 - if (*localptr == '\0'
1.1813 - ||
1.1814 - strcmp(localptr, ".smx") == 0)
1.1815 - {
1.1816 - return true;
1.1817 - } else {
1.1818 - return false;
1.1819 - }
1.1820 - } else if (*aliasptr != *localptr) {
1.1821 - return false;
1.1822 - }
1.1823 - aliasptr++;
1.1824 - localptr++;
1.1825 - }
1.1826 -
1.1827 - return true;
1.1828 -}
1.1829 -
1.1830 -bool CPluginManager::IsLateLoadTime() const
1.1831 -{
1.1832 - return (m_AllPluginsLoaded || !g_SourceMod.IsMapLoading());
1.1833 -}
1.1834 -
1.1835 -void CPluginManager::OnSourceModAllInitialized()
1.1836 -{
1.1837 - m_MyIdent = g_ShareSys.CreateCoreIdentity();
1.1838 -
1.1839 - HandleAccess sec;
1.1840 - g_HandleSys.InitAccessDefaults(NULL, &sec);
1.1841 -
1.1842 - sec.access[HandleAccess_Delete] = HANDLE_RESTRICT_IDENTITY;
1.1843 - sec.access[HandleAccess_Clone] = HANDLE_RESTRICT_IDENTITY;
1.1844 -
1.1845 - g_PluginType = g_HandleSys.CreateType("Plugin", this, 0, NULL, &sec, m_MyIdent, NULL);
1.1846 - g_PluginIdent = g_ShareSys.CreateIdentType("PLUGIN");
1.1847 -
1.1848 - g_RootMenu.AddRootConsoleCommand("plugins", "Manage Plugins", this);
1.1849 -
1.1850 - g_ShareSys.AddInterface(NULL, this);
1.1851 -}
1.1852 -
1.1853 -void CPluginManager::OnSourceModShutdown()
1.1854 -{
1.1855 - g_RootMenu.RemoveRootConsoleCommand("plugins", this);
1.1856 -
1.1857 - UnloadAll();
1.1858 -
1.1859 - g_HandleSys.RemoveType(g_PluginType, m_MyIdent);
1.1860 - g_ShareSys.DestroyIdentType(g_PluginIdent);
1.1861 - g_ShareSys.DestroyIdentity(m_MyIdent);
1.1862 -}
1.1863 -
1.1864 -void CPluginManager::OnHandleDestroy(HandleType_t type, void *object)
1.1865 -{
1.1866 - /* We don't care about the internal object, actually */
1.1867 -}
1.1868 -
1.1869 -bool CPluginManager::GetHandleApproxSize(HandleType_t type, void *object, unsigned int *pSize)
1.1870 -{
1.1871 - *pSize = ((CPlugin *)object)->CalcMemUsage();
1.1872 - return true;
1.1873 -}
1.1874 -
1.1875 -IPlugin *CPluginManager::PluginFromHandle(Handle_t handle, HandleError *err)
1.1876 -{
1.1877 - IPlugin *pPlugin;
1.1878 - HandleError _err;
1.1879 -
1.1880 - HandleSecurity sec;
1.1881 -
1.1882 - sec.pOwner = NULL;
1.1883 - sec.pIdentity = m_MyIdent;
1.1884 -
1.1885 - if ((_err=g_HandleSys.ReadHandle(handle, g_PluginType, &sec, (void **)&pPlugin)) != HandleError_None)
1.1886 - {
1.1887 - pPlugin = NULL;
1.1888 - }
1.1889 -
1.1890 - if (err)
1.1891 - {
1.1892 - *err = _err;
1.1893 - }
1.1894 -
1.1895 - return pPlugin;
1.1896 -}
1.1897 -
1.1898 -CPlugin *CPluginManager::GetPluginByOrder(int num)
1.1899 -{
1.1900 - if (num < 1 || num > (int)GetPluginCount())
1.1901 - {
1.1902 - return NULL;
1.1903 - }
1.1904 -
1.1905 - CPlugin *pl;
1.1906 - int id = 1;
1.1907 -
1.1908 - SourceHook::List<CPlugin *>::iterator iter;
1.1909 - for (iter = m_plugins.begin(); iter != m_plugins.end() && id < num; iter++, id++) {}
1.1910 -
1.1911 - pl = *iter;
1.1912 -
1.1913 - return pl;
1.1914 -}
1.1915 -
1.1916 -const char *CPluginManager::GetStatusText(PluginStatus st)
1.1917 -{
1.1918 - switch (st)
1.1919 - {
1.1920 - case Plugin_Running:
1.1921 - return "Running";
1.1922 - case Plugin_Paused:
1.1923 - return "Paused";
1.1924 - case Plugin_Error:
1.1925 - return "Error";
1.1926 - case Plugin_Uncompiled:
1.1927 - return "Uncompiled";
1.1928 - case Plugin_BadLoad:
1.1929 - return "Bad Load";
1.1930 - case Plugin_Failed:
1.1931 - return "Failed";
1.1932 - default:
1.1933 - assert(false);
1.1934 - return "-";
1.1935 - }
1.1936 -}
1.1937 -
1.1938 -void CPluginManager::OnRootConsoleCommand(const char *cmdname, const CCommand &command)
1.1939 -{
1.1940 - int argcount = command.ArgC();
1.1941 - if (argcount >= 3)
1.1942 - {
1.1943 - const char *cmd = command.Arg(2);
1.1944 - if (strcmp(cmd, "list") == 0)
1.1945 - {
1.1946 - char buffer[256];
1.1947 - unsigned int id = 1;
1.1948 - int plnum = GetPluginCount();
1.1949 -
1.1950 - if (!plnum)
1.1951 - {
1.1952 - g_RootMenu.ConsolePrint("[SM] No plugins loaded");
1.1953 - return;
1.1954 - }
1.1955 - else
1.1956 - {
1.1957 - g_RootMenu.ConsolePrint("[SM] Listing %d plugin%s:", GetPluginCount(), (plnum > 1) ? "s" : "");
1.1958 - }
1.1959 -
1.1960 - CPlugin *pl;
1.1961 - SourceHook::List<CPlugin *>::iterator iter;
1.1962 - SourceHook::List<CPlugin *> m_FailList;
1.1963 -
1.1964 - for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++, id++)
1.1965 - {
1.1966 - pl = (*iter);
1.1967 - assert(pl->GetStatus() != Plugin_Created);
1.1968 - int len = 0;
1.1969 - const sm_plugininfo_t *info = pl->GetPublicInfo();
1.1970 - if (pl->GetStatus() != Plugin_Running)
1.1971 - {
1.1972 - len += UTIL_Format(buffer, sizeof(buffer), " %02d <%s>", id, GetStatusText(pl->GetStatus()));
1.1973 -
1.1974 - if (pl->GetStatus() <= Plugin_Error)
1.1975 - {
1.1976 - /* Plugin has failed to load. */
1.1977 - m_FailList.push_back(pl);
1.1978 - }
1.1979 - }
1.1980 - else
1.1981 - {
1.1982 - len += UTIL_Format(buffer, sizeof(buffer), " %02d", id);
1.1983 - }
1.1984 - len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " \"%s\"", (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
1.1985 - if (IS_STR_FILLED(info->version))
1.1986 - {
1.1987 - len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " (%s)", info->version);
1.1988 - }
1.1989 - if (IS_STR_FILLED(info->author))
1.1990 - {
1.1991 - UTIL_Format(&buffer[len], sizeof(buffer)-len, " by %s", info->author);
1.1992 - }
1.1993 - g_RootMenu.ConsolePrint("%s", buffer);
1.1994 - }
1.1995 -
1.1996 - if (!m_FailList.empty())
1.1997 - {
1.1998 - g_RootMenu.ConsolePrint("Load Errors:");
1.1999 -
1.2000 - SourceHook::List<CPlugin *>::iterator _iter;
1.2001 -
1.2002 - CPlugin *pl;
1.2003 -
1.2004 - for (_iter=m_FailList.begin(); _iter!=m_FailList.end(); _iter++)
1.2005 - {
1.2006 - pl = (CPlugin *)*_iter;
1.2007 - g_RootMenu.ConsolePrint("%s: %s",(IS_STR_FILLED(pl->GetPublicInfo()->name)) ? pl->GetPublicInfo()->name : pl->GetFilename(), pl->m_errormsg);
1.2008 - }
1.2009 - }
1.2010 -
1.2011 - return;
1.2012 - }
1.2013 - else if (strcmp(cmd, "load") == 0)
1.2014 - {
1.2015 - if (argcount < 4)
1.2016 - {
1.2017 - g_RootMenu.ConsolePrint("[SM] Usage: sm plugins load <file>");
1.2018 - return;
1.2019 - }
1.2020 -
1.2021 - char error[128];
1.2022 - bool wasloaded;
1.2023 - const char *filename = command.Arg(3);
1.2024 -
1.2025 - char pluginfile[256];
1.2026 - const char *ext = g_LibSys.GetFileExtension(filename) ? "" : ".smx";
1.2027 - UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", filename, ext);
1.2028 -
1.2029 - IPlugin *pl = LoadPlugin(pluginfile, false, PluginType_MapUpdated, error, sizeof(error), &wasloaded);
1.2030 -
1.2031 - if (wasloaded)
1.2032 - {
1.2033 - g_RootMenu.ConsolePrint("[SM] Plugin %s is already loaded.", pluginfile);
1.2034 - return;
1.2035 - }
1.2036 -
1.2037 - if (pl)
1.2038 - {
1.2039 - g_RootMenu.ConsolePrint("[SM] Loaded plugin %s successfully.", pluginfile);
1.2040 - }
1.2041 - else
1.2042 - {
1.2043 - g_RootMenu.ConsolePrint("[SM] Plugin %s failed to load: %s.", pluginfile, error);
1.2044 - }
1.2045 -
1.2046 - return;
1.2047 - }
1.2048 - else if (strcmp(cmd, "unload") == 0)
1.2049 - {
1.2050 - if (argcount < 4)
1.2051 - {
1.2052 - g_RootMenu.ConsolePrint("[SM] Usage: sm plugins unload <#|file>");
1.2053 - return;
1.2054 - }
1.2055 -
1.2056 - CPlugin *pl;
1.2057 - char *end;
1.2058 - const char *arg = command.Arg(3);
1.2059 - int id = strtol(arg, &end, 10);
1.2060 -
1.2061 - if (*end == '\0')
1.2062 - {
1.2063 - pl = GetPluginByOrder(id);
1.2064 - if (!pl)
1.2065 - {
1.2066 - g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
1.2067 - return;
1.2068 - }
1.2069 - }
1.2070 - else
1.2071 - {
1.2072 - char pluginfile[256];
1.2073 - const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
1.2074 - UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
1.2075 -
1.2076 - if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
1.2077 - {
1.2078 - g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
1.2079 - return;
1.2080 - }
1.2081 - }
1.2082 -
1.2083 - char name[PLATFORM_MAX_PATH];
1.2084 - const sm_plugininfo_t *info = pl->GetPublicInfo();
1.2085 - strcpy(name, (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
1.2086 -
1.2087 - if (UnloadPlugin(pl))
1.2088 - {
1.2089 - g_RootMenu.ConsolePrint("[SM] Plugin %s unloaded successfully.", name);
1.2090 - }
1.2091 - else
1.2092 - {
1.2093 - g_RootMenu.ConsolePrint("[SM] Failed to unload plugin %s.", name);
1.2094 - }
1.2095 -
1.2096 - return;
1.2097 - }
1.2098 - else if (strcmp(cmd, "unload_all") == 0)
1.2099 - {
1.2100 - g_PluginSys.UnloadAll();
1.2101 - g_RootMenu.ConsolePrint("[SM] All plugins have been unloaded.");
1.2102 - return;
1.2103 - }
1.2104 - else if (strcmp(cmd, "load_lock") == 0)
1.2105 - {
1.2106 - if (m_LoadingLocked)
1.2107 - {
1.2108 - g_RootMenu.ConsolePrint("[SM] There is already a loading lock in effect.");
1.2109 - }
1.2110 - else
1.2111 - {
1.2112 - m_LoadingLocked = true;
1.2113 - g_RootMenu.ConsolePrint("[SM] Loading is now locked; no plugins will be loaded or re-loaded.");
1.2114 - }
1.2115 - return;
1.2116 - }
1.2117 - else if (strcmp(cmd, "load_unlock") == 0)
1.2118 - {
1.2119 - if (m_LoadingLocked)
1.2120 - {
1.2121 - m_LoadingLocked = false;
1.2122 - g_RootMenu.ConsolePrint("[SM] The loading lock is no longer in effect.");
1.2123 - }
1.2124 - else
1.2125 - {
1.2126 - g_RootMenu.ConsolePrint("[SM] There was no loading lock in effect.");
1.2127 - }
1.2128 - return;
1.2129 - }
1.2130 - else if (strcmp(cmd, "info") == 0)
1.2131 - {
1.2132 - if (argcount < 4)
1.2133 - {
1.2134 - g_RootMenu.ConsolePrint("[SM] Usage: sm plugins info <#>");
1.2135 - return;
1.2136 - }
1.2137 -
1.2138 - CPlugin *pl;
1.2139 - char *end;
1.2140 - const char *arg = command.Arg(3);
1.2141 - int id = strtol(arg, &end, 10);
1.2142 -
1.2143 - if (*end == '\0')
1.2144 - {
1.2145 - pl = GetPluginByOrder(id);
1.2146 - if (!pl)
1.2147 - {
1.2148 - g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
1.2149 - return;
1.2150 - }
1.2151 - }
1.2152 - else
1.2153 - {
1.2154 - char pluginfile[256];
1.2155 - const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
1.2156 - UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
1.2157 -
1.2158 - if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
1.2159 - {
1.2160 - g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
1.2161 - return;
1.2162 - }
1.2163 - }
1.2164 -
1.2165 - const sm_plugininfo_t *info = pl->GetPublicInfo();
1.2166 -
1.2167 - g_RootMenu.ConsolePrint(" Filename: %s", pl->GetFilename());
1.2168 - if (pl->GetStatus() <= Plugin_Error)
1.2169 - {
1.2170 - if (IS_STR_FILLED(info->name))
1.2171 - {
1.2172 - if (IS_STR_FILLED(info->description))
1.2173 - {
1.2174 - g_RootMenu.ConsolePrint(" Title: %s (%s)", info->name, info->description);
1.2175 - } else {
1.2176 - g_RootMenu.ConsolePrint(" Title: %s", info->name);
1.2177 - }
1.2178 - }
1.2179 - if (IS_STR_FILLED(info->author))
1.2180 - {
1.2181 - g_RootMenu.ConsolePrint(" Author: %s", info->author);
1.2182 - }
1.2183 - if (IS_STR_FILLED(info->version))
1.2184 - {
1.2185 - g_RootMenu.ConsolePrint(" Version: %s", info->version);
1.2186 - }
1.2187 - if (IS_STR_FILLED(info->url))
1.2188 - {
1.2189 - g_RootMenu.ConsolePrint(" URL: %s", info->url);
1.2190 - }
1.2191 - if (pl->GetStatus() == Plugin_Error)
1.2192 - {
1.2193 - g_RootMenu.ConsolePrint(" Error: %s", pl->m_errormsg);
1.2194 - }
1.2195 - else
1.2196 - {
1.2197 - if (pl->GetStatus() == Plugin_Running)
1.2198 - {
1.2199 - if (pl->IsDebugging())
1.2200 - {
1.2201 - g_RootMenu.ConsolePrint(" Status: running, debugging");
1.2202 - }
1.2203 - else
1.2204 - {
1.2205 - g_RootMenu.ConsolePrint(" Status: running");
1.2206 - }
1.2207 - }
1.2208 - else
1.2209 - {
1.2210 - g_RootMenu.ConsolePrint(" Status: not running");
1.2211 - }
1.2212 -
1.2213 - const char *typestr = "";
1.2214 - switch (pl->GetType())
1.2215 - {
1.2216 - case PluginType_MapUpdated:
1.2217 - typestr = "Map Change if Updated";
1.2218 - break;
1.2219 - case PluginType_MapOnly:
1.2220 - typestr = "Map Change";
1.2221 - break;
1.2222 - case PluginType_Private:
1.2223 - case PluginType_Global:
1.2224 - typestr = "Never";
1.2225 - break;
1.2226 - }
1.2227 -
1.2228 - g_RootMenu.ConsolePrint(" Reloads: %s", typestr);
1.2229 - }
1.2230 - if (pl->m_FileVersion >= 3)
1.2231 - {
1.2232 - g_RootMenu.ConsolePrint(" Timestamp: %s", pl->m_DateTime);
1.2233 - }
1.2234 - }
1.2235 - else
1.2236 - {
1.2237 - g_RootMenu.ConsolePrint(" Load error: %s", pl->m_errormsg);
1.2238 - g_RootMenu.ConsolePrint(" File info: (title \"%s\") (version \"%s\")",
1.2239 - info->name ? info->name : "<none>",
1.2240 - info->version ? info->version : "<none>");
1.2241 - if (IS_STR_FILLED(info->url))
1.2242 - {
1.2243 - g_RootMenu.ConsolePrint(" File URL: %s", info->url);
1.2244 - }
1.2245 - }
1.2246 -
1.2247 - return;
1.2248 - }
1.2249 - else if (strcmp(cmd, "debug") == 0)
1.2250 - {
1.2251 - if (argcount < 5)
1.2252 - {
1.2253 - g_RootMenu.ConsolePrint("[SM] Usage: sm plugins debug <#> [on|off]");
1.2254 - return;
1.2255 - }
1.2256 -
1.2257 - CPlugin *pl;
1.2258 - char *end;
1.2259 - const char *arg = command.Arg(3);
1.2260 - int id = strtol(arg, &end, 10);
1.2261 -
1.2262 - if (*end == '\0')
1.2263 - {
1.2264 - pl = GetPluginByOrder(id);
1.2265 - if (!pl)
1.2266 - {
1.2267 - g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
1.2268 - return;
1.2269 - }
1.2270 - }
1.2271 - else
1.2272 - {
1.2273 - char pluginfile[256];
1.2274 - const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
1.2275 - UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
1.2276 -
1.2277 - if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
1.2278 - {
1.2279 - g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
1.2280 - return;
1.2281 - }
1.2282 - }
1.2283 -
1.2284 - int res;
1.2285 - const char *mode = command.Arg(4);
1.2286 - if ((res=strcmp("on", mode)) && strcmp("off", mode))
1.2287 - {
1.2288 - g_RootMenu.ConsolePrint("[SM] The only possible options are \"on\" and \"off.\"");
1.2289 - return;
1.2290 - }
1.2291 -
1.2292 - bool debug;
1.2293 - if (!res)
1.2294 - {
1.2295 - debug = true;
1.2296 - }
1.2297 - else
1.2298 - {
1.2299 - debug = false;
1.2300 - }
1.2301 -
1.2302 - if (debug && pl->IsDebugging())
1.2303 - {
1.2304 - g_RootMenu.ConsolePrint("[SM] This plugin is already in debug mode.");
1.2305 - return;
1.2306 - }
1.2307 - else if (!debug && !pl->IsDebugging())
1.2308 - {
1.2309 - g_RootMenu.ConsolePrint("[SM] Debug mode is already disabled in this plugin.");
1.2310 - return;
1.2311 - }
1.2312 -
1.2313 - char error[256];
1.2314 - if (pl->ToggleDebugMode(debug, error, sizeof(error)))
1.2315 - {
1.2316 - g_RootMenu.ConsolePrint("[SM] Successfully toggled debug mode on plugin %s.", pl->GetFilename());
1.2317 - return;
1.2318 - }
1.2319 - else
1.2320 - {
1.2321 - g_RootMenu.ConsolePrint("[SM] Could not toggle debug mode on plugin %s.", pl->GetFilename());
1.2322 - g_RootMenu.ConsolePrint("[SM] Plugin returned error: %s", error);
1.2323 - return;
1.2324 - }
1.2325 - }
1.2326 - else if (strcmp(cmd, "refresh") == 0)
1.2327 - {
1.2328 - g_SourceMod.DoGlobalPluginLoads();
1.2329 - g_RootMenu.ConsolePrint("[SM] The plugin list has been refreshed and reloaded.");
1.2330 - return;
1.2331 - }
1.2332 - else if (strcmp(cmd, "reload") == 0)
1.2333 - {
1.2334 - if (argcount < 4)
1.2335 - {
1.2336 - g_RootMenu.ConsolePrint("[SM] Usage: sm plugins reload <#|file>");
1.2337 - return;
1.2338 - }
1.2339 -
1.2340 - CPlugin *pl;
1.2341 - char *end;
1.2342 - const char *arg = command.Arg(3);
1.2343 - int id = strtol(arg, &end, 10);
1.2344 -
1.2345 - if (*end == '\0')
1.2346 - {
1.2347 - pl = GetPluginByOrder(id);
1.2348 - if (!pl)
1.2349 - {
1.2350 - g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
1.2351 - return;
1.2352 - }
1.2353 - }
1.2354 - else
1.2355 - {
1.2356 - char pluginfile[256];
1.2357 - const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
1.2358 - UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
1.2359 -
1.2360 - if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
1.2361 - {
1.2362 - g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
1.2363 - return;
1.2364 - }
1.2365 - }
1.2366 -
1.2367 - char name[PLATFORM_MAX_PATH];
1.2368 - const sm_plugininfo_t *info = pl->GetPublicInfo();
1.2369 - strcpy(name, (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
1.2370 -
1.2371 - if (ReloadPlugin(pl))
1.2372 - {
1.2373 - g_RootMenu.ConsolePrint("[SM] Plugin %s reloaded successfully.", name);
1.2374 - }
1.2375 - else
1.2376 - {
1.2377 - g_RootMenu.ConsolePrint("[SM] Failed to reload plugin %s.", name);
1.2378 - }
1.2379 -
1.2380 - return;
1.2381 - }
1.2382 - }
1.2383 -
1.2384 - /* Draw the main menu */
1.2385 - g_RootMenu.ConsolePrint("SourceMod Plugins Menu:");
1.2386 - g_RootMenu.DrawGenericOption("debug", "Toggle debug mode on a plugin");
1.2387 - g_RootMenu.DrawGenericOption("info", "Information about a plugin");
1.2388 - g_RootMenu.DrawGenericOption("list", "Show loaded plugins");
1.2389 - g_RootMenu.DrawGenericOption("load", "Load a plugin");
1.2390 - g_RootMenu.DrawGenericOption("load_lock", "Prevents any more plugins from being loaded");
1.2391 - g_RootMenu.DrawGenericOption("load_unlock", "Re-enables plugin loading");
1.2392 - g_RootMenu.DrawGenericOption("refresh", "Reloads/refreshes all plugins in the plugins folder");
1.2393 - g_RootMenu.DrawGenericOption("reload", "Reloads a plugin");
1.2394 - g_RootMenu.DrawGenericOption("unload", "Unload a plugin");
1.2395 - g_RootMenu.DrawGenericOption("unload_all", "Unloads all plugins");
1.2396 -}
1.2397 -
1.2398 -bool CPluginManager::ReloadPlugin(CPlugin *pl)
1.2399 -{
1.2400 - List<CPlugin *>::iterator iter;
1.2401 - char filename[PLATFORM_MAX_PATH];
1.2402 - bool debug, wasloaded;
1.2403 - PluginType ptype;
1.2404 - IPlugin *newpl;
1.2405 - int id = 1;
1.2406 -
1.2407 - strcpy(filename, pl->m_filename);
1.2408 - debug = pl->IsDebugging();
1.2409 - ptype = pl->GetType();
1.2410 -
1.2411 - for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++, id++)
1.2412 - {
1.2413 - if ((*iter) == pl)
1.2414 - {
1.2415 - break;
1.2416 - }
1.2417 - }
1.2418 -
1.2419 - if (!UnloadPlugin(pl))
1.2420 - {
1.2421 - return false;
1.2422 - }
1.2423 - if (!(newpl=LoadPlugin(filename, debug, ptype, NULL, 0, &wasloaded)))
1.2424 - {
1.2425 - return false;
1.2426 - }
1.2427 -
1.2428 - for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
1.2429 - {
1.2430 - if ((*iter) == (CPlugin *)newpl)
1.2431 - {
1.2432 - m_plugins.erase(iter);
1.2433 - break;
1.2434 - }
1.2435 - }
1.2436 -
1.2437 - int i;
1.2438 - for (i=1, iter=m_plugins.begin(); iter!=m_plugins.end() && i<id; iter++, i++)
1.2439 - {
1.2440 - }
1.2441 - m_plugins.insert(iter, (CPlugin *)newpl);
1.2442 -
1.2443 - return true;
1.2444 -}
1.2445 -
1.2446 -void CPluginManager::ReloadOrUnloadPlugins()
1.2447 -{
1.2448 - /* If we're in a load lock, just skip this whole bit. */
1.2449 - if (m_LoadingLocked)
1.2450 - {
1.2451 - return;
1.2452 - }
1.2453 -
1.2454 - List<CPlugin *>::iterator iter;
1.2455 - List<CPlugin *> tmp_list = m_plugins;
1.2456 - CPlugin *pl;
1.2457 - time_t t;
1.2458 -
1.2459 - for (iter=tmp_list.begin(); iter!=tmp_list.end(); iter++)
1.2460 - {
1.2461 - pl = (*iter);
1.2462 - if (pl->GetType() == PluginType_MapOnly)
1.2463 - {
1.2464 - UnloadPlugin((IPlugin *)pl);
1.2465 - }
1.2466 - else if (pl->GetType() == PluginType_MapUpdated)
1.2467 - {
1.2468 - t = pl->GetFileTimeStamp();
1.2469 - if (!t || t > pl->GetTimeStamp())
1.2470 - {
1.2471 - pl->SetTimeStamp(t);
1.2472 - UnloadPlugin((IPlugin *)pl);
1.2473 - }
1.2474 - }
1.2475 - }
1.2476 -}
1.2477 -
1.2478 -void CPluginManager::_SetPauseState(CPlugin *pl, bool paused)
1.2479 -{
1.2480 - List<IPluginsListener *>::iterator iter;
1.2481 - IPluginsListener *pListener;
1.2482 - for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
1.2483 - {
1.2484 - pListener = (*iter);
1.2485 - pListener->OnPluginPauseChange(pl, paused);
1.2486 - }
1.2487 -}
1.2488 -
1.2489 -void CPluginManager::AddFunctionsToForward(const char *name, IChangeableForward *pForward)
1.2490 -{
1.2491 - List<CPlugin *>::iterator iter;
1.2492 - CPlugin *pPlugin;
1.2493 - IPluginFunction *pFunction;
1.2494 -
1.2495 - for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++)
1.2496 - {
1.2497 - pPlugin = (*iter);
1.2498 -
1.2499 - if (pPlugin->GetStatus() <= Plugin_Paused)
1.2500 - {
1.2501 - pFunction = pPlugin->GetBaseContext()->GetFunctionByName(name);
1.2502 -
1.2503 - if (pFunction)
1.2504 - {
1.2505 - pForward->AddFunction(pFunction);
1.2506 - }
1.2507 - }
1.2508 - }
1.2509 -}
1.2510 -
1.2511 -CPlugin *CPluginManager::GetPluginFromIdentity(IdentityToken_t *pToken)
1.2512 -{
1.2513 - if (pToken->type != g_PluginIdent)
1.2514 - {
1.2515 - return NULL;
1.2516 - }
1.2517 -
1.2518 - return (CPlugin *)(pToken->ptr);
1.2519 -}
1.2520 -
1.2521 -void CPluginManager::OnLibraryAction(const char *lib, bool is_a_plugin, bool drop)
1.2522 -{
1.2523 - List<CPlugin *>::iterator iter;
1.2524 -
1.2525 - struct _pl
1.2526 - {
1.2527 - cell_t name;
1.2528 - cell_t file;
1.2529 - cell_t required;
1.2530 - } *plc;
1.2531 -
1.2532 - struct _ext
1.2533 - {
1.2534 - cell_t name;
1.2535 - cell_t file;
1.2536 - cell_t autoload;
1.2537 - cell_t required;
1.2538 - } *ext;
1.2539 -
1.2540 - const char *name = drop ? "OnLibraryRemoved" : "OnLibraryAdded";
1.2541 -
1.2542 - for (iter=m_plugins.begin();
1.2543 - iter!=m_plugins.end();
1.2544 - iter++)
1.2545 - {
1.2546 - CPlugin *pl = (*iter);
1.2547 - if (pl->GetStatus() != Plugin_Running)
1.2548 - {
1.2549 - continue;
1.2550 - }
1.2551 - IPluginContext *pContext = pl->GetBaseContext();
1.2552 - IPluginFunction *pf = pContext->GetFunctionByName(name);
1.2553 - if (!pf)
1.2554 - {
1.2555 - continue;
1.2556 - }
1.2557 - uint32_t num_vars = pContext->GetPubVarsNum();
1.2558 - for (uint32_t i=0; i<num_vars; i++)
1.2559 - {
1.2560 - sp_pubvar_t *pubvar;
1.2561 - if (pContext->GetPubvarByIndex(i, &pubvar) != SP_ERROR_NONE)
1.2562 - {
1.2563 - continue;
1.2564 - }
1.2565 - if (is_a_plugin && strncmp(pubvar->name, "__pl_", 5) == 0)
1.2566 - {
1.2567 - plc = (_pl *)pubvar->offs;
1.2568 - if (plc->required)
1.2569 - {
1.2570 - continue;
1.2571 - }
1.2572 - char *str;
1.2573 - pContext->LocalToString(plc->name, &str);
1.2574 - if (strcmp(str, lib) != 0)
1.2575 - {
1.2576 - continue;
1.2577 - }
1.2578 - pf->PushString(lib);
1.2579 - pf->Execute(NULL);
1.2580 - }
1.2581 - else if (!is_a_plugin && strncmp(pubvar->name, "__ext_", 6) == 0)
1.2582 - {
1.2583 - ext = (_ext *)pubvar->offs;
1.2584 - if (ext->required)
1.2585 - {
1.2586 - continue;
1.2587 - }
1.2588 - char *str;
1.2589 - pContext->LocalToString(ext->name, &str);
1.2590 - if (strcmp(str, lib) != 0)
1.2591 - {
1.2592 - continue;
1.2593 - }
1.2594 - pf->PushString(lib);
1.2595 - pf->Execute(NULL);
1.2596 - }
1.2597 - }
1.2598 - }
1.2599 -}
1.2600 -
1.2601 -bool CPluginManager::LibraryExists(const char *lib)
1.2602 -{
1.2603 - List<CPlugin *>::iterator iter;
1.2604 -
1.2605 - for (iter=m_plugins.begin();
1.2606 - iter!=m_plugins.end();
1.2607 - iter++)
1.2608 - {
1.2609 - CPlugin *pl = (*iter);
1.2610 - if (pl->GetStatus() != Plugin_Running)
1.2611 - {
1.2612 - continue;
1.2613 - }
1.2614 - List<String>::iterator s_iter;
1.2615 - for (s_iter = pl->m_Libraries.begin();
1.2616 - s_iter != pl->m_Libraries.end();
1.2617 - s_iter++)
1.2618 - {
1.2619 - if ((*s_iter).compare(lib) == 0)
1.2620 - {
1.2621 - return true;
1.2622 - }
1.2623 - }
1.2624 - }
1.2625 -
1.2626 - return false;
1.2627 -}
1.2628 -
1.2629 -void CPluginManager::AllPluginsLoaded()
1.2630 -{
1.2631 - List<CPlugin *>::iterator iter;
1.2632 - CPlugin *pl;
1.2633 -
1.2634 - for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
1.2635 - {
1.2636 - pl = (*iter);
1.2637 - pl->Call_OnAllPluginsLoaded();
1.2638 - }
1.2639 -}
1.2640 -
1.2641 -void CPluginManager::UnloadAll()
1.2642 -{
1.2643 - List<CPlugin *>::iterator iter;
1.2644 - while ( (iter = m_plugins.begin()) != m_plugins.end() )
1.2645 - {
1.2646 - UnloadPlugin((*iter));
1.2647 - }
1.2648 -}
1.2649 -
1.2650 -int CPluginManager::GetOrderOfPlugin(IPlugin *pl)
1.2651 -{
1.2652 - int id = 1;
1.2653 - List<CPlugin *>::iterator iter;
1.2654 -
1.2655 - for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++, id++)
1.2656 - {
1.2657 - if ((*iter) == pl)
1.2658 - {
1.2659 - return id;
1.2660 - }
1.2661 - }
1.2662 -
1.2663 - return -1;
1.2664 -}
1.2665 -
1.2666 -CPlugin *CPluginManager::FindPluginByConsoleArg(const char *arg)
1.2667 -{
1.2668 - int id;
1.2669 - char *end;
1.2670 - CPlugin *pl;
1.2671 -
1.2672 - id = strtol(arg, &end, 10);
1.2673 -
1.2674 - if (*end == '\0')
1.2675 - {
1.2676 - pl = GetPluginByOrder(id);
1.2677 - if (!pl)
1.2678 - {
1.2679 - return NULL;
1.2680 - }
1.2681 - }
1.2682 - else
1.2683 - {
1.2684 - char pluginfile[256];
1.2685 - const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
1.2686 - UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
1.2687 -
1.2688 - if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
1.2689 - {
1.2690 - return NULL;
1.2691 - }
1.2692 - }
1.2693 -
1.2694 - return pl;
1.2695 -}
