core/PluginSys.cpp
author Asher Baker <asherkin@gmail.com>
Fri Dec 14 15:15:12 2012 -0500 (2012-12-14)
changeset 3703 391951cdde71
parent 3700 d2194fab1dd5
child 3790 98b099a406d2
permissions -rw-r--r--
Fixed OnLibraryAdded/Removed not being called in all plugins (bug 5431, r=psychonic).
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
asherkin@3703
   504
void CPlugin::LibraryActions(LibraryAction action)
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
	{
asherkin@3703
   511
		g_PluginSys.OnLibraryAction((*iter).c_str(), action);
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
	{
asherkin@3703
   526
		LibraryActions(LibraryAction_Removed);
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
	{
asherkin@3703
   550
		LibraryActions(LibraryAction_Added);
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
	{
asherkin@3703
  1498
		OnLibraryAction((*s_iter).c_str(), LibraryAction_Added);
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
	{
asherkin@3703
  1600
		OnLibraryAction((*s_iter).c_str(), LibraryAction_Removed);
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@3700
  1608
		/* Notify plugin */
dvander@3700
  1609
		pPlugin->Call_OnPluginEnd();
dvander@3700
  1610
dvander@282
  1611
		/* Notify listeners of unloading */
dvander@282
  1612
		for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
dvander@282
  1613
		{
dvander@282
  1614
			pListener = (*iter);
dvander@282
  1615
			pListener->OnPluginUnloaded(pPlugin);
dvander@282
  1616
		}
dvander@282
  1617
	}
dvander@282
  1618
dvander@2086
  1619
	pPlugin->DropEverything();
faluco@1112
  1620
dvander@174
  1621
	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
dvander@174
  1622
	{
dvander@282
  1623
		/* Notify listeners of destruction */
dvander@174
  1624
		pListener = (*iter);
dvander@174
  1625
		pListener->OnPluginDestroyed(pPlugin);
dvander@174
  1626
	}
dvander@176
  1627
	
dvander@282
  1628
	/* Tell the plugin to delete itself */
dvander@174
  1629
	delete pPlugin;
dvander@174
  1630
dvander@174
  1631
	return true;
dvander@174
  1632
}
dvander@174
  1633
dvander@174
  1634
IPlugin *CPluginManager::FindPluginByContext(const sp_context_t *ctx)
dvander@174
  1635
{
dvander@2193
  1636
	IPlugin *pPlugin;
dvander@2193
  1637
	IPluginContext *pContext;
dvander@2193
  1638
dvander@2193
  1639
	pContext = reinterpret_cast<IPluginContext *>(const_cast<sp_context_t *>(ctx));
dvander@2193
  1640
dvander@2193
  1641
	if (pContext->GetKey(2, (void **)&pPlugin))
dvander@2193
  1642
	{
dvander@2193
  1643
		return pPlugin;
dvander@2193
  1644
	}
dvander@2193
  1645
dvander@2193
  1646
	return NULL;
dvander@2193
  1647
}
dvander@2193
  1648
dvander@2193
  1649
CPlugin *CPluginManager::GetPluginByCtx(const sp_context_t *ctx)
dvander@2193
  1650
{
dvander@2193
  1651
	return (CPlugin *)FindPluginByContext(ctx);
dvander@174
  1652
}
dvander@174
  1653
dvander@174
  1654
unsigned int CPluginManager::GetPluginCount()
dvander@174
  1655
{
dvander@174
  1656
	return m_plugins.size();
dvander@174
  1657
}
dvander@174
  1658
dvander@174
  1659
void CPluginManager::AddPluginsListener(IPluginsListener *listener)
dvander@174
  1660
{
dvander@174
  1661
	m_listeners.push_back(listener);
dvander@174
  1662
}
dvander@174
  1663
dvander@174
  1664
void CPluginManager::RemovePluginsListener(IPluginsListener *listener)
dvander@174
  1665
{
dvander@174
  1666
	m_listeners.remove(listener);
dvander@174
  1667
}
dvander@174
  1668
dvander@174
  1669
IPluginIterator *CPluginManager::GetPluginIterator()
dvander@174
  1670
{
dvander@174
  1671
	if (m_iters.empty())
dvander@174
  1672
	{
dvander@174
  1673
		return new CPluginIterator(&m_plugins);
dvander@174
  1674
	} else {
dvander@174
  1675
		CPluginIterator *iter = m_iters.front();
dvander@174
  1676
		m_iters.pop();
faluco@216
  1677
		iter->Reset();
dvander@174
  1678
		return iter;
dvander@174
  1679
	}
dvander@174
  1680
}
dvander@174
  1681
dvander@174
  1682
void CPluginManager::ReleaseIterator(CPluginIterator *iter)
dvander@174
  1683
{
dvander@174
  1684
	m_iters.push(iter);
dvander@174
  1685
}
dvander@176
  1686
dvander@203
  1687
bool CPluginManager::TestAliasMatch(const char *alias, const char *localpath)
dvander@203
  1688
{
dvander@203
  1689
	/* As an optimization, we do not call strlen, but compute the length in the first pass */
dvander@203
  1690
	size_t alias_len = 0;
dvander@203
  1691
	size_t local_len = 0;
dvander@203
  1692
dvander@203
  1693
	const char *ptr = alias;
dvander@203
  1694
	unsigned int alias_explicit_paths = 0;
dvander@203
  1695
	unsigned int alias_path_end = 0;
dvander@203
  1696
	while (*ptr != '\0')
dvander@203
  1697
	{
dvander@203
  1698
		if (*ptr == '\\' || *ptr == '/')
dvander@203
  1699
		{
dvander@203
  1700
			alias_explicit_paths++;
dvander@203
  1701
			alias_path_end = alias_len;
dvander@203
  1702
		}
dvander@203
  1703
		alias_len++;
dvander@203
  1704
		ptr++;
dvander@203
  1705
	}
dvander@203
  1706
dvander@209
  1707
	if (alias_explicit_paths && alias_path_end == alias_len - 1)
dvander@203
  1708
	{
dvander@203
  1709
		/* Trailing slash is totally invalid here */
dvander@203
  1710
		return false;
dvander@203
  1711
	}
dvander@203
  1712
dvander@203
  1713
	ptr = localpath;
dvander@203
  1714
	unsigned int local_explicit_paths = 0;
dvander@203
  1715
	unsigned int local_path_end = 0;
dvander@203
  1716
	while (*ptr != '\0')
dvander@203
  1717
	{
dvander@203
  1718
		if (*ptr == '\\' || *ptr == '/')
dvander@203
  1719
		{
dvander@203
  1720
			local_explicit_paths++;
dvander@203
  1721
			local_path_end = local_len;
dvander@203
  1722
		}
dvander@203
  1723
		local_len++;
dvander@203
  1724
		ptr++;
dvander@203
  1725
	}
dvander@203
  1726
dvander@203
  1727
	/* If the alias has more explicit paths than the real path,
dvander@203
  1728
	 * no match will be possible.
dvander@203
  1729
	 */
dvander@203
  1730
	if (alias_explicit_paths > local_explicit_paths)
dvander@203
  1731
	{
dvander@203
  1732
		return false;
dvander@203
  1733
	}
dvander@203
  1734
dvander@203
  1735
	if (alias_explicit_paths)
dvander@203
  1736
	{
dvander@203
  1737
		/* We need to find if the paths match now.  For example, these should all match:
dvander@203
  1738
		 * csdm     csdm
dvander@203
  1739
		 * csdm     optional/csdm
dvander@203
  1740
		 * csdm/ban optional/crab/csdm/ban
dvander@203
  1741
		*/
dvander@203
  1742
		const char *aliasptr = alias;
dvander@203
  1743
		const char *localptr = localpath;
dvander@203
  1744
		bool match = true;
dvander@203
  1745
		do
dvander@203
  1746
		{
dvander@203
  1747
			if (*aliasptr != *localptr)
dvander@203
  1748
			{
dvander@203
  1749
				/* We have to knock one path off */
dvander@203
  1750
				local_explicit_paths--;
dvander@203
  1751
				if (alias_explicit_paths > local_explicit_paths)
dvander@203
  1752
				{
dvander@203
  1753
					/* Skip out if we're gonna have an impossible match */
dvander@203
  1754
					return false;
dvander@203
  1755
				}
dvander@203
  1756
				/* Eat up localptr tokens until we get a result */
dvander@203
  1757
				while (((localptr - localpath) < (int)local_path_end)
dvander@203
  1758
					&& *localptr != '/'
dvander@203
  1759
					&& *localptr != '\\')
dvander@203
  1760
				{
dvander@203
  1761
					localptr++;
dvander@203
  1762
				}
dvander@203
  1763
				/* Check if we hit the end of our searchable area.
dvander@203
  1764
				 * This probably isn't possible because of the path 
dvander@203
  1765
				 * count check, but it's a good idea anyway.
dvander@203
  1766
				 */
dvander@203
  1767
				if ((localptr - localpath) >= (int)local_path_end)
dvander@203
  1768
				{
dvander@203
  1769
					return false;
dvander@203
  1770
				} else {
dvander@203
  1771
					/* Consume the slash token */
dvander@203
  1772
					localptr++;
dvander@203
  1773
				}
dvander@203
  1774
				/* Reset the alias pointer so we can continue consuming */
dvander@203
  1775
				aliasptr = alias;
dvander@203
  1776
				match = false;
dvander@203
  1777
				continue;
dvander@203
  1778
			}
dvander@203
  1779
			/* Note:
dvander@203
  1780
			 * This is safe because if localptr terminates early, aliasptr will too
dvander@203
  1781
			 */
dvander@203
  1782
			do
dvander@203
  1783
			{
dvander@203
  1784
				/* We should never reach the end of the string because of this check. */
dvander@203
  1785
				bool aliasend = (aliasptr - alias) > (int)alias_path_end;
dvander@203
  1786
				bool localend = (localptr - localpath) > (int)local_path_end;
dvander@203
  1787
				if (aliasend || localend)
dvander@203
  1788
				{
dvander@203
  1789
					if (aliasend && localend)
dvander@203
  1790
					{
dvander@203
  1791
						/* we matched, and we can break out now */
dvander@203
  1792
						match = true;
dvander@203
  1793
						break;
dvander@203
  1794
					}
dvander@203
  1795
					/* Otherwise, we've hit the end somehow and rest won't match up.  Break out. */
dvander@203
  1796
					match = false;
dvander@203
  1797
					break;
dvander@203
  1798
				}
dvander@203
  1799
dvander@203
  1800
				/* If we got here, it's safe to compare the next two tokens */
dvander@203
  1801
				if (*localptr != *aliasptr)
dvander@203
  1802
				{
dvander@203
  1803
					match = false;
dvander@203
  1804
					break;
dvander@203
  1805
				}
dvander@203
  1806
				localptr++;
dvander@203
  1807
				aliasptr++;
dvander@203
  1808
			} while (true);
dvander@203
  1809
		} while (!match);
dvander@203
  1810
	}
dvander@203
  1811
dvander@203
  1812
	/* If we got here, it's time to compare filenames */
dvander@203
  1813
	const char *aliasptr = alias;
dvander@203
  1814
	const char *localptr = localpath;
dvander@203
  1815
dvander@203
  1816
	if (alias_explicit_paths)
dvander@203
  1817
	{
dvander@203
  1818
		aliasptr = &alias[alias_path_end + 1];
dvander@203
  1819
	}
dvander@203
  1820
dvander@203
  1821
	if (local_explicit_paths)
dvander@203
  1822
	{
dvander@203
  1823
		localptr = &localpath[local_path_end + 1];
dvander@203
  1824
	}
dvander@203
  1825
dvander@203
  1826
	while (true)
dvander@203
  1827
	{
dvander@203
  1828
		if (*aliasptr == '*')
dvander@203
  1829
		{
dvander@203
  1830
			/* First, see if this is the last character */
dvander@351
  1831
			if ((unsigned)(aliasptr - alias) == alias_len - 1)
dvander@203
  1832
			{
dvander@203
  1833
				/* If so, there's no need to match anything else */
dvander@203
  1834
				return true;
dvander@203
  1835
			}
dvander@203
  1836
			/* Otherwise, we need to search for an appropriate matching sequence in local.
dvander@203
  1837
			 * Note that we only need to search up to the next asterisk.
dvander@203
  1838
			 */
dvander@203
  1839
			aliasptr++;
dvander@203
  1840
			bool match = true;
dvander@203
  1841
			const char *local_orig = localptr;
dvander@203
  1842
			do
dvander@203
  1843
			{
dvander@203
  1844
				match = true;
dvander@203
  1845
				while (*aliasptr != '\0' && *aliasptr != '*')
dvander@203
  1846
				{
dvander@203
  1847
					/* Since aliasptr is never '\0', localptr hitting the end will fail */
dvander@203
  1848
					if (*aliasptr != *localptr)
dvander@203
  1849
					{
dvander@203
  1850
						match = false;
dvander@203
  1851
						break;
dvander@203
  1852
					}
dvander@203
  1853
					aliasptr++;
dvander@203
  1854
					localptr++;
dvander@203
  1855
				}
dvander@203
  1856
				if (!match)
dvander@203
  1857
				{
dvander@203
  1858
					/* If we didn't get a match, we need to advance the search stream.
dvander@203
  1859
					 * This will let us skip tokens while still searching for another match.
dvander@203
  1860
					 */
dvander@203
  1861
					localptr = ++local_orig;
dvander@203
  1862
					/* Make sure we don't go out of bounds */
dvander@203
  1863
					if (*localptr == '\0')
dvander@203
  1864
					{
dvander@203
  1865
						break;
dvander@203
  1866
					}
dvander@203
  1867
				}
dvander@203
  1868
			} while (!match);
dvander@203
  1869
dvander@203
  1870
			if (!match)
dvander@203
  1871
			{
dvander@203
  1872
				return false;
dvander@203
  1873
			} else {
dvander@203
  1874
				/* If we got a match, move on to the next token */
dvander@203
  1875
				continue;
dvander@203
  1876
			}
dvander@203
  1877
		} else if (*aliasptr == '\0') {
dvander@203
  1878
			if (*localptr == '\0'
dvander@203
  1879
				||
dvander@203
  1880
				strcmp(localptr, ".smx") == 0)
dvander@203
  1881
			{
dvander@203
  1882
				return true;
dvander@203
  1883
			} else {
dvander@203
  1884
				return false;
dvander@203
  1885
			}
dvander@203
  1886
		} else if (*aliasptr != *localptr) {
dvander@203
  1887
			return false;
dvander@203
  1888
		}
dvander@203
  1889
		aliasptr++;
dvander@203
  1890
		localptr++;
dvander@203
  1891
	}
dvander@203
  1892
dvander@203
  1893
	return true;
dvander@203
  1894
}
dvander@203
  1895
faluco@290
  1896
bool CPluginManager::IsLateLoadTime() const
dvander@176
  1897
{
dvander@315
  1898
	return (m_AllPluginsLoaded || !g_SourceMod.IsMapLoading());
dvander@176
  1899
}
dvander@220
  1900
dvander@220
  1901
void CPluginManager::OnSourceModAllInitialized()
dvander@220
  1902
{
dvander@229
  1903
	m_MyIdent = g_ShareSys.CreateCoreIdentity();
dvander@229
  1904
dvander@262
  1905
	HandleAccess sec;
dvander@262
  1906
	g_HandleSys.InitAccessDefaults(NULL, &sec);
dvander@220
  1907
dvander@262
  1908
	sec.access[HandleAccess_Delete] = HANDLE_RESTRICT_IDENTITY;
dvander@262
  1909
	sec.access[HandleAccess_Clone] = HANDLE_RESTRICT_IDENTITY;
dvander@262
  1910
dvander@262
  1911
	g_PluginType = g_HandleSys.CreateType("Plugin", this, 0, NULL, &sec, m_MyIdent, NULL);
dvander@229
  1912
	g_PluginIdent = g_ShareSys.CreateIdentType("PLUGIN");
dvander@309
  1913
dvander@309
  1914
	g_RootMenu.AddRootConsoleCommand("plugins", "Manage Plugins", this);
dvander@336
  1915
dvander@336
  1916
	g_ShareSys.AddInterface(NULL, this);
asherkin@3703
  1917
	
asherkin@3703
  1918
	m_pOnLibraryAdded = g_Forwards.CreateForward("OnLibraryAdded", ET_Ignore, 1, NULL, Param_String);
asherkin@3703
  1919
	m_pOnLibraryRemoved = g_Forwards.CreateForward("OnLibraryRemoved", ET_Ignore, 1, NULL, Param_String);
dvander@220
  1920
}
dvander@220
  1921
dvander@220
  1922
void CPluginManager::OnSourceModShutdown()
dvander@220
  1923
{
dvander@309
  1924
	g_RootMenu.RemoveRootConsoleCommand("plugins", this);
dvander@309
  1925
dvander@1630
  1926
	UnloadAll();
dvander@298
  1927
dvander@229
  1928
	g_HandleSys.RemoveType(g_PluginType, m_MyIdent);
dvander@229
  1929
	g_ShareSys.DestroyIdentType(g_PluginIdent);
dvander@229
  1930
	g_ShareSys.DestroyIdentity(m_MyIdent);
asherkin@3703
  1931
	
asherkin@3703
  1932
	g_Forwards.ReleaseForward(m_pOnLibraryAdded);
asherkin@3703
  1933
	g_Forwards.ReleaseForward(m_pOnLibraryRemoved);
dvander@220
  1934
}
dvander@220
  1935
asherkin@3601
  1936
ConfigResult CPluginManager::OnSourceModConfigChanged(const char *key, 
asherkin@3601
  1937
												 const char *value, 
asherkin@3601
  1938
												 ConfigSource source, 
asherkin@3601
  1939
												 char *error, 
asherkin@3601
  1940
												 size_t maxlength)
asherkin@3601
  1941
{
asherkin@3601
  1942
	if (strcmp(key, "BlockBadPlugins") == 0) {
asherkin@3601
  1943
		if (strcasecmp(value, "yes") == 0)
asherkin@3601
  1944
		{
asherkin@3601
  1945
			m_bBlockBadPlugins = true;
asherkin@3601
  1946
		} else if (strcasecmp(value, "no") == 0) {
asherkin@3601
  1947
			m_bBlockBadPlugins = false;
asherkin@3601
  1948
		} else {
asherkin@3601
  1949
			UTIL_Format(error, maxlength, "Invalid value: must be \"yes\" or \"no\"");
asherkin@3601
  1950
			return ConfigResult_Reject;
asherkin@3601
  1951
		}
asherkin@3601
  1952
		return ConfigResult_Accept;
asherkin@3601
  1953
	}
asherkin@3601
  1954
	return ConfigResult_Ignore;
asherkin@3601
  1955
}
asherkin@3601
  1956
dvander@220
  1957
void CPluginManager::OnHandleDestroy(HandleType_t type, void *object)
dvander@220
  1958
{
dvander@220
  1959
	/* We don't care about the internal object, actually */
dvander@220
  1960
}
dvander@233
  1961
dvander@1766
  1962
bool CPluginManager::GetHandleApproxSize(HandleType_t type, void *object, unsigned int *pSize)
dvander@1766
  1963
{
dvander@1766
  1964
	*pSize = ((CPlugin *)object)->CalcMemUsage();
dvander@1766
  1965
	return true;
dvander@1766
  1966
}
dvander@1766
  1967
dvander@247
  1968
IPlugin *CPluginManager::PluginFromHandle(Handle_t handle, HandleError *err)
dvander@247
  1969
{
dvander@247
  1970
	IPlugin *pPlugin;
dvander@247
  1971
	HandleError _err;
dvander@247
  1972
dvander@262
  1973
	HandleSecurity sec;
dvander@262
  1974
dvander@262
  1975
	sec.pOwner = NULL;
dvander@262
  1976
	sec.pIdentity = m_MyIdent;
dvander@262
  1977
dvander@262
  1978
	if ((_err=g_HandleSys.ReadHandle(handle, g_PluginType, &sec, (void **)&pPlugin)) != HandleError_None)
dvander@247
  1979
	{
dvander@247
  1980
		pPlugin = NULL;
dvander@247
  1981
	}
dvander@247
  1982
dvander@247
  1983
	if (err)
dvander@247
  1984
	{
dvander@247
  1985
		*err = _err;
dvander@247
  1986
	}
dvander@247
  1987
dvander@247
  1988
	return pPlugin;
dvander@247
  1989
}
faluco@290
  1990
faluco@290
  1991
CPlugin *CPluginManager::GetPluginByOrder(int num)
faluco@290
  1992
{
faluco@290
  1993
	if (num < 1 || num > (int)GetPluginCount())
faluco@290
  1994
	{
faluco@290
  1995
		return NULL;
faluco@290
  1996
	}
faluco@290
  1997
faluco@290
  1998
	CPlugin *pl;
faluco@290
  1999
	int id = 1;
faluco@290
  2000
ds@2113
  2001
	SourceHook::List<CPlugin *>::iterator iter;
ds@2113
  2002
	for (iter = m_plugins.begin(); iter != m_plugins.end() && id < num; iter++, id++) {}
faluco@290
  2003
ds@2113
  2004
	pl = *iter;
faluco@290
  2005
faluco@290
  2006
	return pl;
dvander@298
  2007
}
dvander@309
  2008
dvander@309
  2009
const char *CPluginManager::GetStatusText(PluginStatus st)
dvander@309
  2010
{
dvander@309
  2011
	switch (st)
dvander@309
  2012
	{
dvander@309
  2013
	case Plugin_Running:
dvander@309
  2014
		return "Running";
dvander@309
  2015
	case Plugin_Paused:
dvander@309
  2016
		return "Paused";
dvander@309
  2017
	case Plugin_Error:
dvander@309
  2018
		return "Error";
dvander@309
  2019
	case Plugin_Uncompiled:
dvander@309
  2020
		return "Uncompiled";
dvander@309
  2021
	case Plugin_BadLoad:
dvander@309
  2022
		return "Bad Load";
dvander@309
  2023
	case Plugin_Failed:
dvander@309
  2024
		return "Failed";
dvander@309
  2025
	default:
dvander@309
  2026
		assert(false);
dvander@309
  2027
		return "-";
dvander@309
  2028
	}
dvander@309
  2029
}
dvander@309
  2030
faluco@1541
  2031
void CPluginManager::OnRootConsoleCommand(const char *cmdname, const CCommand &command)
dvander@309
  2032
{
faluco@1541
  2033
	int argcount = command.ArgC();
dvander@309
  2034
	if (argcount >= 3)
dvander@309
  2035
	{
faluco@1541
  2036
		const char *cmd = command.Arg(2);
dvander@309
  2037
		if (strcmp(cmd, "list") == 0)
dvander@309
  2038
		{
dvander@309
  2039
			char buffer[256];
dvander@309
  2040
			unsigned int id = 1;
dvander@309
  2041
			int plnum = GetPluginCount();
dvander@309
  2042
dvander@309
  2043
			if (!plnum)
dvander@309
  2044
			{
dvander@309
  2045
				g_RootMenu.ConsolePrint("[SM] No plugins loaded");
dvander@309
  2046
				return;
dvander@1630
  2047
			}
dvander@1630
  2048
			else
dvander@1630
  2049
			{
ds@582
  2050
				g_RootMenu.ConsolePrint("[SM] Listing %d plugin%s:", GetPluginCount(), (plnum > 1) ? "s" : "");
dvander@309
  2051
			}
dvander@309
  2052
dvander@2105
  2053
			CPlugin *pl;
dvander@2105
  2054
			SourceHook::List<CPlugin *>::iterator iter;
dvander@2105
  2055
			SourceHook::List<CPlugin *> m_FailList;
pred@1920
  2056
ds@2113
  2057
			for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++, id++)
dvander@309
  2058
			{
dvander@2105
  2059
				pl = (*iter);
dvander@309
  2060
				assert(pl->GetStatus() != Plugin_Created);
dvander@309
  2061
				int len = 0;
dvander@309
  2062
				const sm_plugininfo_t *info = pl->GetPublicInfo();
fyrenmoo@2675
  2063
				if (pl->GetStatus() != Plugin_Running && !pl->IsSilentlyFailed())
dvander@340
  2064
				{
dvander@340
  2065
					len += UTIL_Format(buffer, sizeof(buffer), "  %02d <%s>", id, GetStatusText(pl->GetStatus()));
pred@1920
  2066
pred@1920
  2067
					if (pl->GetStatus() <= Plugin_Error)
pred@1920
  2068
					{
pred@1920
  2069
						/* Plugin has failed to load. */
pred@1920
  2070
						m_FailList.push_back(pl);
pred@1920
  2071
					}
dvander@1630
  2072
				}
dvander@1630
  2073
				else
dvander@1630
  2074
				{
dvander@340
  2075
					len += UTIL_Format(buffer, sizeof(buffer), "  %02d", id);
dvander@340
  2076
				}
dvander@2342
  2077
				if (pl->GetStatus() < Plugin_Created)
dvander@309
  2078
				{
fyrenmoo@2675
  2079
					if (pl->IsSilentlyFailed())
fyrenmoo@2675
  2080
						len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " Disabled:");
dvander@2342
  2081
					len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " \"%s\"", (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
dvander@2342
  2082
					if (IS_STR_FILLED(info->version))
dvander@2342
  2083
					{
dvander@2342
  2084
						len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " (%s)", info->version);
dvander@2342
  2085
					}
dvander@2342
  2086
					if (IS_STR_FILLED(info->author))
dvander@2342
  2087
					{
dvander@2342
  2088
						UTIL_Format(&buffer[len], sizeof(buffer)-len, " by %s", info->author);
dvander@2342
  2089
					}
dvander@309
  2090
				}
dvander@2342
  2091
				else
dvander@309
  2092
				{
dvander@2342
  2093
					UTIL_Format(&buffer[len], sizeof(buffer)-len, " %s", pl->m_filename);
dvander@309
  2094
				}
dvander@309
  2095
				g_RootMenu.ConsolePrint("%s", buffer);
dvander@309
  2096
			}
dvander@309
  2097
pred@1920
  2098
			if (!m_FailList.empty())
pred@1920
  2099
			{
pred@1920
  2100
				g_RootMenu.ConsolePrint("Load Errors:");
pred@1920
  2101
dvander@2105
  2102
				SourceHook::List<CPlugin *>::iterator _iter;
pred@1920
  2103
pred@1920
  2104
				CPlugin *pl;
pred@1920
  2105
pred@1920
  2106
				for (_iter=m_FailList.begin(); _iter!=m_FailList.end(); _iter++)
pred@1920
  2107
				{
pred@1920
  2108
					pl = (CPlugin *)*_iter;
pred@1920
  2109
					g_RootMenu.ConsolePrint("%s: %s",(IS_STR_FILLED(pl->GetPublicInfo()->name)) ? pl->GetPublicInfo()->name : pl->GetFilename(), pl->m_errormsg);
pred@1920
  2110
				}
pred@1920
  2111
			}
pred@1920
  2112
dvander@309
  2113
			return;
dvander@1630
  2114
		}
dvander@1630
  2115
		else if (strcmp(cmd, "load") == 0)
dvander@1630
  2116
		{
dvander@309
  2117
			if (argcount < 4)
dvander@309
  2118
			{
dvander@310
  2119
				g_RootMenu.ConsolePrint("[SM] Usage: sm plugins load <file>");
dvander@309
  2120
				return;
dvander@309
  2121
			}
dvander@309
  2122
dvander@309
  2123
			char error[128];
faluco@383
  2124
			bool wasloaded;
faluco@1541
  2125
			const char *filename = command.Arg(3);
ds@794
  2126
ds@794
  2127
			char pluginfile[256];
ds@794
  2128
			const char *ext = g_LibSys.GetFileExtension(filename) ? "" : ".smx";
fyrenmoo@2869
  2129
			g_SourceMod.BuildPath(Path_None, pluginfile, sizeof(pluginfile), "%s%s", filename, ext);
ds@794
  2130
ds@794
  2131
			IPlugin *pl = LoadPlugin(pluginfile, false, PluginType_MapUpdated, error, sizeof(error), &wasloaded);
faluco@383
  2132
faluco@383
  2133
			if (wasloaded)
faluco@383
  2134
			{
ds@794
  2135
				g_RootMenu.ConsolePrint("[SM] Plugin %s is already loaded.", pluginfile);
faluco@383
  2136
				return;
faluco@383
  2137
			}
dvander@309
  2138
dvander@309
  2139
			if (pl)
dvander@309
  2140
			{
ds@794
  2141
				g_RootMenu.ConsolePrint("[SM] Loaded plugin %s successfully.", pluginfile);
dvander@1630
  2142
			}
dvander@1630
  2143
			else
dvander@1630
  2144
			{
ds@794
  2145
				g_RootMenu.ConsolePrint("[SM] Plugin %s failed to load: %s.", pluginfile, error);
dvander@309
  2146
			}
dvander@309
  2147
dvander@309
  2148
			return;
dvander@1630
  2149
		}
dvander@1630
  2150
		else if (strcmp(cmd, "unload") == 0)
dvander@1630
  2151
		{
dvander@309
  2152
			if (argcount < 4)
dvander@309
  2153
			{
ds@1252
  2154
				g_RootMenu.ConsolePrint("[SM] Usage: sm plugins unload <#|file>");
dvander@309
  2155
				return;
dvander@309
  2156
			}
dvander@309
  2157
ds@1252
  2158
			CPlugin *pl;
ds@1252
  2159
			char *end;
faluco@1541
  2160
			const char *arg = command.Arg(3);
ds@1252
  2161
			int id = strtol(arg, &end, 10);
ds@1252
  2162
ds@1252
  2163
			if (*end == '\0')
dvander@309
  2164
			{
ds@1252
  2165
				pl = GetPluginByOrder(id);
ds@1252
  2166
				if (!pl)
ds@1252
  2167
				{
ds@1252
  2168
					g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
ds@1252
  2169
					return;
ds@1252
  2170
				}
dvander@1630
  2171
			}
dvander@1630
  2172
			else
dvander@1630
  2173
			{
ds@1252
  2174
				char pluginfile[256];
ds@1252
  2175
				const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
fyrenmoo@2869
  2176
				g_SourceMod.BuildPath(Path_None, pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
dvander@309
  2177
ds@1252
  2178
				if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
ds@1252
  2179
				{
ds@1252
  2180
					g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
ds@1252
  2181
					return;
ds@1252
  2182
				}
ds@1252
  2183
			}
dvander@309
  2184
ds@652
  2185
			char name[PLATFORM_MAX_PATH];
dvander@2342
  2186
dvander@2342
  2187
			if (pl->GetStatus() < Plugin_Created)
dvander@2342
  2188
			{
dvander@2342
  2189
				const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@2342
  2190
				UTIL_Format(name, sizeof(name), (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
dvander@2342
  2191
			}
dvander@2342
  2192
			else
dvander@2342
  2193
			{
dvander@2342
  2194
				UTIL_Format(name, sizeof(name), "%s", pl->GetFilename());
dvander@2342
  2195
			}
dvander@309
  2196
dvander@309
  2197
			if (UnloadPlugin(pl))
dvander@309
  2198
			{
dvander@310
  2199
				g_RootMenu.ConsolePrint("[SM] Plugin %s unloaded successfully.", name);
dvander@1630
  2200
			}
dvander@1630
  2201
			else
dvander@1630
  2202
			{
dvander@310
  2203
				g_RootMenu.ConsolePrint("[SM] Failed to unload plugin %s.", name);
dvander@309
  2204
			}
dvander@309
  2205
dvander@309
  2206
			return;
dvander@1630
  2207
		}
dvander@1630
  2208
		else if (strcmp(cmd, "unload_all") == 0)
dvander@1630
  2209
		{
dvander@1630
  2210
			g_PluginSys.UnloadAll();
dvander@1630
  2211
			g_RootMenu.ConsolePrint("[SM] All plugins have been unloaded.");
dvander@1630
  2212
			return;
dvander@1630
  2213
		}
dvander@1630
  2214
		else if (strcmp(cmd, "load_lock") == 0)
dvander@1630
  2215
		{
dvander@1630
  2216
			if (m_LoadingLocked)
dvander@1630
  2217
			{
dvander@1630
  2218
				g_RootMenu.ConsolePrint("[SM] There is already a loading lock in effect.");
dvander@1630
  2219
			}
dvander@1630
  2220
			else
dvander@1630
  2221
			{
dvander@1630
  2222
				m_LoadingLocked = true;
dvander@1630
  2223
				g_RootMenu.ConsolePrint("[SM] Loading is now locked; no plugins will be loaded or re-loaded.");
dvander@1630
  2224
			}
dvander@1630
  2225
			return;
dvander@1630
  2226
		}
dvander@1630
  2227
		else if (strcmp(cmd, "load_unlock") == 0)
dvander@1630
  2228
		{
dvander@1630
  2229
			if (m_LoadingLocked)
dvander@1630
  2230
			{
dvander@1630
  2231
				m_LoadingLocked = false;
dvander@1630
  2232
				g_RootMenu.ConsolePrint("[SM] The loading lock is no longer in effect.");
dvander@1630
  2233
			}
dvander@1630
  2234
			else
dvander@1630
  2235
			{
dvander@1630
  2236
				g_RootMenu.ConsolePrint("[SM] There was no loading lock in effect.");
dvander@1630
  2237
			}
dvander@1630
  2238
			return;
dvander@1630
  2239
		}
dvander@1630
  2240
		else if (strcmp(cmd, "info") == 0)
dvander@1630
  2241
		{
dvander@309
  2242
			if (argcount < 4)
dvander@309
  2243
			{
dvander@310
  2244
				g_RootMenu.ConsolePrint("[SM] Usage: sm plugins info <#>");
dvander@309
  2245
				return;
dvander@309
  2246
			}
dvander@309
  2247
dvander@1902
  2248
			CPlugin *pl;
dvander@1902
  2249
			char *end;
dvander@1902
  2250
			const char *arg = command.Arg(3);
dvander@1902
  2251
			int id = strtol(arg, &end, 10);
dvander@1902
  2252
dvander@1902
  2253
			if (*end == '\0')
dvander@309
  2254
			{
dvander@1902
  2255
				pl = GetPluginByOrder(id);
dvander@1902
  2256
				if (!pl)
dvander@1902
  2257
				{
dvander@1902
  2258
					g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
dvander@1902
  2259
					return;
dvander@1902
  2260
				}
dvander@1902
  2261
			}
dvander@1902
  2262
			else
dvander@1902
  2263
			{
dvander@1902
  2264
				char pluginfile[256];
dvander@1902
  2265
				const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
fyrenmoo@2869
  2266
				g_SourceMod.BuildPath(Path_None, pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
dvander@1902
  2267
dvander@1902
  2268
				if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
dvander@1902
  2269
				{
dvander@1902
  2270
					g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
dvander@1902
  2271
					return;
dvander@1902
  2272
				}
dvander@309
  2273
			}
dvander@309
  2274
dvander@309
  2275
			const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@309
  2276
dvander@309
  2277
			g_RootMenu.ConsolePrint("  Filename: %s", pl->GetFilename());
dvander@315
  2278
			if (pl->GetStatus() <= Plugin_Error)
dvander@309
  2279
			{
dvander@315
  2280
				if (IS_STR_FILLED(info->name))
dvander@315
  2281
				{
dvander@605
  2282
					if (IS_STR_FILLED(info->description))
dvander@605
  2283
					{
dvander@605
  2284
						g_RootMenu.ConsolePrint("  Title: %s (%s)", info->name, info->description);
dvander@605
  2285
					} else {
dvander@605
  2286
						g_RootMenu.ConsolePrint("  Title: %s", info->name);
dvander@605
  2287
					}
dvander@605
  2288
				}
dvander@605
  2289
				if (IS_STR_FILLED(info->author))
dvander@605
  2290
				{
dvander@605
  2291
					g_RootMenu.ConsolePrint("  Author: %s", info->author);
dvander@315
  2292
				}
dvander@315
  2293
				if (IS_STR_FILLED(info->version))
dvander@315
  2294
				{
dvander@315
  2295
					g_RootMenu.ConsolePrint("  Version: %s", info->version);
dvander@315
  2296
				}
dvander@315
  2297
				if (IS_STR_FILLED(info->url))
dvander@315
  2298
				{
dvander@315
  2299
					g_RootMenu.ConsolePrint("  URL: %s", info->url);
dvander@315
  2300
				}
dvander@605
  2301
				if (pl->GetStatus() == Plugin_Error)
dvander@315
  2302
				{
dvander@605
  2303
					g_RootMenu.ConsolePrint("  Error: %s", pl->m_errormsg);
dvander@1630
  2304
				}
dvander@1630
  2305
				else
dvander@1630
  2306
				{
dvander@2105
  2307
					if (pl->GetStatus() == Plugin_Running)
dvander@2105
  2308
					{
dvander@2226
  2309
						g_RootMenu.ConsolePrint("  Status: running");
dvander@2105
  2310
					}
dvander@2105
  2311
					else
dvander@2105
  2312
					{
dvander@2105
  2313
						g_RootMenu.ConsolePrint("  Status: not running");
dvander@2105
  2314
					}
ds@860
  2315
ds@861
  2316
					const char *typestr = "";
ds@860
  2317
					switch (pl->GetType())
ds@860
  2318
					{
ds@860
  2319
					case PluginType_MapUpdated:
ds@860
  2320
						typestr = "Map Change if Updated";
ds@860
  2321
						break;
ds@860
  2322
					case PluginType_MapOnly:
ds@860
  2323
						typestr = "Map Change";
ds@860
  2324
						break;
ds@860
  2325
					case PluginType_Private:
ds@860
  2326
					case PluginType_Global:
ds@860
  2327
						typestr = "Never";
ds@860
  2328
						break;
ds@860
  2329
					}
ds@860
  2330
ds@860
  2331
					g_RootMenu.ConsolePrint("  Reloads: %s", typestr);
dvander@315
  2332
				}
dvander@2105
  2333
				if (pl->m_FileVersion >= 3)
dvander@2105
  2334
				{
dvander@2105
  2335
					g_RootMenu.ConsolePrint("  Timestamp: %s", pl->m_DateTime);
dvander@2105
  2336
				}
asherkin@3601
  2337
				
asherkin@3601
  2338
				unsigned char *pCodeHash = pl->m_pRuntime->GetCodeHash();
asherkin@3601
  2339
				unsigned char *pDataHash = pl->m_pRuntime->GetDataHash();
asherkin@3601
  2340
				
asherkin@3601
  2341
				char combinedHash[33];
asherkin@3601
  2342
				for (int i = 0; i < 16; i++)
asherkin@3601
  2343
					UTIL_Format(combinedHash + (i * 2), 3, "%02x", pCodeHash[i] ^ pDataHash[i]);
asherkin@3601
  2344
				
asherkin@3601
  2345
				g_RootMenu.ConsolePrint("  Hash: %s", combinedHash);
dvander@1630
  2346
			}
dvander@1630
  2347
			else
dvander@1630
  2348
			{
dvander@315
  2349
				g_RootMenu.ConsolePrint("  Load error: %s", pl->m_errormsg);
dvander@2342
  2350
				if (pl->GetStatus() < Plugin_Created)
dvander@315
  2351
				{
dvander@2342
  2352
					g_RootMenu.ConsolePrint("  File info: (title \"%s\") (version \"%s\")", 
dvander@2342
  2353
											info->name ? info->name : "<none>",
dvander@2342
  2354
											info->version ? info->version : "<none>");
dvander@2342
  2355
					if (IS_STR_FILLED(info->url))
dvander@2342
  2356
					{
dvander@2342
  2357
						g_RootMenu.ConsolePrint("  File URL: %s", info->url);
dvander@2342
  2358
					}
dvander@315
  2359
				}
dvander@309
  2360
			}
dvander@309
  2361
dvander@309
  2362
			return;
dvander@1630
  2363
		}
dvander@1630
  2364
		else if (strcmp(cmd, "refresh") == 0)
dvander@1630
  2365
		{
dvander@1630
  2366
			g_SourceMod.DoGlobalPluginLoads();
dvander@1630
  2367
			g_RootMenu.ConsolePrint("[SM] The plugin list has been refreshed and reloaded.");
dvander@1630
  2368
			return;
dvander@1630
  2369
		}
dvander@1630
  2370
		else if (strcmp(cmd, "reload") == 0)
dvander@1630
  2371
		{
faluco@1348
  2372
			if (argcount < 4)
faluco@1348
  2373
			{
faluco@1348
  2374
				g_RootMenu.ConsolePrint("[SM] Usage: sm plugins reload <#|file>");
faluco@1348
  2375
				return;
faluco@1348
  2376
			}
faluco@1348
  2377
faluco@1348
  2378
			CPlugin *pl;
faluco@1348
  2379
			char *end;
faluco@1541
  2380
			const char *arg = command.Arg(3);
faluco@1348
  2381
			int id = strtol(arg, &end, 10);
faluco@1348
  2382
faluco@1348
  2383
			if (*end == '\0')
faluco@1348
  2384
			{
faluco@1348
  2385
				pl = GetPluginByOrder(id);
faluco@1348
  2386
				if (!pl)
faluco@1348
  2387
				{
faluco@1348
  2388
					g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
faluco@1348
  2389
					return;
faluco@1348
  2390
				}
dvander@1630
  2391
			}
dvander@1630
  2392
			else
dvander@1630
  2393
			{
faluco@1348
  2394
				char pluginfile[256];
faluco@1348
  2395
				const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
fyrenmoo@2869
  2396
				g_SourceMod.BuildPath(Path_None, pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
faluco@1348
  2397
faluco@1348
  2398
				if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
faluco@1348
  2399
				{
faluco@1348
  2400
					g_RootMenu.ConsolePrint("[SM] Plugin %s is not loaded.", pluginfile);
faluco@1348
  2401
					return;
faluco@1348
  2402
				}
faluco@1348
  2403
			}
faluco@1348
  2404
faluco@1348
  2405
			char name[PLATFORM_MAX_PATH];
faluco@1348
  2406
			const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@2886
  2407
			if (pl->GetStatus() <= Plugin_Paused) 
dvander@2885
  2408
				strcpy(name, (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
dvander@2885
  2409
			else
dvander@2885
  2410
				strcpy(name, pl->GetFilename());
faluco@1348
  2411
faluco@1348
  2412
			if (ReloadPlugin(pl))
faluco@1348
  2413
			{
faluco@1348
  2414
				g_RootMenu.ConsolePrint("[SM] Plugin %s reloaded successfully.", name);
dvander@1630
  2415
			}
dvander@1630
  2416
			else
dvander@1630
  2417
			{
faluco@1348
  2418
				g_RootMenu.ConsolePrint("[SM] Failed to reload plugin %s.", name);
faluco@1348
  2419
			}
faluco@1348
  2420
faluco@1348
  2421
			return;
dvander@309
  2422
		}
dvander@309
  2423
	}
dvander@309
  2424
dvander@309
  2425
	/* Draw the main menu */
dvander@309
  2426
	g_RootMenu.ConsolePrint("SourceMod Plugins Menu:");
dvander@1630
  2427
	g_RootMenu.DrawGenericOption("info", "Information about a plugin");
dvander@309
  2428
	g_RootMenu.DrawGenericOption("list", "Show loaded plugins");
dvander@309
  2429
	g_RootMenu.DrawGenericOption("load", "Load a plugin");
dvander@1630
  2430
	g_RootMenu.DrawGenericOption("load_lock", "Prevents any more plugins from being loaded");
dvander@1630
  2431
	g_RootMenu.DrawGenericOption("load_unlock", "Re-enables plugin loading");
dvander@1630
  2432
	g_RootMenu.DrawGenericOption("refresh", "Reloads/refreshes all plugins in the plugins folder");
dvander@1630
  2433
	g_RootMenu.DrawGenericOption("reload", "Reloads a plugin");
dvander@309
  2434
	g_RootMenu.DrawGenericOption("unload", "Unload a plugin");
dvander@1630
  2435
	g_RootMenu.DrawGenericOption("unload_all", "Unloads all plugins");
faluco@1348
  2436
}
faluco@1348
  2437
faluco@1348
  2438
bool CPluginManager::ReloadPlugin(CPlugin *pl)
faluco@1348
  2439
{
faluco@1348
  2440
	List<CPlugin *>::iterator iter;
faluco@1348
  2441
	char filename[PLATFORM_MAX_PATH];
dvander@2226
  2442
	bool wasloaded;
faluco@1348
  2443
	PluginType ptype;
faluco@1348
  2444
	IPlugin *newpl;
faluco@1348
  2445
	int id = 1;
faluco@1348
  2446
faluco@1348
  2447
	strcpy(filename, pl->m_filename);
faluco@1348
  2448
	ptype = pl->GetType();
faluco@1348
  2449
faluco@1348
  2450
	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++, id++)
faluco@1348
  2451
	{
faluco@1348
  2452
		if ((*iter) == pl)
faluco@1348
  2453
		{
faluco@1348
  2454
			break;
faluco@1348
  2455
		}
faluco@1348
  2456
	}
faluco@1348
  2457
faluco@1348
  2458
	if (!UnloadPlugin(pl))
faluco@1348
  2459
	{
faluco@1348
  2460
		return false;
faluco@1348
  2461
	}
dvander@2226
  2462
	if (!(newpl=LoadPlugin(filename, true, ptype, NULL, 0, &wasloaded)))
faluco@1348
  2463
	{
faluco@1348
  2464
		return false;
faluco@1348
  2465
	}
faluco@1348
  2466
faluco@1348
  2467
	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
faluco@1348
  2468
	{
faluco@1348
  2469
		if ((*iter) == (CPlugin *)newpl)
faluco@1348
  2470
		{
faluco@1348
  2471
			m_plugins.erase(iter);
faluco@1348
  2472
			break;
faluco@1348
  2473
		}
faluco@1348
  2474
	}
faluco@1348
  2475
faluco@1348
  2476
	int i;
dvander@1630
  2477
	for (i=1, iter=m_plugins.begin(); iter!=m_plugins.end() && i<id; iter++, i++)
dvander@1630
  2478
	{
dvander@1630
  2479
	}
faluco@1348
  2480
	m_plugins.insert(iter, (CPlugin *)newpl);
faluco@1348
  2481
faluco@1348
  2482
	return true;
dvander@309
  2483
}
faluco@333
  2484
faluco@333
  2485
void CPluginManager::ReloadOrUnloadPlugins()
faluco@333
  2486
{
dvander@1630
  2487
	/* If we're in a load lock, just skip this whole bit. */
dvander@1630
  2488
	if (m_LoadingLocked)
dvander@1630
  2489
	{
dvander@1630
  2490
		return;
dvander@1630
  2491
	}
dvander@1630
  2492
faluco@333
  2493
	List<CPlugin *>::iterator iter;
faluco@333
  2494
	List<CPlugin *> tmp_list = m_plugins;
faluco@333
  2495
	CPlugin *pl;
faluco@333
  2496
	time_t t;
faluco@333
  2497
faluco@333
  2498
	for (iter=tmp_list.begin(); iter!=tmp_list.end(); iter++)
faluco@333
  2499
	{
faluco@333
  2500
		pl = (*iter);
faluco@333
  2501
		if (pl->GetType() ==  PluginType_MapOnly)
faluco@333
  2502
		{
faluco@333
  2503
			UnloadPlugin((IPlugin *)pl);
dvander@1630
  2504
		}
dvander@1630
  2505
		else if (pl->GetType() == PluginType_MapUpdated)
dvander@1630
  2506
		{
dvander@450
  2507
			t = pl->GetFileTimeStamp();
dvander@450
  2508
			if (!t || t > pl->GetTimeStamp())
faluco@333
  2509
			{
faluco@333
  2510
				pl->SetTimeStamp(t);
faluco@333
  2511
				UnloadPlugin((IPlugin *)pl);
faluco@333
  2512
			}
faluco@333
  2513
		}
faluco@333
  2514
	}
faluco@333
  2515
}
faluco@430
  2516
faluco@430
  2517
void CPluginManager::_SetPauseState(CPlugin *pl, bool paused)
faluco@430
  2518
{
faluco@430
  2519
	List<IPluginsListener *>::iterator iter;
faluco@430
  2520
	IPluginsListener *pListener;
faluco@430
  2521
	for (iter=m_listeners.begin(); iter!=m_listeners.end(); iter++)
faluco@430
  2522
	{
faluco@430
  2523
		pListener = (*iter);
faluco@430
  2524
		pListener->OnPluginPauseChange(pl, paused);
faluco@430
  2525
	}	
dvander@464
  2526
}
dvander@464
  2527
ds@611
  2528
void CPluginManager::AddFunctionsToForward(const char *name, IChangeableForward *pForward)
ds@611
  2529
{
ds@611
  2530
	List<CPlugin *>::iterator iter;
ds@611
  2531
	CPlugin *pPlugin;
ds@611
  2532
	IPluginFunction *pFunction;
ds@611
  2533
ds@611
  2534
	for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++)
ds@611
  2535
	{
ds@611
  2536
		pPlugin = (*iter);
ds@611
  2537
ds@611
  2538
		if (pPlugin->GetStatus() <= Plugin_Paused)
ds@611
  2539
		{
ds@611
  2540
			pFunction = pPlugin->GetBaseContext()->GetFunctionByName(name);
ds@611
  2541
ds@611
  2542
			if (pFunction)
ds@611
  2543
			{
ds@611
  2544
				pForward->AddFunction(pFunction);
ds@611
  2545
			}
ds@611
  2546
		}
ds@611
  2547
	}
ds@611
  2548
}
dvander@900
  2549
dvander@900
  2550
CPlugin *CPluginManager::GetPluginFromIdentity(IdentityToken_t *pToken)
dvander@900
  2551
{
dvander@900
  2552
	if (pToken->type != g_PluginIdent)
dvander@900
  2553
	{
dvander@900
  2554
		return NULL;
dvander@900
  2555
	}
dvander@900
  2556
dvander@900
  2557
	return (CPlugin *)(pToken->ptr);
dvander@900
  2558
}
dvander@978
  2559
asherkin@3703
  2560
void CPluginManager::OnLibraryAction(const char *lib, LibraryAction action)
dvander@978
  2561
{
asherkin@3703
  2562
	switch (action)
dvander@1124
  2563
	{
asherkin@3703
  2564
	case LibraryAction_Removed:
asherkin@3703
  2565
		m_pOnLibraryRemoved->PushString(lib);
asherkin@3703
  2566
		m_pOnLibraryRemoved->Execute(NULL);
asherkin@3703
  2567
		break;
asherkin@3703
  2568
	case LibraryAction_Added:
asherkin@3703
  2569
		m_pOnLibraryAdded->PushString(lib);
asherkin@3703
  2570
		m_pOnLibraryAdded->Execute(NULL);
asherkin@3703
  2571
		break;
dvander@1124
  2572
	}
dvander@978
  2573
}
dvander@1124
  2574
dvander@1124
  2575
bool CPluginManager::LibraryExists(const char *lib)
dvander@1124
  2576
{
dvander@1124
  2577
	List<CPlugin *>::iterator iter;
dvander@1124
  2578
dvander@1124
  2579
	for (iter=m_plugins.begin();
dvander@1124
  2580
		 iter!=m_plugins.end();
dvander@1124
  2581
		 iter++)
dvander@1124
  2582
	{
dvander@1124
  2583
		CPlugin *pl = (*iter);
dvander@1124
  2584
		if (pl->GetStatus() != Plugin_Running)
dvander@1124
  2585
		{
dvander@1124
  2586
			continue;
dvander@1124
  2587
		}
dvander@1124
  2588
		List<String>::iterator s_iter;
dvander@1124
  2589
		for (s_iter = pl->m_Libraries.begin();
dvander@1124
  2590
			 s_iter != pl->m_Libraries.end();
dvander@1124
  2591
			 s_iter++)
dvander@1124
  2592
		{
dvander@1124
  2593
			if ((*s_iter).compare(lib) == 0)
dvander@1124
  2594
			{
dvander@1124
  2595
				return true;
dvander@1124
  2596
			}
dvander@1124
  2597
		}
dvander@1124
  2598
	}
dvander@1124
  2599
dvander@1124
  2600
	return false;
dvander@1124
  2601
}
faluco@1221
  2602
faluco@1221
  2603
void CPluginManager::AllPluginsLoaded()
faluco@1221
  2604
{
faluco@1221
  2605
	List<CPlugin *>::iterator iter;
faluco@1221
  2606
	CPlugin *pl;
faluco@1221
  2607
faluco@1221
  2608
	for (iter=m_plugins.begin(); iter!=m_plugins.end(); iter++)
faluco@1221
  2609
	{
faluco@1221
  2610
		pl = (*iter);
faluco@1221
  2611
		pl->Call_OnAllPluginsLoaded();
faluco@1221
  2612
	}
dvander@1232
  2613
}
dvander@1630
  2614
dvander@1630
  2615
void CPluginManager::UnloadAll()
dvander@1630
  2616
{
dvander@1630
  2617
	List<CPlugin *>::iterator iter;
dvander@1630
  2618
	while ( (iter = m_plugins.begin()) != m_plugins.end() )
dvander@1630
  2619
	{
dvander@1630
  2620
		UnloadPlugin((*iter));
dvander@1630
  2621
	}
dvander@1630
  2622
}
dvander@1638
  2623
dvander@1638
  2624
int CPluginManager::GetOrderOfPlugin(IPlugin *pl)
dvander@1638
  2625
{
dvander@1638
  2626
	int id = 1;
dvander@1638
  2627
	List<CPlugin *>::iterator iter;
dvander@1638
  2628
dvander@1638
  2629
	for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++, id++)
dvander@1638
  2630
	{
dvander@1638
  2631
		if ((*iter) == pl)
dvander@1638
  2632
		{
dvander@1638
  2633
			return id;
dvander@1638
  2634
		}
dvander@1638
  2635
	}
dvander@1638
  2636
dvander@1638
  2637
	return -1;
dvander@1638
  2638
}
dvander@1638
  2639
dvander@1902
  2640
CPlugin *CPluginManager::FindPluginByConsoleArg(const char *arg)
dvander@1902
  2641
{
dvander@1902
  2642
	int id;
dvander@1902
  2643
	char *end;
dvander@1902
  2644
	CPlugin *pl;
dvander@1902
  2645
	
dvander@1902
  2646
	id = strtol(arg, &end, 10);
dvander@1902
  2647
dvander@1902
  2648
	if (*end == '\0')
dvander@1902
  2649
	{
dvander@1902
  2650
		pl = GetPluginByOrder(id);
dvander@1902
  2651
		if (!pl)
dvander@1902
  2652
		{
dvander@1902
  2653
			return NULL;
dvander@1902
  2654
		}
dvander@1902
  2655
	}
dvander@1902
  2656
	else
dvander@1902
  2657
	{
dvander@1902
  2658
		char pluginfile[256];
dvander@1902
  2659
		const char *ext = g_LibSys.GetFileExtension(arg) ? "" : ".smx";
dvander@1902
  2660
		UTIL_Format(pluginfile, sizeof(pluginfile), "%s%s", arg, ext);
dvander@1902
  2661
dvander@1902
  2662
		if (!sm_trie_retrieve(m_LoadLookup, pluginfile, (void **)&pl))
dvander@1902
  2663
		{
dvander@1902
  2664
			return NULL;
dvander@1902
  2665
		}
dvander@1902
  2666
	}
dvander@1902
  2667
dvander@1902
  2668
	return pl;
dvander@1902
  2669
}
dvander@2342
  2670
dvander@2342
  2671
void CPluginManager::OnSourceModMaxPlayersChanged(int newvalue)
dvander@2342
  2672
{
dvander@2342
  2673
	SyncMaxClients(newvalue);
dvander@2342
  2674
}
dvander@2342
  2675
dvander@2342
  2676
void CPluginManager::SyncMaxClients(int max_clients)
dvander@2342
  2677
{
dvander@2342
  2678
	List<CPlugin *>::iterator iter;
dvander@2342
  2679
dvander@2342
  2680
	for (iter = m_plugins.begin(); iter != m_plugins.end(); iter++)
dvander@2342
  2681
	{
dvander@2342
  2682
		(*iter)->SyncMaxClients(max_clients);
dvander@2342
  2683
	}
dvander@2342
  2684
}
dvander@2547
  2685
dvander@2547
  2686
void CPluginManager::ListPluginsToClient(CPlayer *player, const CCommand &args)
dvander@2547
  2687
{
dvander@2547
  2688
	char buffer[256];
dvander@2547
  2689
	unsigned int id = 0;
dvander@2547
  2690
	int plnum = GetPluginCount();
dvander@2547
  2691
	edict_t *e = player->GetEdict();
dvander@2547
  2692
	unsigned int start = 0;
dvander@2547
  2693
dvander@2547
  2694
	if (!plnum)
dvander@2547
  2695
	{
dvander@2547
  2696
		ClientConsolePrint(e, "[SM] No plugins found.");
dvander@2547
  2697
		return;
dvander@2547
  2698
	}
dvander@2547
  2699
dvander@2547
  2700
	if (args.ArgC() > 2)
dvander@2547
  2701
	{
dvander@2547
  2702
		start = atoi(args.Arg(2));
dvander@2547
  2703
	}
dvander@2547
  2704
dvander@2547
  2705
	CPlugin *pl;
dvander@2547
  2706
	SourceHook::List<CPlugin *>::iterator iter;
dvander@2547
  2707
	SourceHook::List<CPlugin *> m_FailList;
dvander@2547
  2708
dvander@2547
  2709
	for (iter = m_plugins.begin();
dvander@2547
  2710
		 iter != m_plugins.end();
dvander@2547
  2711
		 iter++)
dvander@2547
  2712
	{
dvander@2547
  2713
		pl = (*iter);
dvander@2547
  2714
dvander@2547
  2715
		if (pl->GetStatus() != Plugin_Running)
dvander@2547
  2716
		{
dvander@2547
  2717
			continue;
dvander@2547
  2718
		}
dvander@2547
  2719
dvander@2547
  2720
		/* Count valid plugins */
dvander@2547
  2721
		id++;
dvander@2547
  2722
		if (id < start)
dvander@2547
  2723
		{
dvander@2547
  2724
			continue;
dvander@2547
  2725
		}
dvander@2547
  2726
dvander@2547
  2727
		if (id - start > 10)
dvander@2547
  2728
		{
dvander@2547
  2729
			break;
dvander@2547
  2730
		}
dvander@2547
  2731
dvander@2547
  2732
		size_t len;
dvander@2547
  2733
		const sm_plugininfo_t *info = pl->GetPublicInfo();
dvander@2547
  2734
		len = UTIL_Format(buffer, sizeof(buffer), " \"%s\"", (IS_STR_FILLED(info->name)) ? info->name : pl->GetFilename());
dvander@2547
  2735
		if (IS_STR_FILLED(info->version))
dvander@2547
  2736
		{
dvander@2547
  2737
			len += UTIL_Format(&buffer[len], sizeof(buffer)-len, " (%s)", info->version);
dvander@2547
  2738
		}
dvander@2547
  2739
		if (IS_STR_FILLED(info->author))
dvander@2547
  2740
		{
dvander@2547
  2741
			UTIL_Format(&buffer[len], sizeof(buffer)-len, " by %s", info->author);
dvander@2547
  2742
		}
dvander@2547
  2743
		else
dvander@2547
  2744
		{
dvander@2547
  2745
			UTIL_Format(&buffer[len], sizeof(buffer)-len, " %s", pl->m_filename);
dvander@2547
  2746
		}
dvander@2547
  2747
		ClientConsolePrint(e, "%s", buffer);
dvander@2547
  2748
	}
dvander@2547
  2749
dvander@2547
  2750
	/* See if we can get more plugins */
dvander@2547
  2751
	while (iter != m_plugins.end())
dvander@2547
  2752
	{
dvander@2547
  2753
		if ((*iter)->GetStatus() == Plugin_Running)
dvander@2547
  2754
		{
dvander@2547
  2755
			break;
dvander@2547
  2756
		}
dvander@2547
  2757
	}
dvander@2547
  2758
dvander@2547
  2759
	/* Do we actually have more plugins? */
dvander@2547
  2760
	if (iter != m_plugins.end())
dvander@2547
  2761
	{
fyrenmoo@3129
  2762
		ClientConsolePrint(e, "To see more, type \"sm plugins %d\"", id);
dvander@2547
  2763
	}
dvander@2547
  2764
}