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