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

author	David Anderson <dvander@alliedmods.net>
	Sun Jan 13 02:08:40 2008 +0000 (2008-01-13)
changeset 1846	5d0c376864c8
parent 1835	11d60f7219fd
child 1902	ce912ad912a5
permissions	-rw-r--r--