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