sourcemod-central: core/systems/PluginSys.cpp@d30c252089be (annotated)

dvander@370	1	/**
ds@653	2	* vim: set ts=4 :
ds@1330	3	* =============================================================================
dvander@1231	4	* SourceMod
dvander@1231	5	* Copyright (C) 2004-2007 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@978	802	AutoConfig *c = new AutoConfig;
dvander@978	803
dvander@978	804	c->autocfg = cfg;
dvander@978	805	c->folder = folder;
dvander@978	806	c->create = autoCreate;
dvander@978	807
dvander@978	808	m_configs.push_back(c);
dvander@978	809	}
dvander@978	810
dvander@174	811	/*******************
dvander@174	812	* PLUGIN ITERATOR *
dvander@174	813	*******************/
dvander@190	814
dvander@213	815	CPluginManager::CPluginIterator::CPluginIterator(List<CPlugin > _mylist)
dvander@174	816	{
dvander@174	817	mylist = _mylist;
faluco@216	818	Reset();
dvander@174	819	}
dvander@174	820
dvander@174	821	IPlugin *CPluginManager::CPluginIterator::GetPlugin()
dvander@174	822	{
dvander@174	823	return (*current);
dvander@174	824	}
dvander@174	825
dvander@174	826	bool CPluginManager::CPluginIterator::MorePlugins()
dvander@174	827	{
dvander@174	828	return (current != mylist->end());
dvander@174	829	}
dvander@174	830
dvander@174	831	void CPluginManager::CPluginIterator::NextPlugin()
dvander@174	832	{
dvander@174	833	current++;
dvander@174	834	}
dvander@174	835
dvander@174	836	void CPluginManager::CPluginIterator::Release()
dvander@174	837	{
dvander@211	838	g_PluginSys.ReleaseIterator(this);
dvander@174	839	}
dvander@174	840
dvander@174	841	CPluginManager::CPluginIterator::~CPluginIterator()
dvander@174	842	{
dvander@174	843	}
dvander@174	844
dvander@174	845	void CPluginManager::CPluginIterator::Reset()
dvander@174	846	{
dvander@174	847	current = mylist->begin();
dvander@174	848	}
dvander@174	849
dvander@174	850	/******************
dvander@174	851	* PLUGIN MANAGER *
dvander@174	852	******************/
dvander@174	853
dvander@211	854	CPluginManager::CPluginManager()
dvander@211	855	{
dvander@211	856	m_LoadLookup = sm_trie_create();
dvander@211	857	m_AllPluginsLoaded = false;
dvander@229	858	m_MyIdent = NULL;
dvander@601	859	m_pNativeLookup = sm_trie_create();
dvander@1382	860	m_pCoreNatives = sm_trie_create();
dvander@1630	861	m_LoadingLocked = false;
dvander@211	862	}
dvander@211	863
dvander@211	864	CPluginManager::~CPluginManager()
dvander@211	865	{
dvander@282	866	/* :NOTICE:
dvander@282	867	* Ignore the fact that there might be plugins in the cache.
dvander@282	868	* This usually means that Core is not being unloaded properly, and everything
dvander@282	869	* will crash anyway. YAY
dvander@282	870	*/
dvander@211	871	sm_trie_destroy(m_LoadLookup);
dvander@601	872	sm_trie_destroy(m_pNativeLookup);
dvander@1382	873	sm_trie_destroy(m_pCoreNatives);
faluco@641	874
faluco@641	875	CStack<CPluginManager::CPluginIterator *>::iterator iter;
faluco@641	876	for (iter=m_iters.begin(); iter!=m_iters.end(); iter++)
faluco@641	877	{
faluco@641	878	delete (*iter);
faluco@641	879	}
faluco@641	880	m_iters.popall();
dvander@211	881	}
dvander@211	882
dvander@1018	883	void CPluginManager::Shutdown()
dvander@1018	884	{
dvander@1018	885	List<CPlugin *>::iterator iter;
dvander@1018	886
dvander@1018	887	while ((iter = m_plugins.begin()) != m_plugins.end())
dvander@1018	888	{
dvander@1018	889	UnloadPlugin(*iter);
dvander@1018	890	}
dvander@1018	891	}
dvander@1018	892
dvander@213	893	void CPluginManager::LoadAll_FirstPass(const char config, const char basedir)
dvander@190	894	{
dvander@203	895	/* First read in the database of plugin settings */
dvander@1656	896	SMCError err;
dvander@1656	897	SMCStates states;
faluco@216	898	m_AllPluginsLoaded = false;
dvander@1656	899	if ((err=textparsers->ParseFile_SMC(config, &m_PluginInfo, &states)) != SMCError_Okay)
dvander@203	900	{
dvander@305	901	g_Logger.LogError("[SM] Encountered fatal error parsing file \"%s\"", config);
dvander@1406	902	const char *err_msg = textparsers->GetSMCErrorString(err);
dvander@282	903	if (err_msg)
dvander@282	904	{
dvander@305	905	g_Logger.LogError("[SM] Parse error encountered: \"%s\"", err_msg);
dvander@282	906	}
dvander@203	907	}
dvander@203	908
dvander@209	909	LoadPluginsFromDir(basedir, NULL);
dvander@209	910	}
dvander@203	911
dvander@209	912	void CPluginManager::LoadPluginsFromDir(const char basedir, const char localpath)
dvander@209	913	{
ds@652	914	char base_path[PLATFORM_MAX_PATH];
dvander@209	915
dvander@209	916	/* Form the current path to start reading from */
dvander@209	917	if (localpath == NULL)
dvander@209	918	{
dvander@209	919	g_LibSys.PathFormat(base_path, sizeof(base_path), "%s", basedir);
dvander@209	920	} else {
dvander@209	921	g_LibSys.PathFormat(base_path, sizeof(base_path), "%s/%s", basedir, localpath);
dvander@209	922	}
dvander@209	923
dvander@209	924	IDirectory *dir = g_LibSys.OpenDirectory(base_path);
dvander@209	925
dvander@209	926	if (!dir)
dvander@209	927	{
dvander@282	928	char error[256];
dvander@282	929	g_LibSys.GetPlatformError(error, sizeof(error));
dvander@305	930	g_Logger.LogError("[SM] Failure reading from plugins path: %s", localpath);
dvander@305	931	g_Logger.LogError("[SM] Platform returned error: %s", error);
dvander@209	932	return;
dvander@209	933	}
dvander@209	934
dvander@190	935	while (dir->MoreFiles())
dvander@190	936	{
dvander@209	937	if (dir->IsEntryDirectory()
dvander@209	938	&& (strcmp(dir->GetEntryName(), ".") != 0)
dvander@209	939	&& (strcmp(dir->GetEntryName(), "..") != 0)
dvander@209	940	&& (strcmp(dir->GetEntryName(), "disabled") != 0)
dvander@209	941	&& (strcmp(dir->GetEntryName(), "optional") != 0))
dvander@190	942	{
ds@652	943	char new_local[PLATFORM_MAX_PATH];
dvander@209	944	if (localpath == NULL)
dvander@209	945	{
dvander@209	946	/* If no path yet, don't add a former slash */
dvander@209	947	snprintf(new_local, sizeof(new_local), "%s", dir->GetEntryName());
dvander@209	948	} else {
dvander@209	949	g_LibSys.PathFormat(new_local, sizeof(new_local), "%s/%s", localpath, dir->GetEntryName());
dvander@209	950	}
dvander@209	951	LoadPluginsFromDir(basedir, new_local);
dvander@209	952	} else if (dir->IsEntryFile()) {
dvander@209	953	const char *name = dir->GetEntryName();
dvander@209	954	size_t len = strlen(name);
dvander@211	955	if (len >= 4
dvander@211	956	&& strcmp(&name[len-4], ".smx") == 0)
dvander@209	957	{
dvander@211	958	/* If the filename matches, load the plugin */
ds@652	959	char plugin[PLATFORM_MAX_PATH];
dvander@211	960	if (localpath == NULL)
dvander@211	961	{
dvander@211	962	snprintf(plugin, sizeof(plugin), "%s", name);
dvander@211	963	} else {
dvander@211	964	g_LibSys.PathFormat(plugin, sizeof(plugin), "%s/%s", localpath, name);
dvander@211	965	}
dvander@211	966	LoadAutoPlugin(plugin);
dvander@209	967	}
dvander@209	968	}
dvander@209	969	dir->NextEntry();
dvander@209	970	}
dvander@209	971	g_LibSys.CloseDirectory(dir);
dvander@209	972	}
dvander@209	973
dvander@211	974	//well i have discovered that gabe newell is very fat, so i wrote this comment now
ds@755	975	LoadRes CPluginManager::_LoadPlugin(CPlugin *_plugin, const char path, bool debug, PluginType type, char error[], size_t maxlength)
dvander@209	976	{
dvander@1630	977	if (m_LoadingLocked)
dvander@1630	978	{
dvander@1630	979	return LoadRes_NeverLoad;
dvander@1630	980	}
dvander@1630	981
dvander@1436	982	bool no_load = false;
dvander@1436	983	PluginSettings *pset;
dvander@1436	984	unsigned int setcount = m_PluginInfo.GetSettingsNum();
dvander@1436	985	for (unsigned int i=0; i<setcount; i++)
dvander@1436	986	{
dvander@1436	987	if ((pset = m_PluginInfo.GetSettingsIfMatch(i, path)) == NULL)
dvander@1436	988	{
dvander@1436	989	continue;
dvander@1436	990	}
dvander@1436	991	if (pset->blockload_val)
dvander@1436	992	{
dvander@1436	993	no_load = true;
dvander@1436	994	break;
dvander@1436	995	}
dvander@1436	996	}
dvander@1436	997
dvander@211	998	/**
dvander@211	999	* Does this plugin already exist?
dvander@211	1000	*/
dvander@211	1001	CPlugin *pPlugin;
dvander@298	1002	if (sm_trie_retrieve(m_LoadLookup, path, (void **)&pPlugin))
dvander@211	1003	{
dvander@211	1004	/* Check to see if we should try reloading it */
dvander@211	1005	if (pPlugin->GetStatus() == Plugin_BadLoad
dvander@282	1006	\|\| pPlugin->GetStatus() == Plugin_Error
dvander@1436	1007	\|\| pPlugin->GetStatus() == Plugin_Failed
dvander@1436	1008	\|\| no_load)
dvander@211	1009	{
dvander@211	1010	UnloadPlugin(pPlugin);
dvander@1630	1011	}
dvander@1630	1012	else
dvander@1630	1013	{
faluco@383	1014	if (_plugin)
faluco@383	1015	{
faluco@383	1016	*_plugin = pPlugin;
faluco@383	1017	}
faluco@383	1018	return LoadRes_AlreadyLoaded;
dvander@211	1019	}
dvander@211	1020	}
dvander@211	1021
dvander@1436	1022	if (no_load)
dvander@1436	1023	{
dvander@1436	1024	return LoadRes_NeverLoad;
dvander@1436	1025	}
dvander@1436	1026
ds@755	1027	pPlugin = CPlugin::CreatePlugin(path, error, maxlength);
dvander@209	1028
dvander@209	1029	assert(pPlugin != NULL);
dvander@209	1030
dvander@209	1031	pPlugin->m_type = PluginType_MapUpdated;
dvander@209	1032
dvander@209	1033	ICompilation *co = NULL;
dvander@209	1034
dvander@209	1035	if (pPlugin->m_status == Plugin_Uncompiled)
dvander@209	1036	{
dvander@209	1037	co = pPlugin->StartMyCompile(g_pVM);
dvander@209	1038	}
dvander@209	1039
dvander@209	1040	for (unsigned int i=0; i<setcount; i++)
dvander@209	1041	{
dvander@298	1042	if ((pset=m_PluginInfo.GetSettingsIfMatch(i, path)) == NULL)
dvander@209	1043	{
dvander@209	1044	continue;
dvander@209	1045	}
dvander@209	1046	pPlugin->m_type = pset->type_val;
dvander@209	1047	if (co)
dvander@209	1048	{
dvander@209	1049	for (unsigned int j=0; j<pset->opts_num; j++)
dvander@209	1050	{
dvander@209	1051	const char key, val;
dvander@209	1052	m_PluginInfo.GetOptionsForPlugin(pset, j, &key, &val);
dvander@209	1053	if (!key \|\| !val)
dvander@209	1054	{
dvander@209	1055	continue;
dvander@209	1056	}
dvander@209	1057	if (!g_pVM->SetCompilationOption(co, key, val))
dvander@209	1058	{
dvander@298	1059	if (error)
dvander@298	1060	{
ds@755	1061	snprintf(error, maxlength, "Unable to set JIT option (key \"%s\") (value \"%s\")", key, val);
dvander@298	1062	}
dvander@209	1063	pPlugin->CancelMyCompile();
dvander@209	1064	co = NULL;
dvander@209	1065	break;
dvander@209	1066	}
dvander@209	1067	}
dvander@190	1068	}
dvander@190	1069	}
dvander@209	1070
dvander@298	1071	/* Do the actual compiling */
dvander@209	1072	if (co)
dvander@209	1073	{
dvander@803	1074	pPlugin->FinishMyCompile(error, maxlength);
dvander@209	1075	co = NULL;
dvander@209	1076	}
dvander@209	1077
dvander@298	1078	/* Get the status */
dvander@211	1079	if (pPlugin->GetStatus() == Plugin_Created)
dvander@211	1080	{
dvander@211	1081	AddCoreNativesToPlugin(pPlugin);
dvander@229	1082	pPlugin->InitIdentity();
ds@755	1083	if (pPlugin->Call_AskPluginLoad(error, maxlength))
dvander@315	1084	{
dvander@315	1085	/* Autoload any modules */
ds@755	1086	LoadOrRequireExtensions(pPlugin, 1, error, maxlength);
dvander@1572	1087	}
dvander@1572	1088	else
dvander@1572	1089	{
teame06@1200	1090	pPlugin->SetErrorState(Plugin_Failed, "%s", error);
dvander@315	1091	}
dvander@211	1092	}
dvander@211	1093
faluco@333	1094	/* Save the time stamp */
faluco@333	1095	time_t t = pPlugin->GetFileTimeStamp();
faluco@333	1096	pPlugin->SetTimeStamp(t);
faluco@333	1097
dvander@298	1098	if (_plugin)
dvander@298	1099	{
dvander@298	1100	*_plugin = pPlugin;
dvander@298	1101	}
dvander@298	1102
faluco@383	1103	return (pPlugin->GetStatus() == Plugin_Loaded) ? LoadRes_Successful : LoadRes_Failure;
dvander@190	1104	}
dvander@190	1105
ds@755	1106	IPlugin CPluginManager::LoadPlugin(const char path, bool debug, PluginType type, char error[], size_t maxlength, bool *wasloaded)
dvander@174	1107	{
dvander@298	1108	CPlugin *pl;
faluco@383	1109	LoadRes res;
dvander@211	1110
faluco@383	1111	*wasloaded = false;
ds@755	1112	if ((res=_LoadPlugin(&pl, path, debug, type, error, maxlength)) == LoadRes_Failure)
dvander@211	1113	{
dvander@298	1114	delete pl;
dvander@174	1115	return NULL;
dvander@174	1116	}
dvander@174	1117
faluco@383	1118	if (res == LoadRes_AlreadyLoaded)
faluco@383	1119	{
faluco@383	1120	*wasloaded = true;
faluco@383	1121	return pl;
faluco@383	1122	}
faluco@383	1123
dvander@1436	1124	if (res == LoadRes_NeverLoad)
dvander@1436	1125	{
dvander@1630	1126	if (m_LoadingLocked)
dvander@1630	1127	{
dvander@1630	1128	UTIL_Format(error, maxlength, "There is a global plugin loading lock in effect.");
dvander@1630	1129	}
dvander@1630	1130	else
dvander@1630	1131	{
dvander@1630	1132	UTIL_Format(error, maxlength, "This plugin is blocked from loading (see plugin_settings.cfg)");
dvander@1630	1133	}
dvander@1436	1134	return NULL;
dvander@1436	1135	}
dvander@1436	1136
dvander@298	1137	AddPlugin(pl);
dvander@298	1138
dvander@315	1139	/* Run second pass if we need to */
dvander@315	1140	if (IsLateLoadTime() && pl->GetStatus() == Plugin_Loaded)
dvander@315	1141	{
ds@755	1142	if (!RunSecondPass(pl, error, maxlength))
dvander@315	1143	{
dvander@315	1144	UnloadPlugin(pl);
dvander@315	1145	return NULL;
dvander@315	1146	}
faluco@1221	1147	pl->Call_OnAllPluginsLoaded();
dvander@315	1148	}
dvander@315	1149
dvander@298	1150	return pl;
dvander@298	1151	}
dvander@298	1152
dvander@298	1153	void CPluginManager::LoadAutoPlugin(const char *plugin)
dvander@298	1154	{
dvander@1436	1155	CPlugin *pl = NULL;
faluco@383	1156	LoadRes res;
dvander@298	1157	char error[255] = "Unknown error";
dvander@298	1158
faluco@383	1159	if ((res=_LoadPlugin(&pl, plugin, false, PluginType_MapUpdated, error, sizeof(error))) == LoadRes_Failure)
dvander@209	1160	{
dvander@305	1161	g_Logger.LogError("[SM] Failed to load plugin \"%s\": %s", plugin, error);
dvander@298	1162	pl->SetErrorState(Plugin_Failed, "%s", error);
dvander@209	1163	}
dvander@209	1164
faluco@383	1165	if (res == LoadRes_Successful)
faluco@383	1166	{
faluco@383	1167	AddPlugin(pl);
faluco@383	1168	}
dvander@209	1169	}
dvander@209	1170
dvander@211	1171	void CPluginManager::AddPlugin(CPlugin *pPlugin)
dvander@209	1172	{
dvander@174	1173	List<IPluginsListener *>::iterator iter;
dvander@174	1174	IPluginsListener *pListener;
dvander@298	1175
dvander@174	1176	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
dvander@174	1177	{
dvander@174	1178	pListener = (*iter);
dvander@174	1179	pListener->OnPluginCreated(pPlugin);
dvander@174	1180	}
dvander@174	1181
dvander@298	1182	m_plugins.push_back(pPlugin);
dvander@298	1183	sm_trie_insert(m_LoadLookup, pPlugin->m_filename, pPlugin);
dvander@211	1184	}
dvander@211	1185
dvander@213	1186	void CPluginManager::LoadAll_SecondPass()
dvander@213	1187	{
dvander@213	1188	List<CPlugin *>::iterator iter;
dvander@213	1189	CPlugin *pPlugin;
dvander@213	1190
dvander@315	1191	char error[256];
dvander@213	1192	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
dvander@213	1193	{
dvander@213	1194	pPlugin = (*iter);
dvander@213	1195	if (pPlugin->GetStatus() == Plugin_Loaded)
dvander@213	1196	{
dvander@315	1197	error[0] = '\0';
dvander@315	1198	if (!RunSecondPass(pPlugin, error, sizeof(error)))
dvander@315	1199	{
dvander@315	1200	g_Logger.LogError("[SM] Unable to load plugin \"%s\": %s", pPlugin->GetFilename(), error);
dvander@315	1201	pPlugin->SetErrorState(Plugin_Failed, "%s", error);
dvander@315	1202	}
dvander@213	1203	}
dvander@213	1204	}
dvander@298	1205
faluco@216	1206	m_AllPluginsLoaded = true;
dvander@213	1207	}
dvander@213	1208
faluco@1112	1209	bool CPluginManager::FindOrRequirePluginDeps(CPlugin pPlugin, char error, size_t maxlength)
faluco@1112	1210	{
faluco@1112	1211	struct _pl
faluco@1112	1212	{
faluco@1112	1213	cell_t name;
faluco@1112	1214	cell_t file;
faluco@1112	1215	cell_t required;
faluco@1112	1216	} *pl;
faluco@1112	1217
faluco@1112	1218	IPluginContext *pBase = pPlugin->GetBaseContext();
faluco@1112	1219	uint32_t num = pBase->GetPubVarsNum();
faluco@1112	1220	sp_pubvar_t *pubvar;
faluco@1112	1221	char name, file;
faluco@1112	1222	char pathfile[PLATFORM_MAX_PATH];
faluco@1112	1223
faluco@1112	1224	for (uint32_t i=0; i<num; i++)
faluco@1112	1225	{
faluco@1112	1226	if (pBase->GetPubvarByIndex(i, &pubvar) != SP_ERROR_NONE)
faluco@1112	1227	{
faluco@1112	1228	continue;
faluco@1112	1229	}
faluco@1112	1230	if (strncmp(pubvar->name, "__pl_", 5) == 0)
faluco@1112	1231	{
faluco@1112	1232	pl = (_pl *)pubvar->offs;
faluco@1112	1233	if (pBase->LocalToString(pl->file, &file) != SP_ERROR_NONE)
faluco@1112	1234	{
faluco@1112	1235	continue;
faluco@1112	1236	}
faluco@1112	1237	if (pBase->LocalToString(pl->name, &name) != SP_ERROR_NONE)
faluco@1112	1238	{
faluco@1112	1239	continue;
faluco@1112	1240	}
faluco@1112	1241	g_LibSys.GetFileFromPath(pathfile, sizeof(pathfile), pPlugin->GetFilename());
faluco@1112	1242	if (strcmp(pathfile, file) == 0)
faluco@1112	1243	{
faluco@1112	1244	continue;
faluco@1112	1245	}
faluco@1112	1246	if (pl->required == false)
faluco@1112	1247	{
faluco@1112	1248	IPluginFunction *pFunc;
faluco@1112	1249	char buffer[64];
dvander@1123	1250	UTIL_Format(buffer, sizeof(buffer), "__pl_%s_SetNTVOptional", &pubvar->name[5]);
faluco@1112	1251	if ((pFunc=pBase->GetFunctionByName(buffer)))
faluco@1112	1252	{
faluco@1112	1253	cell_t res;
faluco@1112	1254	pFunc->Execute(&res);
faluco@1112	1255	if (pPlugin->GetContext()->n_err != SP_ERROR_NONE)
faluco@1112	1256	{
faluco@1112	1257	if (error)
faluco@1112	1258	{
faluco@1112	1259	snprintf(error, maxlength, "Fatal error during initializing plugin load");
faluco@1112	1260	}
faluco@1112	1261	return false;
faluco@1112	1262	}
faluco@1112	1263	}
dvander@1572	1264	}
dvander@1572	1265	else
dvander@1572	1266	{
faluco@1112	1267	/* Check that we aren't registering the same library twice */
faluco@1112	1268	if (pPlugin->m_RequiredLibs.find(name) == pPlugin->m_RequiredLibs.end())
faluco@1112	1269	{
faluco@1112	1270	pPlugin->m_RequiredLibs.push_back(name);
dvander@1572	1271	}
dvander@1572	1272	else
dvander@1572	1273	{
faluco@1112	1274	continue;
faluco@1112	1275	}
faluco@1112	1276	List<CPlugin *>::iterator iter;
faluco@1112	1277	CPlugin *pl;
faluco@1112	1278	bool found = false;
faluco@1112	1279	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
faluco@1112	1280	{
faluco@1112	1281	pl = (*iter);
faluco@1112	1282	if (pl->m_Libraries.find(name) != pl->m_Libraries.end())
faluco@1112	1283	{
faluco@1112	1284	found = true;
faluco@1112	1285	break;
faluco@1112	1286	}
faluco@1112	1287	}
faluco@1112	1288	if (!found)
faluco@1112	1289	{
faluco@1112	1290	if (error)
faluco@1112	1291	{
faluco@1112	1292	snprintf(error, maxlength, "Could not find required plugin \"%s\"", name);
faluco@1112	1293	}
faluco@1112	1294	return false;
faluco@1112	1295	}
faluco@1112	1296	}
faluco@1112	1297	}
faluco@1112	1298	}
faluco@1112	1299
faluco@1112	1300	return true;
faluco@1112	1301	}
faluco@1112	1302
dvander@315	1303	bool CPluginManager::LoadOrRequireExtensions(CPlugin pPlugin, unsigned int pass, char error, size_t maxlength)
dvander@211	1304	{
dvander@298	1305	/* Find any extensions this plugin needs */
dvander@298	1306	struct _ext
dvander@298	1307	{
dvander@298	1308	cell_t name;
dvander@298	1309	cell_t file;
dvander@298	1310	cell_t autoload;
dvander@298	1311	cell_t required;
dvander@298	1312	} *ext;
dvander@298	1313
dvander@298	1314	IPluginContext *pBase = pPlugin->GetBaseContext();
dvander@298	1315	uint32_t num = pBase->GetPubVarsNum();
dvander@298	1316	sp_pubvar_t *pubvar;
dvander@298	1317	IExtension *pExt;
ds@652	1318	char path[PLATFORM_MAX_PATH];
dvander@298	1319	char file, name;
dvander@298	1320	for (uint32_t i=0; i<num; i++)
dvander@298	1321	{
dvander@298	1322	if (pBase->GetPubvarByIndex(i, &pubvar) != SP_ERROR_NONE)
dvander@298	1323	{
dvander@298	1324	continue;
dvander@298	1325	}
dvander@298	1326	if (strncmp(pubvar->name, "__ext_", 6) == 0)
dvander@298	1327	{
dvander@298	1328	ext = (_ext *)pubvar->offs;
dvander@298	1329	if (!ext->required && !ext->autoload)
dvander@298	1330	{
dvander@298	1331	continue;
dvander@298	1332	}
dvander@298	1333	if (pBase->LocalToString(ext->file, &file) != SP_ERROR_NONE)
dvander@298	1334	{
dvander@298	1335	continue;
dvander@298	1336	}
dvander@298	1337	if (pBase->LocalToString(ext->name, &name) != SP_ERROR_NONE)
dvander@298	1338	{
dvander@298	1339	continue;
dvander@298	1340	}
dvander@315	1341	if (pass == 1)
dvander@298	1342	{
dvander@315	1343	/* Attempt to auto-load if necessary */
dvander@315	1344	if (ext->autoload)
dvander@315	1345	{
dvander@315	1346	g_LibSys.PathFormat(path, PLATFORM_MAX_PATH, "%s", file);
dvander@315	1347	g_Extensions.LoadAutoExtension(path);
dvander@315	1348	}
dvander@1572	1349	}
dvander@1572	1350	else if (pass == 2)
dvander@1572	1351	{
dvander@315	1352	/* Is this required? */
dvander@315	1353	if (ext->required)
dvander@298	1354	{
dvander@315	1355	g_LibSys.PathFormat(path, PLATFORM_MAX_PATH, "%s", file);
dvander@315	1356	if ((pExt = g_Extensions.FindExtensionByFile(path)) == NULL)
dvander@315	1357	{
dvander@315	1358	pExt = g_Extensions.FindExtensionByName(name);
dvander@315	1359	}
dvander@315	1360	/* :TODO: should we bind to unloaded extensions?
dvander@315	1361	* Currently the extension manager will ignore this.
dvander@315	1362	*/
dvander@315	1363	if (!pExt \|\| !pExt->IsRunning(NULL, 0))
dvander@315	1364	{
dvander@315	1365	if (error)
dvander@315	1366	{
dvander@315	1367	snprintf(error, maxlength, "Required extension \"%s\" file(\"%s\") not running", name, file);
dvander@315	1368	}
dvander@315	1369	return false;
dvander@1572	1370	}
dvander@1572	1371	else
dvander@1572	1372	{
dvander@315	1373	g_Extensions.BindChildPlugin(pExt, pPlugin);
dvander@315	1374	}
dvander@298	1375	}
dvander@1572	1376	else
dvander@1572	1377	{
dvander@1572	1378	IPluginFunction *pFunc;
dvander@1572	1379	char buffer[64];
dvander@1572	1380	UTIL_Format(buffer, sizeof(buffer), "__ext_%s_SetNTVOptional", &pubvar->name[6]);
dvander@1572	1381
dvander@1572	1382	if ((pFunc = pBase->GetFunctionByName(buffer)) != NULL)
dvander@1572	1383	{
dvander@1572	1384	cell_t res;
dvander@1572	1385	pFunc->Execute(&res);
dvander@1572	1386	if (pPlugin->GetContext()->n_err != SP_ERROR_NONE)
dvander@1572	1387	{
dvander@1572	1388	if (error)
dvander@1572	1389	{
dvander@1572	1390	UTIL_Format(error, maxlength, "Fatal error during plugin initialization (ext req)");
dvander@1572	1391	}
dvander@1572	1392	return false;
dvander@1572	1393	}
dvander@1572	1394	}
dvander@1572	1395	}
dvander@298	1396	}
dvander@315	1397	}
dvander@315	1398	}
dvander@315	1399
dvander@315	1400	return true;
dvander@315	1401	}
dvander@315	1402
dvander@315	1403	bool CPluginManager::RunSecondPass(CPlugin pPlugin, char error, size_t maxlength)
dvander@315	1404	{
dvander@315	1405	/* Second pass for extension requirements */
dvander@315	1406	if (!LoadOrRequireExtensions(pPlugin, 2, error, maxlength))
dvander@315	1407	{
dvander@315	1408	return false;
dvander@315	1409	}
dvander@315	1410
dvander@315	1411	/* Bind all extra natives */
dvander@315	1412	g_Extensions.BindAllNativesToPlugin(pPlugin);
dvander@315	1413
faluco@1112	1414	if (!FindOrRequirePluginDeps(pPlugin, error, maxlength))
faluco@1112	1415	{
faluco@1112	1416	return false;
faluco@1112	1417	}
faluco@1112	1418
dvander@1572	1419	AddFakeNativesToPlugin(pPlugin);
dvander@1572	1420
dvander@315	1421	/* Find any unbound natives
dvander@315	1422	* Right now, these are not allowed
dvander@315	1423	*/
dvander@315	1424	IPluginContext *pContext = pPlugin->GetBaseContext();
dvander@315	1425	uint32_t num = pContext->GetNativesNum();
dvander@315	1426	sp_native_t *native;
dvander@315	1427	for (unsigned int i=0; i<num; i++)
dvander@315	1428	{
dvander@315	1429	if (pContext->GetNativeByIndex(i, &native) != SP_ERROR_NONE)
dvander@315	1430	{
dvander@315	1431	break;
dvander@315	1432	}
dvander@798	1433	if (native->status == SP_NATIVE_UNBOUND
faluco@1089	1434	&& native->name[0] != '@'
faluco@1089	1435	&& !(native->flags & SP_NTVFLAG_OPTIONAL))
dvander@315	1436	{
dvander@315	1437	if (error)
dvander@298	1438	{
dvander@315	1439	snprintf(error, maxlength, "Native \"%s\" was not found.", native->name);
dvander@298	1440	}
dvander@315	1441	return false;
dvander@298	1442	}
dvander@298	1443	}
dvander@211	1444
dvander@211	1445	/* Finish by telling all listeners */
dvander@211	1446	List<IPluginsListener *>::iterator iter;
dvander@211	1447	IPluginsListener *pListener;
dvander@211	1448	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
dvander@211	1449	{
dvander@211	1450	pListener = (*iter);
dvander@211	1451	pListener->OnPluginLoaded(pPlugin);
dvander@211	1452	}
dvander@298	1453
dvander@298	1454	/* Tell this plugin to finish initializing itself */
ds@426	1455	pPlugin->Call_OnPluginStart();
dvander@298	1456
dvander@605	1457	/* Now, if we have fake natives, go through all plugins that might need rebinding */
dvander@1122	1458	if (pPlugin->GetStatus() <= Plugin_Paused && pPlugin->m_fakeNatives.size())
dvander@605	1459	{
dvander@605	1460	List<CPlugin *>::iterator pl_iter;
dvander@605	1461	CPlugin *pOther;
dvander@605	1462	for (pl_iter = m_plugins.begin();
dvander@605	1463	pl_iter != m_plugins.end();
dvander@605	1464	pl_iter++)
dvander@605	1465	{
dvander@605	1466	pOther = (*pl_iter);
dvander@1122	1467	if ((pOther->GetStatus() == Plugin_Error
faluco@1112	1468	&& (pOther->m_FakeNativesMissing \|\| pOther->m_LibraryMissing))
dvander@1122	1469	\|\| pOther->m_FakeNativesMissing)
dvander@605	1470	{
faluco@1112	1471	TryRefreshDependencies(pOther);
dvander@605	1472	}
dvander@605	1473	}
dvander@605	1474	}
dvander@605	1475
dvander@1124	1476	/* Go through our libraries and tell other plugins they're added */
dvander@1124	1477	List<String>::iterator s_iter;
dvander@1124	1478	for (s_iter = pPlugin->m_Libraries.begin();
dvander@1124	1479	s_iter != pPlugin->m_Libraries.end();
dvander@1124	1480	s_iter++)
dvander@1124	1481	{
dvander@1572	1482	OnLibraryAction((*s_iter).c_str(), true, false);
dvander@1124	1483	}
dvander@1124	1484
dvander@1621	1485	/* Finally, add the core language file */
dvander@1621	1486	pPlugin->AddLangFile(g_pCorePhraseID);
dvander@1621	1487
dvander@298	1488	return true;
dvander@211	1489	}
dvander@211	1490
dvander@211	1491	void CPluginManager::AddCoreNativesToPlugin(CPlugin *pPlugin)
dvander@211	1492	{
dvander@1382	1493	IPluginContext *pContext = pPlugin->GetBaseContext();
dvander@1382	1494
dvander@211	1495
dvander@1382	1496	uint32_t natives = pContext->GetNativesNum();
dvander@1382	1497	sp_native_t *native;
dvander@1382	1498	SPVM_NATIVE_FUNC pfn;
dvander@1382	1499	for (uint32_t i=0; i<natives; i++)
dvander@211	1500	{
dvander@1382	1501	if (pContext->GetNativeByIndex(i, &native) != SP_ERROR_NONE)
dvander@1382	1502	{
dvander@1382	1503	continue;
dvander@1382	1504	}
dvander@1382	1505	if (native->status == SP_NATIVE_BOUND)
dvander@211	1506	{
dvander@1382	1507	continue;
dvander@211	1508	}
dvander@1382	1509	if (!sm_trie_retrieve(m_pCoreNatives, native->name, (void **)&pfn))
dvander@1382	1510	{
dvander@1382	1511	continue;
dvander@1382	1512	}
dvander@1382	1513	pContext->BindNativeToIndex(i, pfn);
dvander@211	1514	}
dvander@174	1515	}
dvander@174	1516
dvander@1689	1517	SPVM_NATIVE_FUNC CPluginManager::FindCoreNative(const char *name)
dvander@1689	1518	{
dvander@1689	1519	SPVM_NATIVE_FUNC pfn;
dvander@1689	1520
dvander@1689	1521	if (!sm_trie_retrieve(m_pCoreNatives, name, (void **)&pfn))
dvander@1689	1522	{
dvander@1689	1523	return NULL;
dvander@1689	1524	}
dvander@1689	1525
dvander@1689	1526	return pfn;
dvander@1689	1527	}
dvander@1689	1528
faluco@1112	1529	void CPluginManager::TryRefreshDependencies(CPlugin *pPlugin)
dvander@605	1530	{
dvander@605	1531	assert(pPlugin->GetBaseContext() != NULL);
dvander@605	1532
dvander@605	1533	AddFakeNativesToPlugin(pPlugin);
dvander@605	1534
faluco@1112	1535	List<String>::iterator lib_iter;
faluco@1112	1536	List<String>::iterator req_iter;
faluco@1112	1537	List<CPlugin *>::iterator pl_iter;
faluco@1112	1538	CPlugin *pl;
faluco@1112	1539	for (req_iter=pPlugin->m_RequiredLibs.begin(); req_iter!=pPlugin->m_RequiredLibs.end(); req_iter++)
faluco@1112	1540	{
faluco@1112	1541	bool found = false;
faluco@1112	1542	for (pl_iter=m_plugins.begin(); pl_iter!=m_plugins.end(); pl_iter++)
faluco@1112	1543	{
faluco@1112	1544	pl = (*pl_iter);
faluco@1112	1545	for (lib_iter=pl->m_Libraries.begin(); lib_iter!=pl->m_Libraries.end(); lib_iter++)
faluco@1112	1546	{
faluco@1112	1547	if ((req_iter) == (lib_iter))
faluco@1112	1548	{
faluco@1112	1549	found = true;
faluco@1112	1550	}
faluco@1112	1551	}
faluco@1112	1552	}
faluco@1112	1553	if (!found)
faluco@1112	1554	{
faluco@1112	1555	pPlugin->SetErrorState(Plugin_Error, "Library not found: %s", (*req_iter).c_str());
faluco@1112	1556	return;
faluco@1112	1557	}
faluco@1112	1558	}
faluco@1112	1559
dvander@605	1560	/* Find any unbound natives
dvander@605	1561	* Right now, these are not allowed
dvander@605	1562	*/
dvander@605	1563	IPluginContext *pContext = pPlugin->GetBaseContext();
dvander@605	1564	uint32_t num = pContext->GetNativesNum();
dvander@605	1565	sp_native_t *native;
dvander@605	1566	for (unsigned int i=0; i<num; i++)
dvander@605	1567	{
dvander@605	1568	if (pContext->GetNativeByIndex(i, &native) != SP_ERROR_NONE)
dvander@605	1569	{
dvander@605	1570	break;
dvander@605	1571	}
faluco@1089	1572	if (native->status == SP_NATIVE_UNBOUND && !(native->flags & SP_NTVFLAG_OPTIONAL))
dvander@605	1573	{
dvander@605	1574	pPlugin->SetErrorState(Plugin_Error, "Native not found: %s", native->name);
dvander@605	1575	return;
dvander@605	1576	}
dvander@605	1577	}
dvander@605	1578
dvander@1122	1579	if (pPlugin->GetStatus() == Plugin_Error)
dvander@605	1580	{
dvander@1122	1581	/* If we got here, all natives are okay again! */
dvander@1122	1582	pPlugin->m_status = Plugin_Running;
dvander@1122	1583	if ((pPlugin->m_ctx.ctx->flags & SPFLAG_PLUGIN_PAUSED) == SPFLAG_PLUGIN_PAUSED)
dvander@1122	1584	{
dvander@1122	1585	pPlugin->m_ctx.ctx->flags &= ~SPFLAG_PLUGIN_PAUSED;
dvander@1122	1586	_SetPauseState(pPlugin, false);
dvander@1122	1587	}
dvander@605	1588	}
dvander@605	1589	}
dvander@605	1590
dvander@601	1591	void CPluginManager::AddFakeNativesToPlugin(CPlugin *pPlugin)
dvander@601	1592	{
dvander@601	1593	IPluginContext *pContext = pPlugin->GetBaseContext();
dvander@601	1594	sp_nativeinfo_t native;
dvander@601	1595
dvander@601	1596	List<FakeNative *>::iterator iter;
dvander@601	1597	FakeNative *pNative;
dvander@601	1598	sp_context_t *ctx;
dvander@601	1599	for (iter = m_Natives.begin(); iter != m_Natives.end(); iter++)
dvander@601	1600	{
dvander@601	1601	pNative = (*iter);
dvander@601	1602	ctx = pNative->ctx->GetContext();
dvander@601	1603	if ((ctx->flags & SPFLAG_PLUGIN_PAUSED) == SPFLAG_PLUGIN_PAUSED)
dvander@601	1604	{
dvander@601	1605	/* Ignore natives in paused plugins */
dvander@601	1606	continue;
dvander@601	1607	}
dvander@601	1608	native.name = pNative->name.c_str();
dvander@601	1609	native.func = pNative->func;
dvander@601	1610	if (pContext->BindNative(&native) == SP_ERROR_NONE)
dvander@601	1611	{
faluco@1089	1612	uint32_t idx;
faluco@1089	1613	pContext->FindNativeByName(native.name, &idx);
faluco@1112	1614	if (pPlugin->GetContext()->natives[idx].flags & SP_NTVFLAG_OPTIONAL)
faluco@1089	1615	{
dvander@1122	1616	WeakNative wkn(pPlugin, idx);
faluco@1112	1617	GetPluginByCtx(ctx)->m_WeakNatives.push_back(wkn);
faluco@1089	1618	continue;
faluco@1089	1619	}
dvander@605	1620	/* Add us as a dependency, but we're careful not to do this circularly! */
dvander@605	1621	if (pNative->ctx != pContext)
dvander@605	1622	{
dvander@605	1623	CPlugin *pOther = GetPluginByCtx(ctx);
dvander@605	1624	if (pOther->m_dependents.find(pPlugin) == pOther->m_dependents.end())
dvander@605	1625	{
dvander@605	1626	pOther->m_dependents.push_back(pPlugin);
dvander@605	1627	}
dvander@605	1628	if (pPlugin->m_dependsOn.find(pOther) == pPlugin->m_dependsOn.end())
dvander@605	1629	{
dvander@605	1630	pPlugin->m_dependsOn.push_back(pOther);
dvander@605	1631	}
dvander@605	1632	}
dvander@601	1633	}
dvander@601	1634	}
dvander@601	1635	}
dvander@601	1636
dvander@174	1637	bool CPluginManager::UnloadPlugin(IPlugin *plugin)
dvander@174	1638	{
dvander@174	1639	CPlugin pPlugin = (CPlugin )plugin;
dvander@298	1640
dvander@298	1641	/* This prevents removal during insertion or anything else weird */
dvander@298	1642	if (m_plugins.find(pPlugin) == m_plugins.end())
dvander@298	1643	{
dvander@298	1644	return false;
dvander@298	1645	}
dvander@298	1646
dvander@1638	1647	IPluginContext *pContext = plugin->GetBaseContext();
dvander@1638	1648	if (pContext->IsInExec())
dvander@1638	1649	{
dvander@1638	1650	char buffer[255];
dvander@1638	1651	UTIL_Format(buffer, sizeof(buffer), "sm plugins unload %s\n", plugin->GetFilename());
dvander@1638	1652	engine->ServerCommand(buffer);
dvander@1638	1653	return false;
dvander@1638	1654	}
dvander@1638	1655
dvander@605	1656	/* Remove us from the lookup table and linked list */
dvander@605	1657	m_plugins.remove(pPlugin);
dvander@605	1658	sm_trie_delete(m_LoadLookup, pPlugin->m_filename);
dvander@605	1659
dvander@1124	1660	/* Go through our libraries and tell other plugins they're gone */
dvander@1124	1661	List<String>::iterator s_iter;
dvander@1124	1662	for (s_iter = pPlugin->m_Libraries.begin();
dvander@1124	1663	s_iter != pPlugin->m_Libraries.end();
dvander@1124	1664	s_iter++)
dvander@1124	1665	{
dvander@1572	1666	OnLibraryAction((*s_iter).c_str(), true, true);
dvander@1124	1667	}
dvander@1124	1668
dvander@605	1669	/* Go through all dependent plugins and tell them this plugin is now gone */
dvander@605	1670	List<CPlugin *>::iterator pl_iter;
dvander@605	1671	CPlugin *pOther;
dvander@605	1672	for (pl_iter = pPlugin->m_dependents.begin();
dvander@605	1673	pl_iter != pPlugin->m_dependents.end();
dvander@605	1674	pl_iter++)
dvander@605	1675	{
dvander@605	1676	pOther = (*pl_iter);
dvander@605	1677	pOther->DependencyDropped(pPlugin);
dvander@605	1678	}
dvander@605	1679
dvander@605	1680	/* Tell everyone we depend on that we no longer exist */
dvander@605	1681	for (pl_iter = pPlugin->m_dependsOn.begin();
dvander@605	1682	pl_iter != pPlugin->m_dependsOn.end();
dvander@605	1683	pl_iter++)
dvander@605	1684	{
dvander@605	1685	pOther = (*pl_iter);
dvander@605	1686	pOther->m_dependents.remove(pPlugin);
dvander@605	1687	}
dvander@605	1688
dvander@1122	1689	/* Remove weak references to us */
dvander@1122	1690	for (pl_iter = m_plugins.begin();
dvander@1122	1691	pl_iter != m_plugins.end();
dvander@1122	1692	pl_iter++)
dvander@1122	1693	{
dvander@1122	1694	pOther = (*pl_iter);
dvander@1122	1695	List<WeakNative>::iterator wk_iter = pOther->m_WeakNatives.begin();
dvander@1122	1696	while (wk_iter != pOther->m_WeakNatives.end())
dvander@1122	1697	{
dvander@1122	1698	if ((*wk_iter).pl == pPlugin)
dvander@1122	1699	{
dvander@1122	1700	wk_iter = pOther->m_WeakNatives.erase(wk_iter);
dvander@1122	1701	} else {
dvander@1122	1702	wk_iter++;
dvander@1122	1703	}
dvander@1122	1704	}
dvander@1122	1705	}
dvander@1122	1706
dvander@174	1707	List<IPluginsListener *>::iterator iter;
dvander@174	1708	IPluginsListener *pListener;
dvander@282	1709
dvander@311	1710	if (pPlugin->GetStatus() <= Plugin_Error)
dvander@282	1711	{
dvander@282	1712	/* Notify listeners of unloading */
dvander@282	1713	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
dvander@282	1714	{
dvander@282	1715	pListener = (*iter);
dvander@282	1716	pListener->OnPluginUnloaded(pPlugin);
dvander@282	1717	}
dvander@282	1718	/* Notify plugin */
ds@426	1719	pPlugin->Call_OnPluginEnd();
dvander@282	1720	}
dvander@282	1721
dvander@1122	1722	/* Unbound weak natives */
dvander@1122	1723	List<WeakNative>::iterator wk_iter;
dvander@1122	1724	for (wk_iter=pPlugin->m_WeakNatives.begin(); wk_iter!=pPlugin->m_WeakNatives.end(); wk_iter++)
dvander@1122	1725	{
dvander@1122	1726	WeakNative & wkn = (*wk_iter);
dvander@1122	1727	sp_context_t *ctx = wkn.pl->GetContext();
dvander@1122	1728	ctx->natives[wkn.idx].status = SP_NATIVE_UNBOUND;
dvander@1122	1729	wkn.pl->m_FakeNativesMissing = true;
dvander@1122	1730	}
dvander@1122	1731
faluco@1112	1732	/* Remove all of our native functions */
faluco@1112	1733	List<FakeNative *>::iterator fn_iter;
faluco@1112	1734	FakeNative *pNative;
faluco@1112	1735	for (fn_iter = pPlugin->m_fakeNatives.begin();
faluco@1112	1736	fn_iter != pPlugin->m_fakeNatives.end();
faluco@1112	1737	fn_iter++)
faluco@1112	1738	{
faluco@1112	1739	pNative = (*fn_iter);
faluco@1112	1740	m_Natives.remove(pNative);
faluco@1112	1741	sm_trie_delete(m_pNativeLookup, pNative->name.c_str());
faluco@1112	1742	g_pVM->DestroyFakeNative(pNative->func);
faluco@1112	1743	delete pNative;
faluco@1112	1744	}
faluco@1112	1745
dvander@174	1746	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
dvander@174	1747	{
dvander@282	1748	/* Notify listeners of destruction */
dvander@174	1749	pListener = (*iter);
dvander@174	1750	pListener->OnPluginDestroyed(pPlugin);
dvander@174	1751	}
dvander@176	1752
dvander@282	1753	/* Tell the plugin to delete itself */
dvander@174	1754	delete pPlugin;
dvander@174	1755
dvander@174	1756	return true;
dvander@174	1757	}
dvander@174	1758
dvander@174	1759	IPlugin CPluginManager::FindPluginByContext(const sp_context_t ctx)
dvander@174	1760	{
dvander@174	1761	IPlugin pl = (IPlugin )ctx->user[SM_CONTEXTVAR_MYSELF];
dvander@174	1762	return pl;
dvander@174	1763	}
dvander@174	1764
dvander@174	1765	unsigned int CPluginManager::GetPluginCount()
dvander@174	1766	{
dvander@174	1767	return m_plugins.size();
dvander@174	1768	}
dvander@174	1769
dvander@174	1770	void CPluginManager::AddPluginsListener(IPluginsListener *listener)
dvander@174	1771	{
dvander@174	1772	m_listeners.push_back(listener);
dvander@174	1773	}
dvander@174	1774
dvander@174	1775	void CPluginManager::RemovePluginsListener(IPluginsListener *listener)
dvander@174	1776	{
dvander@174	1777	m_listeners.remove(listener);
dvander@174	1778	}
dvander@174	1779
dvander@174	1780	IPluginIterator *CPluginManager::GetPluginIterator()
dvander@174	1781	{
dvander@174	1782	if (m_iters.empty())
dvander@174	1783	{
dvander@174	1784	return new CPluginIterator(&m_plugins);
dvander@174	1785	} else {
dvander@174	1786	CPluginIterator *iter = m_iters.front();
dvander@174	1787	m_iters.pop();
faluco@216	1788	iter->Reset();
dvander@174	1789	return iter;
dvander@174	1790	}
dvander@174	1791	}
dvander@174	1792
dvander@174	1793	void CPluginManager::ReleaseIterator(CPluginIterator *iter)
dvander@174	1794	{
dvander@174	1795	m_iters.push(iter);
dvander@174	1796	}
dvander@176	1797
dvander@203	1798	bool CPluginManager::TestAliasMatch(const char alias, const char localpath)
dvander@203	1799	{
dvander@203	1800	/* As an optimization, we do not call strlen, but compute the length in the first pass */
dvander@203	1801	size_t alias_len = 0;
dvander@203	1802	size_t local_len = 0;
dvander@203	1803
dvander@203	1804	const char *ptr = alias;
dvander@203	1805	unsigned int alias_explicit_paths = 0;
dvander@203	1806	unsigned int alias_path_end = 0;
dvander@203	1807	while (*ptr != '\0')
dvander@203	1808	{
dvander@203	1809	if (ptr == '\\' \|\| ptr == '/')
dvander@203	1810	{
dvander@203	1811	alias_explicit_paths++;
dvander@203	1812	alias_path_end = alias_len;
dvander@203	1813	}
dvander@203	1814	alias_len++;
dvander@203	1815	ptr++;
dvander@203	1816	}
dvander@203	1817
dvander@209	1818	if (alias_explicit_paths && alias_path_end == alias_len - 1)
dvander@203	1819	{
dvander@203	1820	/* Trailing slash is totally invalid here */
dvander@203	1821	return false;
dvander@203	1822	}
dvander@203	1823
dvander@203	1824	ptr = localpath;
dvander@203	1825	unsigned int local_explicit_paths = 0;
dvander@203	1826	unsigned int local_path_end = 0;
dvander@203	1827	while (*ptr != '\0')
dvander@203	1828	{
dvander@203	1829	if (ptr == '\\' \|\| ptr == '/')
dvander@203	1830	{
dvander@203	1831	local_explicit_paths++;
dvander@203	1832	local_path_end = local_len;
dvander@203	1833	}
dvander@203	1834	local_len++;
dvander@203	1835	ptr++;
dvander@203	1836	}
dvander@203	1837
dvander@203	1838	/* If the alias has more explicit paths than the real path,
dvander@203	1839	* no match will be possible.
dvander@203	1840	*/
dvander@203	1841	if (alias_explicit_paths > local_explicit_paths)
dvander@203	1842	{
dvander@203	1843	return false;
dvander@203	1844	}
dvander@203	1845
dvander@203	1846	if (alias_explicit_paths)
dvander@203	1847	{
dvander@203	1848	/* We need to find if the paths match now. For example, these should all match:
dvander@203	1849	* csdm csdm
dvander@203	1850	* csdm optional/csdm
dvander@203	1851	* csdm/ban optional/crab/csdm/ban
dvander@203	1852	*/
dvander@203	1853	const char *aliasptr = alias;
dvander@203	1854	const char *localptr = localpath;
dvander@203	1855	bool match = true;
dvander@203	1856	do
dvander@203	1857	{
dvander@203	1858	if (aliasptr != localptr)
dvander@203	1859	{
dvander@203	1860	/* We have to knock one path off */
dvander@203	1861	local_explicit_paths--;
dvander@203	1862	if (alias_explicit_paths > local_explicit_paths)
dvander@203	1863	{
dvander@203	1864	/* Skip out if we're gonna have an impossible match */
dvander@203	1865	return false;
dvander@203	1866	}
dvander@203	1867	/* Eat up localptr tokens until we get a result */
dvander@203	1868	while (((localptr - localpath) < (int)local_path_end)
dvander@203	1869	&& *localptr != '/'
dvander@203	1870	&& *localptr != '\\')
dvander@203	1871	{
dvander@203	1872	localptr++;
dvander@203	1873	}
dvander@203	1874	/* Check if we hit the end of our searchable area.
dvander@203	1875	* This probably isn't possible because of the path
dvander@203	1876	* count check, but it's a good idea anyway.
dvander@203	1877	*/
dvander@203	1878	if ((localptr - localpath) >= (int)local_path_end)
dvander@203	1879	{
dvander@203	1880	return false;
dvander@203	1881	} else {
dvander@203	1882	/* Consume the slash token */
dvander@203	1883	localptr++;
dvander@203	1884	}
dvander@203	1885	/* Reset the alias pointer so we can continue consuming */
dvander@203	1886	aliasptr = alias;
dvander@203	1887	match = false;
dvander@203	1888	continue;
dvander@203	1889	}
dvander@203	1890	/* Note:
dvander@203	1891	* This is safe because if localptr terminates early, aliasptr will too
dvander@203	1892	*/
dvander@203	1893	do
dvander@203	1894	{
dvander@203	1895	/* We should never reach the end of the string because of this check. */
dvander@203	1896	bool aliasend = (aliasptr - alias) > (int)alias_path_end;
dvander@203	1897	bool localend = (localptr - localpath) > (int)local_path_end;
dvander@203	1898	if (aliasend \|\| localend)
dvander@203	1899	{
dvander@203	1900	if (aliasend && localend)
dvander@203	1901	{
dvander@203	1902	/* we matched, and we can break out now */
dvander@203	1903	match = true;
dvander@203	1904	break;
dvander@203	1905	}
dvander@203	1906	/* Otherwise, we've hit the end somehow and rest won't match up. Break out. */
dvander@203	1907	match = false;
dvander@203	1908	break;
dvander@203	1909	}
dvander@203	1910
dvander@203	1911	/* If we got here, it's safe to compare the next two tokens */
dvander@203	1912	if (localptr != aliasptr)
dvander@203	1913	{
dvander@203	1914	match = false;
dvander@203	1915	break;
dvander@203	1916	}
dvander@203	1917	localptr++;
dvander@203	1918	aliasptr++;
dvander@203	1919	} while (true);
dvander@203	1920	} while (!match);
dvander@203	1921	}
dvander@203	1922
dvander@203	1923	/* If we got here, it's time to compare filenames */
dvander@203	1924	const char *aliasptr = alias;
dvander@203	1925	const char *localptr = localpath;
dvander@203	1926
dvander@203	1927	if (alias_explicit_paths)
dvander@203	1928	{
dvander@203	1929	aliasptr = &alias[alias_path_end + 1];
dvander@203	1930	}
dvander@203	1931
dvander@203	1932	if (local_explicit_paths)
dvander@203	1933	{
dvander@203	1934	localptr = &localpath[local_path_end + 1];
dvander@203	1935	}
dvander@203	1936
dvander@203	1937	while (true)
dvander@203	1938	{
dvander@203	1939	if (aliasptr == '')
dvander@203	1940	{
dvander@203	1941	/* First, see if this is the last character */
dvander@351	1942	if ((unsigned)(aliasptr - alias) == alias_len - 1)
dvander@203	1943	{
dvander@203	1944	/* If so, there's no need to match anything else */
dvander@203	1945	return true;
dvander@203	1946	}
dvander@203	1947	/* Otherwise, we need to search for an appropriate matching sequence in local.
dvander@203	1948	* Note that we only need to search up to the next asterisk.
dvander@203	1949	*/
dvander@203	1950	aliasptr++;
dvander@203	1951	bool match = true;
dvander@203	1952	const char *local_orig = localptr;
dvander@203	1953	do
dvander@203	1954	{
dvander@203	1955	match = true;
dvander@203	1956	while (aliasptr != '\0' && aliasptr != '*')
dvander@203	1957	{
dvander@203	1958	/* Since aliasptr is never '\0', localptr hitting the end will fail */
dvander@203	1959	if (aliasptr != localptr)
dvander@203	1960	{
dvander@203	1961	match = false;
dvander@203	1962	break;
dvander@203	1963	}
dvander@203	1964	aliasptr++;
dvander@203	1965	localptr++;
dvander@203	1966	}
dvander@203	1967	if (!match)
dvander@203	1968	{
dvander@203	1969	/* If we didn't get a match, we need to advance the search stream.
dvander@203	1970	* This will let us skip tokens while still searching for another match.
dvander@203	1971	*/
dvander@203	1972	localptr = ++local_orig;
dvander@203	1973	/* Make sure we don't go out of bounds */
dvander@203	1974	if (*localptr == '\0')
dvander@203	1975	{
dvander@203	1976	break;
dvander@203	1977	}
dvander@203	1978	}
dvander@203	1979	} while (!match);
dvander@203	1980
dvander@203	1981	if (!match)
dvander@203	1982	{
dvander@203	1983	return false;
dvander@203	1984	} else {
dvander@203	1985	/* If we got a match, move on to the next token */
dvander@203	1986	continue;
dvander@203	1987	}
dvander@203	1988	} else if (*aliasptr == '\0') {
dvander@203	1989	if (*localptr == '\0'
dvander@203	1990	\|\|
dvander@203	1991	strcmp(localptr, ".smx") == 0)
dvander@203	1992	{
dvander@203	1993	return true;
dvander@203	1994	} else {
dvander@203	1995	return false;
dvander@203	1996	}
dvander@203	1997	} else if (aliasptr != localptr) {
dvander@203	1998	return false;
dvander@203	1999	}
dvander@203	2000	aliasptr++;
dvander@203	2001	localptr++;
dvander@203	2002	}
dvander@203	2003
dvander@203	2004	return true;
dvander@203	2005	}
dvander@203	2006
faluco@290	2007	bool CPluginManager::IsLateLoadTime() const
dvander@176	2008	{
dvander@315	2009	return (m_AllPluginsLoaded \|\| !g_SourceMod.IsMapLoading());
dvander@176	2010	}
dvander@220	2011
dvander@220	2012	void CPluginManager::OnSourceModAllInitialized()
dvander@220	2013	{
dvander@229	2014	m_MyIdent = g_ShareSys.CreateCoreIdentity();
dvander@229	2015
dvander@262	2016	HandleAccess sec;
dvander@262	2017	g_HandleSys.InitAccessDefaults(NULL, &sec);
dvander@220	2018
dvander@262	2019	sec.access[HandleAccess_Delete] = HANDLE_RESTRICT_IDENTITY;
dvander@262	2020	sec.access[HandleAccess_Clone] = HANDLE_RESTRICT_IDENTITY;
dvander@262	2021
dvander@262	2022	g_PluginType = g_HandleSys.CreateType("Plugin", this, 0, NULL, &sec, m_MyIdent, NULL);
dvander@229	2023	g_PluginIdent = g_ShareSys.CreateIdentType("PLUGIN");
dvander@309	2024
dvander@309	2025	g_RootMenu.AddRootConsoleCommand("plugins", "Manage Plugins", this);
dvander@336	2026
dvander@336	2027	g_ShareSys.AddInterface(NULL, this);
dvander@220	2028	}
dvander@220	2029
dvander@220	2030	void CPluginManager::OnSourceModShutdown()
dvander@220	2031	{
dvander@309	2032	g_RootMenu.RemoveRootConsoleCommand("plugins", this);
dvander@309	2033
dvander@1630	2034	UnloadAll();
dvander@298	2035
dvander@229	2036	g_HandleSys.RemoveType(g_PluginType, m_MyIdent);
dvander@229	2037	g_ShareSys.DestroyIdentType(g_PluginIdent);
dvander@229	2038	g_ShareSys.DestroyIdentity(m_MyIdent);
dvander@220	2039	}
dvander@220	2040
dvander@220	2041	void CPluginManager::OnHandleDestroy(HandleType_t type, void *object)
dvander@220	2042	{
dvander@220	2043	/* We don't care about the internal object, actually */
dvander@220	2044	}
dvander@233	2045
dvander@1766	2046	bool CPluginManager::GetHandleApproxSize(HandleType_t type, void object, unsigned int pSize)
dvander@1766	2047	{
dvander@1766	2048	pSize = ((CPlugin )object)->CalcMemUsage();
dvander@1766	2049	return true;
dvander@1766	2050	}
dvander@1766	2051
dvander@233	2052	void CPluginManager::RegisterNativesFromCore(sp_nativeinfo_t *natives)
dvander@233	2053	{
dvander@1382	2054	for (unsigned int i = 0; natives[i].func != NULL; i++)
dvander@1382	2055	{
dvander@1388	2056	sm_trie_insert(m_pCoreNatives, natives[i].name, (void *)natives[i].func);
dvander@1382	2057	}
dvander@233	2058	}
dvander@247	2059
dvander@247	2060	IPlugin CPluginManager::PluginFromHandle(Handle_t handle, HandleError err)
dvander@247	2061	{
dvander@247	2062	IPlugin *pPlugin;
dvander@247	2063	HandleError _err;
dvander@247	2064
dvander@262	2065	HandleSecurity sec;
dvander@262	2066
dvander@262	2067	sec.pOwner = NULL;
dvander@262	2068	sec.pIdentity = m_MyIdent;
dvander@262	2069
dvander@262	2070	if ((_err=g_HandleSys.ReadHandle(handle, g_PluginType, &sec, (void **)&pPlugin)) != HandleError_None)
dvander@247	2071	{
dvander@247	2072	pPlugin = NULL;
dvander@247	2073	}
dvander@247	2074
dvander@247	2075	if (err)
dvander@247	2076	{
dvander@247	2077	*err = _err;
dvander@247	2078	}
dvander@247	2079
dvander@247	2080	return pPlugin;
dvander@247	2081	}
faluco@290	2082
faluco@290	2083	CPlugin *CPluginManager::GetPluginByOrder(int num)
faluco@290	2084	{
faluco@290	2085	if (num < 1 \|\| num > (int)GetPluginCount())
faluco@290	2086	{
faluco@290	2087	return NULL;
faluco@290	2088	}
faluco@290	2089
faluco@290	2090	CPlugin *pl;
faluco@290	2091	int id = 1;
faluco@290	2092
faluco@290	2093	IPluginIterator *iter = GetPluginIterator();
faluco@290	2094	for (; iter->MorePlugins() && id<num; iter->NextPlugin(), id++) {}
faluco@290	2095
faluco@290	2096	pl = (CPlugin *)(iter->GetPlugin());
faluco@290	2097	iter->Release();
faluco@290	2098
faluco@290	2099	return pl;
dvander@298	2100	}
dvander@309	2101
dvander@309	2102	const char *CPluginManager::GetStatusText(PluginStatus st)
dvander@309	2103	{
dvander@309	2104	switch (st)
dvander@309	2105	{
dvander@309	2106	case Plugin_Running:
dvander@309	2107	return "Running";
dvander@309	2108	case Plugin_Paused:
dvander@309	2109	return "Paused";
dvander@309	2110	case Plugin_Error:
dvander@309	2111	return "Error";
dvander@309	2112	case Plugin_Uncompiled:
dvander@309	2113	return "Uncompiled";
dvander@309	2114	case Plugin_BadLoad:
dvander@309	2115	return "Bad Load";
dvander@309	2116	case Plugin_Failed:
dvander@309	2117	return "Failed";
dvander@309	2118	default:
dvander@309	2119	assert(false);
dvander@309	2120	return "-";
dvander@309	2121	}
dvander@309	2122	}
dvander@309	2123
faluco@1541	2124	void CPluginManager::OnRootConsoleCommand(const char *cmdname, const CCommand &command)
dvander@309	2125	{
faluco@1541	2126	int argcount = command.ArgC();
dvander@309	2127	if (argcount >= 3)
dvander@309	2128	{
faluco@1541	2129	const char *cmd = command.Arg(2);
dvander@309	2130	if (strcmp(cmd, "list") == 0)
dvander@309	2131	{
dvander@309	2132	char buffer[256];
dvander@309	2133	unsigned int id = 1;
dvander@309	2134	int plnum = GetPluginCount();
dvander@309	2135
dvander@309	2136	if (!plnum)
dvander@309	2137	{
dvander@309	2138	g_RootMenu.ConsolePrint("[SM] No plugins loaded");
dvander@309	2139	return;
dvander@1630	2140	}
dvander@1630	2141	else
dvander@1630	2142	{
ds@582	2143	g_RootMenu.ConsolePrint("[SM] Listing %d plugin%s:", GetPluginCount(), (plnum > 1) ? "s" : "");
dvander@309	2144	}
dvander@309	2145
dvander@309	2146	IPluginIterator *iter = GetPluginIterator();
dvander@309	2147	for (; iter->MorePlugins(); iter->NextPlugin(), id++)
dvander@309	2148	{
dvander@309	2149	IPlugin *pl = iter->GetPlugin();
dvander@309	2150	assert(pl->GetStatus() != Plugin_Created);
dvander@309	2151	int len = 0;
dvander@309	2152	const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@351	2153	if (pl->GetStatus() != Plugin_Running)
dvander@340	2154	{
dvander@340	2155	len += UTIL_Format(buffer, sizeof(buffer), " %02d <%s>", id, GetStatusText(pl->GetStatus()));
dvander@1630	2156	}
dvander@1630	2157	else
dvander@1630	2158	{
dvander@340	2159	len += UTIL_Format(buffer, sizeof(buffer), " %02d", id);
dvander@340	2160	}
dvander@309	2161	len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " \"%s\"", (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
dvander@309	2162	if (IS_STR_FILLED(info->version))
dvander@309	2163	{
dvander@309	2164	len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " (%s)", info->version);
dvander@309	2165	}
dvander@309	2166	if (IS_STR_FILLED(info->author))
dvander@309	2167	{
dvander@309	2168	UTIL_Format(&buffer[len], sizeof(buffer)-len, " by %s", info->author);
dvander@309	2169	}
dvander@309	2170	g_RootMenu.ConsolePrint("%s", buffer);
dvander@309	2171	}
dvander@309	2172
dvander@309	2173	iter->Release();
dvander@309	2174	return;
dvander@1630	2175	}
dvander@1630	2176	else if (strcmp(cmd, "load") == 0)
dvander@1630	2177	{
dvander@309	2178	if (argcount < 4)
dvander@309	2179	{
dvander@310	2180	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins load <file>");
dvander@309	2181	return;
dvander@309	2182	}
dvander@309	2183
dvander@309	2184	char error[128];
faluco@383	2185	bool wasloaded;
faluco@1541	2186	const char *filename = command.Arg(3);
ds@794	2187
ds@794	2188	char pluginfile[256];
ds@794	2189	const char *ext = g_LibSys.GetFileExtension(filename) ? "" : ".smx";
ds@794	2190	UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", filename, ext);
ds@794	2191
ds@794	2192	IPlugin *pl = LoadPlugin(pluginfile, false, PluginType_MapUpdated, error, sizeof(error), &wasloaded);
faluco@383	2193
faluco@383	2194	if (wasloaded)
faluco@383	2195	{
ds@794	2196	g_RootMenu.ConsolePrint("[SM] Plugin %s is already loaded.", pluginfile);
faluco@383	2197	return;
faluco@383	2198	}
dvander@309	2199
dvander@309	2200	if (pl)
dvander@309	2201	{
ds@794	2202	g_RootMenu.ConsolePrint("[SM] Loaded plugin %s successfully.", pluginfile);
dvander@1630	2203	}
dvander@1630	2204	else
dvander@1630	2205	{
ds@794	2206	g_RootMenu.ConsolePrint("[SM] Plugin %s failed to load: %s.", pluginfile, error);
dvander@309	2207	}
dvander@309	2208
dvander@309	2209	return;
dvander@1630	2210	}
dvander@1630	2211	else if (strcmp(cmd, "unload") == 0)
dvander@1630	2212	{
dvander@309	2213	if (argcount < 4)
dvander@309	2214	{
ds@1252	2215	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins unload <#\|file>");
dvander@309	2216	return;
dvander@309	2217	}
dvander@309	2218
ds@1252	2219	CPlugin *pl;
ds@1252	2220	char *end;
faluco@1541	2221	const char *arg = command.Arg(3);
ds@1252	2222	int id = strtol(arg, &end, 10);
ds@1252	2223
ds@1252	2224	if (*end == '\0')
dvander@309	2225	{
ds@1252	2226	pl = GetPluginByOrder(id);
ds@1252	2227	if (!pl)
ds@1252	2228	{
ds@1252	2229	g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
ds@1252	2230	return;
ds@1252	2231	}
dvander@1630	2232	}
dvander@1630	2233	else
dvander@1630	2234	{
ds@1252	2235	char pluginfile[256];
ds@1252	2236	const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
ds@1252	2237	UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
dvander@309	2238
ds@1252	2239	if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
ds@1252	2240	{
ds@1252	2241	g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
ds@1252	2242	return;
ds@1252	2243	}
ds@1252	2244	}
dvander@309	2245
ds@652	2246	char name[PLATFORM_MAX_PATH];
dvander@309	2247	const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@309	2248	strcpy(name, (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
dvander@309	2249
dvander@309	2250	if (UnloadPlugin(pl))
dvander@309	2251	{
dvander@310	2252	g_RootMenu.ConsolePrint("[SM] Plugin %s unloaded successfully.", name);
dvander@1630	2253	}
dvander@1630	2254	else
dvander@1630	2255	{
dvander@310	2256	g_RootMenu.ConsolePrint("[SM] Failed to unload plugin %s.", name);
dvander@309	2257	}
dvander@309	2258
dvander@309	2259	return;
dvander@1630	2260	}
dvander@1630	2261	else if (strcmp(cmd, "unload_all") == 0)
dvander@1630	2262	{
dvander@1630	2263	g_PluginSys.UnloadAll();
dvander@1630	2264	g_RootMenu.ConsolePrint("[SM] All plugins have been unloaded.");
dvander@1630	2265	return;
dvander@1630	2266	}
dvander@1630	2267	else if (strcmp(cmd, "load_lock") == 0)
dvander@1630	2268	{
dvander@1630	2269	if (m_LoadingLocked)
dvander@1630	2270	{
dvander@1630	2271	g_RootMenu.ConsolePrint("[SM] There is already a loading lock in effect.");
dvander@1630	2272	}
dvander@1630	2273	else
dvander@1630	2274	{
dvander@1630	2275	m_LoadingLocked = true;
dvander@1630	2276	g_RootMenu.ConsolePrint("[SM] Loading is now locked; no plugins will be loaded or re-loaded.");
dvander@1630	2277	}
dvander@1630	2278	return;
dvander@1630	2279	}
dvander@1630	2280	else if (strcmp(cmd, "load_unlock") == 0)
dvander@1630	2281	{
dvander@1630	2282	if (m_LoadingLocked)
dvander@1630	2283	{
dvander@1630	2284	m_LoadingLocked = false;
dvander@1630	2285	g_RootMenu.ConsolePrint("[SM] The loading lock is no longer in effect.");
dvander@1630	2286	}
dvander@1630	2287	else
dvander@1630	2288	{
dvander@1630	2289	g_RootMenu.ConsolePrint("[SM] There was no loading lock in effect.");
dvander@1630	2290	}
dvander@1630	2291	return;
dvander@1630	2292	}
dvander@1630	2293	else if (strcmp(cmd, "info") == 0)
dvander@1630	2294	{
dvander@309	2295	if (argcount < 4)
dvander@309	2296	{
dvander@310	2297	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins info <#>");
dvander@309	2298	return;
dvander@309	2299	}
dvander@309	2300
faluco@1541	2301	int num = atoi(command.Arg(3));
dvander@309	2302	if (num < 1 \|\| num > (int)GetPluginCount())
dvander@309	2303	{
dvander@310	2304	g_RootMenu.ConsolePrint("[SM] Plugin index not found.");
dvander@309	2305	return;
dvander@309	2306	}
dvander@309	2307
dvander@315	2308	CPlugin *pl = GetPluginByOrder(num);
dvander@309	2309	const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@309	2310
dvander@309	2311	g_RootMenu.ConsolePrint(" Filename: %s", pl->GetFilename());
dvander@315	2312	if (pl->GetStatus() <= Plugin_Error)
dvander@309	2313	{
dvander@315	2314	if (IS_STR_FILLED(info->name))
dvander@315	2315	{
dvander@605	2316	if (IS_STR_FILLED(info->description))
dvander@605	2317	{
dvander@605	2318	g_RootMenu.ConsolePrint(" Title: %s (%s)", info->name, info->description);
dvander@605	2319	} else {
dvander@605	2320	g_RootMenu.ConsolePrint(" Title: %s", info->name);
dvander@605	2321	}
dvander@605	2322	}
dvander@605	2323	if (IS_STR_FILLED(info->author))
dvander@605	2324	{
dvander@605	2325	g_RootMenu.ConsolePrint(" Author: %s", info->author);
dvander@315	2326	}
dvander@315	2327	if (IS_STR_FILLED(info->version))
dvander@315	2328	{
dvander@315	2329	g_RootMenu.ConsolePrint(" Version: %s", info->version);
dvander@315	2330	}
dvander@315	2331	if (IS_STR_FILLED(info->url))
dvander@315	2332	{
dvander@315	2333	g_RootMenu.ConsolePrint(" URL: %s", info->url);
dvander@315	2334	}
dvander@605	2335	if (pl->GetStatus() == Plugin_Error)
dvander@315	2336	{
dvander@605	2337	g_RootMenu.ConsolePrint(" Error: %s", pl->m_errormsg);
dvander@1630	2338	}
dvander@1630	2339	else
dvander@1630	2340	{
ds@860	2341	g_RootMenu.ConsolePrint(" Debugging: %s", pl->IsDebugging() ? "Yes" : "No");
ds@860	2342	g_RootMenu.ConsolePrint(" Running: %s", pl->GetStatus() == Plugin_Running ? "Yes" : "No");
ds@860	2343
ds@861	2344	const char *typestr = "";
ds@860	2345	switch (pl->GetType())
ds@860	2346	{
ds@860	2347	case PluginType_MapUpdated:
ds@860	2348	typestr = "Map Change if Updated";
ds@860	2349	break;
ds@860	2350	case PluginType_MapOnly:
ds@860	2351	typestr = "Map Change";
ds@860	2352	break;
ds@860	2353	case PluginType_Private:
ds@860	2354	case PluginType_Global:
ds@860	2355	typestr = "Never";
ds@860	2356	break;
ds@860	2357	}
ds@860	2358
ds@860	2359	g_RootMenu.ConsolePrint(" Reloads: %s", typestr);
dvander@315	2360	}
dvander@1630	2361	}
dvander@1630	2362	else
dvander@1630	2363	{
dvander@315	2364	g_RootMenu.ConsolePrint(" Load error: %s", pl->m_errormsg);
dvander@315	2365	g_RootMenu.ConsolePrint(" File info: (title \"%s\") (version \"%s\")",
dvander@315	2366	info->name ? info->name : "<none>",
dvander@315	2367	info->version ? info->version : "<none>");
dvander@315	2368	if (IS_STR_FILLED(info->url))
dvander@315	2369	{
dvander@315	2370	g_RootMenu.ConsolePrint(" File URL: %s", info->url);
dvander@315	2371	}
dvander@309	2372	}
dvander@309	2373
dvander@309	2374	return;
dvander@1630	2375	}
dvander@1630	2376	else if (strcmp(cmd, "debug") == 0)
dvander@1630	2377	{
dvander@309	2378	if (argcount < 5)
dvander@309	2379	{
dvander@310	2380	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins debug <#> [on\|off]");
dvander@309	2381	return;
dvander@309	2382	}
dvander@309	2383
faluco@1541	2384	int num = atoi(command.Arg(3));
dvander@309	2385	if (num < 1 \|\| num > (int)GetPluginCount())
dvander@309	2386	{
dvander@310	2387	g_RootMenu.ConsolePrint("[SM] Plugin index not found.");
dvander@309	2388	return;
dvander@309	2389	}
dvander@309	2390
dvander@309	2391	int res;
faluco@1541	2392	const char *mode = command.Arg(4);
dvander@309	2393	if ((res=strcmp("on", mode)) && strcmp("off", mode))
dvander@309	2394	{
dvander@310	2395	g_RootMenu.ConsolePrint("[SM] The only possible options are \"on\" and \"off.\"");
dvander@309	2396	return;
dvander@309	2397	}
dvander@309	2398
dvander@309	2399	bool debug;
dvander@309	2400	if (!res)
dvander@309	2401	{
dvander@309	2402	debug = true;
dvander@1630	2403	}
dvander@1630	2404	else
dvander@1630	2405	{
dvander@309	2406	debug = false;
dvander@309	2407	}
dvander@309	2408
dvander@309	2409	CPlugin *pl = GetPluginByOrder(num);
dvander@309	2410	if (debug && pl->IsDebugging())
dvander@309	2411	{
dvander@309	2412	g_RootMenu.ConsolePrint("[SM] This plugin is already in debug mode.");
dvander@309	2413	return;
dvander@1630	2414	}
dvander@1630	2415	else if (!debug && !pl->IsDebugging())
dvander@1630	2416	{
dvander@309	2417	g_RootMenu.ConsolePrint("[SM] Debug mode is already disabled in this plugin.");
dvander@309	2418	return;
dvander@309	2419	}
dvander@309	2420
dvander@309	2421	char error[256];
dvander@309	2422	if (pl->ToggleDebugMode(debug, error, sizeof(error)))
dvander@309	2423	{
dvander@310	2424	g_RootMenu.ConsolePrint("[SM] Successfully toggled debug mode on plugin %s.", pl->GetFilename());
dvander@309	2425	return;
dvander@1630	2426	}
dvander@1630	2427	else
dvander@1630	2428	{
dvander@309	2429	g_RootMenu.ConsolePrint("[SM] Could not toggle debug mode on plugin %s.", pl->GetFilename());
dvander@309	2430	g_RootMenu.ConsolePrint("[SM] Plugin returned error: %s", error);
dvander@309	2431	return;
dvander@309	2432	}
dvander@1630	2433	}
dvander@1630	2434	else if (strcmp(cmd, "refresh") == 0)
dvander@1630	2435	{
dvander@1630	2436	g_SourceMod.DoGlobalPluginLoads();
dvander@1630	2437	g_RootMenu.ConsolePrint("[SM] The plugin list has been refreshed and reloaded.");
dvander@1630	2438	return;
dvander@1630	2439	}
dvander@1630	2440	else if (strcmp(cmd, "reload") == 0)
dvander@1630	2441	{
faluco@1348	2442	if (argcount < 4)
faluco@1348	2443	{
faluco@1348	2444	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins reload <#\|file>");
faluco@1348	2445	return;
faluco@1348	2446	}
faluco@1348	2447
faluco@1348	2448	CPlugin *pl;
faluco@1348	2449	char *end;
faluco@1541	2450	const char *arg = command.Arg(3);
faluco@1348	2451	int id = strtol(arg, &end, 10);
faluco@1348	2452
faluco@1348	2453	if (*end == '\0')
faluco@1348	2454	{
faluco@1348	2455	pl = GetPluginByOrder(id);
faluco@1348	2456	if (!pl)
faluco@1348	2457	{
faluco@1348	2458	g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
faluco@1348	2459	return;
faluco@1348	2460	}
dvander@1630	2461	}
dvander@1630	2462	else
dvander@1630	2463	{
faluco@1348	2464	char pluginfile[256];
faluco@1348	2465	const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
faluco@1348	2466	UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
faluco@1348	2467
faluco@1348	2468	if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
faluco@1348	2469	{
faluco@1348	2470	g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
faluco@1348	2471	return;
faluco@1348	2472	}
faluco@1348	2473	}
faluco@1348	2474
faluco@1348	2475	char name[PLATFORM_MAX_PATH];
faluco@1348	2476	const sm_plugininfo_t *info = pl->GetPublicInfo();
faluco@1348	2477	strcpy(name, (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
faluco@1348	2478
faluco@1348	2479	if (ReloadPlugin(pl))
faluco@1348	2480	{
faluco@1348	2481	g_RootMenu.ConsolePrint("[SM] Plugin %s reloaded successfully.", name);
dvander@1630	2482	}
dvander@1630	2483	else
dvander@1630	2484	{
faluco@1348	2485	g_RootMenu.ConsolePrint("[SM] Failed to reload plugin %s.", name);
faluco@1348	2486	}
faluco@1348	2487
faluco@1348	2488	return;
dvander@309	2489	}
dvander@309	2490	}
dvander@309	2491
dvander@309	2492	/* Draw the main menu */
dvander@309	2493	g_RootMenu.ConsolePrint("SourceMod Plugins Menu:");
dvander@1630	2494	g_RootMenu.DrawGenericOption("debug", "Toggle debug mode on a plugin");
dvander@1630	2495	g_RootMenu.DrawGenericOption("info", "Information about a plugin");
dvander@309	2496	g_RootMenu.DrawGenericOption("list", "Show loaded plugins");
dvander@309	2497	g_RootMenu.DrawGenericOption("load", "Load a plugin");
dvander@1630	2498	g_RootMenu.DrawGenericOption("load_lock", "Prevents any more plugins from being loaded");
dvander@1630	2499	g_RootMenu.DrawGenericOption("load_unlock", "Re-enables plugin loading");
dvander@1630	2500	g_RootMenu.DrawGenericOption("refresh", "Reloads/refreshes all plugins in the plugins folder");
dvander@1630	2501	g_RootMenu.DrawGenericOption("reload", "Reloads a plugin");
dvander@309	2502	g_RootMenu.DrawGenericOption("unload", "Unload a plugin");
dvander@1630	2503	g_RootMenu.DrawGenericOption("unload_all", "Unloads all plugins");
faluco@1348	2504	}
faluco@1348	2505
faluco@1348	2506	bool CPluginManager::ReloadPlugin(CPlugin *pl)
faluco@1348	2507	{
faluco@1348	2508	List<CPlugin *>::iterator iter;
faluco@1348	2509	char filename[PLATFORM_MAX_PATH];
faluco@1348	2510	bool debug, wasloaded;
faluco@1348	2511	PluginType ptype;
faluco@1348	2512	IPlugin *newpl;
faluco@1348	2513	int id = 1;
faluco@1348	2514
faluco@1348	2515	strcpy(filename, pl->m_filename);
faluco@1348	2516	debug = pl->IsDebugging();
faluco@1348	2517	ptype = pl->GetType();
faluco@1348	2518
faluco@1348	2519	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++, id++)
faluco@1348	2520	{
faluco@1348	2521	if ((*iter) == pl)
faluco@1348	2522	{
faluco@1348	2523	break;
faluco@1348	2524	}
faluco@1348	2525	}
faluco@1348	2526
faluco@1348	2527	if (!UnloadPlugin(pl))
faluco@1348	2528	{
faluco@1348	2529	return false;
faluco@1348	2530	}
faluco@1348	2531	if (!(newpl=LoadPlugin(filename, debug, ptype, NULL, 0, &wasloaded)))
faluco@1348	2532	{
faluco@1348	2533	return false;
faluco@1348	2534	}
faluco@1348	2535
faluco@1348	2536	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
faluco@1348	2537	{
faluco@1348	2538	if ((iter) == (CPlugin )newpl)
faluco@1348	2539	{
faluco@1348	2540	m_plugins.erase(iter);
faluco@1348	2541	break;
faluco@1348	2542	}
faluco@1348	2543	}
faluco@1348	2544
faluco@1348	2545	int i;
dvander@1630	2546	for (i=1, iter=m_plugins.begin(); iter!=m_plugins.end() && i<id; iter++, i++)
dvander@1630	2547	{
dvander@1630	2548	}
faluco@1348	2549	m_plugins.insert(iter, (CPlugin *)newpl);
faluco@1348	2550
faluco@1348	2551	return true;
dvander@309	2552	}
faluco@333	2553
faluco@333	2554	void CPluginManager::ReloadOrUnloadPlugins()
faluco@333	2555	{
dvander@1630	2556	/* If we're in a load lock, just skip this whole bit. */
dvander@1630	2557	if (m_LoadingLocked)
dvander@1630	2558	{
dvander@1630	2559	return;
dvander@1630	2560	}
dvander@1630	2561
faluco@333	2562	List<CPlugin *>::iterator iter;
faluco@333	2563	List<CPlugin *> tmp_list = m_plugins;
faluco@333	2564	CPlugin *pl;
faluco@333	2565	time_t t;
faluco@333	2566
faluco@333	2567	for (iter=tmp_list.begin(); iter!=tmp_list.end(); iter++)
faluco@333	2568	{
faluco@333	2569	pl = (*iter);
faluco@333	2570	if (pl->GetType() == PluginType_MapOnly)
faluco@333	2571	{
faluco@333	2572	UnloadPlugin((IPlugin *)pl);
dvander@1630	2573	}
dvander@1630	2574	else if (pl->GetType() == PluginType_MapUpdated)
dvander@1630	2575	{
dvander@450	2576	t = pl->GetFileTimeStamp();
dvander@450	2577	if (!t \|\| t > pl->GetTimeStamp())
faluco@333	2578	{
faluco@333	2579	pl->SetTimeStamp(t);
faluco@333	2580	UnloadPlugin((IPlugin *)pl);
faluco@333	2581	}
faluco@333	2582	}
faluco@333	2583	}
faluco@333	2584	}
faluco@430	2585
faluco@430	2586	void CPluginManager::_SetPauseState(CPlugin *pl, bool paused)
faluco@430	2587	{
faluco@430	2588	List<IPluginsListener *>::iterator iter;
faluco@430	2589	IPluginsListener *pListener;
faluco@430	2590	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
faluco@430	2591	{
faluco@430	2592	pListener = (*iter);
faluco@430	2593	pListener->OnPluginPauseChange(pl, paused);
faluco@430	2594	}
dvander@464	2595	}
dvander@464	2596
dvander@601	2597	bool CPluginManager::AddFakeNative(IPluginFunction pFunction, const char name, SPVM_FAKENATIVE_FUNC func)
dvander@601	2598	{
dvander@601	2599	if (sm_trie_retrieve(m_pNativeLookup, name, NULL))
dvander@601	2600	{
dvander@601	2601	return false;
dvander@601	2602	}
dvander@601	2603
dvander@601	2604	FakeNative *pNative = new FakeNative;
dvander@601	2605
dvander@601	2606	pNative->func = g_pVM->CreateFakeNative(func, pNative);
dvander@601	2607	if (!pNative->func)
dvander@601	2608	{
dvander@601	2609	delete pNative;
dvander@601	2610	return false;
dvander@601	2611	}
dvander@601	2612
dvander@601	2613	pNative->call = pFunction;
dvander@601	2614	pNative->name.assign(name);
dvander@601	2615	pNative->ctx = pFunction->GetParentContext();
dvander@601	2616
dvander@601	2617	m_Natives.push_back(pNative);
dvander@601	2618	sm_trie_insert(m_pNativeLookup, name,pNative);
dvander@601	2619
dvander@605	2620	CPlugin *pPlugin = GetPluginByCtx(pNative->ctx->GetContext());
dvander@605	2621	pPlugin->m_fakeNatives.push_back(pNative);
dvander@605	2622
dvander@601	2623	return true;
dvander@601	2624	}
ds@611	2625
ds@611	2626	void CPluginManager::AddFunctionsToForward(const char name, IChangeableForward pForward)
ds@611	2627	{
ds@611	2628	List<CPlugin *>::iterator iter;
ds@611	2629	CPlugin *pPlugin;
ds@611	2630	IPluginFunction *pFunction;
ds@611	2631
ds@611	2632	for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++)
ds@611	2633	{
ds@611	2634	pPlugin = (*iter);
ds@611	2635
ds@611	2636	if (pPlugin->GetStatus() <= Plugin_Paused)
ds@611	2637	{
ds@611	2638	pFunction = pPlugin->GetBaseContext()->GetFunctionByName(name);
ds@611	2639
ds@611	2640	if (pFunction)
ds@611	2641	{
ds@611	2642	pForward->AddFunction(pFunction);
ds@611	2643	}
ds@611	2644	}
ds@611	2645	}
ds@611	2646	}
dvander@900	2647
dvander@900	2648	CPlugin CPluginManager::GetPluginFromIdentity(IdentityToken_t pToken)
dvander@900	2649	{
dvander@900	2650	if (pToken->type != g_PluginIdent)
dvander@900	2651	{
dvander@900	2652	return NULL;
dvander@900	2653	}
dvander@900	2654
dvander@900	2655	return (CPlugin *)(pToken->ptr);
dvander@900	2656	}
dvander@978	2657
dvander@1572	2658	void CPluginManager::OnLibraryAction(const char *lib, bool is_a_plugin, bool drop)
dvander@978	2659	{
dvander@978	2660	List<CPlugin *>::iterator iter;
dvander@978	2661
dvander@1124	2662	struct _pl
dvander@1124	2663	{
dvander@1124	2664	cell_t name;
dvander@1124	2665	cell_t file;
dvander@1124	2666	cell_t required;
dvander@1124	2667	} *plc;
dvander@1124	2668
dvander@1572	2669	struct _ext
dvander@1572	2670	{
dvander@1572	2671	cell_t name;
dvander@1572	2672	cell_t file;
dvander@1572	2673	cell_t autoload;
dvander@1572	2674	cell_t required;
dvander@1572	2675	} *ext;
dvander@1572	2676
dvander@1124	2677	const char *name = drop ? "OnLibraryRemoved" : "OnLibraryAdded";
dvander@1124	2678
dvander@1124	2679	for (iter=m_plugins.begin();
dvander@1124	2680	iter!=m_plugins.end();
dvander@1124	2681	iter++)
dvander@1124	2682	{
dvander@1124	2683	CPlugin pl = (iter);
dvander@1124	2684	if (pl->GetStatus() != Plugin_Running)
dvander@1124	2685	{
dvander@1124	2686	continue;
dvander@1124	2687	}
dvander@1124	2688	IPluginContext *pContext = pl->GetBaseContext();
dvander@1124	2689	IPluginFunction *pf = pContext->GetFunctionByName(name);
dvander@1124	2690	if (!pf)
dvander@1124	2691	{
dvander@1124	2692	continue;
dvander@1124	2693	}
dvander@1124	2694	uint32_t num_vars = pContext->GetPubVarsNum();
dvander@1124	2695	for (uint32_t i=0; i<num_vars; i++)
dvander@1124	2696	{
dvander@1124	2697	sp_pubvar_t *pubvar;
dvander@1124	2698	if (pContext->GetPubvarByIndex(i, &pubvar) != SP_ERROR_NONE)
dvander@1124	2699	{
dvander@1124	2700	continue;
dvander@1124	2701	}
dvander@1572	2702	if (is_a_plugin && strncmp(pubvar->name, "__pl_", 5) == 0)
dvander@1124	2703	{
dvander@1572	2704	plc = (_pl *)pubvar->offs;
dvander@1572	2705	if (plc->required)
dvander@1572	2706	{
dvander@1572	2707	continue;
dvander@1572	2708	}
dvander@1572	2709	char *str;
dvander@1572	2710	pContext->LocalToString(plc->name, &str);
dvander@1572	2711	if (strcmp(str, lib) != 0)
dvander@1572	2712	{
dvander@1572	2713	continue;
dvander@1572	2714	}
dvander@1572	2715	pf->PushString(lib);
dvander@1572	2716	pf->Execute(NULL);
dvander@1124	2717	}
dvander@1572	2718	else if (!is_a_plugin && strncmp(pubvar->name, "__ext_", 6) == 0)
dvander@1124	2719	{
dvander@1572	2720	ext = (_ext *)pubvar->offs;
dvander@1572	2721	if (ext->required)
dvander@1572	2722	{
dvander@1572	2723	continue;
dvander@1572	2724	}
dvander@1572	2725	char *str;
dvander@1572	2726	pContext->LocalToString(ext->name, &str);
dvander@1572	2727	if (strcmp(str, lib) != 0)
dvander@1572	2728	{
dvander@1572	2729	continue;
dvander@1572	2730	}
dvander@1572	2731	pf->PushString(lib);
dvander@1572	2732	pf->Execute(NULL);
dvander@1124	2733	}
dvander@1124	2734	}
dvander@1124	2735	}
dvander@978	2736	}
dvander@1124	2737
dvander@1124	2738	bool CPluginManager::LibraryExists(const char *lib)
dvander@1124	2739	{
dvander@1124	2740	List<CPlugin *>::iterator iter;
dvander@1124	2741
dvander@1124	2742	for (iter=m_plugins.begin();
dvander@1124	2743	iter!=m_plugins.end();
dvander@1124	2744	iter++)
dvander@1124	2745	{
dvander@1124	2746	CPlugin pl = (iter);
dvander@1124	2747	if (pl->GetStatus() != Plugin_Running)
dvander@1124	2748	{
dvander@1124	2749	continue;
dvander@1124	2750	}
dvander@1124	2751	List<String>::iterator s_iter;
dvander@1124	2752	for (s_iter = pl->m_Libraries.begin();
dvander@1124	2753	s_iter != pl->m_Libraries.end();
dvander@1124	2754	s_iter++)
dvander@1124	2755	{
dvander@1124	2756	if ((*s_iter).compare(lib) == 0)
dvander@1124	2757	{
dvander@1124	2758	return true;
dvander@1124	2759	}
dvander@1124	2760	}
dvander@1124	2761	}
dvander@1124	2762
dvander@1124	2763	return false;
dvander@1124	2764	}
faluco@1221	2765
faluco@1221	2766	void CPluginManager::AllPluginsLoaded()
faluco@1221	2767	{
faluco@1221	2768	List<CPlugin *>::iterator iter;
faluco@1221	2769	CPlugin *pl;
faluco@1221	2770
faluco@1221	2771	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
faluco@1221	2772	{
faluco@1221	2773	pl = (*iter);
faluco@1221	2774	pl->Call_OnAllPluginsLoaded();
faluco@1221	2775	}
dvander@1232	2776	}
dvander@1630	2777
dvander@1630	2778	void CPluginManager::UnloadAll()
dvander@1630	2779	{
dvander@1630	2780	List<CPlugin *>::iterator iter;
dvander@1630	2781	while ( (iter = m_plugins.begin()) != m_plugins.end() )
dvander@1630	2782	{
dvander@1630	2783	UnloadPlugin((*iter));
dvander@1630	2784	}
dvander@1630	2785	}
dvander@1638	2786
dvander@1638	2787	int CPluginManager::GetOrderOfPlugin(IPlugin *pl)
dvander@1638	2788	{
dvander@1638	2789	int id = 1;
dvander@1638	2790	List<CPlugin *>::iterator iter;
dvander@1638	2791
dvander@1638	2792	for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++, id++)
dvander@1638	2793	{
dvander@1638	2794	if ((*iter) == pl)
dvander@1638	2795	{
dvander@1638	2796	return id;
dvander@1638	2797	}
dvander@1638	2798	}
dvander@1638	2799
dvander@1638	2800	return -1;
dvander@1638	2801	}
dvander@1638	2802

author	David Anderson <dvander@alliedmods.net>
	Wed Dec 05 18:07:18 2007 +0000 (2007-12-05)
changeset 1766	d30c252089be
parent 1689	451c11ef473a
child 1784	72b2eec6b55f
permissions	-rw-r--r--