sourcemod-central: core/PluginSys.cpp@2fe3c4c0eb76 (annotated)

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

author	Anthony Iacono <anthonyiacono@gmail.com>
	Sun Aug 26 23:33:04 2012 -0400 (2012-08-26)
changeset 3632	2fe3c4c0eb76
parent 3601	d3a2fb7b3a1e
child 3700	d2194fab1dd5
permissions	-rw-r--r--