sourcemod-central: core/systems/PluginSys.cpp@5399d6f24417 (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@273	810	if ((err=g_TextParser.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@282	813	const char *err_msg = g_TextParser.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	sp_context_t *ctx = pContext->GetContext();
dvander@1382	1332
dvander@211	1333
dvander@1382	1334	uint32_t natives = pContext->GetNativesNum();
dvander@1382	1335	sp_native_t *native;
dvander@1382	1336	SPVM_NATIVE_FUNC pfn;
dvander@1382	1337	for (uint32_t i=0; i<natives; i++)
dvander@211	1338	{
dvander@1382	1339	if (pContext->GetNativeByIndex(i, &native) != SP_ERROR_NONE)
dvander@1382	1340	{
dvander@1382	1341	continue;
dvander@1382	1342	}
dvander@1382	1343	if (native->status == SP_NATIVE_BOUND)
dvander@211	1344	{
dvander@1382	1345	continue;
dvander@211	1346	}
dvander@1382	1347	if (!sm_trie_retrieve(m_pCoreNatives, native->name, (void **)&pfn))
dvander@1382	1348	{
dvander@1382	1349	continue;
dvander@1382	1350	}
dvander@1382	1351	pContext->BindNativeToIndex(i, pfn);
dvander@211	1352	}
dvander@174	1353	}
dvander@174	1354
faluco@1112	1355	void CPluginManager::TryRefreshDependencies(CPlugin *pPlugin)
dvander@605	1356	{
dvander@605	1357	assert(pPlugin->GetBaseContext() != NULL);
dvander@605	1358
dvander@605	1359	AddFakeNativesToPlugin(pPlugin);
dvander@605	1360
faluco@1112	1361	List<String>::iterator lib_iter;
faluco@1112	1362	List<String>::iterator req_iter;
faluco@1112	1363	List<CPlugin *>::iterator pl_iter;
faluco@1112	1364	CPlugin *pl;
faluco@1112	1365	for (req_iter=pPlugin->m_RequiredLibs.begin(); req_iter!=pPlugin->m_RequiredLibs.end(); req_iter++)
faluco@1112	1366	{
faluco@1112	1367	bool found = false;
faluco@1112	1368	for (pl_iter=m_plugins.begin(); pl_iter!=m_plugins.end(); pl_iter++)
faluco@1112	1369	{
faluco@1112	1370	pl = (*pl_iter);
faluco@1112	1371	for (lib_iter=pl->m_Libraries.begin(); lib_iter!=pl->m_Libraries.end(); lib_iter++)
faluco@1112	1372	{
faluco@1112	1373	if ((req_iter) == (lib_iter))
faluco@1112	1374	{
faluco@1112	1375	found = true;
faluco@1112	1376	}
faluco@1112	1377	}
faluco@1112	1378	}
faluco@1112	1379	if (!found)
faluco@1112	1380	{
faluco@1112	1381	pPlugin->SetErrorState(Plugin_Error, "Library not found: %s", (*req_iter).c_str());
faluco@1112	1382	return;
faluco@1112	1383	}
faluco@1112	1384	}
faluco@1112	1385
dvander@605	1386	/* Find any unbound natives
dvander@605	1387	* Right now, these are not allowed
dvander@605	1388	*/
dvander@605	1389	IPluginContext *pContext = pPlugin->GetBaseContext();
dvander@605	1390	uint32_t num = pContext->GetNativesNum();
dvander@605	1391	sp_native_t *native;
dvander@605	1392	for (unsigned int i=0; i<num; i++)
dvander@605	1393	{
dvander@605	1394	if (pContext->GetNativeByIndex(i, &native) != SP_ERROR_NONE)
dvander@605	1395	{
dvander@605	1396	break;
dvander@605	1397	}
faluco@1089	1398	if (native->status == SP_NATIVE_UNBOUND && !(native->flags & SP_NTVFLAG_OPTIONAL))
dvander@605	1399	{
dvander@605	1400	pPlugin->SetErrorState(Plugin_Error, "Native not found: %s", native->name);
dvander@605	1401	return;
dvander@605	1402	}
dvander@605	1403	}
dvander@605	1404
dvander@1122	1405	if (pPlugin->GetStatus() == Plugin_Error)
dvander@605	1406	{
dvander@1122	1407	/* If we got here, all natives are okay again! */
dvander@1122	1408	pPlugin->m_status = Plugin_Running;
dvander@1122	1409	if ((pPlugin->m_ctx.ctx->flags & SPFLAG_PLUGIN_PAUSED) == SPFLAG_PLUGIN_PAUSED)
dvander@1122	1410	{
dvander@1122	1411	pPlugin->m_ctx.ctx->flags &= ~SPFLAG_PLUGIN_PAUSED;
dvander@1122	1412	_SetPauseState(pPlugin, false);
dvander@1122	1413	}
dvander@605	1414	}
dvander@605	1415	}
dvander@605	1416
dvander@601	1417	void CPluginManager::AddFakeNativesToPlugin(CPlugin *pPlugin)
dvander@601	1418	{
dvander@601	1419	IPluginContext *pContext = pPlugin->GetBaseContext();
dvander@601	1420	sp_nativeinfo_t native;
dvander@601	1421
dvander@601	1422	List<FakeNative *>::iterator iter;
dvander@601	1423	FakeNative *pNative;
dvander@601	1424	sp_context_t *ctx;
dvander@601	1425	for (iter = m_Natives.begin(); iter != m_Natives.end(); iter++)
dvander@601	1426	{
dvander@601	1427	pNative = (*iter);
dvander@601	1428	ctx = pNative->ctx->GetContext();
dvander@601	1429	if ((ctx->flags & SPFLAG_PLUGIN_PAUSED) == SPFLAG_PLUGIN_PAUSED)
dvander@601	1430	{
dvander@601	1431	/* Ignore natives in paused plugins */
dvander@601	1432	continue;
dvander@601	1433	}
dvander@601	1434	native.name = pNative->name.c_str();
dvander@601	1435	native.func = pNative->func;
dvander@601	1436	if (pContext->BindNative(&native) == SP_ERROR_NONE)
dvander@601	1437	{
faluco@1089	1438	uint32_t idx;
faluco@1089	1439	pContext->FindNativeByName(native.name, &idx);
faluco@1112	1440	if (pPlugin->GetContext()->natives[idx].flags & SP_NTVFLAG_OPTIONAL)
faluco@1089	1441	{
dvander@1122	1442	WeakNative wkn(pPlugin, idx);
faluco@1112	1443	GetPluginByCtx(ctx)->m_WeakNatives.push_back(wkn);
faluco@1089	1444	continue;
faluco@1089	1445	}
dvander@605	1446	/* Add us as a dependency, but we're careful not to do this circularly! */
dvander@605	1447	if (pNative->ctx != pContext)
dvander@605	1448	{
dvander@605	1449	CPlugin *pOther = GetPluginByCtx(ctx);
dvander@605	1450	if (pOther->m_dependents.find(pPlugin) == pOther->m_dependents.end())
dvander@605	1451	{
dvander@605	1452	pOther->m_dependents.push_back(pPlugin);
dvander@605	1453	}
dvander@605	1454	if (pPlugin->m_dependsOn.find(pOther) == pPlugin->m_dependsOn.end())
dvander@605	1455	{
dvander@605	1456	pPlugin->m_dependsOn.push_back(pOther);
dvander@605	1457	}
dvander@605	1458	}
dvander@601	1459	}
dvander@601	1460	}
dvander@601	1461	}
dvander@601	1462
dvander@174	1463	bool CPluginManager::UnloadPlugin(IPlugin *plugin)
dvander@174	1464	{
dvander@174	1465	CPlugin pPlugin = (CPlugin )plugin;
dvander@298	1466
dvander@298	1467	/* This prevents removal during insertion or anything else weird */
dvander@298	1468	if (m_plugins.find(pPlugin) == m_plugins.end())
dvander@298	1469	{
dvander@298	1470	return false;
dvander@298	1471	}
dvander@298	1472
dvander@605	1473	/* Remove us from the lookup table and linked list */
dvander@605	1474	m_plugins.remove(pPlugin);
dvander@605	1475	sm_trie_delete(m_LoadLookup, pPlugin->m_filename);
dvander@605	1476
dvander@1124	1477	/* Go through our libraries and tell other plugins they're gone */
dvander@1124	1478	List<String>::iterator s_iter;
dvander@1124	1479	for (s_iter = pPlugin->m_Libraries.begin();
dvander@1124	1480	s_iter != pPlugin->m_Libraries.end();
dvander@1124	1481	s_iter++)
dvander@1124	1482	{
dvander@1124	1483	OnLibraryAction((*s_iter).c_str(), true);
dvander@1124	1484	}
dvander@1124	1485
dvander@605	1486	/* Go through all dependent plugins and tell them this plugin is now gone */
dvander@605	1487	List<CPlugin *>::iterator pl_iter;
dvander@605	1488	CPlugin *pOther;
dvander@605	1489	for (pl_iter = pPlugin->m_dependents.begin();
dvander@605	1490	pl_iter != pPlugin->m_dependents.end();
dvander@605	1491	pl_iter++)
dvander@605	1492	{
dvander@605	1493	pOther = (*pl_iter);
dvander@605	1494	pOther->DependencyDropped(pPlugin);
dvander@605	1495	}
dvander@605	1496
dvander@605	1497	/* Tell everyone we depend on that we no longer exist */
dvander@605	1498	for (pl_iter = pPlugin->m_dependsOn.begin();
dvander@605	1499	pl_iter != pPlugin->m_dependsOn.end();
dvander@605	1500	pl_iter++)
dvander@605	1501	{
dvander@605	1502	pOther = (*pl_iter);
dvander@605	1503	pOther->m_dependents.remove(pPlugin);
dvander@605	1504	}
dvander@605	1505
dvander@1122	1506	/* Remove weak references to us */
dvander@1122	1507	for (pl_iter = m_plugins.begin();
dvander@1122	1508	pl_iter != m_plugins.end();
dvander@1122	1509	pl_iter++)
dvander@1122	1510	{
dvander@1122	1511	pOther = (*pl_iter);
dvander@1122	1512	List<WeakNative>::iterator wk_iter = pOther->m_WeakNatives.begin();
dvander@1122	1513	while (wk_iter != pOther->m_WeakNatives.end())
dvander@1122	1514	{
dvander@1122	1515	if ((*wk_iter).pl == pPlugin)
dvander@1122	1516	{
dvander@1122	1517	wk_iter = pOther->m_WeakNatives.erase(wk_iter);
dvander@1122	1518	} else {
dvander@1122	1519	wk_iter++;
dvander@1122	1520	}
dvander@1122	1521	}
dvander@1122	1522	}
dvander@1122	1523
dvander@174	1524	List<IPluginsListener *>::iterator iter;
dvander@174	1525	IPluginsListener *pListener;
dvander@282	1526
dvander@311	1527	if (pPlugin->GetStatus() <= Plugin_Error)
dvander@282	1528	{
dvander@282	1529	/* Notify listeners of unloading */
dvander@282	1530	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
dvander@282	1531	{
dvander@282	1532	pListener = (*iter);
dvander@282	1533	pListener->OnPluginUnloaded(pPlugin);
dvander@282	1534	}
dvander@282	1535	/* Notify plugin */
ds@426	1536	pPlugin->Call_OnPluginEnd();
dvander@282	1537	}
dvander@282	1538
dvander@1122	1539	/* Unbound weak natives */
dvander@1122	1540	List<WeakNative>::iterator wk_iter;
dvander@1122	1541	for (wk_iter=pPlugin->m_WeakNatives.begin(); wk_iter!=pPlugin->m_WeakNatives.end(); wk_iter++)
dvander@1122	1542	{
dvander@1122	1543	WeakNative & wkn = (*wk_iter);
dvander@1122	1544	sp_context_t *ctx = wkn.pl->GetContext();
dvander@1122	1545	ctx->natives[wkn.idx].status = SP_NATIVE_UNBOUND;
dvander@1122	1546	wkn.pl->m_FakeNativesMissing = true;
dvander@1122	1547	}
dvander@1122	1548
faluco@1112	1549	/* Remove all of our native functions */
faluco@1112	1550	List<FakeNative *>::iterator fn_iter;
faluco@1112	1551	FakeNative *pNative;
faluco@1112	1552	for (fn_iter = pPlugin->m_fakeNatives.begin();
faluco@1112	1553	fn_iter != pPlugin->m_fakeNatives.end();
faluco@1112	1554	fn_iter++)
faluco@1112	1555	{
faluco@1112	1556	pNative = (*fn_iter);
faluco@1112	1557	m_Natives.remove(pNative);
faluco@1112	1558	sm_trie_delete(m_pNativeLookup, pNative->name.c_str());
faluco@1112	1559	g_pVM->DestroyFakeNative(pNative->func);
faluco@1112	1560	delete pNative;
faluco@1112	1561	}
faluco@1112	1562
dvander@174	1563	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
dvander@174	1564	{
dvander@282	1565	/* Notify listeners of destruction */
dvander@174	1566	pListener = (*iter);
dvander@174	1567	pListener->OnPluginDestroyed(pPlugin);
dvander@174	1568	}
dvander@176	1569
dvander@282	1570	/* Tell the plugin to delete itself */
dvander@174	1571	delete pPlugin;
dvander@174	1572
dvander@174	1573	return true;
dvander@174	1574	}
dvander@174	1575
dvander@174	1576	IPlugin CPluginManager::FindPluginByContext(const sp_context_t ctx)
dvander@174	1577	{
dvander@174	1578	IPlugin pl = (IPlugin )ctx->user[SM_CONTEXTVAR_MYSELF];
dvander@174	1579	return pl;
dvander@174	1580	}
dvander@174	1581
dvander@174	1582	unsigned int CPluginManager::GetPluginCount()
dvander@174	1583	{
dvander@174	1584	return m_plugins.size();
dvander@174	1585	}
dvander@174	1586
dvander@174	1587	void CPluginManager::AddPluginsListener(IPluginsListener *listener)
dvander@174	1588	{
dvander@174	1589	m_listeners.push_back(listener);
dvander@174	1590	}
dvander@174	1591
dvander@174	1592	void CPluginManager::RemovePluginsListener(IPluginsListener *listener)
dvander@174	1593	{
dvander@174	1594	m_listeners.remove(listener);
dvander@174	1595	}
dvander@174	1596
dvander@174	1597	IPluginIterator *CPluginManager::GetPluginIterator()
dvander@174	1598	{
dvander@174	1599	if (m_iters.empty())
dvander@174	1600	{
dvander@174	1601	return new CPluginIterator(&m_plugins);
dvander@174	1602	} else {
dvander@174	1603	CPluginIterator *iter = m_iters.front();
dvander@174	1604	m_iters.pop();
faluco@216	1605	iter->Reset();
dvander@174	1606	return iter;
dvander@174	1607	}
dvander@174	1608	}
dvander@174	1609
dvander@174	1610	void CPluginManager::ReleaseIterator(CPluginIterator *iter)
dvander@174	1611	{
dvander@174	1612	m_iters.push(iter);
dvander@174	1613	}
dvander@176	1614
dvander@203	1615	bool CPluginManager::TestAliasMatch(const char alias, const char localpath)
dvander@203	1616	{
dvander@203	1617	/* As an optimization, we do not call strlen, but compute the length in the first pass */
dvander@203	1618	size_t alias_len = 0;
dvander@203	1619	size_t local_len = 0;
dvander@203	1620
dvander@203	1621	const char *ptr = alias;
dvander@203	1622	unsigned int alias_explicit_paths = 0;
dvander@203	1623	unsigned int alias_path_end = 0;
dvander@203	1624	while (*ptr != '\0')
dvander@203	1625	{
dvander@203	1626	if (ptr == '\\' \|\| ptr == '/')
dvander@203	1627	{
dvander@203	1628	alias_explicit_paths++;
dvander@203	1629	alias_path_end = alias_len;
dvander@203	1630	}
dvander@203	1631	alias_len++;
dvander@203	1632	ptr++;
dvander@203	1633	}
dvander@203	1634
dvander@209	1635	if (alias_explicit_paths && alias_path_end == alias_len - 1)
dvander@203	1636	{
dvander@203	1637	/* Trailing slash is totally invalid here */
dvander@203	1638	return false;
dvander@203	1639	}
dvander@203	1640
dvander@203	1641	ptr = localpath;
dvander@203	1642	unsigned int local_explicit_paths = 0;
dvander@203	1643	unsigned int local_path_end = 0;
dvander@203	1644	while (*ptr != '\0')
dvander@203	1645	{
dvander@203	1646	if (ptr == '\\' \|\| ptr == '/')
dvander@203	1647	{
dvander@203	1648	local_explicit_paths++;
dvander@203	1649	local_path_end = local_len;
dvander@203	1650	}
dvander@203	1651	local_len++;
dvander@203	1652	ptr++;
dvander@203	1653	}
dvander@203	1654
dvander@203	1655	/* If the alias has more explicit paths than the real path,
dvander@203	1656	* no match will be possible.
dvander@203	1657	*/
dvander@203	1658	if (alias_explicit_paths > local_explicit_paths)
dvander@203	1659	{
dvander@203	1660	return false;
dvander@203	1661	}
dvander@203	1662
dvander@203	1663	if (alias_explicit_paths)
dvander@203	1664	{
dvander@203	1665	/* We need to find if the paths match now. For example, these should all match:
dvander@203	1666	* csdm csdm
dvander@203	1667	* csdm optional/csdm
dvander@203	1668	* csdm/ban optional/crab/csdm/ban
dvander@203	1669	*/
dvander@203	1670	const char *aliasptr = alias;
dvander@203	1671	const char *localptr = localpath;
dvander@203	1672	bool match = true;
dvander@203	1673	do
dvander@203	1674	{
dvander@203	1675	if (aliasptr != localptr)
dvander@203	1676	{
dvander@203	1677	/* We have to knock one path off */
dvander@203	1678	local_explicit_paths--;
dvander@203	1679	if (alias_explicit_paths > local_explicit_paths)
dvander@203	1680	{
dvander@203	1681	/* Skip out if we're gonna have an impossible match */
dvander@203	1682	return false;
dvander@203	1683	}
dvander@203	1684	/* Eat up localptr tokens until we get a result */
dvander@203	1685	while (((localptr - localpath) < (int)local_path_end)
dvander@203	1686	&& *localptr != '/'
dvander@203	1687	&& *localptr != '\\')
dvander@203	1688	{
dvander@203	1689	localptr++;
dvander@203	1690	}
dvander@203	1691	/* Check if we hit the end of our searchable area.
dvander@203	1692	* This probably isn't possible because of the path
dvander@203	1693	* count check, but it's a good idea anyway.
dvander@203	1694	*/
dvander@203	1695	if ((localptr - localpath) >= (int)local_path_end)
dvander@203	1696	{
dvander@203	1697	return false;
dvander@203	1698	} else {
dvander@203	1699	/* Consume the slash token */
dvander@203	1700	localptr++;
dvander@203	1701	}
dvander@203	1702	/* Reset the alias pointer so we can continue consuming */
dvander@203	1703	aliasptr = alias;
dvander@203	1704	match = false;
dvander@203	1705	continue;
dvander@203	1706	}
dvander@203	1707	/* Note:
dvander@203	1708	* This is safe because if localptr terminates early, aliasptr will too
dvander@203	1709	*/
dvander@203	1710	do
dvander@203	1711	{
dvander@203	1712	/* We should never reach the end of the string because of this check. */
dvander@203	1713	bool aliasend = (aliasptr - alias) > (int)alias_path_end;
dvander@203	1714	bool localend = (localptr - localpath) > (int)local_path_end;
dvander@203	1715	if (aliasend \|\| localend)
dvander@203	1716	{
dvander@203	1717	if (aliasend && localend)
dvander@203	1718	{
dvander@203	1719	/* we matched, and we can break out now */
dvander@203	1720	match = true;
dvander@203	1721	break;
dvander@203	1722	}
dvander@203	1723	/* Otherwise, we've hit the end somehow and rest won't match up. Break out. */
dvander@203	1724	match = false;
dvander@203	1725	break;
dvander@203	1726	}
dvander@203	1727
dvander@203	1728	/* If we got here, it's safe to compare the next two tokens */
dvander@203	1729	if (localptr != aliasptr)
dvander@203	1730	{
dvander@203	1731	match = false;
dvander@203	1732	break;
dvander@203	1733	}
dvander@203	1734	localptr++;
dvander@203	1735	aliasptr++;
dvander@203	1736	} while (true);
dvander@203	1737	} while (!match);
dvander@203	1738	}
dvander@203	1739
dvander@203	1740	/* If we got here, it's time to compare filenames */
dvander@203	1741	const char *aliasptr = alias;
dvander@203	1742	const char *localptr = localpath;
dvander@203	1743
dvander@203	1744	if (alias_explicit_paths)
dvander@203	1745	{
dvander@203	1746	aliasptr = &alias[alias_path_end + 1];
dvander@203	1747	}
dvander@203	1748
dvander@203	1749	if (local_explicit_paths)
dvander@203	1750	{
dvander@203	1751	localptr = &localpath[local_path_end + 1];
dvander@203	1752	}
dvander@203	1753
dvander@203	1754	while (true)
dvander@203	1755	{
dvander@203	1756	if (aliasptr == '')
dvander@203	1757	{
dvander@203	1758	/* First, see if this is the last character */
dvander@351	1759	if ((unsigned)(aliasptr - alias) == alias_len - 1)
dvander@203	1760	{
dvander@203	1761	/* If so, there's no need to match anything else */
dvander@203	1762	return true;
dvander@203	1763	}
dvander@203	1764	/* Otherwise, we need to search for an appropriate matching sequence in local.
dvander@203	1765	* Note that we only need to search up to the next asterisk.
dvander@203	1766	*/
dvander@203	1767	aliasptr++;
dvander@203	1768	bool match = true;
dvander@203	1769	const char *local_orig = localptr;
dvander@203	1770	do
dvander@203	1771	{
dvander@203	1772	match = true;
dvander@203	1773	while (aliasptr != '\0' && aliasptr != '*')
dvander@203	1774	{
dvander@203	1775	/* Since aliasptr is never '\0', localptr hitting the end will fail */
dvander@203	1776	if (aliasptr != localptr)
dvander@203	1777	{
dvander@203	1778	match = false;
dvander@203	1779	break;
dvander@203	1780	}
dvander@203	1781	aliasptr++;
dvander@203	1782	localptr++;
dvander@203	1783	}
dvander@203	1784	if (!match)
dvander@203	1785	{
dvander@203	1786	/* If we didn't get a match, we need to advance the search stream.
dvander@203	1787	* This will let us skip tokens while still searching for another match.
dvander@203	1788	*/
dvander@203	1789	localptr = ++local_orig;
dvander@203	1790	/* Make sure we don't go out of bounds */
dvander@203	1791	if (*localptr == '\0')
dvander@203	1792	{
dvander@203	1793	break;
dvander@203	1794	}
dvander@203	1795	}
dvander@203	1796	} while (!match);
dvander@203	1797
dvander@203	1798	if (!match)
dvander@203	1799	{
dvander@203	1800	return false;
dvander@203	1801	} else {
dvander@203	1802	/* If we got a match, move on to the next token */
dvander@203	1803	continue;
dvander@203	1804	}
dvander@203	1805	} else if (*aliasptr == '\0') {
dvander@203	1806	if (*localptr == '\0'
dvander@203	1807	\|\|
dvander@203	1808	strcmp(localptr, ".smx") == 0)
dvander@203	1809	{
dvander@203	1810	return true;
dvander@203	1811	} else {
dvander@203	1812	return false;
dvander@203	1813	}
dvander@203	1814	} else if (aliasptr != localptr) {
dvander@203	1815	return false;
dvander@203	1816	}
dvander@203	1817	aliasptr++;
dvander@203	1818	localptr++;
dvander@203	1819	}
dvander@203	1820
dvander@203	1821	return true;
dvander@203	1822	}
dvander@203	1823
faluco@290	1824	bool CPluginManager::IsLateLoadTime() const
dvander@176	1825	{
dvander@315	1826	return (m_AllPluginsLoaded \|\| !g_SourceMod.IsMapLoading());
dvander@176	1827	}
dvander@220	1828
dvander@220	1829	void CPluginManager::OnSourceModAllInitialized()
dvander@220	1830	{
dvander@229	1831	m_MyIdent = g_ShareSys.CreateCoreIdentity();
dvander@229	1832
dvander@262	1833	HandleAccess sec;
dvander@262	1834	g_HandleSys.InitAccessDefaults(NULL, &sec);
dvander@220	1835
dvander@262	1836	sec.access[HandleAccess_Delete] = HANDLE_RESTRICT_IDENTITY;
dvander@262	1837	sec.access[HandleAccess_Clone] = HANDLE_RESTRICT_IDENTITY;
dvander@262	1838
dvander@262	1839	g_PluginType = g_HandleSys.CreateType("Plugin", this, 0, NULL, &sec, m_MyIdent, NULL);
dvander@229	1840	g_PluginIdent = g_ShareSys.CreateIdentType("PLUGIN");
dvander@309	1841
dvander@309	1842	g_RootMenu.AddRootConsoleCommand("plugins", "Manage Plugins", this);
dvander@336	1843
dvander@336	1844	g_ShareSys.AddInterface(NULL, this);
dvander@220	1845	}
dvander@220	1846
dvander@220	1847	void CPluginManager::OnSourceModShutdown()
dvander@220	1848	{
dvander@309	1849	g_RootMenu.RemoveRootConsoleCommand("plugins", this);
dvander@309	1850
dvander@298	1851	List<CPlugin *>::iterator iter;
dvander@298	1852	while ( (iter = m_plugins.begin()) != m_plugins.end() )
dvander@298	1853	{
dvander@298	1854	UnloadPlugin((*iter));
dvander@298	1855	}
dvander@298	1856
dvander@229	1857	g_HandleSys.RemoveType(g_PluginType, m_MyIdent);
dvander@229	1858	g_ShareSys.DestroyIdentType(g_PluginIdent);
dvander@229	1859	g_ShareSys.DestroyIdentity(m_MyIdent);
dvander@220	1860	}
dvander@220	1861
dvander@220	1862	void CPluginManager::OnHandleDestroy(HandleType_t type, void *object)
dvander@220	1863	{
dvander@220	1864	/* We don't care about the internal object, actually */
dvander@220	1865	}
dvander@233	1866
dvander@233	1867	void CPluginManager::RegisterNativesFromCore(sp_nativeinfo_t *natives)
dvander@233	1868	{
dvander@1382	1869	for (unsigned int i = 0; natives[i].func != NULL; i++)
dvander@1382	1870	{
dvander@1382	1871	sm_trie_insert(m_pCoreNatives, natives[i].name, natives[i].func);
dvander@1382	1872	}
dvander@233	1873	}
dvander@247	1874
dvander@247	1875	IPlugin CPluginManager::PluginFromHandle(Handle_t handle, HandleError err)
dvander@247	1876	{
dvander@247	1877	IPlugin *pPlugin;
dvander@247	1878	HandleError _err;
dvander@247	1879
dvander@262	1880	HandleSecurity sec;
dvander@262	1881
dvander@262	1882	sec.pOwner = NULL;
dvander@262	1883	sec.pIdentity = m_MyIdent;
dvander@262	1884
dvander@262	1885	if ((_err=g_HandleSys.ReadHandle(handle, g_PluginType, &sec, (void **)&pPlugin)) != HandleError_None)
dvander@247	1886	{
dvander@247	1887	pPlugin = NULL;
dvander@247	1888	}
dvander@247	1889
dvander@247	1890	if (err)
dvander@247	1891	{
dvander@247	1892	*err = _err;
dvander@247	1893	}
dvander@247	1894
dvander@247	1895	return pPlugin;
dvander@247	1896	}
faluco@290	1897
faluco@290	1898	CPlugin *CPluginManager::GetPluginByOrder(int num)
faluco@290	1899	{
faluco@290	1900	if (num < 1 \|\| num > (int)GetPluginCount())
faluco@290	1901	{
faluco@290	1902	return NULL;
faluco@290	1903	}
faluco@290	1904
faluco@290	1905	CPlugin *pl;
faluco@290	1906	int id = 1;
faluco@290	1907
faluco@290	1908	IPluginIterator *iter = GetPluginIterator();
faluco@290	1909	for (; iter->MorePlugins() && id<num; iter->NextPlugin(), id++) {}
faluco@290	1910
faluco@290	1911	pl = (CPlugin *)(iter->GetPlugin());
faluco@290	1912	iter->Release();
faluco@290	1913
faluco@290	1914	return pl;
dvander@298	1915	}
dvander@309	1916
dvander@309	1917	const char *CPluginManager::GetStatusText(PluginStatus st)
dvander@309	1918	{
dvander@309	1919	switch (st)
dvander@309	1920	{
dvander@309	1921	case Plugin_Running:
dvander@309	1922	return "Running";
dvander@309	1923	case Plugin_Paused:
dvander@309	1924	return "Paused";
dvander@309	1925	case Plugin_Error:
dvander@309	1926	return "Error";
dvander@309	1927	case Plugin_Uncompiled:
dvander@309	1928	return "Uncompiled";
dvander@309	1929	case Plugin_BadLoad:
dvander@309	1930	return "Bad Load";
dvander@309	1931	case Plugin_Failed:
dvander@309	1932	return "Failed";
dvander@309	1933	default:
dvander@309	1934	assert(false);
dvander@309	1935	return "-";
dvander@309	1936	}
dvander@309	1937	}
dvander@309	1938
dvander@309	1939	void CPluginManager::OnRootConsoleCommand(const char *command, unsigned int argcount)
dvander@309	1940	{
dvander@309	1941	if (argcount >= 3)
dvander@309	1942	{
dvander@309	1943	const char *cmd = g_RootMenu.GetArgument(2);
dvander@309	1944	if (strcmp(cmd, "list") == 0)
dvander@309	1945	{
dvander@309	1946	char buffer[256];
dvander@309	1947	unsigned int id = 1;
dvander@309	1948	int plnum = GetPluginCount();
dvander@309	1949
dvander@309	1950	if (!plnum)
dvander@309	1951	{
dvander@309	1952	g_RootMenu.ConsolePrint("[SM] No plugins loaded");
dvander@309	1953	return;
dvander@309	1954	} else {
ds@582	1955	g_RootMenu.ConsolePrint("[SM] Listing %d plugin%s:", GetPluginCount(), (plnum > 1) ? "s" : "");
dvander@309	1956	}
dvander@309	1957
dvander@309	1958	IPluginIterator *iter = GetPluginIterator();
dvander@309	1959	for (; iter->MorePlugins(); iter->NextPlugin(), id++)
dvander@309	1960	{
dvander@309	1961	IPlugin *pl = iter->GetPlugin();
dvander@309	1962	assert(pl->GetStatus() != Plugin_Created);
dvander@309	1963	int len = 0;
dvander@309	1964	const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@351	1965	if (pl->GetStatus() != Plugin_Running)
dvander@340	1966	{
dvander@340	1967	len += UTIL_Format(buffer, sizeof(buffer), " %02d <%s>", id, GetStatusText(pl->GetStatus()));
dvander@340	1968	} else {
dvander@340	1969	len += UTIL_Format(buffer, sizeof(buffer), " %02d", id);
dvander@340	1970	}
dvander@309	1971	len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " \"%s\"", (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
dvander@309	1972	if (IS_STR_FILLED(info->version))
dvander@309	1973	{
dvander@309	1974	len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " (%s)", info->version);
dvander@309	1975	}
dvander@309	1976	if (IS_STR_FILLED(info->author))
dvander@309	1977	{
dvander@309	1978	UTIL_Format(&buffer[len], sizeof(buffer)-len, " by %s", info->author);
dvander@309	1979	}
dvander@309	1980	g_RootMenu.ConsolePrint("%s", buffer);
dvander@309	1981	}
dvander@309	1982
dvander@309	1983	iter->Release();
dvander@309	1984	return;
dvander@309	1985	} else if (strcmp(cmd, "load") == 0) {
dvander@309	1986	if (argcount < 4)
dvander@309	1987	{
dvander@310	1988	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins load <file>");
dvander@309	1989	return;
dvander@309	1990	}
dvander@309	1991
dvander@309	1992	char error[128];
faluco@383	1993	bool wasloaded;
dvander@309	1994	const char *filename = g_RootMenu.GetArgument(3);
ds@794	1995
ds@794	1996	char pluginfile[256];
ds@794	1997	const char *ext = g_LibSys.GetFileExtension(filename) ? "" : ".smx";
ds@794	1998	UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", filename, ext);
ds@794	1999
ds@794	2000	IPlugin *pl = LoadPlugin(pluginfile, false, PluginType_MapUpdated, error, sizeof(error), &wasloaded);
faluco@383	2001
faluco@383	2002	if (wasloaded)
faluco@383	2003	{
ds@794	2004	g_RootMenu.ConsolePrint("[SM] Plugin %s is already loaded.", pluginfile);
faluco@383	2005	return;
faluco@383	2006	}
dvander@309	2007
dvander@309	2008	if (pl)
dvander@309	2009	{
ds@794	2010	g_RootMenu.ConsolePrint("[SM] Loaded plugin %s successfully.", pluginfile);
dvander@309	2011	} else {
ds@794	2012	g_RootMenu.ConsolePrint("[SM] Plugin %s failed to load: %s.", pluginfile, error);
dvander@309	2013	}
dvander@309	2014
dvander@309	2015	return;
dvander@309	2016	} else if (strcmp(cmd, "unload") == 0) {
dvander@309	2017	if (argcount < 4)
dvander@309	2018	{
ds@1252	2019	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins unload <#\|file>");
dvander@309	2020	return;
dvander@309	2021	}
dvander@309	2022
ds@1252	2023	CPlugin *pl;
ds@1252	2024	char *end;
ds@1252	2025	const char *arg = g_RootMenu.GetArgument(3);
ds@1252	2026	int id = strtol(arg, &end, 10);
ds@1252	2027
ds@1252	2028	if (*end == '\0')
dvander@309	2029	{
ds@1252	2030	pl = GetPluginByOrder(id);
ds@1252	2031	if (!pl)
ds@1252	2032	{
ds@1252	2033	g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
ds@1252	2034	return;
ds@1252	2035	}
ds@1252	2036	} else {
ds@1252	2037	char pluginfile[256];
ds@1252	2038	const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
ds@1252	2039	UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
dvander@309	2040
ds@1252	2041	if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
ds@1252	2042	{
ds@1252	2043	g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
ds@1252	2044	return;
ds@1252	2045	}
ds@1252	2046	}
dvander@309	2047
ds@652	2048	char name[PLATFORM_MAX_PATH];
dvander@309	2049	const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@309	2050	strcpy(name, (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
dvander@309	2051
dvander@309	2052	if (UnloadPlugin(pl))
dvander@309	2053	{
dvander@310	2054	g_RootMenu.ConsolePrint("[SM] Plugin %s unloaded successfully.", name);
dvander@309	2055	} else {
dvander@310	2056	g_RootMenu.ConsolePrint("[SM] Failed to unload plugin %s.", name);
dvander@309	2057	}
dvander@309	2058
dvander@309	2059	return;
dvander@309	2060	} else if (strcmp(cmd, "info") == 0) {
dvander@309	2061	if (argcount < 4)
dvander@309	2062	{
dvander@310	2063	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins info <#>");
dvander@309	2064	return;
dvander@309	2065	}
dvander@309	2066
dvander@309	2067	int num = atoi(g_RootMenu.GetArgument(3));
dvander@309	2068	if (num < 1 \|\| num > (int)GetPluginCount())
dvander@309	2069	{
dvander@310	2070	g_RootMenu.ConsolePrint("[SM] Plugin index not found.");
dvander@309	2071	return;
dvander@309	2072	}
dvander@309	2073
dvander@315	2074	CPlugin *pl = GetPluginByOrder(num);
dvander@309	2075	const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@309	2076
dvander@309	2077	g_RootMenu.ConsolePrint(" Filename: %s", pl->GetFilename());
dvander@315	2078	if (pl->GetStatus() <= Plugin_Error)
dvander@309	2079	{
dvander@315	2080	if (IS_STR_FILLED(info->name))
dvander@315	2081	{
dvander@605	2082	if (IS_STR_FILLED(info->description))
dvander@605	2083	{
dvander@605	2084	g_RootMenu.ConsolePrint(" Title: %s (%s)", info->name, info->description);
dvander@605	2085	} else {
dvander@605	2086	g_RootMenu.ConsolePrint(" Title: %s", info->name);
dvander@605	2087	}
dvander@605	2088	}
dvander@605	2089	if (IS_STR_FILLED(info->author))
dvander@605	2090	{
dvander@605	2091	g_RootMenu.ConsolePrint(" Author: %s", info->author);
dvander@315	2092	}
dvander@315	2093	if (IS_STR_FILLED(info->version))
dvander@315	2094	{
dvander@315	2095	g_RootMenu.ConsolePrint(" Version: %s", info->version);
dvander@315	2096	}
dvander@315	2097	if (IS_STR_FILLED(info->url))
dvander@315	2098	{
dvander@315	2099	g_RootMenu.ConsolePrint(" URL: %s", info->url);
dvander@315	2100	}
dvander@605	2101	if (pl->GetStatus() == Plugin_Error)
dvander@315	2102	{
dvander@605	2103	g_RootMenu.ConsolePrint(" Error: %s", pl->m_errormsg);
dvander@605	2104	} else {
ds@860	2105	g_RootMenu.ConsolePrint(" Debugging: %s", pl->IsDebugging() ? "Yes" : "No");
ds@860	2106	g_RootMenu.ConsolePrint(" Running: %s", pl->GetStatus() == Plugin_Running ? "Yes" : "No");
ds@860	2107
ds@861	2108	const char *typestr = "";
ds@860	2109	switch (pl->GetType())
ds@860	2110	{
ds@860	2111	case PluginType_MapUpdated:
ds@860	2112	typestr = "Map Change if Updated";
ds@860	2113	break;
ds@860	2114	case PluginType_MapOnly:
ds@860	2115	typestr = "Map Change";
ds@860	2116	break;
ds@860	2117	case PluginType_Private:
ds@860	2118	case PluginType_Global:
ds@860	2119	typestr = "Never";
ds@860	2120	break;
ds@860	2121	}
ds@860	2122
ds@860	2123	g_RootMenu.ConsolePrint(" Reloads: %s", typestr);
dvander@315	2124	}
dvander@315	2125	} else {
dvander@315	2126	g_RootMenu.ConsolePrint(" Load error: %s", pl->m_errormsg);
dvander@315	2127	g_RootMenu.ConsolePrint(" File info: (title \"%s\") (version \"%s\")",
dvander@315	2128	info->name ? info->name : "<none>",
dvander@315	2129	info->version ? info->version : "<none>");
dvander@315	2130	if (IS_STR_FILLED(info->url))
dvander@315	2131	{
dvander@315	2132	g_RootMenu.ConsolePrint(" File URL: %s", info->url);
dvander@315	2133	}
dvander@309	2134	}
dvander@309	2135
dvander@309	2136	return;
dvander@309	2137	} else if (strcmp(cmd, "debug") == 0) {
dvander@309	2138	if (argcount < 5)
dvander@309	2139	{
dvander@310	2140	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins debug <#> [on\|off]");
dvander@309	2141	return;
dvander@309	2142	}
dvander@309	2143
dvander@309	2144	int num = atoi(g_RootMenu.GetArgument(3));
dvander@309	2145	if (num < 1 \|\| num > (int)GetPluginCount())
dvander@309	2146	{
dvander@310	2147	g_RootMenu.ConsolePrint("[SM] Plugin index not found.");
dvander@309	2148	return;
dvander@309	2149	}
dvander@309	2150
dvander@309	2151	int res;
dvander@309	2152	const char *mode = g_RootMenu.GetArgument(4);
dvander@309	2153	if ((res=strcmp("on", mode)) && strcmp("off", mode))
dvander@309	2154	{
dvander@310	2155	g_RootMenu.ConsolePrint("[SM] The only possible options are \"on\" and \"off.\"");
dvander@309	2156	return;
dvander@309	2157	}
dvander@309	2158
dvander@309	2159	bool debug;
dvander@309	2160	if (!res)
dvander@309	2161	{
dvander@309	2162	debug = true;
dvander@309	2163	} else {
dvander@309	2164	debug = false;
dvander@309	2165	}
dvander@309	2166
dvander@309	2167	CPlugin *pl = GetPluginByOrder(num);
dvander@309	2168	if (debug && pl->IsDebugging())
dvander@309	2169	{
dvander@309	2170	g_RootMenu.ConsolePrint("[SM] This plugin is already in debug mode.");
dvander@309	2171	return;
dvander@309	2172	} else if (!debug && !pl->IsDebugging()) {
dvander@309	2173	g_RootMenu.ConsolePrint("[SM] Debug mode is already disabled in this plugin.");
dvander@309	2174	return;
dvander@309	2175	}
dvander@309	2176
dvander@309	2177	char error[256];
dvander@309	2178	if (pl->ToggleDebugMode(debug, error, sizeof(error)))
dvander@309	2179	{
dvander@310	2180	g_RootMenu.ConsolePrint("[SM] Successfully toggled debug mode on plugin %s.", pl->GetFilename());
dvander@309	2181	return;
dvander@309	2182	} else {
dvander@309	2183	g_RootMenu.ConsolePrint("[SM] Could not toggle debug mode on plugin %s.", pl->GetFilename());
dvander@309	2184	g_RootMenu.ConsolePrint("[SM] Plugin returned error: %s", error);
dvander@309	2185	return;
dvander@309	2186	}
faluco@1348	2187	} else if (strcmp(cmd, "reload") == 0) {
faluco@1348	2188	if (argcount < 4)
faluco@1348	2189	{
faluco@1348	2190	g_RootMenu.ConsolePrint("[SM] Usage: sm plugins reload <#\|file>");
faluco@1348	2191	return;
faluco@1348	2192	}
faluco@1348	2193
faluco@1348	2194	CPlugin *pl;
faluco@1348	2195	char *end;
faluco@1348	2196	const char *arg = g_RootMenu.GetArgument(3);
faluco@1348	2197	int id = strtol(arg, &end, 10);
faluco@1348	2198
faluco@1348	2199	if (*end == '\0')
faluco@1348	2200	{
faluco@1348	2201	pl = GetPluginByOrder(id);
faluco@1348	2202	if (!pl)
faluco@1348	2203	{
faluco@1348	2204	g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
faluco@1348	2205	return;
faluco@1348	2206	}
faluco@1348	2207	} else {
faluco@1348	2208	char pluginfile[256];
faluco@1348	2209	const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
faluco@1348	2210	UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
faluco@1348	2211
faluco@1348	2212	if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
faluco@1348	2213	{
faluco@1348	2214	g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
faluco@1348	2215	return;
faluco@1348	2216	}
faluco@1348	2217	}
faluco@1348	2218
faluco@1348	2219	char name[PLATFORM_MAX_PATH];
faluco@1348	2220	const sm_plugininfo_t *info = pl->GetPublicInfo();
faluco@1348	2221	strcpy(name, (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
faluco@1348	2222
faluco@1348	2223	if (ReloadPlugin(pl))
faluco@1348	2224	{
faluco@1348	2225	g_RootMenu.ConsolePrint("[SM] Plugin %s reloaded successfully.", name);
faluco@1348	2226	} else {
faluco@1348	2227	g_RootMenu.ConsolePrint("[SM] Failed to reload plugin %s.", name);
faluco@1348	2228	}
faluco@1348	2229
faluco@1348	2230	return;
dvander@309	2231	}
dvander@309	2232	}
dvander@309	2233
dvander@309	2234	/* Draw the main menu */
dvander@309	2235	g_RootMenu.ConsolePrint("SourceMod Plugins Menu:");
dvander@309	2236	g_RootMenu.DrawGenericOption("list", "Show loaded plugins");
dvander@309	2237	g_RootMenu.DrawGenericOption("load", "Load a plugin");
dvander@309	2238	g_RootMenu.DrawGenericOption("unload", "Unload a plugin");
dvander@309	2239	g_RootMenu.DrawGenericOption("info", "Information about a plugin");
dvander@309	2240	g_RootMenu.DrawGenericOption("debug", "Toggle debug mode on a plugin");
faluco@1348	2241	g_RootMenu.DrawGenericOption("reload", "Reloads a plugin");
faluco@1348	2242	}
faluco@1348	2243
faluco@1348	2244	bool CPluginManager::ReloadPlugin(CPlugin *pl)
faluco@1348	2245	{
faluco@1348	2246	List<CPlugin *>::iterator iter;
faluco@1348	2247	char filename[PLATFORM_MAX_PATH];
faluco@1348	2248	bool debug, wasloaded;
faluco@1348	2249	PluginType ptype;
faluco@1348	2250	IPlugin *newpl;
faluco@1348	2251	int id = 1;
faluco@1348	2252
faluco@1348	2253	strcpy(filename, pl->m_filename);
faluco@1348	2254	debug = pl->IsDebugging();
faluco@1348	2255	ptype = pl->GetType();
faluco@1348	2256
faluco@1348	2257	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++, id++)
faluco@1348	2258	{
faluco@1348	2259	if ((*iter) == pl)
faluco@1348	2260	{
faluco@1348	2261	break;
faluco@1348	2262	}
faluco@1348	2263	}
faluco@1348	2264
faluco@1348	2265	if (!UnloadPlugin(pl))
faluco@1348	2266	{
faluco@1348	2267	return false;
faluco@1348	2268	}
faluco@1348	2269	if (!(newpl=LoadPlugin(filename, debug, ptype, NULL, 0, &wasloaded)))
faluco@1348	2270	{
faluco@1348	2271	return false;
faluco@1348	2272	}
faluco@1348	2273
faluco@1348	2274	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
faluco@1348	2275	{
faluco@1348	2276	if ((iter) == (CPlugin )newpl)
faluco@1348	2277	{
faluco@1348	2278	m_plugins.erase(iter);
faluco@1348	2279	break;
faluco@1348	2280	}
faluco@1348	2281	}
faluco@1348	2282
faluco@1348	2283	int i;
faluco@1348	2284	for (i=1, iter=m_plugins.begin(); iter!=m_plugins.end() && i<id; iter++, i++) {}
faluco@1348	2285	m_plugins.insert(iter, (CPlugin *)newpl);
faluco@1348	2286
faluco@1348	2287	return true;
dvander@309	2288	}
faluco@333	2289
faluco@333	2290	void CPluginManager::ReloadOrUnloadPlugins()
faluco@333	2291	{
faluco@333	2292	List<CPlugin *>::iterator iter;
faluco@333	2293	List<CPlugin *> tmp_list = m_plugins;
faluco@333	2294	CPlugin *pl;
faluco@333	2295	time_t t;
faluco@333	2296
faluco@333	2297	for (iter=tmp_list.begin(); iter!=tmp_list.end(); iter++)
faluco@333	2298	{
faluco@333	2299	pl = (*iter);
faluco@333	2300	if (pl->GetType() == PluginType_MapOnly)
faluco@333	2301	{
faluco@333	2302	UnloadPlugin((IPlugin *)pl);
faluco@333	2303	} else if (pl->GetType() == PluginType_MapUpdated) {
dvander@450	2304	t = pl->GetFileTimeStamp();
dvander@450	2305	if (!t \|\| t > pl->GetTimeStamp())
faluco@333	2306	{
faluco@333	2307	pl->SetTimeStamp(t);
faluco@333	2308	UnloadPlugin((IPlugin *)pl);
faluco@333	2309	}
faluco@333	2310	}
faluco@333	2311	}
faluco@333	2312	}
faluco@430	2313
faluco@430	2314	void CPluginManager::_SetPauseState(CPlugin *pl, bool paused)
faluco@430	2315	{
faluco@430	2316	List<IPluginsListener *>::iterator iter;
faluco@430	2317	IPluginsListener *pListener;
faluco@430	2318	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
faluco@430	2319	{
faluco@430	2320	pListener = (*iter);
faluco@430	2321	pListener->OnPluginPauseChange(pl, paused);
faluco@430	2322	}
dvander@464	2323	}
dvander@464	2324
dvander@601	2325	bool CPluginManager::AddFakeNative(IPluginFunction pFunction, const char name, SPVM_FAKENATIVE_FUNC func)
dvander@601	2326	{
dvander@601	2327	if (sm_trie_retrieve(m_pNativeLookup, name, NULL))
dvander@601	2328	{
dvander@601	2329	return false;
dvander@601	2330	}
dvander@601	2331
dvander@601	2332	FakeNative *pNative = new FakeNative;
dvander@601	2333
dvander@601	2334	pNative->func = g_pVM->CreateFakeNative(func, pNative);
dvander@601	2335	if (!pNative->func)
dvander@601	2336	{
dvander@601	2337	delete pNative;
dvander@601	2338	return false;
dvander@601	2339	}
dvander@601	2340
dvander@601	2341	pNative->call = pFunction;
dvander@601	2342	pNative->name.assign(name);
dvander@601	2343	pNative->ctx = pFunction->GetParentContext();
dvander@601	2344
dvander@601	2345	m_Natives.push_back(pNative);
dvander@601	2346	sm_trie_insert(m_pNativeLookup, name,pNative);
dvander@601	2347
dvander@605	2348	CPlugin *pPlugin = GetPluginByCtx(pNative->ctx->GetContext());
dvander@605	2349	pPlugin->m_fakeNatives.push_back(pNative);
dvander@605	2350
dvander@601	2351	return true;
dvander@601	2352	}
ds@611	2353
ds@611	2354	void CPluginManager::AddFunctionsToForward(const char name, IChangeableForward pForward)
ds@611	2355	{
ds@611	2356	List<CPlugin *>::iterator iter;
ds@611	2357	CPlugin *pPlugin;
ds@611	2358	IPluginFunction *pFunction;
ds@611	2359
ds@611	2360	for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++)
ds@611	2361	{
ds@611	2362	pPlugin = (*iter);
ds@611	2363
ds@611	2364	if (pPlugin->GetStatus() <= Plugin_Paused)
ds@611	2365	{
ds@611	2366	pFunction = pPlugin->GetBaseContext()->GetFunctionByName(name);
ds@611	2367
ds@611	2368	if (pFunction)
ds@611	2369	{
ds@611	2370	pForward->AddFunction(pFunction);
ds@611	2371	}
ds@611	2372	}
ds@611	2373	}
ds@611	2374	}
dvander@900	2375
dvander@900	2376	CPlugin CPluginManager::GetPluginFromIdentity(IdentityToken_t pToken)
dvander@900	2377	{
dvander@900	2378	if (pToken->type != g_PluginIdent)
dvander@900	2379	{
dvander@900	2380	return NULL;
dvander@900	2381	}
dvander@900	2382
dvander@900	2383	return (CPlugin *)(pToken->ptr);
dvander@900	2384	}
dvander@978	2385
dvander@1124	2386	void CPluginManager::OnLibraryAction(const char *lib, bool drop)
dvander@978	2387	{
dvander@978	2388	List<CPlugin *>::iterator iter;
dvander@978	2389
dvander@1124	2390	struct _pl
dvander@1124	2391	{
dvander@1124	2392	cell_t name;
dvander@1124	2393	cell_t file;
dvander@1124	2394	cell_t required;
dvander@1124	2395	} *plc;
dvander@1124	2396
dvander@1124	2397	const char *name = drop ? "OnLibraryRemoved" : "OnLibraryAdded";
dvander@1124	2398
dvander@1124	2399	for (iter=m_plugins.begin();
dvander@1124	2400	iter!=m_plugins.end();
dvander@1124	2401	iter++)
dvander@1124	2402	{
dvander@1124	2403	CPlugin pl = (iter);
dvander@1124	2404	if (pl->GetStatus() != Plugin_Running)
dvander@1124	2405	{
dvander@1124	2406	continue;
dvander@1124	2407	}
dvander@1124	2408	IPluginContext *pContext = pl->GetBaseContext();
dvander@1124	2409	IPluginFunction *pf = pContext->GetFunctionByName(name);
dvander@1124	2410	if (!pf)
dvander@1124	2411	{
dvander@1124	2412	continue;
dvander@1124	2413	}
dvander@1124	2414	uint32_t num_vars = pContext->GetPubVarsNum();
dvander@1124	2415	for (uint32_t i=0; i<num_vars; i++)
dvander@1124	2416	{
dvander@1124	2417	sp_pubvar_t *pubvar;
dvander@1124	2418	if (pContext->GetPubvarByIndex(i, &pubvar) != SP_ERROR_NONE)
dvander@1124	2419	{
dvander@1124	2420	continue;
dvander@1124	2421	}
dvander@1124	2422	if (strncmp(pubvar->name, "__pl_", 5) != 0)
dvander@1124	2423	{
dvander@1124	2424	continue;
dvander@1124	2425	}
dvander@1124	2426	plc = (_pl *)pubvar->offs;
dvander@1124	2427	if (plc->required)
dvander@1124	2428	{
dvander@1124	2429	continue;
dvander@1124	2430	}
dvander@1124	2431	char *str;
dvander@1124	2432	pContext->LocalToString(plc->name, &str);
dvander@1124	2433	if (strcmp(str, lib) != 0)
dvander@1124	2434	{
dvander@1124	2435	continue;
dvander@1124	2436	}
dvander@1124	2437	pf->PushString(lib);
dvander@1124	2438	pf->Execute(NULL);
dvander@1124	2439	}
dvander@1124	2440	}
dvander@978	2441	}
dvander@1124	2442
dvander@1124	2443	bool CPluginManager::LibraryExists(const char *lib)
dvander@1124	2444	{
dvander@1124	2445	List<CPlugin *>::iterator iter;
dvander@1124	2446
dvander@1124	2447	for (iter=m_plugins.begin();
dvander@1124	2448	iter!=m_plugins.end();
dvander@1124	2449	iter++)
dvander@1124	2450	{
dvander@1124	2451	CPlugin pl = (iter);
dvander@1124	2452	if (pl->GetStatus() != Plugin_Running)
dvander@1124	2453	{
dvander@1124	2454	continue;
dvander@1124	2455	}
dvander@1124	2456	List<String>::iterator s_iter;
dvander@1124	2457	for (s_iter = pl->m_Libraries.begin();
dvander@1124	2458	s_iter != pl->m_Libraries.end();
dvander@1124	2459	s_iter++)
dvander@1124	2460	{
dvander@1124	2461	if ((*s_iter).compare(lib) == 0)
dvander@1124	2462	{
dvander@1124	2463	return true;
dvander@1124	2464	}
dvander@1124	2465	}
dvander@1124	2466	}
dvander@1124	2467
dvander@1124	2468	return false;
dvander@1124	2469	}
faluco@1221	2470
faluco@1221	2471	void CPluginManager::AllPluginsLoaded()
faluco@1221	2472	{
faluco@1221	2473	List<CPlugin *>::iterator iter;
faluco@1221	2474	CPlugin *pl;
faluco@1221	2475
faluco@1221	2476	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
faluco@1221	2477	{
faluco@1221	2478	pl = (*iter);
faluco@1221	2479	pl->Call_OnAllPluginsLoaded();
faluco@1221	2480	}
dvander@1232	2481	}

author	David Anderson <dvander@alliedmods.net>
	Wed Aug 29 00:47:05 2007 +0000 (2007-08-29)
changeset 1382	5399d6f24417
parent 1348	ad1199b211ac
child 1383	9e05176c1e92
permissions	-rw-r--r--