sourcemod-central: core/PluginSys.cpp@07a1b8a8a2ba (annotated)

dvander@370	1	/**
ds@653	2	* vim: set ts=4 :
ds@1330	3	* =============================================================================
dvander@1231	4	* SourceMod
ds@1947	5	* Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved.
ds@1330	6	* =============================================================================
dvander@370	7	*
ds@1330	8	* This program is free software; you can redistribute it and/or modify it under
ds@1330	9	* the terms of the GNU General Public License, version 3.0, as published by the
ds@1330	10	* Free Software Foundation.
dvander@1231	11	*
ds@1330	12	* This program is distributed in the hope that it will be useful, but WITHOUT
ds@1330	13	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
ds@1330	14	* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
ds@1330	15	* details.
dvander@1231	16	*
ds@1330	17	* You should have received a copy of the GNU General Public License along with
ds@1330	18	* this program. If not, see <http://www.gnu.org/licenses/>.
dvander@1231	19	*
ds@1330	20	* As a special exception, AlliedModders LLC gives you permission to link the
ds@1330	21	* code of this program (as well as its derivative works) to "Half-Life 2," the
ds@1330	22	* "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
ds@1330	23	* by the Valve Corporation. You must obey the GNU General Public License in
ds@1330	24	* all respects for all other code used. Additionally, AlliedModders LLC grants
ds@1330	25	* this exception to all derivative works. AlliedModders LLC defines further
ds@1330	26	* exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
ds@1330	27	* or <http://www.sourcemod.net/license.php>.
dvander@370	28	*
dvander@370	29	* Version: $Id$
dvander@370	30	*/
dvander@370	31
dvander@173	32	#include <stdio.h>
dvander@173	33	#include "PluginSys.h"
dvander@229	34	#include "ShareSys.h"
dvander@190	35	#include "LibrarySys.h"
dvander@220	36	#include "HandleSys.h"
ds@611	37	#include "ForwardSys.h"
dvander@197	38	#include "sourcemm_api.h"
dvander@209	39	#include "sourcemod.h"
dvander@592	40	#include "Logger.h"
dvander@298	41	#include "ExtensionSys.h"
dvander@309	42	#include "sm_srvcmds.h"
dvander@309	43	#include "sm_stringutil.h"
dvander@848	44	#include "ConCmdManager.h"
dvander@848	45	#include "PlayerManager.h"
dvander@978	46	#include "CoreConfig.h"
dvander@3004	47	#include "logic_bridge.h"
dvander@173	48
dvander@211	49	CPluginManager g_PluginSys;
dvander@220	50	HandleType_t g_PluginType = 0;
dvander@229	51	IdentityType_t g_PluginIdent = 0;
dvander@173	52
dvander@209	53	CPlugin::CPlugin(const char *file)
dvander@209	54	{
dvander@209	55	static int MySerial = 0;
dvander@209	56
dvander@209	57	m_type = PluginType_Private;
dvander@209	58	m_status = Plugin_Uncompiled;
fyrenmoo@2675	59	m_bSilentlyFailed = false;
dvander@209	60	m_serial = ++MySerial;
dvander@2193	61	m_pRuntime = NULL;
dvander@2798	62	m_errormsg[sizeof(m_errormsg) - 1] = '\0';
dvander@2814	63	UTIL_Format(m_filename, sizeof(m_filename), "%s", file);
dvander@229	64	m_handle = 0;
dvander@229	65	m_ident = NULL;
dvander@486	66	m_pProps = sm_trie_create();
dvander@605	67	m_FakeNativesMissing = false;
faluco@1112	68	m_LibraryMissing = false;
dvander@1630	69	m_bGotAllLoaded = false;
dvander@3004	70	m_pPhrases = translator->CreatePhraseCollection();
dvander@2342	71	m_MaxClientsVar = NULL;
dvander@209	72	}
dvander@209	73
dvander@209	74	CPlugin::~CPlugin()
dvander@173	75	{
dvander@298	76	if (m_handle)
dvander@298	77	{
dvander@298	78	HandleSecurity sec;
dvander@298	79	sec.pOwner = g_PluginSys.GetIdentity();
dvander@298	80	sec.pIdentity = sec.pOwner;
dvander@298	81
dvander@298	82	g_HandleSys.FreeHandle(m_handle, &sec);
dvander@298	83	g_ShareSys.DestroyIdentity(m_ident);
dvander@298	84	}
dvander@298	85
dvander@2193	86	if (m_pRuntime != NULL)
dvander@209	87	{
dvander@2193	88	delete m_pRuntime;
dvander@2193	89	m_pRuntime = NULL;
dvander@209	90	}
dvander@209	91
faluco@641	92	if (m_pProps)
dvander@486	93	{
dvander@486	94	sm_trie_destroy(m_pProps);
dvander@486	95	}
dvander@978	96	for (size_t i=0; i<m_configs.size(); i++)
dvander@978	97	{
dvander@978	98	delete m_configs[i];
dvander@978	99	}
dvander@978	100	m_configs.clear();
dvander@2042	101	if (m_pPhrases != NULL)
dvander@2042	102	{
dvander@2042	103	m_pPhrases->Destroy();
dvander@2042	104	m_pPhrases = NULL;
dvander@2042	105	}
dvander@229	106	}
dvander@229	107
dvander@229	108	void CPlugin::InitIdentity()
dvander@229	109	{
dvander@229	110	if (!m_handle)
dvander@229	111	{
dvander@868	112	m_ident = g_ShareSys.CreateIdentity(g_PluginIdent, this);
dvander@262	113	m_handle = g_HandleSys.CreateHandle(g_PluginType, this, g_PluginSys.GetIdentity(), g_PluginSys.GetIdentity(), NULL);
dvander@2193	114	m_pRuntime->GetDefaultContext()->SetKey(1, m_ident);
dvander@2193	115	m_pRuntime->GetDefaultContext()->SetKey(2, (IPlugin *)this);
dvander@229	116	}
dvander@209	117	}
dvander@209	118
dvander@1766	119	unsigned int CPlugin::CalcMemUsage()
dvander@1766	120	{
dvander@1766	121	unsigned int base_size =
dvander@1766	122	sizeof(CPlugin)
dvander@1766	123	+ sizeof(IdentityToken_t)
dvander@1766	124	+ (m_configs.size() * (sizeof(AutoConfig *) + sizeof(AutoConfig)))
dvander@1766	125	+ sm_trie_mem_usage(m_pProps);
dvander@1766	126
dvander@1766	127	for (unsigned int i = 0; i < m_configs.size(); i++)
dvander@1766	128	{
dvander@1766	129	base_size += m_configs[i]->autocfg.size();
dvander@1766	130	base_size += m_configs[i]->folder.size();
dvander@1766	131	}
dvander@1766	132
dvander@1766	133	for (List<String>::iterator i = m_Libraries.begin();
dvander@1766	134	i != m_Libraries.end();
dvander@1766	135	i++)
dvander@1766	136	{
dvander@1766	137	base_size += (*i).size();
dvander@1766	138	}
dvander@1766	139
dvander@1766	140	for (List<String>::iterator i = m_RequiredLibs.begin();
dvander@1766	141	i != m_RequiredLibs.end();
dvander@1766	142	i++)
dvander@1766	143	{
dvander@1766	144	base_size += (*i).size();
dvander@1766	145	}
dvander@1766	146
dvander@1766	147	return base_size;
dvander@1766	148	}
dvander@1766	149
dvander@536	150	Handle_t CPlugin::GetMyHandle()
dvander@536	151	{
dvander@536	152	return m_handle;
dvander@536	153	}
dvander@536	154
dvander@209	155	CPlugin CPlugin::CreatePlugin(const char file, char *error, size_t maxlength)
dvander@209	156	{
ds@652	157	char fullpath[PLATFORM_MAX_PATH];
dvander@317	158	g_SourceMod.BuildPath(Path_SM, fullpath, sizeof(fullpath), "plugins/%s", file);
dvander@209	159	FILE *fp = fopen(fullpath, "rb");
dvander@173	160
dvander@298	161	CPlugin *pPlugin = new CPlugin(file);
dvander@298	162
dvander@173	163	if (!fp)
dvander@173	164	{
dvander@209	165	if (error)
dvander@209	166	{
dvander@2814	167	UTIL_Format(error, maxlength, "Unable to open file");
dvander@209	168	}
dvander@298	169	pPlugin->m_status = Plugin_BadLoad;
dvander@298	170	return pPlugin;
dvander@173	171	}
dvander@173	172
dvander@173	173	fclose(fp);
dvander@173	174
dvander@209	175	return pPlugin;
dvander@209	176	}
dvander@209	177
dvander@486	178	bool CPlugin::GetProperty(const char prop, void ptr, bool remove/ =false */)
dvander@486	179	{
dvander@486	180	bool exists = sm_trie_retrieve(m_pProps, prop, ptr);
dvander@486	181
dvander@486	182	if (exists && remove)
dvander@486	183	{
dvander@486	184	sm_trie_delete(m_pProps, prop);
dvander@486	185	}
dvander@486	186
dvander@486	187	return exists;
dvander@486	188	}
dvander@486	189
dvander@486	190	bool CPlugin::SetProperty(const char prop, void ptr)
dvander@486	191	{
dvander@486	192	return sm_trie_insert(m_pProps, prop, ptr);
dvander@486	193	}
dvander@486	194
dvander@2193	195	IPluginRuntime *CPlugin::GetRuntime()
dvander@209	196	{
dvander@2193	197	return m_pRuntime;
dvander@209	198	}
dvander@209	199
dvander@209	200	void CPlugin::SetErrorState(PluginStatus status, const char *error_fmt, ...)
dvander@209	201	{
dvander@605	202	PluginStatus old_status = m_status;
dvander@209	203	m_status = status;
dvander@209	204
dvander@605	205	if (old_status == Plugin_Running)
dvander@605	206	{
dvander@605	207	/* Tell everyone we're now paused */
dvander@605	208	g_PluginSys._SetPauseState(this, true);
dvander@605	209	}
dvander@605	210
dvander@209	211	va_list ap;
dvander@209	212	va_start(ap, error_fmt);
dvander@2814	213	UTIL_FormatArgs(m_errormsg, sizeof(m_errormsg), error_fmt, ap);
dvander@209	214	va_end(ap);
dvander@173	215
dvander@2193	216	if (m_pRuntime != NULL)
dvander@176	217	{
dvander@2193	218	m_pRuntime->SetPauseState(true);
dvander@176	219	}
dvander@176	220	}
dvander@176	221
dvander@2342	222	bool CPlugin::UpdateInfo()
dvander@173	223	{
dvander@173	224	/* Now grab the info */
dvander@173	225	uint32_t idx;
dvander@173	226	IPluginContext *base = GetBaseContext();
dvander@173	227	int err = base->FindPubvarByName("myinfo", &idx);
dvander@173	228
dvander@413	229	memset(&m_info, 0, sizeof(m_info));
dvander@413	230
dvander@173	231	if (err == SP_ERROR_NONE)
dvander@173	232	{
dvander@173	233	struct sm_plugininfo_c_t
dvander@173	234	{
dvander@173	235	cell_t name;
dvander@173	236	cell_t description;
dvander@173	237	cell_t author;
dvander@173	238	cell_t version;
dvander@173	239	cell_t url;
dvander@173	240	};
dvander@173	241	sm_plugininfo_c_t *cinfo;
dvander@173	242	cell_t local_addr;
dvander@173	243
dvander@173	244	base->GetPubvarAddrs(idx, &local_addr, (cell_t **)&cinfo);
dvander@173	245	base->LocalToString(cinfo->name, (char **)&m_info.name);
dvander@173	246	base->LocalToString(cinfo->description, (char **)&m_info.description);
dvander@173	247	base->LocalToString(cinfo->author, (char **)&m_info.author);
dvander@173	248	base->LocalToString(cinfo->url, (char **)&m_info.url);
dvander@173	249	base->LocalToString(cinfo->version, (char **)&m_info.version);
dvander@173	250	}
dvander@173	251
dvander@173	252	m_info.author = m_info.author ? m_info.author : "";
dvander@173	253	m_info.description = m_info.description ? m_info.description : "";
dvander@173	254	m_info.name = m_info.name ? m_info.name : "";
dvander@173	255	m_info.url = m_info.url ? m_info.url : "";
dvander@173	256	m_info.version = m_info.version ? m_info.version : "";
dvander@2105	257
dvander@2105	258	if ((err = base->FindPubvarByName("__version", &idx)) == SP_ERROR_NONE)
dvander@2105	259	{
dvander@2105	260	struct __version_info
dvander@2105	261	{
dvander@2105	262	cell_t version;
dvander@2105	263	cell_t filevers;
dvander@2105	264	cell_t date;
dvander@2105	265	cell_t time;
dvander@2105	266	};
dvander@2105	267	__version_info *info;
dvander@2105	268	cell_t local_addr;
dvander@2342	269	const char pDate, pTime, *pFileVers;
dvander@2105	270
dvander@2105	271	pDate = "";
dvander@2105	272	pTime = "";
dvander@2105	273
dvander@2105	274	base->GetPubvarAddrs(idx, &local_addr, (cell_t **)&info);
dvander@2105	275	m_FileVersion = info->version;
dvander@2636	276	if (m_FileVersion >= 4)
dvander@2105	277	{
dvander@2105	278	base->LocalToString(info->date, (char **)&pDate);
dvander@2105	279	base->LocalToString(info->time, (char **)&pTime);
dvander@2105	280	UTIL_Format(m_DateTime, sizeof(m_DateTime), "%s %s", pDate, pTime);
dvander@2105	281	}
dvander@2636	282	if (m_FileVersion > 5)
dvander@2342	283	{
dvander@2342	284	base->LocalToString(info->filevers, (char **)&pFileVers);
dvander@2342	285	SetErrorState(Plugin_Failed, "Newer SourceMod required (%s or higher)", pFileVers);
dvander@2342	286	return false;
dvander@2342	287	}
dvander@2105	288	}
dvander@2105	289	else
dvander@2105	290	{
dvander@2105	291	m_FileVersion = 0;
dvander@2105	292	}
dvander@2342	293
dvander@2342	294	if ((err = base->FindPubvarByName("MaxClients", &idx)) == SP_ERROR_NONE)
dvander@2342	295	{
dvander@2342	296	base->GetPubvarByIndex(idx, &m_MaxClientsVar);
dvander@2342	297	}
dvander@2342	298
dvander@2342	299	return true;
dvander@2342	300	}
dvander@2342	301
dvander@2342	302	void CPlugin::SyncMaxClients(int max_clients)
dvander@2342	303	{
dvander@2342	304	if (m_MaxClientsVar == NULL)
dvander@2342	305	{
dvander@2342	306	return;
dvander@2342	307	}
dvander@2342	308
dvander@2342	309	*m_MaxClientsVar->offs = max_clients;
dvander@173	310	}
dvander@173	311
ds@426	312	void CPlugin::Call_OnPluginStart()
dvander@211	313	{
dvander@211	314	if (m_status != Plugin_Loaded)
dvander@211	315	{
dvander@211	316	return;
dvander@211	317	}
dvander@211	318
dvander@211	319	m_status = Plugin_Running;
dvander@211	320
dvander@2342	321	SyncMaxClients(g_Players.MaxClients());
dvander@2342	322
dvander@211	323	cell_t result;
dvander@2193	324	IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("OnPluginStart");
dvander@211	325	if (!pFunction)
dvander@211	326	{
dvander@211	327	return;
dvander@211	328	}
dvander@211	329
dvander@838	330	int err;
dvander@838	331	if ((err=pFunction->Execute(&result)) != SP_ERROR_NONE)
dvander@838	332	{
dvander@838	333	SetErrorState(Plugin_Error, "Error detected in plugin startup (see error logs)");
dvander@838	334	}
dvander@282	335	}
dvander@282	336
ds@426	337	void CPlugin::Call_OnPluginEnd()
dvander@282	338	{
faluco@736	339	if (m_status > Plugin_Paused)
dvander@211	340	{
dvander@282	341	return;
dvander@211	342	}
dvander@282	343
dvander@282	344	cell_t result;
dvander@2193	345	IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("OnPluginEnd");
dvander@282	346	if (!pFunction)
dvander@282	347	{
dvander@282	348	return;
dvander@282	349	}
dvander@282	350
dvander@282	351	pFunction->Execute(&result);
dvander@211	352	}
dvander@211	353
faluco@1221	354	void CPlugin::Call_OnAllPluginsLoaded()
faluco@1221	355	{
faluco@1221	356	if (m_status > Plugin_Paused)
faluco@1221	357	{
faluco@1221	358	return;
faluco@1221	359	}
faluco@1221	360
dvander@1630	361	if (m_bGotAllLoaded)
dvander@1630	362	{
dvander@1630	363	return;
dvander@1630	364	}
dvander@1630	365
dvander@1630	366	m_bGotAllLoaded = true;
dvander@1630	367
faluco@1221	368	cell_t result;
dvander@2193	369	IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("OnAllPluginsLoaded");
dvander@1631	370	if (pFunction != NULL)
faluco@1221	371	{
dvander@1631	372	pFunction->Execute(&result);
faluco@1221	373	}
faluco@1221	374
skamonkey@3401	375	if (g_SourceMod.IsMapRunning())
dvander@1630	376	{
dvander@2193	377	if ((pFunction = m_pRuntime->GetFunctionByName("OnMapStart")) != NULL)
dvander@1630	378	{
dvander@1630	379	pFunction->Execute(NULL);
dvander@1630	380	}
dvander@1630	381	}
dvander@1630	382
dvander@1630	383	if (SM_AreConfigsExecuted())
dvander@1630	384	{
dvander@1630	385	SM_ExecuteForPlugin(GetBaseContext());
dvander@1630	386	}
faluco@1221	387	}
faluco@1221	388
fyrenmoo@2675	389	APLRes CPlugin::Call_AskPluginLoad(char *error, size_t maxlength)
dvander@211	390	{
dvander@211	391	if (m_status != Plugin_Created)
dvander@211	392	{
fyrenmoo@2675	393	return APLRes_Failure;
dvander@211	394	}
dvander@211	395
faluco@216	396	m_status = Plugin_Loaded;
faluco@216	397
dvander@211	398	int err;
dvander@211	399	cell_t result;
fyrenmoo@2675	400	bool haveNewAPL = false;
fyrenmoo@2675	401	IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("AskPluginLoad2");
dvander@211	402
fyrenmoo@2675	403	if (pFunction)
dvander@211	404	{
fyrenmoo@2675	405	haveNewAPL = true;
fyrenmoo@2675	406	}
fyrenmoo@2675	407	else if (!(pFunction = m_pRuntime->GetFunctionByName("AskPluginLoad")))
fyrenmoo@2675	408	{
fyrenmoo@2675	409	return APLRes_Success;
dvander@211	410	}
dvander@211	411
dvander@222	412	pFunction->PushCell(m_handle);
dvander@211	413	pFunction->PushCell(g_PluginSys.IsLateLoadTime() ? 1 : 0);
dvander@211	414	pFunction->PushStringEx(error, maxlength, 0, SM_PARAM_COPYBACK);
dvander@211	415	pFunction->PushCell(maxlength);
dvander@211	416	if ((err=pFunction->Execute(&result)) != SP_ERROR_NONE)
dvander@211	417	{
fyrenmoo@2675	418	return APLRes_Failure;
dvander@211	419	}
dvander@211	420
fyrenmoo@2675	421	if (haveNewAPL)
fyrenmoo@2675	422	{
fyrenmoo@2675	423	return (APLRes)result;
fyrenmoo@2675	424	}
fyrenmoo@2675	425	else if (result)
dvander@211	426	{
fyrenmoo@2675	427	return APLRes_Success;
dvander@211	428	}
fyrenmoo@2675	429	else
fyrenmoo@2675	430	{
fyrenmoo@2675	431	return APLRes_Failure;
fyrenmoo@2675	432	}
dvander@211	433	}
dvander@211	434
dvander@2193	435	void *CPlugin::GetPluginStructure()
dvander@173	436	{
dvander@2193	437	return NULL;
dvander@173	438	}
dvander@173	439
dvander@615	440	IPluginContext *CPlugin::GetBaseContext()
dvander@173	441	{
dvander@2193	442	if (m_pRuntime == NULL)
dvander@2193	443	{
dvander@2193	444	return NULL;
dvander@2193	445	}
dvander@2193	446
dvander@2193	447	return m_pRuntime->GetDefaultContext();
dvander@173	448	}
dvander@173	449
dvander@615	450	sp_context_t *CPlugin::GetContext()
dvander@173	451	{
dvander@2193	452	return NULL;
dvander@173	453	}
dvander@173	454
dvander@615	455	const char *CPlugin::GetFilename()
dvander@173	456	{
dvander@173	457	return m_filename;
dvander@173	458	}
dvander@173	459
dvander@615	460	PluginType CPlugin::GetType()
dvander@173	461	{
dvander@174	462	return m_type;
dvander@173	463	}
dvander@173	464
dvander@615	465	const sm_plugininfo_t *CPlugin::GetPublicInfo()
dvander@173	466	{
dvander@2342	467	if (GetStatus() >= Plugin_Created)
dvander@2342	468	{
dvander@2342	469	return NULL;
dvander@2342	470	}
dvander@173	471	return &m_info;
dvander@173	472	}
dvander@173	473
dvander@615	474	unsigned int CPlugin::GetSerial()
dvander@173	475	{
dvander@173	476	return m_serial;
dvander@173	477	}
dvander@173	478
dvander@615	479	PluginStatus CPlugin::GetStatus()
dvander@173	480	{
dvander@173	481	return m_status;
dvander@173	482	}
dvander@173	483
fyrenmoo@2675	484	bool CPlugin::IsSilentlyFailed()
fyrenmoo@2675	485	{
fyrenmoo@2675	486	return m_bSilentlyFailed;
fyrenmoo@2675	487	}
fyrenmoo@2675	488
dvander@615	489	bool CPlugin::IsDebugging()
dvander@173	490	{
dvander@2193	491	if (m_pRuntime == NULL)
dvander@209	492	{
dvander@209	493	return false;
dvander@209	494	}
dvander@209	495
dvander@2226	496	return true;
dvander@173	497	}
dvander@173	498
fyrenmoo@2675	499	void CPlugin::SetSilentlyFailed(bool sf)
fyrenmoo@2675	500	{
fyrenmoo@2675	501	m_bSilentlyFailed = sf;
fyrenmoo@2675	502	}
fyrenmoo@2675	503
dvander@1124	504	void CPlugin::LibraryActions(bool dropping)
dvander@1124	505	{
dvander@1124	506	List<String>::iterator iter;
dvander@1124	507	for (iter = m_Libraries.begin();
dvander@1124	508	iter != m_Libraries.end();
dvander@1124	509	iter++)
dvander@1124	510	{
dvander@1572	511	g_PluginSys.OnLibraryAction((*iter).c_str(), true, dropping);
dvander@1124	512	}
dvander@1124	513	}
dvander@1124	514
dvander@173	515	bool CPlugin::SetPauseState(bool paused)
dvander@173	516	{
dvander@1124	517	if (paused && GetStatus() != Plugin_Running)
dvander@173	518	{
dvander@173	519	return false;
dvander@1124	520	} else if (!paused && GetStatus() != Plugin_Paused) {
dvander@173	521	return false;
dvander@173	522	}
dvander@282	523
dvander@1124	524	if (paused)
dvander@1124	525	{
dvander@1124	526	LibraryActions(true);
dvander@1124	527	}
dvander@1124	528
dvander@2193	529	IPluginFunction *pFunction = m_pRuntime->GetFunctionByName("OnPluginPauseChange");
dvander@282	530	if (pFunction)
dvander@282	531	{
dvander@282	532	cell_t result;
dvander@282	533	pFunction->PushCell(paused ? 1 : 0);
dvander@282	534	pFunction->Execute(&result);
dvander@282	535	}
dvander@173	536
dvander@525	537	if (paused)
dvander@525	538	{
dvander@525	539	m_status = Plugin_Paused;
dvander@2193	540	m_pRuntime->SetPauseState(true);
dvander@525	541	} else {
dvander@525	542	m_status = Plugin_Running;
dvander@2193	543	m_pRuntime->SetPauseState(false);
dvander@525	544	}
dvander@525	545
faluco@430	546	g_PluginSys._SetPauseState(this, paused);
faluco@430	547
dvander@1124	548	if (!paused)
dvander@1124	549	{
dvander@1124	550	LibraryActions(false);
dvander@1124	551	}
dvander@1124	552
dvander@173	553	return true;
dvander@173	554	}
dvander@174	555
dvander@615	556	IdentityToken_t *CPlugin::GetIdentity()
dvander@220	557	{
dvander@229	558	return m_ident;
dvander@220	559	}
dvander@220	560
dvander@615	561	bool CPlugin::IsRunnable()
faluco@290	562	{
faluco@290	563	return (m_status <= Plugin_Paused) ? true : false;
faluco@290	564	}
faluco@290	565
faluco@333	566	time_t CPlugin::GetFileTimeStamp()
faluco@333	567	{
ds@652	568	char path[PLATFORM_MAX_PATH];
faluco@333	569	g_SourceMod.BuildPath(Path_SM, path, sizeof(path), "plugins/%s", m_filename);
faluco@333	570	#ifdef PLATFORM_WINDOWS
faluco@333	571	struct _stat s;
faluco@333	572	if (_stat(path, &s) != 0)
dvander@351	573	#elif defined PLATFORM_POSIX
faluco@333	574	struct stat s;
faluco@333	575	if (stat(path, &s) != 0)
faluco@333	576	#endif
faluco@333	577	{
faluco@333	578	return 0;
dvander@1630	579	}
dvander@1630	580	else
dvander@1630	581	{
faluco@333	582	return s.st_mtime;
faluco@333	583	}
faluco@333	584	}
faluco@333	585
dvander@615	586	time_t CPlugin::GetTimeStamp()
faluco@333	587	{
faluco@333	588	return m_LastAccess;
faluco@333	589	}
faluco@333	590
faluco@333	591	void CPlugin::SetTimeStamp(time_t t)
faluco@333	592	{
faluco@333	593	m_LastAccess = t;
faluco@333	594	}
faluco@333	595
dvander@2042	596	IPhraseCollection *CPlugin::GetPhrases()
faluco@389	597	{
dvander@2042	598	return m_pPhrases;
faluco@389	599	}
faluco@389	600
dvander@605	601	void CPlugin::DependencyDropped(CPlugin *pOwner)
dvander@605	602	{
dvander@2193	603	if (!m_pRuntime)
dvander@605	604	{
dvander@605	605	return;
dvander@605	606	}
dvander@605	607
faluco@1112	608	List<String>::iterator reqlib_iter;
faluco@1112	609	List<String>::iterator lib_iter;
faluco@1112	610	for (lib_iter=pOwner->m_Libraries.begin(); lib_iter!=pOwner->m_Libraries.end(); lib_iter++)
faluco@1112	611	{
faluco@1112	612	for (reqlib_iter=m_RequiredLibs.begin(); reqlib_iter!=m_RequiredLibs.end(); reqlib_iter++)
faluco@1112	613	{
faluco@1112	614	if ((reqlib_iter) == (lib_iter))
faluco@1112	615	{
faluco@1112	616	m_LibraryMissing = true;
faluco@1112	617	}
faluco@1112	618	}
faluco@1112	619	}
faluco@1112	620
dvander@2086	621	List<NativeEntry *>::iterator iter;
dvander@2086	622	NativeEntry *pNative;
dvander@605	623	sp_native_t *native;
dvander@605	624	uint32_t idx;
dvander@605	625	unsigned int unbound = 0;
dvander@605	626
dvander@2086	627	for (iter = pOwner->m_Natives.begin();
dvander@2086	628	iter != pOwner->m_Natives.end();
dvander@605	629	iter++)
dvander@605	630	{
dvander@605	631	pNative = (*iter);
dvander@605	632	/* Find this native! */
dvander@2193	633	if (m_pRuntime->FindNativeByName(pNative->name, &idx) != SP_ERROR_NONE)
dvander@605	634	{
dvander@605	635	continue;
dvander@605	636	}
dvander@605	637	/* Unbind it */
dvander@2193	638	m_pRuntime->GetNativeByIndex(idx, &native);
dvander@605	639	native->pfn = NULL;
dvander@605	640	native->status = SP_NATIVE_UNBOUND;
dvander@605	641	unbound++;
dvander@605	642	}
dvander@605	643
dvander@605	644	if (unbound)
dvander@605	645	{
dvander@605	646	m_FakeNativesMissing = true;
faluco@1112	647	}
faluco@1112	648
faluco@1112	649	/* :IDEA: in the future, add native trapping? */
faluco@1112	650	if (m_FakeNativesMissing \|\| m_LibraryMissing)
faluco@1112	651	{
dvander@605	652	SetErrorState(Plugin_Error, "Depends on plugin: %s", pOwner->GetFilename());
dvander@605	653	}
dvander@605	654	}
dvander@605	655
dvander@978	656	unsigned int CPlugin::GetConfigCount()
dvander@978	657	{
dvander@978	658	return (unsigned int)m_configs.size();
dvander@978	659	}
dvander@978	660
dvander@978	661	AutoConfig *CPlugin::GetConfig(unsigned int i)
dvander@978	662	{
dvander@978	663	if (i >= GetConfigCount())
dvander@978	664	{
dvander@978	665	return NULL;
dvander@978	666	}
dvander@978	667
dvander@978	668	return m_configs[i];
dvander@978	669	}
dvander@978	670
dvander@978	671	void CPlugin::AddConfig(bool autoCreate, const char cfg, const char folder)
dvander@978	672	{
dvander@1846	673	/* Do a check for duplicates to prevent double-execution */
dvander@1846	674	for (size_t i = 0; i < m_configs.size(); i++)
dvander@1846	675	{
dvander@1846	676	if (m_configs[i]->autocfg.compare(cfg) == 0
dvander@1846	677	&& m_configs[i]->folder.compare(folder) == 0
dvander@1846	678	&& m_configs[i]->create == autoCreate)
dvander@1846	679	{
dvander@1846	680	return;
dvander@1846	681	}
dvander@1846	682	}
dvander@1846	683
dvander@978	684	AutoConfig *c = new AutoConfig;
dvander@978	685
dvander@978	686	c->autocfg = cfg;
dvander@978	687	c->folder = folder;
dvander@978	688	c->create = autoCreate;
dvander@978	689
dvander@978	690	m_configs.push_back(c);
dvander@978	691	}
dvander@978	692
dvander@2086	693	void CPlugin::DropEverything()
dvander@2086	694	{
dvander@2086	695	CPlugin *pOther;
dvander@2086	696	List<CPlugin *>::iterator iter;
dvander@2086	697	List<WeakNative>::iterator wk_iter;
dvander@2086	698
dvander@2086	699	/* Tell everyone that depends on us that we're about to drop */
dvander@2086	700	for (iter = m_Dependents.begin();
dvander@2086	701	iter != m_Dependents.end();
dvander@2086	702	iter++)
dvander@2086	703	{
dvander@2086	704	pOther = (*iter);
dvander@2086	705	pOther->DependencyDropped(this);
dvander@2086	706	}
dvander@2086	707
dvander@2086	708	/* Note: we don't care about things we depend on.
dvander@2086	709	* The reason is that extensions have their own cleanup
dvander@2086	710	* code for plugins. Although the "right" design would be
dvander@2086	711	* to centralize that here, i'm omitting it for now. Thus,
dvander@2086	712	* the code below to walk the plugins list will suffice.
dvander@2086	713	*/
dvander@2086	714
dvander@2086	715	/* Other plugins could be holding weak references that were
dvander@2086	716	* added by us. We need to clean all of those up now.
dvander@2086	717	*/
dvander@2086	718	for (iter = g_PluginSys.m_plugins.begin();
dvander@2086	719	iter != g_PluginSys.m_plugins.end();
dvander@2086	720	iter++)
dvander@2086	721	{
dvander@2086	722	(*iter)->DropRefsTo(this);
dvander@2086	723	}
dvander@2086	724
dvander@2086	725	/* Proceed with the rest of the necessities. */
dvander@2086	726	CNativeOwner::DropEverything();
dvander@2086	727	}
dvander@2086	728
dvander@2086	729	bool CPlugin::AddFakeNative(IPluginFunction pFunc, const char name, SPVM_FAKENATIVE_FUNC func)
dvander@2086	730	{
dvander@2086	731	NativeEntry *pEntry;
dvander@2086	732
dvander@2086	733	if ((pEntry = g_ShareSys.AddFakeNative(pFunc, name, func)) == NULL)
dvander@2086	734	{
dvander@2086	735	return false;
dvander@2086	736	}
dvander@2086	737
dvander@2086	738	m_Natives.push_back(pEntry);
dvander@2086	739
dvander@2086	740	return true;
dvander@2086	741	}
dvander@2086	742
dvander@174	743	/*******************
dvander@174	744	* PLUGIN ITERATOR *
dvander@174	745	*******************/
dvander@190	746
dvander@213	747	CPluginManager::CPluginIterator::CPluginIterator(List<CPlugin > _mylist)
dvander@174	748	{
dvander@174	749	mylist = _mylist;
faluco@216	750	Reset();
dvander@174	751	}
dvander@174	752
dvander@174	753	IPlugin *CPluginManager::CPluginIterator::GetPlugin()
dvander@174	754	{
dvander@174	755	return (*current);
dvander@174	756	}
dvander@174	757
dvander@174	758	bool CPluginManager::CPluginIterator::MorePlugins()
dvander@174	759	{
dvander@174	760	return (current != mylist->end());
dvander@174	761	}
dvander@174	762
dvander@174	763	void CPluginManager::CPluginIterator::NextPlugin()
dvander@174	764	{
dvander@174	765	current++;
dvander@174	766	}
dvander@174	767
dvander@174	768	void CPluginManager::CPluginIterator::Release()
dvander@174	769	{
dvander@211	770	g_PluginSys.ReleaseIterator(this);
dvander@174	771	}
dvander@174	772
dvander@174	773	CPluginManager::CPluginIterator::~CPluginIterator()
dvander@174	774	{
dvander@174	775	}
dvander@174	776
dvander@174	777	void CPluginManager::CPluginIterator::Reset()
dvander@174	778	{
dvander@174	779	current = mylist->begin();
dvander@174	780	}
dvander@174	781
dvander@174	782	/******************
dvander@174	783	* PLUGIN MANAGER *
dvander@174	784	******************/
dvander@174	785
dvander@211	786	CPluginManager::CPluginManager()
dvander@211	787	{
dvander@211	788	m_LoadLookup = sm_trie_create();
dvander@211	789	m_AllPluginsLoaded = false;
dvander@229	790	m_MyIdent = NULL;
dvander@1630	791	m_LoadingLocked = false;
dvander@211	792	}
dvander@211	793
dvander@211	794	CPluginManager::~CPluginManager()
dvander@211	795	{
dvander@282	796	/* :NOTICE:
dvander@282	797	* Ignore the fact that there might be plugins in the cache.
dvander@282	798	* This usually means that Core is not being unloaded properly, and everything
dvander@282	799	* will crash anyway. YAY
dvander@282	800	*/
dvander@211	801	sm_trie_destroy(m_LoadLookup);
faluco@641	802
faluco@641	803	CStack<CPluginManager::CPluginIterator *>::iterator iter;
faluco@641	804	for (iter=m_iters.begin(); iter!=m_iters.end(); iter++)
faluco@641	805	{
faluco@641	806	delete (*iter);
faluco@641	807	}
faluco@641	808	m_iters.popall();
dvander@211	809	}
dvander@211	810
dvander@1018	811	void CPluginManager::Shutdown()
dvander@1018	812	{
dvander@1018	813	List<CPlugin *>::iterator iter;
dvander@1018	814
dvander@1018	815	while ((iter = m_plugins.begin()) != m_plugins.end())
dvander@1018	816	{
dvander@1018	817	UnloadPlugin(*iter);
dvander@1018	818	}
dvander@1018	819	}
dvander@1018	820
dvander@213	821	void CPluginManager::LoadAll_FirstPass(const char config, const char basedir)
dvander@190	822	{
dvander@203	823	/* First read in the database of plugin settings */
dvander@1656	824	SMCError err;
dvander@1656	825	SMCStates states;
faluco@216	826	m_AllPluginsLoaded = false;
dvander@1656	827	if ((err=textparsers->ParseFile_SMC(config, &m_PluginInfo, &states)) != SMCError_Okay)
dvander@203	828	{
dvander@305	829	g_Logger.LogError("[SM] Encountered fatal error parsing file \"%s\"", config);
dvander@1406	830	const char *err_msg = textparsers->GetSMCErrorString(err);
dvander@282	831	if (err_msg)
dvander@282	832	{
dvander@305	833	g_Logger.LogError("[SM] Parse error encountered: \"%s\"", err_msg);
dvander@282	834	}
dvander@203	835	}
dvander@203	836
dvander@209	837	LoadPluginsFromDir(basedir, NULL);
dvander@209	838	}
dvander@203	839
dvander@209	840	void CPluginManager::LoadPluginsFromDir(const char basedir, const char localpath)
dvander@209	841	{
ds@652	842	char base_path[PLATFORM_MAX_PATH];
dvander@209	843
dvander@209	844	/* Form the current path to start reading from */
dvander@209	845	if (localpath == NULL)
dvander@209	846	{
dvander@209	847	g_LibSys.PathFormat(base_path, sizeof(base_path), "%s", basedir);
dvander@209	848	} else {
dvander@209	849	g_LibSys.PathFormat(base_path, sizeof(base_path), "%s/%s", basedir, localpath);
dvander@209	850	}
dvander@209	851
dvander@209	852	IDirectory *dir = g_LibSys.OpenDirectory(base_path);
dvander@209	853
dvander@209	854	if (!dir)
dvander@209	855	{
dvander@282	856	char error[256];
dvander@282	857	g_LibSys.GetPlatformError(error, sizeof(error));
dvander@305	858	g_Logger.LogError("[SM] Failure reading from plugins path: %s", localpath);
dvander@305	859	g_Logger.LogError("[SM] Platform returned error: %s", error);
dvander@209	860	return;
dvander@209	861	}
dvander@209	862
dvander@190	863	while (dir->MoreFiles())
dvander@190	864	{
dvander@209	865	if (dir->IsEntryDirectory()
dvander@209	866	&& (strcmp(dir->GetEntryName(), ".") != 0)
dvander@209	867	&& (strcmp(dir->GetEntryName(), "..") != 0)
dvander@209	868	&& (strcmp(dir->GetEntryName(), "disabled") != 0)
dvander@209	869	&& (strcmp(dir->GetEntryName(), "optional") != 0))
dvander@190	870	{
ds@652	871	char new_local[PLATFORM_MAX_PATH];
dvander@209	872	if (localpath == NULL)
dvander@209	873	{
dvander@209	874	/* If no path yet, don't add a former slash */
dvander@2814	875	UTIL_Format(new_local, sizeof(new_local), "%s", dir->GetEntryName());
dvander@209	876	} else {
dvander@209	877	g_LibSys.PathFormat(new_local, sizeof(new_local), "%s/%s", localpath, dir->GetEntryName());
dvander@209	878	}
dvander@209	879	LoadPluginsFromDir(basedir, new_local);
dvander@209	880	} else if (dir->IsEntryFile()) {
dvander@209	881	const char *name = dir->GetEntryName();
dvander@209	882	size_t len = strlen(name);
dvander@211	883	if (len >= 4
dvander@211	884	&& strcmp(&name[len-4], ".smx") == 0)
dvander@209	885	{
dvander@211	886	/* If the filename matches, load the plugin */
ds@652	887	char plugin[PLATFORM_MAX_PATH];
dvander@211	888	if (localpath == NULL)
dvander@211	889	{
dvander@2814	890	UTIL_Format(plugin, sizeof(plugin), "%s", name);
dvander@211	891	} else {
dvander@211	892	g_LibSys.PathFormat(plugin, sizeof(plugin), "%s/%s", localpath, name);
dvander@211	893	}
dvander@211	894	LoadAutoPlugin(plugin);
dvander@209	895	}
dvander@209	896	}
dvander@209	897	dir->NextEntry();
dvander@209	898	}
dvander@209	899	g_LibSys.CloseDirectory(dir);
dvander@209	900	}
dvander@209	901
ds@755	902	LoadRes CPluginManager::_LoadPlugin(CPlugin *_plugin, const char path, bool debug, PluginType type, char error[], size_t maxlength)
dvander@209	903	{
dvander@1630	904	if (m_LoadingLocked)
dvander@1630	905	{
dvander@1630	906	return LoadRes_NeverLoad;
dvander@1630	907	}
dvander@1630	908
dvander@2193	909	int err;
dvander@1436	910	bool no_load = false;
dvander@1436	911	PluginSettings *pset;
dvander@1436	912	unsigned int setcount = m_PluginInfo.GetSettingsNum();
dvander@1436	913	for (unsigned int i=0; i<setcount; i++)
dvander@1436	914	{
dvander@1436	915	if ((pset = m_PluginInfo.GetSettingsIfMatch(i, path)) == NULL)
dvander@1436	916	{
dvander@1436	917	continue;
dvander@1436	918	}
dvander@1436	919	if (pset->blockload_val)
dvander@1436	920	{
dvander@1436	921	no_load = true;
dvander@1436	922	break;
dvander@1436	923	}
dvander@1436	924	}
dvander@1436	925
dvander@211	926	/**
dvander@211	927	* Does this plugin already exist?
dvander@211	928	*/
dvander@211	929	CPlugin *pPlugin;
dvander@298	930	if (sm_trie_retrieve(m_LoadLookup, path, (void **)&pPlugin))
dvander@211	931	{
dvander@211	932	/* Check to see if we should try reloading it */
dvander@211	933	if (pPlugin->GetStatus() == Plugin_BadLoad
dvander@282	934	\|\| pPlugin->GetStatus() == Plugin_Error
dvander@1436	935	\|\| pPlugin->GetStatus() == Plugin_Failed
dvander@1436	936	\|\| no_load)
dvander@211	937	{
dvander@211	938	UnloadPlugin(pPlugin);
dvander@1630	939	}
dvander@1630	940	else
dvander@1630	941	{
faluco@383	942	if (_plugin)
faluco@383	943	{
faluco@383	944	*_plugin = pPlugin;
faluco@383	945	}
faluco@383	946	return LoadRes_AlreadyLoaded;
dvander@211	947	}
dvander@211	948	}
dvander@211	949
dvander@1436	950	if (no_load)
dvander@1436	951	{
dvander@1436	952	return LoadRes_NeverLoad;
dvander@1436	953	}
dvander@1436	954
ds@755	955	pPlugin = CPlugin::CreatePlugin(path, error, maxlength);
dvander@209	956
dvander@209	957	assert(pPlugin != NULL);
dvander@209	958
dvander@209	959	pPlugin->m_type = PluginType_MapUpdated;
dvander@209	960
dvander@209	961	ICompilation *co = NULL;
dvander@209	962
dvander@209	963	if (pPlugin->m_status == Plugin_Uncompiled)
dvander@209	964	{
dvander@2193	965	co = g_pSourcePawn2->StartCompilation();
dvander@209	966	}
dvander@209	967
dvander@209	968	for (unsigned int i=0; i<setcount; i++)
dvander@209	969	{
dvander@298	970	if ((pset=m_PluginInfo.GetSettingsIfMatch(i, path)) == NULL)
dvander@209	971	{
dvander@209	972	continue;
dvander@209	973	}
dvander@209	974	pPlugin->m_type = pset->type_val;
dvander@209	975	if (co)
dvander@209	976	{
dvander@209	977	for (unsigned int j=0; j<pset->opts_num; j++)
dvander@209	978	{
dvander@209	979	const char key, val;
dvander@209	980	m_PluginInfo.GetOptionsForPlugin(pset, j, &key, &val);
dvander@209	981	if (!key \|\| !val)
dvander@209	982	{
dvander@209	983	continue;
dvander@209	984	}
dvander@2193	985	if ((err = co->SetOption(key, val)) == SP_ERROR_NONE)
dvander@209	986	{
dvander@298	987	if (error)
dvander@298	988	{
dvander@2814	989	UTIL_Format(error, maxlength, "Unable to set JIT option (key \"%s\") (value \"%s\")", key, val);
dvander@298	990	}
dvander@2193	991	co->Abort();
dvander@209	992	co = NULL;
dvander@209	993	break;
dvander@209	994	}
dvander@209	995	}
dvander@190	996	}
dvander@190	997	}
dvander@209	998
dvander@298	999	/* Do the actual compiling */
dvander@2193	1000	if (co != NULL)
dvander@209	1001	{
dvander@2193	1002	char fullpath[PLATFORM_MAX_PATH];
dvander@2193	1003	g_SourceMod.BuildPath(Path_SM, fullpath, sizeof(fullpath), "plugins/%s", pPlugin->m_filename);
dvander@2193	1004
dvander@2193	1005	pPlugin->m_pRuntime = g_pSourcePawn2->LoadPlugin(co, fullpath, &err);
dvander@2193	1006	if (pPlugin->m_pRuntime == NULL)
dvander@2193	1007	{
dvander@2814	1008	UTIL_Format(error,
dvander@2193	1009	maxlength,
dvander@2193	1010	"Unable to load plugin (error %d: %s)",
dvander@2193	1011	err,
dvander@2193	1012	g_pSourcePawn2->GetErrorString(err));
dvander@2342	1013	pPlugin->m_status = Plugin_BadLoad;
dvander@2193	1014	}
dvander@2193	1015	else
dvander@2193	1016	{
dvander@2342	1017	if (pPlugin->UpdateInfo())
dvander@2342	1018	{
dvander@2342	1019	pPlugin->m_status = Plugin_Created;
dvander@2342	1020	}
dvander@2342	1021	else
dvander@2342	1022	{
dvander@2342	1023	UTIL_Format(error, maxlength, "%s", pPlugin->m_errormsg);
dvander@2342	1024	}
dvander@2193	1025	}
dvander@209	1026	}
dvander@209	1027
fyrenmoo@2675	1028	LoadRes loadFailure = LoadRes_Failure;
dvander@298	1029	/* Get the status */
dvander@211	1030	if (pPlugin->GetStatus() == Plugin_Created)
dvander@211	1031	{
dvander@2086	1032	/* First native pass - add anything from Core */
dvander@2086	1033	g_ShareSys.BindNativesToPlugin(pPlugin, true);
dvander@229	1034	pPlugin->InitIdentity();
fyrenmoo@2675	1035	APLRes result = pPlugin->Call_AskPluginLoad(error, maxlength);
fyrenmoo@2675	1036	switch (result)
dvander@315	1037	{
fyrenmoo@2675	1038	case APLRes_Success:
dvander@315	1039	/* Autoload any modules */
ds@755	1040	LoadOrRequireExtensions(pPlugin, 1, error, maxlength);
fyrenmoo@2675	1041	break;
fyrenmoo@2675	1042	case APLRes_Failure:
teame06@1200	1043	pPlugin->SetErrorState(Plugin_Failed, "%s", error);
fyrenmoo@2675	1044	loadFailure = LoadRes_Failure;
fyrenmoo@2675	1045	break;
fyrenmoo@2675	1046	case APLRes_SilentFailure:
fyrenmoo@2675	1047	pPlugin->SetErrorState(Plugin_Failed, "%s", error);
fyrenmoo@2675	1048	loadFailure = LoadRes_SilentFailure;
fyrenmoo@2675	1049	pPlugin->SetSilentlyFailed(true);
fyrenmoo@2675	1050	break;
fyrenmoo@2675	1051	default:
fyrenmoo@2675	1052	assert(false);
dvander@315	1053	}
dvander@211	1054	}
dvander@211	1055
faluco@333	1056	/* Save the time stamp */
faluco@333	1057	time_t t = pPlugin->GetFileTimeStamp();
faluco@333	1058	pPlugin->SetTimeStamp(t);
faluco@333	1059
dvander@298	1060	if (_plugin)
dvander@298	1061	{
dvander@298	1062	*_plugin = pPlugin;
dvander@298	1063	}
dvander@298	1064
fyrenmoo@2675	1065	return (pPlugin->GetStatus() == Plugin_Loaded) ? LoadRes_Successful : loadFailure;
dvander@190	1066	}
dvander@190	1067
ds@755	1068	IPlugin CPluginManager::LoadPlugin(const char path, bool debug, PluginType type, char error[], size_t maxlength, bool *wasloaded)
dvander@174	1069	{
dvander@298	1070	CPlugin *pl;
faluco@383	1071	LoadRes res;
dvander@211	1072
faluco@383	1073	*wasloaded = false;
dvander@2226	1074	if ((res=_LoadPlugin(&pl, path, true, type, error, maxlength)) == LoadRes_Failure)
dvander@211	1075	{
dvander@298	1076	delete pl;
dvander@174	1077	return NULL;
dvander@174	1078	}
dvander@174	1079
faluco@383	1080	if (res == LoadRes_AlreadyLoaded)
faluco@383	1081	{
faluco@383	1082	*wasloaded = true;
faluco@383	1083	return pl;
faluco@383	1084	}
faluco@383	1085
dvander@1436	1086	if (res == LoadRes_NeverLoad)
dvander@1436	1087	{
dvander@2814	1088	if (error)
dvander@1630	1089	{
dvander@2814	1090	if (m_LoadingLocked)
dvander@2814	1091	{
dvander@2814	1092	UTIL_Format(error, maxlength, "There is a global plugin loading lock in effect");
dvander@2814	1093	}
dvander@2814	1094	else
dvander@2814	1095	{
dvander@2814	1096	UTIL_Format(error, maxlength, "This plugin is blocked from loading (see plugin_settings.cfg)");
dvander@2814	1097	}
dvander@1630	1098	}
dvander@1436	1099	return NULL;
dvander@1436	1100	}
dvander@1436	1101
dvander@298	1102	AddPlugin(pl);
dvander@298	1103
dvander@315	1104	/* Run second pass if we need to */
dvander@315	1105	if (IsLateLoadTime() && pl->GetStatus() == Plugin_Loaded)
dvander@315	1106	{
ds@755	1107	if (!RunSecondPass(pl, error, maxlength))
dvander@315	1108	{
dvander@315	1109	UnloadPlugin(pl);
dvander@315	1110	return NULL;
dvander@315	1111	}
faluco@1221	1112	pl->Call_OnAllPluginsLoaded();
dvander@315	1113	}
dvander@315	1114
dvander@298	1115	return pl;
dvander@298	1116	}
dvander@298	1117
dvander@298	1118	void CPluginManager::LoadAutoPlugin(const char *plugin)
dvander@298	1119	{
dvander@1436	1120	CPlugin *pl = NULL;
faluco@383	1121	LoadRes res;
dvander@298	1122	char error[255] = "Unknown error";
dvander@298	1123
faluco@383	1124	if ((res=_LoadPlugin(&pl, plugin, false, PluginType_MapUpdated, error, sizeof(error))) == LoadRes_Failure)
dvander@209	1125	{
dvander@305	1126	g_Logger.LogError("[SM] Failed to load plugin \"%s\": %s", plugin, error);
dvander@2342	1127	pl->SetErrorState(
dvander@2885	1128	pl->GetStatus() <= Plugin_Created ? Plugin_BadLoad : pl->GetStatus(),
dvander@2342	1129	"%s",
dvander@2342	1130	error);
dvander@209	1131	}
dvander@209	1132
fyrenmoo@2675	1133	if (res == LoadRes_Successful \|\| res == LoadRes_Failure \|\| res == LoadRes_SilentFailure)
faluco@383	1134	{
faluco@383	1135	AddPlugin(pl);
faluco@383	1136	}
dvander@209	1137	}
dvander@209	1138
dvander@211	1139	void CPluginManager::AddPlugin(CPlugin *pPlugin)
dvander@209	1140	{
dvander@174	1141	List<IPluginsListener *>::iterator iter;
dvander@174	1142	IPluginsListener *pListener;
dvander@298	1143
dvander@174	1144	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
dvander@174	1145	{
dvander@174	1146	pListener = (*iter);
dvander@174	1147	pListener->OnPluginCreated(pPlugin);
dvander@174	1148	}
dvander@174	1149
dvander@298	1150	m_plugins.push_back(pPlugin);
dvander@298	1151	sm_trie_insert(m_LoadLookup, pPlugin->m_filename, pPlugin);
dvander@211	1152	}
dvander@211	1153
dvander@213	1154	void CPluginManager::LoadAll_SecondPass()
dvander@213	1155	{
dvander@213	1156	List<CPlugin *>::iterator iter;
dvander@213	1157	CPlugin *pPlugin;
dvander@213	1158
dvander@315	1159	char error[256];
dvander@213	1160	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
dvander@213	1161	{
dvander@213	1162	pPlugin = (*iter);
dvander@213	1163	if (pPlugin->GetStatus() == Plugin_Loaded)
dvander@213	1164	{
dvander@315	1165	error[0] = '\0';
dvander@315	1166	if (!RunSecondPass(pPlugin, error, sizeof(error)))
dvander@315	1167	{
dvander@315	1168	g_Logger.LogError("[SM] Unable to load plugin \"%s\": %s", pPlugin->GetFilename(), error);
dvander@315	1169	pPlugin->SetErrorState(Plugin_Failed, "%s", error);
dvander@315	1170	}
dvander@213	1171	}
dvander@213	1172	}
dvander@298	1173
faluco@216	1174	m_AllPluginsLoaded = true;
dvander@213	1175	}
dvander@213	1176
faluco@1112	1177	bool CPluginManager::FindOrRequirePluginDeps(CPlugin pPlugin, char error, size_t maxlength)
faluco@1112	1178	{
faluco@1112	1179	struct _pl
faluco@1112	1180	{
faluco@1112	1181	cell_t name;
faluco@1112	1182	cell_t file;
faluco@1112	1183	cell_t required;
faluco@1112	1184	} *pl;
faluco@1112	1185
faluco@1112	1186	IPluginContext *pBase = pPlugin->GetBaseContext();
faluco@1112	1187	uint32_t num = pBase->GetPubVarsNum();
faluco@1112	1188	sp_pubvar_t *pubvar;
faluco@1112	1189	char name, file;
faluco@1112	1190	char pathfile[PLATFORM_MAX_PATH];
faluco@1112	1191
faluco@1112	1192	for (uint32_t i=0; i<num; i++)
faluco@1112	1193	{
faluco@1112	1194	if (pBase->GetPubvarByIndex(i, &pubvar) != SP_ERROR_NONE)
faluco@1112	1195	{
faluco@1112	1196	continue;
faluco@1112	1197	}
faluco@1112	1198	if (strncmp(pubvar->name, "__pl_", 5) == 0)
faluco@1112	1199	{
faluco@1112	1200	pl = (_pl *)pubvar->offs;
faluco@1112	1201	if (pBase->LocalToString(pl->file, &file) != SP_ERROR_NONE)
faluco@1112	1202	{
faluco@1112	1203	continue;
faluco@1112	1204	}
faluco@1112	1205	if (pBase->LocalToString(pl->name, &name) != SP_ERROR_NONE)
faluco@1112	1206	{
faluco@1112	1207	continue;
faluco@1112	1208	}
faluco@1112	1209	g_LibSys.GetFileFromPath(pathfile, sizeof(pathfile), pPlugin->GetFilename());
faluco@1112	1210	if (strcmp(pathfile, file) == 0)
faluco@1112	1211	{
faluco@1112	1212	continue;
faluco@1112	1213	}
faluco@1112	1214	if (pl->required == false)
faluco@1112	1215	{
faluco@1112	1216	IPluginFunction *pFunc;
faluco@1112	1217	char buffer[64];
dvander@1123	1218	UTIL_Format(buffer, sizeof(buffer), "__pl_%s_SetNTVOptional", &pubvar->name[5]);
faluco@1112	1219	if ((pFunc=pBase->GetFunctionByName(buffer)))
faluco@1112	1220	{
faluco@1112	1221	cell_t res;
faluco@1112	1222	pFunc->Execute(&res);
dvander@2193	1223	if (pPlugin->GetBaseContext()->GetLastNativeError() != SP_ERROR_NONE)
faluco@1112	1224	{
faluco@1112	1225	if (error)
faluco@1112	1226	{
dvander@2814	1227	UTIL_Format(error, maxlength, "Fatal error during initializing plugin load");
faluco@1112	1228	}
faluco@1112	1229	return false;
faluco@1112	1230	}
faluco@1112	1231	}
dvander@1572	1232	}
dvander@1572	1233	else
dvander@1572	1234	{
faluco@1112	1235	/* Check that we aren't registering the same library twice */
faluco@1112	1236	if (pPlugin->m_RequiredLibs.find(name) == pPlugin->m_RequiredLibs.end())
faluco@1112	1237	{
faluco@1112	1238	pPlugin->m_RequiredLibs.push_back(name);
dvander@1572	1239	}
dvander@1572	1240	else
dvander@1572	1241	{
faluco@1112	1242	continue;
faluco@1112	1243	}
faluco@1112	1244	List<CPlugin *>::iterator iter;
faluco@1112	1245	CPlugin *pl;
faluco@1112	1246	bool found = false;
faluco@1112	1247	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
faluco@1112	1248	{
faluco@1112	1249	pl = (*iter);
faluco@1112	1250	if (pl->m_Libraries.find(name) != pl->m_Libraries.end())
faluco@1112	1251	{
faluco@1112	1252	found = true;
faluco@1112	1253	break;
faluco@1112	1254	}
faluco@1112	1255	}
faluco@1112	1256	if (!found)
faluco@1112	1257	{
faluco@1112	1258	if (error)
faluco@1112	1259	{
dvander@2814	1260	UTIL_Format(error, maxlength, "Could not find required plugin \"%s\"", name);
faluco@1112	1261	}
faluco@1112	1262	return false;
faluco@1112	1263	}
faluco@1112	1264	}
faluco@1112	1265	}
faluco@1112	1266	}
faluco@1112	1267
faluco@1112	1268	return true;
faluco@1112	1269	}
faluco@1112	1270
dvander@315	1271	bool CPluginManager::LoadOrRequireExtensions(CPlugin pPlugin, unsigned int pass, char error, size_t maxlength)
dvander@211	1272	{
dvander@298	1273	/* Find any extensions this plugin needs */
dvander@298	1274	struct _ext
dvander@298	1275	{
dvander@298	1276	cell_t name;
dvander@298	1277	cell_t file;
dvander@298	1278	cell_t autoload;
dvander@298	1279	cell_t required;
dvander@298	1280	} *ext;
dvander@298	1281
dvander@298	1282	IPluginContext *pBase = pPlugin->GetBaseContext();
dvander@298	1283	uint32_t num = pBase->GetPubVarsNum();
dvander@298	1284	sp_pubvar_t *pubvar;
dvander@298	1285	IExtension *pExt;
ds@652	1286	char path[PLATFORM_MAX_PATH];
dvander@298	1287	char file, name;
dvander@298	1288	for (uint32_t i=0; i<num; i++)
dvander@298	1289	{
dvander@298	1290	if (pBase->GetPubvarByIndex(i, &pubvar) != SP_ERROR_NONE)
dvander@298	1291	{
dvander@298	1292	continue;
dvander@298	1293	}
dvander@298	1294	if (strncmp(pubvar->name, "__ext_", 6) == 0)
dvander@298	1295	{
dvander@298	1296	ext = (_ext *)pubvar->offs;
dvander@298	1297	if (pBase->LocalToString(ext->file, &file) != SP_ERROR_NONE)
dvander@298	1298	{
dvander@298	1299	continue;
dvander@298	1300	}
dvander@298	1301	if (pBase->LocalToString(ext->name, &name) != SP_ERROR_NONE)
dvander@298	1302	{
dvander@298	1303	continue;
dvander@298	1304	}
dvander@315	1305	if (pass == 1)
dvander@298	1306	{
dvander@315	1307	/* Attempt to auto-load if necessary */
dvander@315	1308	if (ext->autoload)
dvander@315	1309	{
dvander@315	1310	g_LibSys.PathFormat(path, PLATFORM_MAX_PATH, "%s", file);
skamonkey@3402	1311	bool bErrorOnMissing = ext->required ? true : false;
skamonkey@3402	1312	g_Extensions.LoadAutoExtension(path, bErrorOnMissing);
dvander@315	1313	}
dvander@1572	1314	}
dvander@1572	1315	else if (pass == 2)
dvander@1572	1316	{
dvander@315	1317	/* Is this required? */
dvander@315	1318	if (ext->required)
dvander@298	1319	{
dvander@315	1320	g_LibSys.PathFormat(path, PLATFORM_MAX_PATH, "%s", file);
dvander@315	1321	if ((pExt = g_Extensions.FindExtensionByFile(path)) == NULL)
dvander@315	1322	{
dvander@315	1323	pExt = g_Extensions.FindExtensionByName(name);
dvander@315	1324	}
dvander@315	1325	/* :TODO: should we bind to unloaded extensions?
dvander@315	1326	* Currently the extension manager will ignore this.
dvander@315	1327	*/
dvander@315	1328	if (!pExt \|\| !pExt->IsRunning(NULL, 0))
dvander@315	1329	{
dvander@315	1330	if (error)
dvander@315	1331	{
dvander@2814	1332	UTIL_Format(error, maxlength, "Required extension \"%s\" file(\"%s\") not running", name, file);
dvander@315	1333	}
dvander@315	1334	return false;
dvander@1572	1335	}
dvander@1572	1336	else
dvander@1572	1337	{
dvander@315	1338	g_Extensions.BindChildPlugin(pExt, pPlugin);
dvander@315	1339	}
dvander@298	1340	}
dvander@1572	1341	else
dvander@1572	1342	{
dvander@1572	1343	IPluginFunction *pFunc;
dvander@1572	1344	char buffer[64];
dvander@1572	1345	UTIL_Format(buffer, sizeof(buffer), "__ext_%s_SetNTVOptional", &pubvar->name[6]);
dvander@1572	1346
dvander@1572	1347	if ((pFunc = pBase->GetFunctionByName(buffer)) != NULL)
dvander@1572	1348	{
dvander@1572	1349	cell_t res;
dvander@1572	1350	pFunc->Execute(&res);
dvander@2193	1351	if (pPlugin->GetBaseContext()->GetLastNativeError() != SP_ERROR_NONE)
dvander@1572	1352	{
dvander@1572	1353	if (error)
dvander@1572	1354	{
dvander@1572	1355	UTIL_Format(error, maxlength, "Fatal error during plugin initialization (ext req)");
dvander@1572	1356	}
dvander@1572	1357	return false;
dvander@1572	1358	}
dvander@1572	1359	}
dvander@1572	1360	}
dvander@298	1361	}
dvander@315	1362	}
dvander@315	1363	}
dvander@315	1364
dvander@315	1365	return true;
dvander@315	1366	}
dvander@315	1367
dvander@315	1368	bool CPluginManager::RunSecondPass(CPlugin pPlugin, char error, size_t maxlength)
dvander@315	1369	{
dvander@315	1370	/* Second pass for extension requirements */
dvander@315	1371	if (!LoadOrRequireExtensions(pPlugin, 2, error, maxlength))
dvander@315	1372	{
dvander@315	1373	return false;
dvander@315	1374	}
dvander@315	1375
faluco@1112	1376	if (!FindOrRequirePluginDeps(pPlugin, error, maxlength))
faluco@1112	1377	{
faluco@1112	1378	return false;
faluco@1112	1379	}
faluco@1112	1380
dvander@2086	1381	/* Run another binding pass */
dvander@2086	1382	g_ShareSys.BindNativesToPlugin(pPlugin, false);
dvander@1572	1383
dvander@2086	1384	/* Find any unbound natives. Right now, these are not allowed. */
dvander@315	1385	IPluginContext *pContext = pPlugin->GetBaseContext();
dvander@315	1386	uint32_t num = pContext->GetNativesNum();
dvander@315	1387	sp_native_t *native;
dvander@315	1388	for (unsigned int i=0; i<num; i++)
dvander@315	1389	{
dvander@315	1390	if (pContext->GetNativeByIndex(i, &native) != SP_ERROR_NONE)
dvander@315	1391	{
dvander@315	1392	break;
dvander@315	1393	}
dvander@798	1394	if (native->status == SP_NATIVE_UNBOUND
faluco@1089	1395	&& native->name[0] != '@'
faluco@1089	1396	&& !(native->flags & SP_NTVFLAG_OPTIONAL))
dvander@315	1397	{
dvander@315	1398	if (error)
dvander@298	1399	{
dvander@2814	1400	UTIL_Format(error, maxlength, "Native \"%s\" was not found", native->name);
dvander@298	1401	}
dvander@315	1402	return false;
dvander@298	1403	}
dvander@298	1404	}
dvander@211	1405
dvander@211	1406	/* Finish by telling all listeners */
dvander@211	1407	List<IPluginsListener *>::iterator iter;
dvander@211	1408	IPluginsListener *pListener;
dvander@211	1409	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
dvander@211	1410	{
dvander@211	1411	pListener = (*iter);
dvander@211	1412	pListener->OnPluginLoaded(pPlugin);
dvander@211	1413	}
dvander@298	1414
dvander@298	1415	/* Tell this plugin to finish initializing itself */
ds@426	1416	pPlugin->Call_OnPluginStart();
dvander@298	1417
dvander@605	1418	/* Now, if we have fake natives, go through all plugins that might need rebinding */
dvander@2086	1419	if (pPlugin->GetStatus() <= Plugin_Paused && pPlugin->m_Natives.size())
dvander@605	1420	{
dvander@605	1421	List<CPlugin *>::iterator pl_iter;
dvander@605	1422	CPlugin *pOther;
dvander@605	1423	for (pl_iter = m_plugins.begin();
dvander@605	1424	pl_iter != m_plugins.end();
dvander@605	1425	pl_iter++)
dvander@605	1426	{
dvander@605	1427	pOther = (*pl_iter);
dvander@1122	1428	if ((pOther->GetStatus() == Plugin_Error
faluco@1112	1429	&& (pOther->m_FakeNativesMissing \|\| pOther->m_LibraryMissing))
dvander@1122	1430	\|\| pOther->m_FakeNativesMissing)
dvander@605	1431	{
faluco@1112	1432	TryRefreshDependencies(pOther);
dvander@605	1433	}
dvander@2086	1434	else if ((pOther->GetStatus() == Plugin_Running
dvander@2086	1435	\|\| pOther->GetStatus() == Plugin_Paused)
dvander@2086	1436	&& pOther != pPlugin)
dvander@2086	1437	{
dvander@2086	1438	List<NativeEntry *>::iterator nv_iter;
dvander@2086	1439	for (nv_iter = pPlugin->m_Natives.begin();
dvander@2086	1440	nv_iter != pPlugin->m_Natives.end();
dvander@2086	1441	nv_iter++)
dvander@2086	1442	{
dvander@2086	1443	g_ShareSys.BindNativeToPlugin(pOther, (*nv_iter));
dvander@2086	1444	}
dvander@2086	1445	}
dvander@605	1446	}
dvander@605	1447	}
dvander@605	1448
dvander@1124	1449	/* Go through our libraries and tell other plugins they're added */
dvander@1124	1450	List<String>::iterator s_iter;
dvander@1124	1451	for (s_iter = pPlugin->m_Libraries.begin();
dvander@1124	1452	s_iter != pPlugin->m_Libraries.end();
dvander@1124	1453	s_iter++)
dvander@1124	1454	{
dvander@1572	1455	OnLibraryAction((*s_iter).c_str(), true, false);
dvander@1124	1456	}
dvander@1124	1457
dvander@2042	1458	/* :TODO: optimize? does this even matter? */
dvander@2042	1459	pPlugin->GetPhrases()->AddPhraseFile("core.phrases");
dvander@1621	1460
dvander@298	1461	return true;
dvander@211	1462	}
dvander@211	1463
faluco@1112	1464	void CPluginManager::TryRefreshDependencies(CPlugin *pPlugin)
dvander@605	1465	{
dvander@605	1466	assert(pPlugin->GetBaseContext() != NULL);
dvander@605	1467
dvander@2086	1468	g_ShareSys.BindNativesToPlugin(pPlugin, false);
dvander@605	1469
faluco@1112	1470	List<String>::iterator lib_iter;
faluco@1112	1471	List<String>::iterator req_iter;
faluco@1112	1472	List<CPlugin *>::iterator pl_iter;
faluco@1112	1473	CPlugin *pl;
faluco@1112	1474	for (req_iter=pPlugin->m_RequiredLibs.begin(); req_iter!=pPlugin->m_RequiredLibs.end(); req_iter++)
faluco@1112	1475	{
faluco@1112	1476	bool found = false;
faluco@1112	1477	for (pl_iter=m_plugins.begin(); pl_iter!=m_plugins.end(); pl_iter++)
faluco@1112	1478	{
faluco@1112	1479	pl = (*pl_iter);
faluco@1112	1480	for (lib_iter=pl->m_Libraries.begin(); lib_iter!=pl->m_Libraries.end(); lib_iter++)
faluco@1112	1481	{
faluco@1112	1482	if ((req_iter) == (lib_iter))
faluco@1112	1483	{
faluco@1112	1484	found = true;
faluco@1112	1485	}
faluco@1112	1486	}
faluco@1112	1487	}
faluco@1112	1488	if (!found)
faluco@1112	1489	{
faluco@1112	1490	pPlugin->SetErrorState(Plugin_Error, "Library not found: %s", (*req_iter).c_str());
faluco@1112	1491	return;
faluco@1112	1492	}
faluco@1112	1493	}
faluco@1112	1494
dvander@605	1495	/* Find any unbound natives
dvander@605	1496	* Right now, these are not allowed
dvander@605	1497	*/
dvander@605	1498	IPluginContext *pContext = pPlugin->GetBaseContext();
dvander@605	1499	uint32_t num = pContext->GetNativesNum();
dvander@605	1500	sp_native_t *native;
dvander@605	1501	for (unsigned int i=0; i<num; i++)
dvander@605	1502	{
dvander@605	1503	if (pContext->GetNativeByIndex(i, &native) != SP_ERROR_NONE)
dvander@605	1504	{
dvander@605	1505	break;
dvander@605	1506	}
faluco@1089	1507	if (native->status == SP_NATIVE_UNBOUND && !(native->flags & SP_NTVFLAG_OPTIONAL))
dvander@605	1508	{
dvander@605	1509	pPlugin->SetErrorState(Plugin_Error, "Native not found: %s", native->name);
dvander@605	1510	return;
dvander@605	1511	}
dvander@605	1512	}
dvander@605	1513
dvander@1122	1514	if (pPlugin->GetStatus() == Plugin_Error)
dvander@605	1515	{
dvander@1122	1516	/* If we got here, all natives are okay again! */
dvander@1122	1517	pPlugin->m_status = Plugin_Running;
dvander@2193	1518	if (pPlugin->m_pRuntime->IsPaused())
dvander@601	1519	{
dvander@2193	1520	pPlugin->m_pRuntime->SetPauseState(false);
ds@1952	1521	_SetPauseState(pPlugin, false);
dvander@601	1522	}
dvander@601	1523	}
dvander@601	1524	}
dvander@601	1525
dvander@174	1526	bool CPluginManager::UnloadPlugin(IPlugin *plugin)
dvander@174	1527	{
dvander@174	1528	CPlugin pPlugin = (CPlugin )plugin;
dvander@298	1529
dvander@298	1530	/* This prevents removal during insertion or anything else weird */
dvander@298	1531	if (m_plugins.find(pPlugin) == m_plugins.end())
dvander@298	1532	{
dvander@298	1533	return false;
dvander@298	1534	}
dvander@298	1535
dvander@1638	1536	IPluginContext *pContext = plugin->GetBaseContext();
dvander@2342	1537	if (pContext != NULL && pContext->IsInExec())
dvander@1638	1538	{
dvander@1638	1539	char buffer[255];
dvander@1638	1540	UTIL_Format(buffer, sizeof(buffer), "sm plugins unload %s\n", plugin->GetFilename());
dvander@1638	1541	engine->ServerCommand(buffer);
dvander@1638	1542	return false;
dvander@1638	1543	}
dvander@1638	1544
dvander@605	1545	/* Remove us from the lookup table and linked list */
dvander@605	1546	m_plugins.remove(pPlugin);
dvander@605	1547	sm_trie_delete(m_LoadLookup, pPlugin->m_filename);
dvander@605	1548
dvander@1124	1549	/* Go through our libraries and tell other plugins they're gone */
dvander@1124	1550	List<String>::iterator s_iter;
dvander@1124	1551	for (s_iter = pPlugin->m_Libraries.begin();
dvander@1124	1552	s_iter != pPlugin->m_Libraries.end();
dvander@1124	1553	s_iter++)
dvander@1124	1554	{
dvander@1572	1555	OnLibraryAction((*s_iter).c_str(), true, true);
dvander@1124	1556	}
dvander@1124	1557
dvander@174	1558	List<IPluginsListener *>::iterator iter;
dvander@174	1559	IPluginsListener *pListener;
dvander@282	1560
dvander@311	1561	if (pPlugin->GetStatus() <= Plugin_Error)
dvander@282	1562	{
dvander@282	1563	/* Notify listeners of unloading */
dvander@282	1564	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
dvander@282	1565	{
dvander@282	1566	pListener = (*iter);
dvander@282	1567	pListener->OnPluginUnloaded(pPlugin);
dvander@282	1568	}
dvander@282	1569	/* Notify plugin */
ds@426	1570	pPlugin->Call_OnPluginEnd();
dvander@282	1571	}
dvander@282	1572
dvander@2086	1573	pPlugin->DropEverything();
faluco@1112	1574
dvander@174	1575	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
dvander@174	1576	{
dvander@282	1577	/* Notify listeners of destruction */
dvander@174	1578	pListener = (*iter);
dvander@174	1579	pListener->OnPluginDestroyed(pPlugin);
dvander@174	1580	}
dvander@176	1581
dvander@282	1582	/* Tell the plugin to delete itself */
dvander@174	1583	delete pPlugin;
dvander@174	1584
dvander@174	1585	return true;
dvander@174	1586	}
dvander@174	1587
dvander@174	1588	IPlugin CPluginManager::FindPluginByContext(const sp_context_t ctx)
dvander@174	1589	{
dvander@2193	1590	IPlugin *pPlugin;
dvander@2193	1591	IPluginContext *pContext;
dvander@2193	1592
dvander@2193	1593	pContext = reinterpret_cast<IPluginContext >(const_cast<sp_context_t >(ctx));
dvander@2193	1594
dvander@2193	1595	if (pContext->GetKey(2, (void **)&pPlugin))
dvander@2193	1596	{
dvander@2193	1597	return pPlugin;
dvander@2193	1598	}
dvander@2193	1599
dvander@2193	1600	return NULL;
dvander@2193	1601	}
dvander@2193	1602
dvander@2193	1603	CPlugin CPluginManager::GetPluginByCtx(const sp_context_t ctx)
dvander@2193	1604	{
dvander@2193	1605	return (CPlugin *)FindPluginByContext(ctx);
dvander@174	1606	}
dvander@174	1607
dvander@174	1608	unsigned int CPluginManager::GetPluginCount()
dvander@174	1609	{
dvander@174	1610	return m_plugins.size();
dvander@174	1611	}
dvander@174	1612
dvander@174	1613	void CPluginManager::AddPluginsListener(IPluginsListener *listener)
dvander@174	1614	{
dvander@174	1615	m_listeners.push_back(listener);
dvander@174	1616	}
dvander@174	1617
dvander@174	1618	void CPluginManager::RemovePluginsListener(IPluginsListener *listener)
dvander@174	1619	{
dvander@174	1620	m_listeners.remove(listener);
dvander@174	1621	}
dvander@174	1622
dvander@174	1623	IPluginIterator *CPluginManager::GetPluginIterator()
dvander@174	1624	{
dvander@174	1625	if (m_iters.empty())
dvander@174	1626	{
dvander@174	1627	return new CPluginIterator(&m_plugins);
dvander@174	1628	} else {
dvander@174	1629	CPluginIterator *iter = m_iters.front();
dvander@174	1630	m_iters.pop();
faluco@216	1631	iter->Reset();
dvander@174	1632	return iter;
dvander@174	1633	}
dvander@174	1634	}
dvander@174	1635
dvander@174	1636	void CPluginManager::ReleaseIterator(CPluginIterator *iter)
dvander@174	1637	{
dvander@174	1638	m_iters.push(iter);
dvander@174	1639	}
dvander@176	1640
dvander@203	1641	bool CPluginManager::TestAliasMatch(const char alias, const char localpath)
dvander@203	1642	{
dvander@203	1643	/* As an optimization, we do not call strlen, but compute the length in the first pass */
dvander@203	1644	size_t alias_len = 0;
dvander@203	1645	size_t local_len = 0;
dvander@203	1646
dvander@203	1647	const char *ptr = alias;
dvander@203	1648	unsigned int alias_explicit_paths = 0;
dvander@203	1649	unsigned int alias_path_end = 0;
dvander@203	1650	while (*ptr != '\0')
dvander@203	1651	{
dvander@203	1652	if (ptr == '\\' \|\| ptr == '/')
dvander@203	1653	{
dvander@203	1654	alias_explicit_paths++;
dvander@203	1655	alias_path_end = alias_len;
dvander@203	1656	}
dvander@203	1657	alias_len++;
dvander@203	1658	ptr++;
dvander@203	1659	}
dvander@203	1660
dvander@209	1661	if (alias_explicit_paths && alias_path_end == alias_len - 1)
dvander@203	1662	{
dvander@203	1663	/* Trailing slash is totally invalid here */
dvander@203	1664	return false;
dvander@203	1665	}
dvander@203	1666
dvander@203	1667	ptr = localpath;
dvander@203	1668	unsigned int local_explicit_paths = 0;
dvander@203	1669	unsigned int local_path_end = 0;
dvander@203	1670	while (*ptr != '\0')
dvander@203	1671	{
dvander@203	1672	if (ptr == '\\' \|\| ptr == '/')
dvander@203	1673	{
dvander@203	1674	local_explicit_paths++;
dvander@203	1675	local_path_end = local_len;
dvander@203	1676	}
dvander@203	1677	local_len++;
dvander@203	1678	ptr++;
dvander@203	1679	}
dvander@203	1680
dvander@203	1681	/* If the alias has more explicit paths than the real path,
dvander@203	1682	* no match will be possible.
dvander@203	1683	*/
dvander@203	1684	if (alias_explicit_paths > local_explicit_paths)
dvander@203	1685	{
dvander@203	1686	return false;
dvander@203	1687	}
dvander@203	1688
dvander@203	1689	if (alias_explicit_paths)
dvander@203	1690	{
dvander@203	1691	/* We need to find if the paths match now. For example, these should all match:
dvander@203	1692	* csdm csdm
dvander@203	1693	* csdm optional/csdm
dvander@203	1694	* csdm/ban optional/crab/csdm/ban
dvander@203	1695	*/
dvander@203	1696	const char *aliasptr = alias;
dvander@203	1697	const char *localptr = localpath;
dvander@203	1698	bool match = true;
dvander@203	1699	do
dvander@203	1700	{
dvander@203	1701	if (aliasptr != localptr)
dvander@203	1702	{
dvander@203	1703	/* We have to knock one path off */
dvander@203	1704	local_explicit_paths--;
dvander@203	1705	if (alias_explicit_paths > local_explicit_paths)
dvander@203	1706	{
dvander@203	1707	/* Skip out if we're gonna have an impossible match */
dvander@203	1708	return false;
dvander@203	1709	}
dvander@203	1710	/* Eat up localptr tokens until we get a result */
dvander@203	1711	while (((localptr - localpath) < (int)local_path_end)
dvander@203	1712	&& *localptr != '/'
dvander@203	1713	&& *localptr != '\\')
dvander@203	1714	{
dvander@203	1715	localptr++;
dvander@203	1716	}
dvander@203	1717	/* Check if we hit the end of our searchable area.
dvander@203	1718	* This probably isn't possible because of the path
dvander@203	1719	* count check, but it's a good idea anyway.
dvander@203	1720	*/
dvander@203	1721	if ((localptr - localpath) >= (int)local_path_end)
dvander@203	1722	{
dvander@203	1723	return false;
dvander@203	1724	} else {
dvander@203	1725	/* Consume the slash token */
dvander@203	1726	localptr++;
dvander@203	1727	}
dvander@203	1728	/* Reset the alias pointer so we can continue consuming */
dvander@203	1729	aliasptr = alias;
dvander@203	1730	match = false;
dvander@203	1731	continue;
dvander@203	1732	}
dvander@203	1733	/* Note:
dvander@203	1734	* This is safe because if localptr terminates early, aliasptr will too
dvander@203	1735	*/
dvander@203	1736	do
dvander@203	1737	{
dvander@203	1738	/* We should never reach the end of the string because of this check. */
dvander@203	1739	bool aliasend = (aliasptr - alias) > (int)alias_path_end;
dvander@203	1740	bool localend = (localptr - localpath) > (int)local_path_end;
dvander@203	1741	if (aliasend \|\| localend)
dvander@203	1742	{
dvander@203	1743	if (aliasend && localend)
dvander@203	1744	{
dvander@203	1745	/* we matched, and we can break out now */
dvander@203	1746	match = true;
dvander@203	1747	break;
dvander@203	1748	}
dvander@203	1749	/* Otherwise, we've hit the end somehow and rest won't match up. Break out. */
dvander@203	1750	match = false;
dvander@203	1751	break;
dvander@203	1752	}
dvander@203	1753
dvander@203	1754	/* If we got here, it's safe to compare the next two tokens */
dvander@203	1755	if (localptr != aliasptr)
dvander@203	1756	{
dvander@203	1757	match = false;
dvander@203	1758	break;
dvander@203	1759	}
dvander@203	1760	localptr++;
dvander@203	1761	aliasptr++;
dvander@203	1762	} while (true);
dvander@203	1763	} while (!match);
dvander@203	1764	}
dvander@203	1765
dvander@203	1766	/* If we got here, it's time to compare filenames */
dvander@203	1767	const char *aliasptr = alias;
dvander@203	1768	const char *localptr = localpath;
dvander@203	1769
dvander@203	1770	if (alias_explicit_paths)
dvander@203	1771	{
dvander@203	1772	aliasptr = &alias[alias_path_end + 1];
dvander@203	1773	}
dvander@203	1774
dvander@203	1775	if (local_explicit_paths)
dvander@203	1776	{
dvander@203	1777	localptr = &localpath[local_path_end + 1];
dvander@203	1778	}
dvander@203	1779
dvander@203	1780	while (true)
dvander@203	1781	{
dvander@203	1782	if (aliasptr == '')
dvander@203	1783	{
dvander@203	1784	/* First, see if this is the last character */
dvander@351	1785	if ((unsigned)(aliasptr - alias) == alias_len - 1)
dvander@203	1786	{
dvander@203	1787	/* If so, there's no need to match anything else */
dvander@203	1788	return true;
dvander@203	1789	}
dvander@203	1790	/* Otherwise, we need to search for an appropriate matching sequence in local.
dvander@203	1791	* Note that we only need to search up to the next asterisk.
dvander@203	1792	*/
dvander@203	1793	aliasptr++;
dvander@203	1794	bool match = true;
dvander@203	1795	const char *local_orig = localptr;
dvander@203	1796	do
dvander@203	1797	{
dvander@203	1798	match = true;
dvander@203	1799	while (aliasptr != '\0' && aliasptr != '*')
dvander@203	1800	{
dvander@203	1801	/* Since aliasptr is never '\0', localptr hitting the end will fail */
dvander@203	1802	if (aliasptr != localptr)
dvander@203	1803	{
dvander@203	1804	match = false;
dvander@203	1805	break;
dvander@203	1806	}
dvander@203	1807	aliasptr++;
dvander@203	1808	localptr++;
dvander@203	1809	}
dvander@203	1810	if (!match)
dvander@203	1811	{
dvander@203	1812	/* If we didn't get a match, we need to advance the search stream.
dvander@203	1813	* This will let us skip tokens while still searching for another match.
dvander@203	1814	*/
dvander@203	1815	localptr = ++local_orig;
dvander@203	1816	/* Make sure we don't go out of bounds */
dvander@203	1817	if (*localptr == '\0')
dvander@203	1818	{
dvander@203	1819	break;
dvander@203	1820	}
dvander@203	1821	}
dvander@203	1822	} while (!match);
dvander@203	1823
dvander@203	1824	if (!match)
dvander@203	1825	{
dvander@203	1826	return false;
dvander@203	1827	} else {
dvander@203	1828	/* If we got a match, move on to the next token */
dvander@203	1829	continue;
dvander@203	1830	}
dvander@203	1831	} else if (*aliasptr == '\0') {
dvander@203	1832	if (*localptr == '\0'
dvander@203	1833	\|\|
dvander@203	1834	strcmp(localptr, ".smx") == 0)
dvander@203	1835	{
dvander@203	1836	return true;
dvander@203	1837	} else {
dvander@203	1838	return false;
dvander@203	1839	}
dvander@203	1840	} else if (aliasptr != localptr) {
dvander@203	1841	return false;
dvander@203	1842	}
dvander@203	1843	aliasptr++;
dvander@203	1844	localptr++;
dvander@203	1845	}
dvander@203	1846
dvander@203	1847	return true;
dvander@203	1848	}
dvander@203	1849
faluco@290	1850	bool CPluginManager::IsLateLoadTime() const
dvander@176	1851	{
dvander@315	1852	return (m_AllPluginsLoaded \|\| !g_SourceMod.IsMapLoading());
dvander@176	1853	}
dvander@220	1854
dvander@220	1855	void CPluginManager::OnSourceModAllInitialized()
dvander@220	1856	{
dvander@229	1857	m_MyIdent = g_ShareSys.CreateCoreIdentity();
dvander@229	1858
dvander@262	1859	HandleAccess sec;
dvander@262	1860	g_HandleSys.InitAccessDefaults(NULL, &sec);
dvander@220	1861
dvander@262	1862	sec.access[HandleAccess_Delete] = HANDLE_RESTRICT_IDENTITY;
dvander@262	1863	sec.access[HandleAccess_Clone] = HANDLE_RESTRICT_IDENTITY;
dvander@262	1864
dvander@262	1865	g_PluginType = g_HandleSys.CreateType("Plugin", this, 0, NULL, &sec, m_MyIdent, NULL);
dvander@229	1866	g_PluginIdent = g_ShareSys.CreateIdentType("PLUGIN");
dvander@309	1867
dvander@309	1868	g_RootMenu.AddRootConsoleCommand("plugins", "Manage Plugins", this);
dvander@336	1869
dvander@336	1870	g_ShareSys.AddInterface(NULL, this);
dvander@220	1871	}
dvander@220	1872
dvander@220	1873	void CPluginManager::OnSourceModShutdown()
dvander@220	1874	{
dvander@309	1875	g_RootMenu.RemoveRootConsoleCommand("plugins", this);
dvander@309	1876
dvander@1630	1877	UnloadAll();
dvander@298	1878
dvander@229	1879	g_HandleSys.RemoveType(g_PluginType, m_MyIdent);
dvander@229	1880	g_ShareSys.DestroyIdentType(g_PluginIdent);
dvander@229	1881	g_ShareSys.DestroyIdentity(m_MyIdent);
dvander@220	1882	}
dvander@220	1883
dvander@220	1884	void CPluginManager::OnHandleDestroy(HandleType_t type, void *object)
dvander@220	1885	{
dvander@220	1886	/* We don't care about the internal object, actually */
dvander@220	1887	}
dvander@233	1888
dvander@1766	1889	bool CPluginManager::GetHandleApproxSize(HandleType_t type, void object, unsigned int pSize)
dvander@1766	1890	{
dvander@1766	1891	pSize = ((CPlugin )object)->CalcMemUsage();
dvander@1766	1892	return true;
dvander@1766	1893	}
dvander@1766	1894
dvander@247	1895	IPlugin CPluginManager::PluginFromHandle(Handle_t handle, HandleError err)
dvander@247	1896	{
dvander@247	1897	IPlugin *pPlugin;
dvander@247	1898	HandleError _err;
dvander@247	1899
dvander@262	1900	HandleSecurity sec;
dvander@262	1901
dvander@262	1902	sec.pOwner = NULL;
dvander@262	1903	sec.pIdentity = m_MyIdent;
dvander@262	1904
dvander@262	1905	if ((_err=g_HandleSys.ReadHandle(handle, g_PluginType, &sec, (void **)&pPlugin)) != HandleError_None)
dvander@247	1906	{
dvander@247	1907	pPlugin = NULL;
dvander@247	1908	}
dvander@247	1909
dvander@247	1910	if (err)
dvander@247	1911	{
dvander@247	1912	*err = _err;
dvander@247	1913	}
dvander@247	1914
dvander@247	1915	return pPlugin;
dvander@247	1916	}
faluco@290	1917
faluco@290	1918	CPlugin *CPluginManager::GetPluginByOrder(int num)
faluco@290	1919	{
faluco@290	1920	if (num < 1 \|\| num > (int)GetPluginCount())
faluco@290	1921	{
faluco@290	1922	return NULL;
faluco@290	1923	}
faluco@290	1924
faluco@290	1925	CPlugin *pl;
faluco@290	1926	int id = 1;
faluco@290	1927
ds@2113	1928	SourceHook::List<CPlugin *>::iterator iter;
ds@2113	1929	for (iter = m_plugins.begin(); iter != m_plugins.end() && id < num; iter++, id++) {}
faluco@290	1930
ds@2113	1931	pl = *iter;
faluco@290	1932
faluco@290	1933	return pl;
dvander@298	1934	}
dvander@309	1935
dvander@309	1936	const char *CPluginManager::GetStatusText(PluginStatus st)
dvander@309	1937	{
dvander@309	1938	switch (st)
dvander@309	1939	{
dvander@309	1940	case Plugin_Running:
dvander@309	1941	return "Running";
dvander@309	1942	case Plugin_Paused:
dvander@309	1943	return "Paused";
dvander@309	1944	case Plugin_Error:
dvander@309	1945	return "Error";
dvander@309	1946	case Plugin_Uncompiled:
dvander@309	1947	return "Uncompiled";
dvander@309	1948	case Plugin_BadLoad:
dvander@309	1949	return "Bad Load";
dvander@309	1950	case Plugin_Failed:
dvander@309	1951	return "Failed";
dvander@309	1952	default:
dvander@309	1953	assert(false);
dvander@309	1954	return "-";
dvander@309	1955	}
dvander@309	1956	}
dvander@309	1957
faluco@1541	1958	void CPluginManager::OnRootConsoleCommand(const char *cmdname, const CCommand &command)
dvander@309	1959	{
faluco@1541	1960	int argcount = command.ArgC();
dvander@309	1961	if (argcount >= 3)
dvander@309	1962	{
faluco@1541	1963	const char *cmd = command.Arg(2);
dvander@309	1964	if (strcmp(cmd, "list") == 0)
dvander@309	1965	{
dvander@309	1966	char buffer[256];
dvander@309	1967	unsigned int id = 1;
dvander@309	1968	int plnum = GetPluginCount();
dvander@309	1969
dvander@309	1970	if (!plnum)
dvander@309	1971	{
dvander@309	1972	g_RootMenu.ConsolePrint("[SM] No plugins loaded");
dvander@309	1973	return;
dvander@1630	1974	}
dvander@1630	1975	else
dvander@1630	1976	{
ds@582	1977	g_RootMenu.ConsolePrint("[SM] Listing %d plugin%s:", GetPluginCount(), (plnum > 1) ? "s" : "");
dvander@309	1978	}
dvander@309	1979
dvander@2105	1980	CPlugin *pl;
dvander@2105	1981	SourceHook::List<CPlugin *>::iterator iter;
dvander@2105	1982	SourceHook::List<CPlugin *> m_FailList;
pred@1920	1983
ds@2113	1984	for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++, id++)
dvander@309	1985	{
dvander@2105	1986	pl = (*iter);
dvander@309	1987	assert(pl->GetStatus() != Plugin_Created);
dvander@309	1988	int len = 0;
dvander@309	1989	const sm_plugininfo_t *info = pl->GetPublicInfo();
fyrenmoo@2675	1990	if (pl->GetStatus() != Plugin_Running && !pl->IsSilentlyFailed())
dvander@340	1991	{
dvander@340	1992	len += UTIL_Format(buffer, sizeof(buffer), " %02d <%s>", id, GetStatusText(pl->GetStatus()));
pred@1920	1993
pred@1920	1994	if (pl->GetStatus() <= Plugin_Error)
pred@1920	1995	{
pred@1920	1996	/* Plugin has failed to load. */
pred@1920	1997	m_FailList.push_back(pl);
pred@1920	1998	}
dvander@1630	1999	}
dvander@1630	2000	else
dvander@1630	2001	{
dvander@340	2002	len += UTIL_Format(buffer, sizeof(buffer), " %02d", id);
dvander@340	2003	}
dvander@2342	2004	if (pl->GetStatus() < Plugin_Created)
dvander@309	2005	{
fyrenmoo@2675	2006	if (pl->IsSilentlyFailed())
fyrenmoo@2675	2007	len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " Disabled:");
dvander@2342	2008	len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " \"%s\"", (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
dvander@2342	2009	if (IS_STR_FILLED(info->version))
dvander@2342	2010	{
dvander@2342	2011	len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " (%s)", info->version);
dvander@2342	2012	}
dvander@2342	2013	if (IS_STR_FILLED(info->author))
dvander@2342	2014	{
dvander@2342	2015	UTIL_Format(&buffer[len], sizeof(buffer)-len, " by %s", info->author);
dvander@2342	2016	}
dvander@309	2017	}
dvander@2342	2018	else
dvander@309	2019	{
dvander@2342	2020	UTIL_Format(&buffer[len], sizeof(buffer)-len, " %s", pl->m_filename);
dvander@309	2021	}
dvander@309	2022	g_RootMenu.ConsolePrint("%s", buffer);
dvander@309	2023	}
dvander@309	2024
pred@1920	2025	if (!m_FailList.empty())
pred@1920	2026	{
pred@1920	2027	g_RootMenu.ConsolePrint("Load Errors:");
pred@1920	2028
dvander@2105	2029	SourceHook::List<CPlugin *>::iterator _iter;
pred@1920	2030
pred@1920	2031	CPlugin *pl;
pred@1920	2032
pred@1920	2033	for (_iter=m_FailList.begin(); _iter!=m_FailList.end(); _iter++)
pred@1920	2034	{
pred@1920	2035	pl = (CPlugin )_iter;
pred@1920	2036	g_RootMenu.ConsolePrint("%s: %s",(IS_STR_FILLED(pl->GetPublicInfo()->name)) ? pl->GetPublicInfo()->name : pl->GetFilename(), pl->m_errormsg);
pred@1920	2037	}
pred@1920	2038	}
pred@1920	2039
dvander@309	2040	return;
dvander@1630	2041	}
dvander@1630	2042	else if (strcmp(cmd, "load") == 0)
dvander@1630	2043	{
dvander@309	2044	if (argcount < 4)
dvander@309	2045	{
dvander@310	2046	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins load <file>");
dvander@309	2047	return;
dvander@309	2048	}
dvander@309	2049
dvander@309	2050	char error[128];
faluco@383	2051	bool wasloaded;
faluco@1541	2052	const char *filename = command.Arg(3);
ds@794	2053
ds@794	2054	char pluginfile[256];
ds@794	2055	const char *ext = g_LibSys.GetFileExtension(filename) ? "" : ".smx";
fyrenmoo@2869	2056	g_SourceMod.BuildPath(Path_None, pluginfile, sizeof(pluginfile), "%s%s", filename, ext);
ds@794	2057
ds@794	2058	IPlugin *pl = LoadPlugin(pluginfile, false, PluginType_MapUpdated, error, sizeof(error), &wasloaded);
faluco@383	2059
faluco@383	2060	if (wasloaded)
faluco@383	2061	{
ds@794	2062	g_RootMenu.ConsolePrint("[SM] Plugin %s is already loaded.", pluginfile);
faluco@383	2063	return;
faluco@383	2064	}
dvander@309	2065
dvander@309	2066	if (pl)
dvander@309	2067	{
ds@794	2068	g_RootMenu.ConsolePrint("[SM] Loaded plugin %s successfully.", pluginfile);
dvander@1630	2069	}
dvander@1630	2070	else
dvander@1630	2071	{
ds@794	2072	g_RootMenu.ConsolePrint("[SM] Plugin %s failed to load: %s.", pluginfile, error);
dvander@309	2073	}
dvander@309	2074
dvander@309	2075	return;
dvander@1630	2076	}
dvander@1630	2077	else if (strcmp(cmd, "unload") == 0)
dvander@1630	2078	{
dvander@309	2079	if (argcount < 4)
dvander@309	2080	{
ds@1252	2081	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins unload <#\|file>");
dvander@309	2082	return;
dvander@309	2083	}
dvander@309	2084
ds@1252	2085	CPlugin *pl;
ds@1252	2086	char *end;
faluco@1541	2087	const char *arg = command.Arg(3);
ds@1252	2088	int id = strtol(arg, &end, 10);
ds@1252	2089
ds@1252	2090	if (*end == '\0')
dvander@309	2091	{
ds@1252	2092	pl = GetPluginByOrder(id);
ds@1252	2093	if (!pl)
ds@1252	2094	{
ds@1252	2095	g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
ds@1252	2096	return;
ds@1252	2097	}
dvander@1630	2098	}
dvander@1630	2099	else
dvander@1630	2100	{
ds@1252	2101	char pluginfile[256];
ds@1252	2102	const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
fyrenmoo@2869	2103	g_SourceMod.BuildPath(Path_None, pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
dvander@309	2104
ds@1252	2105	if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
ds@1252	2106	{
ds@1252	2107	g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
ds@1252	2108	return;
ds@1252	2109	}
ds@1252	2110	}
dvander@309	2111
ds@652	2112	char name[PLATFORM_MAX_PATH];
dvander@2342	2113
dvander@2342	2114	if (pl->GetStatus() < Plugin_Created)
dvander@2342	2115	{
dvander@2342	2116	const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@2342	2117	UTIL_Format(name, sizeof(name), (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
dvander@2342	2118	}
dvander@2342	2119	else
dvander@2342	2120	{
dvander@2342	2121	UTIL_Format(name, sizeof(name), "%s", pl->GetFilename());
dvander@2342	2122	}
dvander@309	2123
dvander@309	2124	if (UnloadPlugin(pl))
dvander@309	2125	{
dvander@310	2126	g_RootMenu.ConsolePrint("[SM] Plugin %s unloaded successfully.", name);
dvander@1630	2127	}
dvander@1630	2128	else
dvander@1630	2129	{
dvander@310	2130	g_RootMenu.ConsolePrint("[SM] Failed to unload plugin %s.", name);
dvander@309	2131	}
dvander@309	2132
dvander@309	2133	return;
dvander@1630	2134	}
dvander@1630	2135	else if (strcmp(cmd, "unload_all") == 0)
dvander@1630	2136	{
dvander@1630	2137	g_PluginSys.UnloadAll();
dvander@1630	2138	g_RootMenu.ConsolePrint("[SM] All plugins have been unloaded.");
dvander@1630	2139	return;
dvander@1630	2140	}
dvander@1630	2141	else if (strcmp(cmd, "load_lock") == 0)
dvander@1630	2142	{
dvander@1630	2143	if (m_LoadingLocked)
dvander@1630	2144	{
dvander@1630	2145	g_RootMenu.ConsolePrint("[SM] There is already a loading lock in effect.");
dvander@1630	2146	}
dvander@1630	2147	else
dvander@1630	2148	{
dvander@1630	2149	m_LoadingLocked = true;
dvander@1630	2150	g_RootMenu.ConsolePrint("[SM] Loading is now locked; no plugins will be loaded or re-loaded.");
dvander@1630	2151	}
dvander@1630	2152	return;
dvander@1630	2153	}
dvander@1630	2154	else if (strcmp(cmd, "load_unlock") == 0)
dvander@1630	2155	{
dvander@1630	2156	if (m_LoadingLocked)
dvander@1630	2157	{
dvander@1630	2158	m_LoadingLocked = false;
dvander@1630	2159	g_RootMenu.ConsolePrint("[SM] The loading lock is no longer in effect.");
dvander@1630	2160	}
dvander@1630	2161	else
dvander@1630	2162	{
dvander@1630	2163	g_RootMenu.ConsolePrint("[SM] There was no loading lock in effect.");
dvander@1630	2164	}
dvander@1630	2165	return;
dvander@1630	2166	}
dvander@1630	2167	else if (strcmp(cmd, "info") == 0)
dvander@1630	2168	{
dvander@309	2169	if (argcount < 4)
dvander@309	2170	{
dvander@310	2171	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins info <#>");
dvander@309	2172	return;
dvander@309	2173	}
dvander@309	2174
dvander@1902	2175	CPlugin *pl;
dvander@1902	2176	char *end;
dvander@1902	2177	const char *arg = command.Arg(3);
dvander@1902	2178	int id = strtol(arg, &end, 10);
dvander@1902	2179
dvander@1902	2180	if (*end == '\0')
dvander@309	2181	{
dvander@1902	2182	pl = GetPluginByOrder(id);
dvander@1902	2183	if (!pl)
dvander@1902	2184	{
dvander@1902	2185	g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
dvander@1902	2186	return;
dvander@1902	2187	}
dvander@1902	2188	}
dvander@1902	2189	else
dvander@1902	2190	{
dvander@1902	2191	char pluginfile[256];
dvander@1902	2192	const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
fyrenmoo@2869	2193	g_SourceMod.BuildPath(Path_None, pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
dvander@1902	2194
dvander@1902	2195	if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
dvander@1902	2196	{
dvander@1902	2197	g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
dvander@1902	2198	return;
dvander@1902	2199	}
dvander@309	2200	}
dvander@309	2201
dvander@309	2202	const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@309	2203
dvander@309	2204	g_RootMenu.ConsolePrint(" Filename: %s", pl->GetFilename());
dvander@315	2205	if (pl->GetStatus() <= Plugin_Error)
dvander@309	2206	{
dvander@315	2207	if (IS_STR_FILLED(info->name))
dvander@315	2208	{
dvander@605	2209	if (IS_STR_FILLED(info->description))
dvander@605	2210	{
dvander@605	2211	g_RootMenu.ConsolePrint(" Title: %s (%s)", info->name, info->description);
dvander@605	2212	} else {
dvander@605	2213	g_RootMenu.ConsolePrint(" Title: %s", info->name);
dvander@605	2214	}
dvander@605	2215	}
dvander@605	2216	if (IS_STR_FILLED(info->author))
dvander@605	2217	{
dvander@605	2218	g_RootMenu.ConsolePrint(" Author: %s", info->author);
dvander@315	2219	}
dvander@315	2220	if (IS_STR_FILLED(info->version))
dvander@315	2221	{
dvander@315	2222	g_RootMenu.ConsolePrint(" Version: %s", info->version);
dvander@315	2223	}
dvander@315	2224	if (IS_STR_FILLED(info->url))
dvander@315	2225	{
dvander@315	2226	g_RootMenu.ConsolePrint(" URL: %s", info->url);
dvander@315	2227	}
dvander@605	2228	if (pl->GetStatus() == Plugin_Error)
dvander@315	2229	{
dvander@605	2230	g_RootMenu.ConsolePrint(" Error: %s", pl->m_errormsg);
dvander@1630	2231	}
dvander@1630	2232	else
dvander@1630	2233	{
dvander@2105	2234	if (pl->GetStatus() == Plugin_Running)
dvander@2105	2235	{
dvander@2226	2236	g_RootMenu.ConsolePrint(" Status: running");
dvander@2105	2237	}
dvander@2105	2238	else
dvander@2105	2239	{
dvander@2105	2240	g_RootMenu.ConsolePrint(" Status: not running");
dvander@2105	2241	}
ds@860	2242
ds@861	2243	const char *typestr = "";
ds@860	2244	switch (pl->GetType())
ds@860	2245	{
ds@860	2246	case PluginType_MapUpdated:
ds@860	2247	typestr = "Map Change if Updated";
ds@860	2248	break;
ds@860	2249	case PluginType_MapOnly:
ds@860	2250	typestr = "Map Change";
ds@860	2251	break;
ds@860	2252	case PluginType_Private:
ds@860	2253	case PluginType_Global:
ds@860	2254	typestr = "Never";
ds@860	2255	break;
ds@860	2256	}
ds@860	2257
ds@860	2258	g_RootMenu.ConsolePrint(" Reloads: %s", typestr);
dvander@315	2259	}
dvander@2105	2260	if (pl->m_FileVersion >= 3)
dvander@2105	2261	{
dvander@2105	2262	g_RootMenu.ConsolePrint(" Timestamp: %s", pl->m_DateTime);
dvander@2105	2263	}
dvander@1630	2264	}
dvander@1630	2265	else
dvander@1630	2266	{
dvander@315	2267	g_RootMenu.ConsolePrint(" Load error: %s", pl->m_errormsg);
dvander@2342	2268	if (pl->GetStatus() < Plugin_Created)
dvander@315	2269	{
dvander@2342	2270	g_RootMenu.ConsolePrint(" File info: (title \"%s\") (version \"%s\")",
dvander@2342	2271	info->name ? info->name : "<none>",
dvander@2342	2272	info->version ? info->version : "<none>");
dvander@2342	2273	if (IS_STR_FILLED(info->url))
dvander@2342	2274	{
dvander@2342	2275	g_RootMenu.ConsolePrint(" File URL: %s", info->url);
dvander@2342	2276	}
dvander@315	2277	}
dvander@309	2278	}
dvander@309	2279
dvander@309	2280	return;
dvander@1630	2281	}
dvander@1630	2282	else if (strcmp(cmd, "refresh") == 0)
dvander@1630	2283	{
dvander@1630	2284	g_SourceMod.DoGlobalPluginLoads();
dvander@1630	2285	g_RootMenu.ConsolePrint("[SM] The plugin list has been refreshed and reloaded.");
dvander@1630	2286	return;
dvander@1630	2287	}
dvander@1630	2288	else if (strcmp(cmd, "reload") == 0)
dvander@1630	2289	{
faluco@1348	2290	if (argcount < 4)
faluco@1348	2291	{
faluco@1348	2292	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins reload <#\|file>");
faluco@1348	2293	return;
faluco@1348	2294	}
faluco@1348	2295
faluco@1348	2296	CPlugin *pl;
faluco@1348	2297	char *end;
faluco@1541	2298	const char *arg = command.Arg(3);
faluco@1348	2299	int id = strtol(arg, &end, 10);
faluco@1348	2300
faluco@1348	2301	if (*end == '\0')
faluco@1348	2302	{
faluco@1348	2303	pl = GetPluginByOrder(id);
faluco@1348	2304	if (!pl)
faluco@1348	2305	{
faluco@1348	2306	g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
faluco@1348	2307	return;
faluco@1348	2308	}
dvander@1630	2309	}
dvander@1630	2310	else
dvander@1630	2311	{
faluco@1348	2312	char pluginfile[256];
faluco@1348	2313	const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
fyrenmoo@2869	2314	g_SourceMod.BuildPath(Path_None, pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
faluco@1348	2315
faluco@1348	2316	if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
faluco@1348	2317	{
faluco@1348	2318	g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
faluco@1348	2319	return;
faluco@1348	2320	}
faluco@1348	2321	}
faluco@1348	2322
faluco@1348	2323	char name[PLATFORM_MAX_PATH];
faluco@1348	2324	const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@2886	2325	if (pl->GetStatus() <= Plugin_Paused)
dvander@2885	2326	strcpy(name, (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
dvander@2885	2327	else
dvander@2885	2328	strcpy(name, pl->GetFilename());
faluco@1348	2329
faluco@1348	2330	if (ReloadPlugin(pl))
faluco@1348	2331	{
faluco@1348	2332	g_RootMenu.ConsolePrint("[SM] Plugin %s reloaded successfully.", name);
dvander@1630	2333	}
dvander@1630	2334	else
dvander@1630	2335	{
faluco@1348	2336	g_RootMenu.ConsolePrint("[SM] Failed to reload plugin %s.", name);
faluco@1348	2337	}
faluco@1348	2338
faluco@1348	2339	return;
dvander@309	2340	}
dvander@309	2341	}
dvander@309	2342
dvander@309	2343	/* Draw the main menu */
dvander@309	2344	g_RootMenu.ConsolePrint("SourceMod Plugins Menu:");
dvander@1630	2345	g_RootMenu.DrawGenericOption("info", "Information about a plugin");
dvander@309	2346	g_RootMenu.DrawGenericOption("list", "Show loaded plugins");
dvander@309	2347	g_RootMenu.DrawGenericOption("load", "Load a plugin");
dvander@1630	2348	g_RootMenu.DrawGenericOption("load_lock", "Prevents any more plugins from being loaded");
dvander@1630	2349	g_RootMenu.DrawGenericOption("load_unlock", "Re-enables plugin loading");
dvander@1630	2350	g_RootMenu.DrawGenericOption("refresh", "Reloads/refreshes all plugins in the plugins folder");
dvander@1630	2351	g_RootMenu.DrawGenericOption("reload", "Reloads a plugin");
dvander@309	2352	g_RootMenu.DrawGenericOption("unload", "Unload a plugin");
dvander@1630	2353	g_RootMenu.DrawGenericOption("unload_all", "Unloads all plugins");
faluco@1348	2354	}
faluco@1348	2355
faluco@1348	2356	bool CPluginManager::ReloadPlugin(CPlugin *pl)
faluco@1348	2357	{
faluco@1348	2358	List<CPlugin *>::iterator iter;
faluco@1348	2359	char filename[PLATFORM_MAX_PATH];
dvander@2226	2360	bool wasloaded;
faluco@1348	2361	PluginType ptype;
faluco@1348	2362	IPlugin *newpl;
faluco@1348	2363	int id = 1;
faluco@1348	2364
faluco@1348	2365	strcpy(filename, pl->m_filename);
faluco@1348	2366	ptype = pl->GetType();
faluco@1348	2367
faluco@1348	2368	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++, id++)
faluco@1348	2369	{
faluco@1348	2370	if ((*iter) == pl)
faluco@1348	2371	{
faluco@1348	2372	break;
faluco@1348	2373	}
faluco@1348	2374	}
faluco@1348	2375
faluco@1348	2376	if (!UnloadPlugin(pl))
faluco@1348	2377	{
faluco@1348	2378	return false;
faluco@1348	2379	}
dvander@2226	2380	if (!(newpl=LoadPlugin(filename, true, ptype, NULL, 0, &wasloaded)))
faluco@1348	2381	{
faluco@1348	2382	return false;
faluco@1348	2383	}
faluco@1348	2384
faluco@1348	2385	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
faluco@1348	2386	{
faluco@1348	2387	if ((iter) == (CPlugin )newpl)
faluco@1348	2388	{
faluco@1348	2389	m_plugins.erase(iter);
faluco@1348	2390	break;
faluco@1348	2391	}
faluco@1348	2392	}
faluco@1348	2393
faluco@1348	2394	int i;
dvander@1630	2395	for (i=1, iter=m_plugins.begin(); iter!=m_plugins.end() && i<id; iter++, i++)
dvander@1630	2396	{
dvander@1630	2397	}
faluco@1348	2398	m_plugins.insert(iter, (CPlugin *)newpl);
faluco@1348	2399
faluco@1348	2400	return true;
dvander@309	2401	}
faluco@333	2402
faluco@333	2403	void CPluginManager::ReloadOrUnloadPlugins()
faluco@333	2404	{
dvander@1630	2405	/* If we're in a load lock, just skip this whole bit. */
dvander@1630	2406	if (m_LoadingLocked)
dvander@1630	2407	{
dvander@1630	2408	return;
dvander@1630	2409	}
dvander@1630	2410
faluco@333	2411	List<CPlugin *>::iterator iter;
faluco@333	2412	List<CPlugin *> tmp_list = m_plugins;
faluco@333	2413	CPlugin *pl;
faluco@333	2414	time_t t;
faluco@333	2415
faluco@333	2416	for (iter=tmp_list.begin(); iter!=tmp_list.end(); iter++)
faluco@333	2417	{
faluco@333	2418	pl = (*iter);
faluco@333	2419	if (pl->GetType() == PluginType_MapOnly)
faluco@333	2420	{
faluco@333	2421	UnloadPlugin((IPlugin *)pl);
dvander@1630	2422	}
dvander@1630	2423	else if (pl->GetType() == PluginType_MapUpdated)
dvander@1630	2424	{
dvander@450	2425	t = pl->GetFileTimeStamp();
dvander@450	2426	if (!t \|\| t > pl->GetTimeStamp())
faluco@333	2427	{
faluco@333	2428	pl->SetTimeStamp(t);
faluco@333	2429	UnloadPlugin((IPlugin *)pl);
faluco@333	2430	}
faluco@333	2431	}
faluco@333	2432	}
faluco@333	2433	}
faluco@430	2434
faluco@430	2435	void CPluginManager::_SetPauseState(CPlugin *pl, bool paused)
faluco@430	2436	{
faluco@430	2437	List<IPluginsListener *>::iterator iter;
faluco@430	2438	IPluginsListener *pListener;
faluco@430	2439	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
faluco@430	2440	{
faluco@430	2441	pListener = (*iter);
faluco@430	2442	pListener->OnPluginPauseChange(pl, paused);
faluco@430	2443	}
dvander@464	2444	}
dvander@464	2445
ds@611	2446	void CPluginManager::AddFunctionsToForward(const char name, IChangeableForward pForward)
ds@611	2447	{
ds@611	2448	List<CPlugin *>::iterator iter;
ds@611	2449	CPlugin *pPlugin;
ds@611	2450	IPluginFunction *pFunction;
ds@611	2451
ds@611	2452	for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++)
ds@611	2453	{
ds@611	2454	pPlugin = (*iter);
ds@611	2455
ds@611	2456	if (pPlugin->GetStatus() <= Plugin_Paused)
ds@611	2457	{
ds@611	2458	pFunction = pPlugin->GetBaseContext()->GetFunctionByName(name);
ds@611	2459
ds@611	2460	if (pFunction)
ds@611	2461	{
ds@611	2462	pForward->AddFunction(pFunction);
ds@611	2463	}
ds@611	2464	}
ds@611	2465	}
ds@611	2466	}
dvander@900	2467
dvander@900	2468	CPlugin CPluginManager::GetPluginFromIdentity(IdentityToken_t pToken)
dvander@900	2469	{
dvander@900	2470	if (pToken->type != g_PluginIdent)
dvander@900	2471	{
dvander@900	2472	return NULL;
dvander@900	2473	}
dvander@900	2474
dvander@900	2475	return (CPlugin *)(pToken->ptr);
dvander@900	2476	}
dvander@978	2477
dvander@1572	2478	void CPluginManager::OnLibraryAction(const char *lib, bool is_a_plugin, bool drop)
dvander@978	2479	{
dvander@978	2480	List<CPlugin *>::iterator iter;
dvander@978	2481
dvander@1124	2482	struct _pl
dvander@1124	2483	{
dvander@1124	2484	cell_t name;
dvander@1124	2485	cell_t file;
dvander@1124	2486	cell_t required;
dvander@1124	2487	} *plc;
dvander@1124	2488
dvander@1572	2489	struct _ext
dvander@1572	2490	{
dvander@1572	2491	cell_t name;
dvander@1572	2492	cell_t file;
dvander@1572	2493	cell_t autoload;
dvander@1572	2494	cell_t required;
dvander@1572	2495	} *ext;
dvander@1572	2496
dvander@1124	2497	const char *name = drop ? "OnLibraryRemoved" : "OnLibraryAdded";
dvander@1124	2498
dvander@1124	2499	for (iter=m_plugins.begin();
dvander@1124	2500	iter!=m_plugins.end();
dvander@1124	2501	iter++)
dvander@1124	2502	{
dvander@1124	2503	CPlugin pl = (iter);
dvander@1124	2504	if (pl->GetStatus() != Plugin_Running)
dvander@1124	2505	{
dvander@1124	2506	continue;
dvander@1124	2507	}
dvander@1124	2508	IPluginContext *pContext = pl->GetBaseContext();
dvander@1124	2509	IPluginFunction *pf = pContext->GetFunctionByName(name);
dvander@1124	2510	if (!pf)
dvander@1124	2511	{
dvander@1124	2512	continue;
dvander@1124	2513	}
dvander@1124	2514	uint32_t num_vars = pContext->GetPubVarsNum();
dvander@1124	2515	for (uint32_t i=0; i<num_vars; i++)
dvander@1124	2516	{
dvander@1124	2517	sp_pubvar_t *pubvar;
dvander@1124	2518	if (pContext->GetPubvarByIndex(i, &pubvar) != SP_ERROR_NONE)
dvander@1124	2519	{
dvander@1124	2520	continue;
dvander@1124	2521	}
dvander@1572	2522	if (is_a_plugin && strncmp(pubvar->name, "__pl_", 5) == 0)
dvander@1124	2523	{
dvander@1572	2524	plc = (_pl *)pubvar->offs;
dvander@1572	2525	if (plc->required)
dvander@1572	2526	{
dvander@1572	2527	continue;
dvander@1572	2528	}
dvander@1572	2529	char *str;
dvander@1572	2530	pContext->LocalToString(plc->name, &str);
dvander@1572	2531	if (strcmp(str, lib) != 0)
dvander@1572	2532	{
dvander@1572	2533	continue;
dvander@1572	2534	}
dvander@1572	2535	pf->PushString(lib);
dvander@1572	2536	pf->Execute(NULL);
dvander@1124	2537	}
dvander@1572	2538	else if (!is_a_plugin && strncmp(pubvar->name, "__ext_", 6) == 0)
dvander@1124	2539	{
dvander@1572	2540	ext = (_ext *)pubvar->offs;
dvander@1572	2541	if (ext->required)
dvander@1572	2542	{
dvander@1572	2543	continue;
dvander@1572	2544	}
dvander@1572	2545	char *str;
dvander@1572	2546	pContext->LocalToString(ext->name, &str);
dvander@1572	2547	if (strcmp(str, lib) != 0)
dvander@1572	2548	{
dvander@1572	2549	continue;
dvander@1572	2550	}
dvander@1572	2551	pf->PushString(lib);
dvander@1572	2552	pf->Execute(NULL);
dvander@1124	2553	}
dvander@1124	2554	}
dvander@1124	2555	}
dvander@978	2556	}
dvander@1124	2557
dvander@1124	2558	bool CPluginManager::LibraryExists(const char *lib)
dvander@1124	2559	{
dvander@1124	2560	List<CPlugin *>::iterator iter;
dvander@1124	2561
dvander@1124	2562	for (iter=m_plugins.begin();
dvander@1124	2563	iter!=m_plugins.end();
dvander@1124	2564	iter++)
dvander@1124	2565	{
dvander@1124	2566	CPlugin pl = (iter);
dvander@1124	2567	if (pl->GetStatus() != Plugin_Running)
dvander@1124	2568	{
dvander@1124	2569	continue;
dvander@1124	2570	}
dvander@1124	2571	List<String>::iterator s_iter;
dvander@1124	2572	for (s_iter = pl->m_Libraries.begin();
dvander@1124	2573	s_iter != pl->m_Libraries.end();
dvander@1124	2574	s_iter++)
dvander@1124	2575	{
dvander@1124	2576	if ((*s_iter).compare(lib) == 0)
dvander@1124	2577	{
dvander@1124	2578	return true;
dvander@1124	2579	}
dvander@1124	2580	}
dvander@1124	2581	}
dvander@1124	2582
dvander@1124	2583	return false;
dvander@1124	2584	}
faluco@1221	2585
faluco@1221	2586	void CPluginManager::AllPluginsLoaded()
faluco@1221	2587	{
faluco@1221	2588	List<CPlugin *>::iterator iter;
faluco@1221	2589	CPlugin *pl;
faluco@1221	2590
faluco@1221	2591	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
faluco@1221	2592	{
faluco@1221	2593	pl = (*iter);
faluco@1221	2594	pl->Call_OnAllPluginsLoaded();
faluco@1221	2595	}
dvander@1232	2596	}
dvander@1630	2597
dvander@1630	2598	void CPluginManager::UnloadAll()
dvander@1630	2599	{
dvander@1630	2600	List<CPlugin *>::iterator iter;
dvander@1630	2601	while ( (iter = m_plugins.begin()) != m_plugins.end() )
dvander@1630	2602	{
dvander@1630	2603	UnloadPlugin((*iter));
dvander@1630	2604	}
dvander@1630	2605	}
dvander@1638	2606
dvander@1638	2607	int CPluginManager::GetOrderOfPlugin(IPlugin *pl)
dvander@1638	2608	{
dvander@1638	2609	int id = 1;
dvander@1638	2610	List<CPlugin *>::iterator iter;
dvander@1638	2611
dvander@1638	2612	for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++, id++)
dvander@1638	2613	{
dvander@1638	2614	if ((*iter) == pl)
dvander@1638	2615	{
dvander@1638	2616	return id;
dvander@1638	2617	}
dvander@1638	2618	}
dvander@1638	2619
dvander@1638	2620	return -1;
dvander@1638	2621	}
dvander@1638	2622
dvander@1902	2623	CPlugin CPluginManager::FindPluginByConsoleArg(const char arg)
dvander@1902	2624	{
dvander@1902	2625	int id;
dvander@1902	2626	char *end;
dvander@1902	2627	CPlugin *pl;
dvander@1902	2628
dvander@1902	2629	id = strtol(arg, &end, 10);
dvander@1902	2630
dvander@1902	2631	if (*end == '\0')
dvander@1902	2632	{
dvander@1902	2633	pl = GetPluginByOrder(id);
dvander@1902	2634	if (!pl)
dvander@1902	2635	{
dvander@1902	2636	return NULL;
dvander@1902	2637	}
dvander@1902	2638	}
dvander@1902	2639	else
dvander@1902	2640	{
dvander@1902	2641	char pluginfile[256];
dvander@1902	2642	const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
dvander@1902	2643	UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
dvander@1902	2644
dvander@1902	2645	if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
dvander@1902	2646	{
dvander@1902	2647	return NULL;
dvander@1902	2648	}
dvander@1902	2649	}
dvander@1902	2650
dvander@1902	2651	return pl;
dvander@1902	2652	}
dvander@2342	2653
dvander@2342	2654	void CPluginManager::OnSourceModMaxPlayersChanged(int newvalue)
dvander@2342	2655	{
dvander@2342	2656	SyncMaxClients(newvalue);
dvander@2342	2657	}
dvander@2342	2658
dvander@2342	2659	void CPluginManager::SyncMaxClients(int max_clients)
dvander@2342	2660	{
dvander@2342	2661	List<CPlugin *>::iterator iter;
dvander@2342	2662
dvander@2342	2663	for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++)
dvander@2342	2664	{
dvander@2342	2665	(*iter)->SyncMaxClients(max_clients);
dvander@2342	2666	}
dvander@2342	2667	}
dvander@2547	2668
dvander@2547	2669	void CPluginManager::ListPluginsToClient(CPlayer *player, const CCommand &args)
dvander@2547	2670	{
dvander@2547	2671	char buffer[256];
dvander@2547	2672	unsigned int id = 0;
dvander@2547	2673	int plnum = GetPluginCount();
dvander@2547	2674	edict_t *e = player->GetEdict();
dvander@2547	2675	unsigned int start = 0;
dvander@2547	2676
dvander@2547	2677	if (!plnum)
dvander@2547	2678	{
dvander@2547	2679	ClientConsolePrint(e, "[SM] No plugins found.");
dvander@2547	2680	return;
dvander@2547	2681	}
dvander@2547	2682
dvander@2547	2683	if (args.ArgC() > 2)
dvander@2547	2684	{
dvander@2547	2685	start = atoi(args.Arg(2));
dvander@2547	2686	}
dvander@2547	2687
dvander@2547	2688	CPlugin *pl;
dvander@2547	2689	SourceHook::List<CPlugin *>::iterator iter;
dvander@2547	2690	SourceHook::List<CPlugin *> m_FailList;
dvander@2547	2691
dvander@2547	2692	for (iter = m_plugins.begin();
dvander@2547	2693	iter != m_plugins.end();
dvander@2547	2694	iter++)
dvander@2547	2695	{
dvander@2547	2696	pl = (*iter);
dvander@2547	2697
dvander@2547	2698	if (pl->GetStatus() != Plugin_Running)
dvander@2547	2699	{
dvander@2547	2700	continue;
dvander@2547	2701	}
dvander@2547	2702
dvander@2547	2703	/* Count valid plugins */
dvander@2547	2704	id++;
dvander@2547	2705	if (id < start)
dvander@2547	2706	{
dvander@2547	2707	continue;
dvander@2547	2708	}
dvander@2547	2709
dvander@2547	2710	if (id - start > 10)
dvander@2547	2711	{
dvander@2547	2712	break;
dvander@2547	2713	}
dvander@2547	2714
dvander@2547	2715	size_t len;
dvander@2547	2716	const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@2547	2717	len = UTIL_Format(buffer, sizeof(buffer), " \"%s\"", (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
dvander@2547	2718	if (IS_STR_FILLED(info->version))
dvander@2547	2719	{
dvander@2547	2720	len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " (%s)", info->version);
dvander@2547	2721	}
dvander@2547	2722	if (IS_STR_FILLED(info->author))
dvander@2547	2723	{
dvander@2547	2724	UTIL_Format(&buffer[len], sizeof(buffer)-len, " by %s", info->author);
dvander@2547	2725	}
dvander@2547	2726	else
dvander@2547	2727	{
dvander@2547	2728	UTIL_Format(&buffer[len], sizeof(buffer)-len, " %s", pl->m_filename);
dvander@2547	2729	}
dvander@2547	2730	ClientConsolePrint(e, "%s", buffer);
dvander@2547	2731	}
dvander@2547	2732
dvander@2547	2733	/* See if we can get more plugins */
dvander@2547	2734	while (iter != m_plugins.end())
dvander@2547	2735	{
dvander@2547	2736	if ((*iter)->GetStatus() == Plugin_Running)
dvander@2547	2737	{
dvander@2547	2738	break;
dvander@2547	2739	}
dvander@2547	2740	}
dvander@2547	2741
dvander@2547	2742	/* Do we actually have more plugins? */
dvander@2547	2743	if (iter != m_plugins.end())
dvander@2547	2744	{
fyrenmoo@3129	2745	ClientConsolePrint(e, "To see more, type \"sm plugins %d\"", id);
dvander@2547	2746	}
dvander@2547	2747	}

author	Nicholas Hastings <skamonkey@gmail.com>
	Fri Nov 04 10:01:33 2011 -0400 (2011-11-04)
changeset 3438	07a1b8a8a2ba
parent 3402	a459183ce707
child 3507	d1b6fde18611
permissions	-rw-r--r--