extensions/cstrike/extension.cpp
author David Anderson <dvander@alliedmods.net>
Sun Aug 30 01:54:45 2009 -0700 (2009-08-30)
changeset 2790 6dd851ba15fd
parent 2783 5f333027ef04
child 3049 e5d9da0530cf
permissions -rw-r--r--
Fixed extension version reporting.
     1 /**
     2  * vim: set ts=4 :
     3  * =============================================================================
     4  * SourceMod Counter-Strike:Source Extension
     5  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
     6  * =============================================================================
     7  *
     8  * This program is free software; you can redistribute it and/or modify it under
     9  * the terms of the GNU General Public License, version 3.0, as published by the
    10  * Free Software Foundation.
    11  * 
    12  * This program is distributed in the hope that it will be useful, but WITHOUT
    13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
    14  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
    15  * details.
    16  *
    17  * You should have received a copy of the GNU General Public License along with
    18  * this program.  If not, see <http://www.gnu.org/licenses/>.
    19  *
    20  * As a special exception, AlliedModders LLC gives you permission to link the
    21  * code of this program (as well as its derivative works) to "Half-Life 2," the
    22  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
    23  * by the Valve Corporation.  You must obey the GNU General Public License in
    24  * all respects for all other code used.  Additionally, AlliedModders LLC grants
    25  * this exception to all derivative works.  AlliedModders LLC defines further
    26  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
    27  * or <http://www.sourcemod.net/license.php>.
    28  *
    29  * Version: $Id$
    30  */
    31 
    32 #include <sourcemod_version.h>
    33 #include "extension.h"
    34 #include "RegNatives.h"
    35 #include "timeleft.h"
    36 #include "iplayerinfo.h"
    37 
    38 /**
    39  * @file extension.cpp
    40  * @brief Implement extension code here.
    41  */
    42 
    43 SH_DECL_HOOK6(IServerGameDLL, LevelInit, SH_NOATTRIB, false, bool, const char *, const char *, const char *, const char *, bool, bool);
    44 
    45 CStrike g_CStrike;		/**< Global singleton for extension's main interface */
    46 IBinTools *g_pBinTools = NULL;
    47 IGameConfig *g_pGameConf = NULL;
    48 IGameEventManager2 *gameevents = NULL;
    49 bool hooked_everything = false;
    50 int g_msgHintText = -1;
    51 
    52 SMEXT_LINK(&g_CStrike);
    53 
    54 extern sp_nativeinfo_t g_CSNatives[];
    55 
    56 bool CStrike::SDK_OnLoad(char *error, size_t maxlength, bool late)
    57 {
    58 	if (strcmp(g_pSM->GetGameFolderName(), "cstrike") != 0)
    59 	{
    60 		snprintf(error, maxlength, "Cannot Load Cstrike Extension on mods other than CS:S");
    61 		return false;
    62 	}
    63 
    64 	sharesys->AddDependency(myself, "bintools.ext", true, true);
    65 
    66 	char conf_error[255];
    67 	if (!gameconfs->LoadGameConfigFile("sm-cstrike.games", &g_pGameConf, conf_error, sizeof(conf_error)))
    68 	{
    69 		if (error)
    70 		{
    71 			snprintf(error, maxlength, "Could not read sm-cstrike.games.txt: %s", conf_error);
    72 		}
    73 		return false;
    74 	}
    75 
    76 	sharesys->AddNatives(myself, g_CSNatives);
    77 	sharesys->RegisterLibrary(myself, "cstrike");
    78 
    79 	if ((g_msgHintText = usermsgs->GetMessageIndex("HintText")) != -1)
    80 	{
    81 		sharesys->OverrideNatives(myself, g_CS_PrintHintText);
    82 	}
    83 
    84 	playerhelpers->RegisterCommandTargetProcessor(this);
    85 
    86 	return true;
    87 }
    88 
    89 bool CStrike::SDK_OnMetamodLoad(ISmmAPI *ismm, char *error, size_t maxlen, bool late)
    90 {
    91 	GET_V_IFACE_CURRENT(GetEngineFactory, gameevents, IGameEventManager2, INTERFACEVERSION_GAMEEVENTSMANAGER2);
    92 	GET_V_IFACE_CURRENT(GetEngineFactory, engine, IVEngineServer, INTERFACEVERSION_VENGINESERVER);
    93 
    94 	return true;
    95 }
    96 
    97 void CStrike::SDK_OnUnload()
    98 {
    99 	if (hooked_everything)
   100 	{
   101 		gameevents->RemoveListener(&g_TimeLeftEvents);
   102 		SH_REMOVE_HOOK_MEMFUNC(IServerGameDLL, LevelInit, gamedll, &g_TimeLeftEvents, &TimeLeftEvents::LevelInit, true);
   103 		hooked_everything = false;
   104 	}
   105 	g_RegNatives.UnregisterAll();
   106 	gameconfs->CloseGameConfigFile(g_pGameConf);
   107 	playerhelpers->UnregisterCommandTargetProcessor(this);
   108 }
   109 
   110 void CStrike::SDK_OnAllLoaded()
   111 {
   112 	gameevents->AddListener(&g_TimeLeftEvents, "round_start", true);
   113 	gameevents->AddListener(&g_TimeLeftEvents, "round_end", true);
   114 	SH_ADD_HOOK_MEMFUNC(IServerGameDLL, LevelInit, gamedll, &g_TimeLeftEvents, &TimeLeftEvents::LevelInit, true);
   115 	hooked_everything = true;
   116 
   117 	SM_GET_LATE_IFACE(BINTOOLS, g_pBinTools);
   118 }
   119 
   120 bool CStrike::QueryRunning(char *error, size_t maxlength)
   121 {
   122 	SM_CHECK_IFACE(BINTOOLS, g_pBinTools);
   123 
   124 	return true;
   125 }
   126 
   127 bool CStrike::QueryInterfaceDrop(SMInterface *pInterface)
   128 {
   129 	if (pInterface == g_pBinTools)
   130 	{
   131 		return false;
   132 	}
   133 
   134 	return IExtensionInterface::QueryInterfaceDrop(pInterface);
   135 }
   136 
   137 void CStrike::NotifyInterfaceDrop(SMInterface *pInterface)
   138 {
   139 	g_RegNatives.UnregisterAll();
   140 }
   141 
   142 size_t UTIL_Format(char *buffer, size_t maxlength, const char *fmt, ...)
   143 {
   144 	va_list ap;
   145 	va_start(ap, fmt);
   146 	size_t len = vsnprintf(buffer, maxlength, fmt, ap);
   147 	va_end(ap);
   148 
   149 	if (len >= maxlength)
   150 	{
   151 		buffer[maxlength - 1] = '\0';
   152 		return (maxlength - 1);
   153 	}
   154 	else
   155 	{
   156 		return len;
   157 	}
   158 }
   159 
   160 bool CStrike::ProcessCommandTarget(cmd_target_info_t *info)
   161 {
   162 	int max_clients;
   163 	IPlayerInfo *pInfo;
   164 	unsigned int team_index = 0;
   165 	IGamePlayer *pPlayer, *pAdmin;
   166 
   167 	if ((info->flags & COMMAND_FILTER_NO_MULTI) == COMMAND_FILTER_NO_MULTI)
   168 	{
   169 		return false;
   170 	}
   171 
   172 	if (info->admin)
   173 	{
   174 		if ((pAdmin = playerhelpers->GetGamePlayer(info->admin)) == NULL)
   175 		{
   176 			return false;
   177 		}
   178 		if (!pAdmin->IsInGame())
   179 		{
   180 			return false;
   181 		}
   182 	}
   183 	else
   184 	{
   185 		pAdmin = NULL;
   186 	}
   187 
   188 	if (strcmp(info->pattern, "@ct") == 0 || strcmp(info->pattern, "@cts") == 0)
   189 	{
   190 		team_index = 3;
   191 	}
   192 	else if (strcmp(info->pattern, "@t") == 0 || strcmp(info->pattern, "@ts") == 0)
   193 	{
   194 		team_index = 2;
   195 	}
   196 	else
   197 	{
   198 		return false;
   199 	}
   200 
   201 	info->num_targets = 0;
   202 
   203 	max_clients = playerhelpers->GetMaxClients();
   204 	for (int i = 1; 
   205 		 i <= max_clients && (cell_t)info->num_targets < info->max_targets; 
   206 		 i++)
   207 	{
   208 		if ((pPlayer = playerhelpers->GetGamePlayer(i)) == NULL)
   209 		{
   210 			continue;
   211 		}
   212 		if (!pPlayer->IsInGame())
   213 		{
   214 			continue;
   215 		}
   216 		if ((pInfo = pPlayer->GetPlayerInfo()) == NULL)
   217 		{
   218 			continue;
   219 		}
   220 		if (pInfo->GetTeamIndex() != (int)team_index)
   221 		{
   222 			continue;
   223 		}
   224 		if (playerhelpers->FilterCommandTarget(pAdmin, pPlayer, info->flags) 
   225 			!= COMMAND_TARGET_VALID)
   226 		{
   227 			continue;
   228 		}
   229 		info->targets[info->num_targets] = i;
   230 		info->num_targets++;
   231 	}
   232 
   233 	if (info->num_targets == 0)
   234 	{
   235 		info->reason = COMMAND_TARGET_EMPTY_FILTER;
   236 	}
   237 	else
   238 	{
   239 		info->reason = COMMAND_TARGET_VALID;
   240 	}
   241 
   242 	info->target_name_style = COMMAND_TARGETNAME_RAW;
   243 	if (team_index == 2)
   244 	{
   245 		UTIL_Format(info->target_name, info->target_name_maxlength, "Terrorists");
   246 	}
   247 	else if (team_index == 3)
   248 	{
   249 		UTIL_Format(info->target_name, info->target_name_maxlength, "CTs");
   250 	}
   251 
   252 	return true;
   253 }
   254 
   255 const char *CStrike::GetExtensionVerString()
   256 {
   257 	return SM_FULL_VERSION;
   258 }
   259 
   260 const char *CStrike::GetExtensionDateString()
   261 {
   262 	return SM_BUILD_TIMESTAMP;
   263 }
   264