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