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