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

author	Scott Ehlert <ds@alliedmods.net>
	Fri Jun 01 06:33:54 2007 +0000 (2007-06-01)
changeset 873	a2136b3be10d
parent 868	bdbf922e246f
child 900	b38c83f9d526
permissions	-rw-r--r--