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

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

author	David Anderson <dvander@alliedmods.net>
	Wed Aug 01 02:12:47 2007 +0000 (2007-08-01)
changeset 1231	17badaf66ce4
parent 1221	f2dc55c0ab6f
child 1232	9f123e1a6c28
permissions	-rw-r--r--