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

author	David Anderson <dvander@alliedmods.net>
	Tue Sep 11 15:10:18 2007 +0000 (2007-09-11)
changeset 1406	f5c07ec3c052
parent 1388	fe8f00446d07
child 1436	38994f8f28c3
permissions	-rw-r--r--