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