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

author	Scott Ehlert <ds@alliedmods.net>
	Thu Mar 27 04:54:56 2008 +0000 (2008-03-27)
changeset 1947	e73a1e51833b
parent 1920	e6e470b49d10
child 2289	44045ead61cd
permissions	-rw-r--r--