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

author	David Anderson <dvander@alliedmods.net>
	Tue Sep 18 17:44:17 2007 +0000 (2007-09-18)
changeset 1436	38994f8f28c3
parent 1406	f5c07ec3c052
child 1541	e46ad208b9b4
permissions	-rw-r--r--