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

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

author	Borja Ferrer <faluco@alliedmods.net>
	Tue Jul 31 22:34:52 2007 +0000 (2007-07-31)
changeset 1221	f2dc55c0ab6f
parent 1200	623603b49096
child 1231	17badaf66ce4
permissions	-rw-r--r--