extensions/cstrike/extension.cpp
author David Anderson <dvander@alliedmods.net>
Tue Dec 04 01:08:03 2007 +0000 (2007-12-04)
changeset 1756 33f9fa5adfe3
parent 1692 dc5dcaaeb744
child 1758 ac2ed05893c9
permissions -rw-r--r--
added @t, @ts, @ct, @cts target patterns
     1 /**
     2  * vim: set ts=4 :
     3  * =============================================================================
     4  * SourceMod Counter-Strike:Source Extension
     5  * Copyright (C) 2004-2007 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 	sharesys->AddDependency(myself, "bintools.ext", true, true);
    58 
    59 	char conf_error[255];
    60 	if (!gameconfs->LoadGameConfigFile("sm-cstrike.games", &g_pGameConf, conf_error, sizeof(conf_error)))
    61 	{
    62 		if (error)
    63 		{
    64 			snprintf(error, maxlength, "Could not read sm-cstrike.games.txt: %s", conf_error);
    65 		}
    66 		return false;
    67 	}
    68 
    69 	sharesys->AddNatives(myself, g_CSNatives);
    70 	sharesys->RegisterLibrary(myself, "cstrike");
    71 
    72 	if ((g_msgHintText = usermsgs->GetMessageIndex("HintText")) != -1)
    73 	{
    74 		sharesys->OverrideNatives(myself, g_CS_PrintHintText);
    75 	}
    76 
    77 	playerhelpers->RegisterCommandTargetProcessor(this);
    78 
    79 	return true;
    80 }
    81 
    82 bool CStrike::SDK_OnMetamodLoad(ISmmAPI *ismm, char *error, size_t maxlen, bool late)
    83 {
    84 	GET_V_IFACE_CURRENT(GetEngineFactory, gameevents, IGameEventManager2, INTERFACEVERSION_GAMEEVENTSMANAGER2);
    85 	GET_V_IFACE_CURRENT(GetEngineFactory, engine, IVEngineServer, INTERFACEVERSION_VENGINESERVER);
    86 
    87 	return true;
    88 }
    89 
    90 void CStrike::SDK_OnUnload()
    91 {
    92 	if (hooked_everything)
    93 	{
    94 		gameevents->RemoveListener(&g_TimeLeftEvents);
    95 		SH_REMOVE_HOOK_MEMFUNC(IServerGameDLL, LevelInit, gamedll, &g_TimeLeftEvents, &TimeLeftEvents::LevelInit, true);
    96 		hooked_everything = false;
    97 	}
    98 	g_RegNatives.UnregisterAll();
    99 	gameconfs->CloseGameConfigFile(g_pGameConf);
   100 	playerhelpers->UnregisterCommandTargetProcessor(this);
   101 }
   102 
   103 void CStrike::SDK_OnAllLoaded()
   104 {
   105 	gameevents->AddListener(&g_TimeLeftEvents, "round_start", true);
   106 	gameevents->AddListener(&g_TimeLeftEvents, "round_end", true);
   107 	SH_ADD_HOOK_MEMFUNC(IServerGameDLL, LevelInit, gamedll, &g_TimeLeftEvents, &TimeLeftEvents::LevelInit, true);
   108 	hooked_everything = true;
   109 
   110 	SM_GET_LATE_IFACE(BINTOOLS, g_pBinTools);
   111 }
   112 
   113 bool CStrike::QueryRunning(char *error, size_t maxlength)
   114 {
   115 	SM_CHECK_IFACE(BINTOOLS, g_pBinTools);
   116 
   117 	return true;
   118 }
   119 
   120 bool CStrike::QueryInterfaceDrop(SMInterface *pInterface)
   121 {
   122 	if (pInterface == g_pBinTools)
   123 	{
   124 		return false;
   125 	}
   126 
   127 	return IExtensionInterface::QueryInterfaceDrop(pInterface);
   128 }
   129 
   130 void CStrike::NotifyInterfaceDrop(SMInterface *pInterface)
   131 {
   132 	g_RegNatives.UnregisterAll();
   133 }
   134 
   135 size_t UTIL_Format(char *buffer, size_t maxlength, const char *fmt, ...)
   136 {
   137 	va_list ap;
   138 	va_start(ap, fmt);
   139 	size_t len = vsnprintf(buffer, maxlength, fmt, ap);
   140 	va_end(ap);
   141 
   142 	if (len >= maxlength)
   143 	{
   144 		buffer[maxlength - 1] = '\0';
   145 		return (maxlength - 1);
   146 	}
   147 	else
   148 	{
   149 		return len;
   150 	}
   151 }
   152 
   153 bool CStrike::ProcessCommandTarget(cmd_target_info_t *info)
   154 {
   155 	int max_clients;
   156 	IPlayerInfo *pInfo;
   157 	unsigned int team_index = 0;
   158 	IGamePlayer *pPlayer, *pAdmin;
   159 
   160 	if ((info->flags & COMMAND_FILTER_NO_MULTI) == COMMAND_FILTER_NO_MULTI)
   161 	{
   162 		return false;
   163 	}
   164 
   165 	if (info->admin)
   166 	{
   167 		if ((pAdmin = playerhelpers->GetGamePlayer(info->admin)) == NULL)
   168 		{
   169 			return false;
   170 		}
   171 		if (!pAdmin->IsInGame())
   172 		{
   173 			return false;
   174 		}
   175 	}
   176 	else
   177 	{
   178 		pAdmin = NULL;
   179 	}
   180 
   181 	if (strcmp(info->pattern, "@ct") == 0 || strcmp(info->pattern, "@cts") == 0)
   182 	{
   183 		team_index = 3;
   184 	}
   185 	else if (strcmp(info->pattern, "@t") == 0 || strcmp(info->pattern, "@ts") == 0)
   186 	{
   187 		team_index = 2;
   188 	}
   189 
   190 	info->num_targets = 0;
   191 
   192 	max_clients = playerhelpers->GetMaxClients();
   193 	for (int i = 1; 
   194 		 i <= max_clients && (cell_t)info->num_targets < info->max_targets; 
   195 		 i++)
   196 	{
   197 		if ((pPlayer = playerhelpers->GetGamePlayer(i)) == NULL)
   198 		{
   199 			continue;
   200 		}
   201 		if (!pPlayer->IsInGame())
   202 		{
   203 			continue;
   204 		}
   205 		if ((pInfo = pPlayer->GetPlayerInfo()) == NULL)
   206 		{
   207 			continue;
   208 		}
   209 		if (pInfo->GetTeamIndex() != team_index)
   210 		{
   211 			continue;
   212 		}
   213 		if (playerhelpers->FilterCommandTarget(pAdmin, pPlayer, info->flags) 
   214 			!= COMMAND_TARGET_VALID)
   215 		{
   216 			continue;
   217 		}
   218 		info->targets[info->num_targets] = i;
   219 		info->num_targets++;
   220 	}
   221 
   222 	if (info->num_targets == 0)
   223 	{
   224 		info->reason = COMMAND_TARGET_EMPTY_FILTER;
   225 	}
   226 	else
   227 	{
   228 		info->reason = COMMAND_TARGET_VALID;
   229 	}
   230 
   231 	info->target_name_style = COMMAND_TARGETNAME_RAW;
   232 	if (team_index == 2)
   233 	{
   234 		UTIL_Format(info->target_name, info->target_name_maxlength, "Terrorists");
   235 	}
   236 	else if (team_index == 3)
   237 	{
   238 		UTIL_Format(info->target_name, info->target_name_maxlength, "CTs");
   239 	}
   240 
   241 	return true;
   242 }