extensions/cstrike/extension.cpp
author Scott Ehlert <ds@alliedmods.net>
Sun Sep 14 03:01:05 2008 -0500 (2008-09-14)
changeset 2289 44045ead61cd
parent 2069 80d56f1d58fe
parent 1947 e73a1e51833b
child 2783 5f333027ef04
permissions -rw-r--r--
Eliminated head: da55a80522cf, a remnant of sourcemod-1.1.0 branch
     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 "extension.h"
    33 #include "RegNatives.h"
    34 #include "timeleft.h"
    35 #include "iplayerinfo.h"
    36 
    37 /**
    38  * @file extension.cpp
    39  * @brief Implement extension code here.
    40  */
    41 
    42 SH_DECL_HOOK6(IServerGameDLL, LevelInit, SH_NOATTRIB, false, bool, const char *, const char *, const char *, const char *, bool, bool);
    43 
    44 CStrike g_CStrike;		/**< Global singleton for extension's main interface */
    45 IBinTools *g_pBinTools = NULL;
    46 IGameConfig *g_pGameConf = NULL;
    47 IGameEventManager2 *gameevents = NULL;
    48 bool hooked_everything = false;
    49 int g_msgHintText = -1;
    50 
    51 SMEXT_LINK(&g_CStrike);
    52 
    53 extern sp_nativeinfo_t g_CSNatives[];
    54 
    55 bool CStrike::SDK_OnLoad(char *error, size_t maxlength, bool late)
    56 {
    57 	if (strcmp(g_pSM->GetGameFolderName(), "cstrike") != 0)
    58 	{
    59 		snprintf(error, maxlength, "Cannot Load Cstrike Extension on mods other than CS:S");
    60 		return false;
    61 	}
    62 
    63 	sharesys->AddDependency(myself, "bintools.ext", true, true);
    64 
    65 	char conf_error[255];
    66 	if (!gameconfs->LoadGameConfigFile("sm-cstrike.games", &g_pGameConf, conf_error, sizeof(conf_error)))
    67 	{
    68 		if (error)
    69 		{
    70 			snprintf(error, maxlength, "Could not read sm-cstrike.games.txt: %s", conf_error);
    71 		}
    72 		return false;
    73 	}
    74 
    75 	sharesys->AddNatives(myself, g_CSNatives);
    76 	sharesys->RegisterLibrary(myself, "cstrike");
    77 
    78 	if ((g_msgHintText = usermsgs->GetMessageIndex("HintText")) != -1)
    79 	{
    80 		sharesys->OverrideNatives(myself, g_CS_PrintHintText);
    81 	}
    82 
    83 	playerhelpers->RegisterCommandTargetProcessor(this);
    84 
    85 	return true;
    86 }
    87 
    88 bool CStrike::SDK_OnMetamodLoad(ISmmAPI *ismm, char *error, size_t maxlen, bool late)
    89 {
    90 	GET_V_IFACE_CURRENT(GetEngineFactory, gameevents, IGameEventManager2, INTERFACEVERSION_GAMEEVENTSMANAGER2);
    91 	GET_V_IFACE_CURRENT(GetEngineFactory, engine, IVEngineServer, INTERFACEVERSION_VENGINESERVER);
    92 
    93 	return true;
    94 }
    95 
    96 void CStrike::SDK_OnUnload()
    97 {
    98 	if (hooked_everything)
    99 	{
   100 		gameevents->RemoveListener(&g_TimeLeftEvents);
   101 		SH_REMOVE_HOOK_MEMFUNC(IServerGameDLL, LevelInit, gamedll, &g_TimeLeftEvents, &TimeLeftEvents::LevelInit, true);
   102 		hooked_everything = false;
   103 	}
   104 	g_RegNatives.UnregisterAll();
   105 	gameconfs->CloseGameConfigFile(g_pGameConf);
   106 	playerhelpers->UnregisterCommandTargetProcessor(this);
   107 }
   108 
   109 void CStrike::SDK_OnAllLoaded()
   110 {
   111 	gameevents->AddListener(&g_TimeLeftEvents, "round_start", true);
   112 	gameevents->AddListener(&g_TimeLeftEvents, "round_end", true);
   113 	SH_ADD_HOOK_MEMFUNC(IServerGameDLL, LevelInit, gamedll, &g_TimeLeftEvents, &TimeLeftEvents::LevelInit, true);
   114 	hooked_everything = true;
   115 
   116 	SM_GET_LATE_IFACE(BINTOOLS, g_pBinTools);
   117 }
   118 
   119 bool CStrike::QueryRunning(char *error, size_t maxlength)
   120 {
   121 	SM_CHECK_IFACE(BINTOOLS, g_pBinTools);
   122 
   123 	return true;
   124 }
   125 
   126 bool CStrike::QueryInterfaceDrop(SMInterface *pInterface)
   127 {
   128 	if (pInterface == g_pBinTools)
   129 	{
   130 		return false;
   131 	}
   132 
   133 	return IExtensionInterface::QueryInterfaceDrop(pInterface);
   134 }
   135 
   136 void CStrike::NotifyInterfaceDrop(SMInterface *pInterface)
   137 {
   138 	g_RegNatives.UnregisterAll();
   139 }
   140 
   141 size_t UTIL_Format(char *buffer, size_t maxlength, const char *fmt, ...)
   142 {
   143 	va_list ap;
   144 	va_start(ap, fmt);
   145 	size_t len = vsnprintf(buffer, maxlength, fmt, ap);
   146 	va_end(ap);
   147 
   148 	if (len >= maxlength)
   149 	{
   150 		buffer[maxlength - 1] = '\0';
   151 		return (maxlength - 1);
   152 	}
   153 	else
   154 	{
   155 		return len;
   156 	}
   157 }
   158 
   159 bool CStrike::ProcessCommandTarget(cmd_target_info_t *info)
   160 {
   161 	int max_clients;
   162 	IPlayerInfo *pInfo;
   163 	unsigned int team_index = 0;
   164 	IGamePlayer *pPlayer, *pAdmin;
   165 
   166 	if ((info->flags & COMMAND_FILTER_NO_MULTI) == COMMAND_FILTER_NO_MULTI)
   167 	{
   168 		return false;
   169 	}
   170 
   171 	if (info->admin)
   172 	{
   173 		if ((pAdmin = playerhelpers->GetGamePlayer(info->admin)) == NULL)
   174 		{
   175 			return false;
   176 		}
   177 		if (!pAdmin->IsInGame())
   178 		{
   179 			return false;
   180 		}
   181 	}
   182 	else
   183 	{
   184 		pAdmin = NULL;
   185 	}
   186 
   187 	if (strcmp(info->pattern, "@ct") == 0 || strcmp(info->pattern, "@cts") == 0)
   188 	{
   189 		team_index = 3;
   190 	}
   191 	else if (strcmp(info->pattern, "@t") == 0 || strcmp(info->pattern, "@ts") == 0)
   192 	{
   193 		team_index = 2;
   194 	}
   195 	else
   196 	{
   197 		return false;
   198 	}
   199 
   200 	info->num_targets = 0;
   201 
   202 	max_clients = playerhelpers->GetMaxClients();
   203 	for (int i = 1; 
   204 		 i <= max_clients && (cell_t)info->num_targets < info->max_targets; 
   205 		 i++)
   206 	{
   207 		if ((pPlayer = playerhelpers->GetGamePlayer(i)) == NULL)
   208 		{
   209 			continue;
   210 		}
   211 		if (!pPlayer->IsInGame())
   212 		{
   213 			continue;
   214 		}
   215 		if ((pInfo = pPlayer->GetPlayerInfo()) == NULL)
   216 		{
   217 			continue;
   218 		}
   219 		if (pInfo->GetTeamIndex() != (int)team_index)
   220 		{
   221 			continue;
   222 		}
   223 		if (playerhelpers->FilterCommandTarget(pAdmin, pPlayer, info->flags) 
   224 			!= COMMAND_TARGET_VALID)
   225 		{
   226 			continue;
   227 		}
   228 		info->targets[info->num_targets] = i;
   229 		info->num_targets++;
   230 	}
   231 
   232 	if (info->num_targets == 0)
   233 	{
   234 		info->reason = COMMAND_TARGET_EMPTY_FILTER;
   235 	}
   236 	else
   237 	{
   238 		info->reason = COMMAND_TARGET_VALID;
   239 	}
   240 
   241 	info->target_name_style = COMMAND_TARGETNAME_RAW;
   242 	if (team_index == 2)
   243 	{
   244 		UTIL_Format(info->target_name, info->target_name_maxlength, "Terrorists");
   245 	}
   246 	else if (team_index == 3)
   247 	{
   248 		UTIL_Format(info->target_name, info->target_name_maxlength, "CTs");
   249 	}
   250 
   251 	return true;
   252 }