Home - sourcemod-central - file revision - core/sourcemod.cpp@18fc0da83775 Mercurial
summary | shortlog | changelog | graph | tags | branches | files | changeset | file | latest | revisions | annotate | diff | raw | help
core/sourcemod.cpp
author David Anderson <dvander@alliedmods.net>
Sat Jan 27 09:25:07 2007 +0000 (2007-01-27)
changeset 401 18fc0da83775
parent 397 d78fcaa24fbb
child 422 5ef0f5610c2a
permissions -rw-r--r--
fixed this using /MD in release mode 
     1 /**
     2  * ===============================================================
     3  * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.
     4  * ===============================================================
     5  *
     6  * This file is not open source and may not be copied without explicit
     7  * written permission of AlliedModders LLC.  This file may not be redistributed 
     8  * in whole or significant part.
     9  * For information, see LICENSE.txt or http://www.sourcemod.net/license.php
    10  *
    11  * Version: $Id$
    12  */
    13 
    14 #include <stdio.h>
    15 #include "sourcemod.h"
    16 #include "sourcemm_api.h"
    17 #include "systems/LibrarySys.h"
    18 #include "vm/sp_vm_engine.h"
    19 #include <sh_string.h>
    20 #include "PluginSys.h"
    21 #include "ShareSys.h"
    22 #include "CLogger.h"
    23 #include "ExtensionSys.h"
    24 #include "AdminCache.h"
    25 #include "sm_stringutil.h"
    26 
    27 SH_DECL_HOOK6(IServerGameDLL, LevelInit, SH_NOATTRIB, false, bool, const char *, const char *, const char *, const char *, bool, bool);
    28 SH_DECL_HOOK0_void(IServerGameDLL, LevelShutdown, SH_NOATTRIB, false);
    29 
    30 SourcePawnEngine g_SourcePawn;
    31 SourceModBase g_SourceMod;
    32 
    33 ILibrary *g_pJIT = NULL;
    34 SourceHook::String g_BaseDir;
    35 ISourcePawnEngine *g_pSourcePawn = &g_SourcePawn;
    36 IVirtualMachine *g_pVM;
    37 IdentityToken_t *g_pCoreIdent = NULL;
    38 
    39 typedef int (*GIVEENGINEPOINTER)(ISourcePawnEngine *);
    40 typedef unsigned int (*GETEXPORTCOUNT)();
    41 typedef IVirtualMachine *(*GETEXPORT)(unsigned int);
    42 typedef void (*NOTIFYSHUTDOWN)();
    43 
    44 void ShutdownJIT()
    45 {
    46 	NOTIFYSHUTDOWN notify = (NOTIFYSHUTDOWN)g_pJIT->GetSymbolAddress("NotifyShutdown");
    47 	if (notify)
    48 	{
    49 		notify();
    50 	}
    51 
    52 	g_pJIT->CloseLibrary();
    53 }
    54 
    55 SourceModBase::SourceModBase()
    56 {
    57 	m_IsMapLoading = false;
    58 	m_ExecPluginReload = false;
    59 }
    60 
    61 bool SourceModBase::InitializeSourceMod(char *error, size_t err_max, bool late)
    62 {
    63 	g_BaseDir.assign(g_SMAPI->GetBaseDir());
    64 	g_LibSys.PathFormat(m_SMBaseDir, sizeof(m_SMBaseDir), "%s/addons/sourcemod", g_BaseDir.c_str());
    65 	g_LibSys.PathFormat(m_SMRelDir, sizeof(m_SMRelDir), "addons/sourcemod");
    66 
    67 	/* Attempt to load the JIT! */
    68 	char file[PLATFORM_MAX_PATH];
    69 	char myerror[255];
    70 	g_SMAPI->PathFormat(file, sizeof(file), "%s/bin/sourcepawn.jit.x86.%s",
    71 		GetSourceModPath(),
    72 		PLATFORM_LIB_EXT
    73 		);
    74 
    75 	g_pJIT = g_LibSys.OpenLibrary(file, myerror, sizeof(myerror));
    76 	if (!g_pJIT)
    77 	{
    78 		if (error && err_max)
    79 		{
    80 			snprintf(error, err_max, "%s (failed to load bin/sourcepawn.jit.x86.%s)", 
    81 				myerror,
    82 				PLATFORM_LIB_EXT);
    83 		}
    84 		return false;
    85 	}
    86 
    87 	int err;
    88 	GIVEENGINEPOINTER jit_init = (GIVEENGINEPOINTER)g_pJIT->GetSymbolAddress("GiveEnginePointer");
    89 	if (!jit_init)
    90 	{
    91 		ShutdownJIT();
    92 		if (error && err_max)
    93 		{
    94 			snprintf(error, err_max, "Failed to find GiveEnginePointer in JIT!");
    95 		}
    96 		return false;
    97 	}
    98 
    99 	if ((err=jit_init(g_pSourcePawn)) != 0)
   100 	{
   101 		ShutdownJIT();
   102 		if (error && err_max)
   103 		{
   104 			snprintf(error, err_max, "GiveEnginePointer returned %d in the JIT", err);
   105 		}
   106 		return false;
   107 	}
   108 
   109 	GETEXPORTCOUNT jit_getnum = (GETEXPORTCOUNT)g_pJIT->GetSymbolAddress("GetExportCount");
   110 	GETEXPORT jit_get = (GETEXPORT)g_pJIT->GetSymbolAddress("GetExport");
   111 	if (!jit_get)
   112 	{
   113 		ShutdownJIT();
   114 		if (error && err_max)
   115 		{
   116 			snprintf(error, err_max, "JIT is missing a necessary export!");
   117 		}
   118 		return false;
   119 	}
   120 
   121 	unsigned int num = jit_getnum();
   122 	if (!num || ((g_pVM=jit_get(0)) == NULL))
   123 	{
   124 		ShutdownJIT();
   125 		if (error && err_max)
   126 		{
   127 			snprintf(error, err_max, "JIT did not export any virtual machines!");
   128 		}
   129 		return false;
   130 	}
   131 
   132 	unsigned int api = g_pVM->GetAPIVersion();
   133 	if (api != SOURCEPAWN_VM_API_VERSION)
   134 	{
   135 		ShutdownJIT();
   136 		if (error && err_max)
   137 		{
   138 			snprintf(error, err_max, "JIT is not a compatible version");
   139 		}
   140 		return false;
   141 	}
   142 
   143 	StartSourceMod(late);
   144 
   145 	return true;
   146 }
   147 
   148 void SourceModBase::StartSourceMod(bool late)
   149 {
   150 	/* First initialize the global hooks we need */
   151 	SH_ADD_HOOK_MEMFUNC(IServerGameDLL, LevelInit, gamedll, this, &SourceModBase::LevelInit, false);
   152 	SH_ADD_HOOK_MEMFUNC(IServerGameDLL, LevelShutdown, gamedll, this, &SourceModBase::LevelShutdown, false);
   153 
   154 	/* Notify! */
   155 	SMGlobalClass *pBase = SMGlobalClass::head;
   156 	while (pBase)
   157 	{
   158 		pBase->OnSourceModStartup(false);
   159 		pBase = pBase->m_pGlobalClassNext;
   160 	}
   161 
   162 	/* Make the global core identity */
   163 	g_pCoreIdent = g_ShareSys.CreateCoreIdentity();
   164 
   165 	/* Notify! */
   166 	pBase = SMGlobalClass::head;
   167 	while (pBase)
   168 	{
   169 		pBase->OnSourceModAllInitialized();
   170 		pBase = pBase->m_pGlobalClassNext;
   171 	}
   172 
   173 	/* Add us now... */
   174 	g_ShareSys.AddInterface(NULL, this);
   175 }
   176 
   177 bool SourceModBase::LevelInit(char const *pMapName, char const *pMapEntities, char const *pOldLevel, char const *pLandmarkName, bool loadGame, bool background)
   178 {
   179 	m_IsMapLoading = true;
   180 	m_ExecPluginReload = true;
   181 
   182 	g_Logger.MapChange(pMapName);
   183 
   184 	DoGlobalPluginLoads();
   185 
   186 	m_IsMapLoading = false;
   187 
   188 	g_Admins.DumpAdminCache(ADMIN_CACHE_GROUPS|ADMIN_CACHE_ADMINS|ADMIN_CACHE_OVERRIDES, true);
   189 
   190 	RETURN_META_VALUE(MRES_IGNORED, true);
   191 }
   192 
   193 void SourceModBase::LevelShutdown()
   194 {
   195 	if (m_ExecPluginReload)
   196 	{
   197 		g_PluginSys.ReloadOrUnloadPlugins();
   198 		m_ExecPluginReload = false;
   199 	}
   200 }
   201 
   202 bool SourceModBase::IsMapLoading()
   203 {
   204 	return m_IsMapLoading;
   205 }
   206 
   207 void SourceModBase::DoGlobalPluginLoads()
   208 {
   209 	char config_path[PLATFORM_MAX_PATH];
   210 	char plugins_path[PLATFORM_MAX_PATH];
   211 
   212 	BuildPath(Path_SM, config_path, 
   213 		sizeof(config_path),
   214 		"configs/plugin_settings.cfg");
   215 	BuildPath(Path_SM, plugins_path,
   216 		sizeof(plugins_path),
   217 		"plugins");
   218 
   219 	/* Load any auto extensions */
   220 	g_Extensions.TryAutoload();
   221 
   222 	/* Run the first pass */
   223 	g_PluginSys.LoadAll_FirstPass(config_path, plugins_path);
   224 
   225 	/* Mark any extensions as loaded */
   226 	g_Extensions.MarkAllLoaded();
   227 
   228 	/* No modules yet, it's safe to call this from here */
   229 	g_PluginSys.LoadAll_SecondPass();
   230 
   231 	/* Re-mark any extensions as loaded */
   232 	g_Extensions.MarkAllLoaded();
   233 }
   234 
   235 size_t SourceModBase::BuildPath(PathType type, char *buffer, size_t maxlength, char *format, ...)
   236 {
   237 	char _buffer[PLATFORM_MAX_PATH+1];
   238 	va_list ap;
   239 
   240 	va_start(ap, format);
   241 	vsnprintf(_buffer, PLATFORM_MAX_PATH, format, ap);
   242 	va_end(ap);
   243 
   244 	const char *base = NULL;
   245 	if (type == Path_Game)
   246 	{
   247 		base = GetModPath();
   248 	} else if (type == Path_SM) {
   249 		base = GetSourceModPath();
   250 	} else if (type == Path_SM_Rel) {
   251 		base = m_SMRelDir;
   252 	}
   253 
   254 	if (base)
   255 	{
   256 		return g_LibSys.PathFormat(buffer, maxlength, "%s/%s", base, _buffer);
   257 	} else {
   258 		return g_LibSys.PathFormat(buffer, maxlength, "%s", _buffer);
   259 	}
   260 }
   261 
   262 void SourceModBase::CloseSourceMod()
   263 {
   264 	/* Notify! */
   265 	SMGlobalClass *pBase = SMGlobalClass::head;
   266 	while (pBase)
   267 	{
   268 		pBase->OnSourceModShutdown();
   269 		pBase = pBase->m_pGlobalClassNext;
   270 	}
   271 
   272 	/* Notify! */
   273 	pBase = SMGlobalClass::head;
   274 	while (pBase)
   275 	{
   276 		pBase->OnSourceModAllShutdown();
   277 		pBase = pBase->m_pGlobalClassNext;
   278 	}
   279 
   280 	SH_REMOVE_HOOK_MEMFUNC(IServerGameDLL, LevelInit, gamedll, this, &SourceModBase::LevelInit, false);
   281 	SH_REMOVE_HOOK_MEMFUNC(IServerGameDLL, LevelShutdown, gamedll, this, &SourceModBase::LevelShutdown, false);
   282 
   283 	/* Rest In Peace */
   284 	ShutdownJIT();
   285 }
   286 
   287 void SourceModBase::LogMessage(IExtension *pExt, const char *format, ...)
   288 {
   289 	IExtensionInterface *pAPI = pExt->GetAPI();
   290 	const char *tag = pAPI->GetExtensionTag();
   291 	char buffer[2048];
   292 	va_list ap;
   293 
   294 	va_start(ap, format);
   295 	vsnprintf(buffer, sizeof(buffer), format, ap);
   296 	va_end(ap);
   297 
   298 	if (tag)
   299 	{
   300 		g_Logger.LogMessage("[%s] %s", tag, buffer);
   301 	} else {
   302 		g_Logger.LogMessage("%s", buffer);
   303 	}
   304 }
   305 
   306 void SourceModBase::LogError(IExtension *pExt, const char *format, ...)
   307 {
   308 	IExtensionInterface *pAPI = pExt->GetAPI();
   309 	const char *tag = pAPI->GetExtensionTag();
   310 	char buffer[2048];
   311 	va_list ap;
   312 
   313 	va_start(ap, format);
   314 	vsnprintf(buffer, sizeof(buffer), format, ap);
   315 	va_end(ap);
   316 
   317 	if (tag)
   318 	{
   319 		g_Logger.LogError("[%s] %s", tag, buffer);
   320 	} else {
   321 		g_Logger.LogError("%s", buffer);
   322 	}
   323 }
   324 
   325 size_t SourceModBase::FormatString(char *buffer, size_t maxlength, IPluginContext *pContext, const cell_t *params, unsigned int param)
   326 {
   327 	char *fmt;
   328 
   329 	pContext->LocalToString(params[param], &fmt);
   330 
   331 	int lparam = ++param;
   332 
   333 	return atcprintf(buffer, maxlength, fmt, pContext, params, &lparam);
   334 }
   335 
   336 const char *SourceModBase::GetSourceModPath()
   337 {
   338 	return m_SMBaseDir;
   339 }
   340 
   341 const char *SourceModBase::GetModPath()
   342 {
   343 	return g_BaseDir.c_str();
   344 }
   345 
   346 SMGlobalClass *SMGlobalClass::head = NULL;
   347 
   348 SMGlobalClass::SMGlobalClass()
   349 {
   350 	m_pGlobalClassNext = SMGlobalClass::head;
   351 	SMGlobalClass::head = this;
   352 }
sourcemod-central
Feed