core/PlayerManager.cpp
author David Anderson <dvander@alliedmods.net>
Mon Oct 01 18:41:39 2007 +0000 (2007-10-01)
changeset 1509 cbb82af65663
parent 1493 cdf3fce388d7
child 1519 c325cd4b060a
permissions -rw-r--r--
added IsClientInKickQueue() to deal with a bot issue
     1 /**
     2  * vim: set ts=4 :
     3  * =============================================================================
     4  * SourceMod
     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 "PlayerManager.h"
    33 #include "ForwardSys.h"
    34 #include "ShareSys.h"
    35 #include "AdminCache.h"
    36 #include "ConCmdManager.h"
    37 #include "MenuStyle_Valve.h"
    38 #include "MenuStyle_Radio.h"
    39 #include "sm_stringutil.h"
    40 #include "CoreConfig.h"
    41 #include <inetchannel.h>
    42 #include <iclient.h>
    43 #include "TimerSys.h"
    44 #include "Translator.h"
    45 #include "Logger.h"
    46 
    47 PlayerManager g_Players;
    48 bool g_OnMapStarted = false;
    49 IForward *PreAdminCheck = NULL;
    50 IForward *PostAdminCheck = NULL;
    51 const unsigned int *g_NumPlayersToAuth = NULL;
    52 
    53 SH_DECL_HOOK5(IServerGameClients, ClientConnect, SH_NOATTRIB, 0, bool, edict_t *, const char *, const char *, char *, int);
    54 SH_DECL_HOOK2_void(IServerGameClients, ClientPutInServer, SH_NOATTRIB, 0, edict_t *, const char *);
    55 SH_DECL_HOOK1_void(IServerGameClients, ClientDisconnect, SH_NOATTRIB, 0, edict_t *);
    56 SH_DECL_HOOK1_void(IServerGameClients, ClientCommand, SH_NOATTRIB, 0, edict_t *);
    57 SH_DECL_HOOK1_void(IServerGameClients, ClientSettingsChanged, SH_NOATTRIB, 0, edict_t *);
    58 SH_DECL_HOOK3_void(IServerGameDLL, ServerActivate, SH_NOATTRIB, 0, edict_t *, int, int);
    59 
    60 class KickPlayerTimer : public ITimedEvent
    61 {
    62 public:
    63 	ResultType OnTimer(ITimer *pTimer, void *pData)
    64 	{
    65 		int userid = (int)pData;
    66 		int client = g_Players.GetClientOfUserId(userid);
    67 		if (client)
    68 		{
    69 			CPlayer *player = g_Players.GetPlayerByIndex(client);
    70 			player->Kick("Your name is reserved by SourceMod; set your password to use it.");
    71 		}
    72 		return Pl_Stop;
    73 	}
    74 	void OnTimerEnd(ITimer *pTimer, void *pData)
    75 	{
    76 	}
    77 } s_KickPlayerTimer;
    78 
    79 PlayerManager::PlayerManager()
    80 {
    81 	m_AuthQueue = NULL;
    82 	m_FirstPass = false;
    83 
    84 	m_UserIdLookUp = new int[USHRT_MAX+1];
    85 	memset(m_UserIdLookUp, 0, sizeof(int) * (USHRT_MAX+1));
    86 }
    87 
    88 PlayerManager::~PlayerManager()
    89 {
    90 	g_NumPlayersToAuth = NULL;
    91 
    92 	delete [] m_AuthQueue;
    93 	delete [] m_UserIdLookUp;
    94 }
    95 
    96 void PlayerManager::OnSourceModAllInitialized()
    97 {
    98 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientConnect, serverClients, this, &PlayerManager::OnClientConnect, false);
    99 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientConnect, serverClients, this, &PlayerManager::OnClientConnect_Post, true);
   100 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientPutInServer, serverClients, this, &PlayerManager::OnClientPutInServer, true);
   101 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientDisconnect, serverClients, this, &PlayerManager::OnClientDisconnect, false);
   102 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientDisconnect, serverClients, this, &PlayerManager::OnClientDisconnect_Post, true);
   103 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientCommand, serverClients, this, &PlayerManager::OnClientCommand, false);
   104 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientSettingsChanged, serverClients, this, &PlayerManager::OnClientSettingsChanged, true);
   105 	SH_ADD_HOOK_MEMFUNC(IServerGameDLL, ServerActivate, gamedll, this, &PlayerManager::OnServerActivate, true);
   106 
   107 	g_ShareSys.AddInterface(NULL, this);
   108 
   109 	ParamType p1[] = {Param_Cell, Param_String, Param_Cell};
   110 	ParamType p2[] = {Param_Cell};
   111 
   112 	m_clconnect = g_Forwards.CreateForward("OnClientConnect", ET_Event, 3, p1);
   113 	m_clputinserver = g_Forwards.CreateForward("OnClientPutInServer", ET_Ignore, 1, p2);
   114 	m_cldisconnect = g_Forwards.CreateForward("OnClientDisconnect", ET_Ignore, 1, p2);
   115 	m_cldisconnect_post = g_Forwards.CreateForward("OnClientDisconnect_Post", ET_Ignore, 1, p2);
   116 	m_clcommand = g_Forwards.CreateForward("OnClientCommand", ET_Hook, 2, NULL, Param_Cell, Param_Cell);
   117 	m_clinfochanged = g_Forwards.CreateForward("OnClientSettingsChanged", ET_Ignore, 1, p2);
   118 	m_clauth = g_Forwards.CreateForward("OnClientAuthorized", ET_Ignore, 2, NULL, Param_Cell, Param_String);
   119 	m_onActivate = g_Forwards.CreateForward("OnServerLoad", ET_Ignore, 0, NULL);
   120 	m_onActivate2 = g_Forwards.CreateForward("OnMapStart", ET_Ignore, 0, NULL);
   121 
   122 	PreAdminCheck = g_Forwards.CreateForward("OnClientPreAdminCheck", ET_Event, 1, p1);
   123 	PostAdminCheck = g_Forwards.CreateForward("OnClientPostAdminCheck", ET_Ignore, 1, p1);
   124 }
   125 
   126 void PlayerManager::OnSourceModShutdown()
   127 {
   128 	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientConnect, serverClients, this, &PlayerManager::OnClientConnect, false);
   129 	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientPutInServer, serverClients, this, &PlayerManager::OnClientPutInServer, true);
   130 	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientDisconnect, serverClients, this, &PlayerManager::OnClientDisconnect, false);
   131 	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientDisconnect, serverClients, this, &PlayerManager::OnClientDisconnect_Post, true);
   132 	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientCommand, serverClients, this, &PlayerManager::OnClientCommand, false);
   133 	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientSettingsChanged, serverClients, this, &PlayerManager::OnClientSettingsChanged, true);
   134 	SH_REMOVE_HOOK_MEMFUNC(IServerGameDLL, ServerActivate, gamedll, this, &PlayerManager::OnServerActivate, true);
   135 
   136 	/* Release forwards */
   137 	g_Forwards.ReleaseForward(m_clconnect);
   138 	g_Forwards.ReleaseForward(m_clputinserver);
   139 	g_Forwards.ReleaseForward(m_cldisconnect);
   140 	g_Forwards.ReleaseForward(m_cldisconnect_post);
   141 	g_Forwards.ReleaseForward(m_clcommand);
   142 	g_Forwards.ReleaseForward(m_clinfochanged);
   143 	g_Forwards.ReleaseForward(m_clauth);
   144 	g_Forwards.ReleaseForward(m_onActivate);
   145 	g_Forwards.ReleaseForward(m_onActivate2);
   146 
   147 	g_Forwards.ReleaseForward(PreAdminCheck);
   148 	g_Forwards.ReleaseForward(PostAdminCheck);
   149 
   150 	delete [] m_Players;
   151 }
   152 
   153 ConfigResult PlayerManager::OnSourceModConfigChanged(const char *key, 
   154 												 const char *value, 
   155 												 ConfigSource source, 
   156 												 char *error, 
   157 												 size_t maxlength)
   158 {
   159 	if (strcmp(key, "PassInfoVar") == 0)
   160 	{
   161 		if (strcmp(value, "_password") != 0)
   162 		{
   163 			m_PassInfoVar.assign(value);
   164 		}
   165 		return ConfigResult_Accept;
   166 	} else if (strcmp(key, "AllowClLanguageVar") == 0) {
   167 		if (strcasecmp(value, "on") == 0)
   168 		{
   169 			m_QueryLang = true;
   170 		} else if (strcasecmp(value, "off") == 0) {
   171 			m_QueryLang = false;
   172 		} else {
   173 			UTIL_Format(error, maxlength, "Invalid value: must be \"on\" or \"off\"");
   174 			return ConfigResult_Reject;
   175 		}
   176 		return ConfigResult_Accept;
   177 	}
   178 	return ConfigResult_Ignore;
   179 }
   180 
   181 void PlayerManager::OnServerActivate(edict_t *pEdictList, int edictCount, int clientMax)
   182 {
   183 	if (!m_FirstPass)
   184 	{
   185 		/* Initialize all players */
   186 		m_maxClients = clientMax;
   187 		m_PlayerCount = 0;
   188 		m_Players = new CPlayer[m_maxClients + 1];
   189 		m_AuthQueue = new unsigned int[m_maxClients + 1];
   190 		m_FirstPass = true;
   191 
   192 		memset(m_AuthQueue, 0, sizeof(unsigned int) * (m_maxClients + 1));
   193 
   194 		g_NumPlayersToAuth = &m_AuthQueue[0];
   195 	}
   196 	m_onActivate->Execute(NULL);
   197 	m_onActivate2->Execute(NULL);
   198 
   199 	List<IClientListener *>::iterator iter;
   200 	for (iter = m_hooks.begin(); iter != m_hooks.end(); iter++)
   201 	{
   202 		if ((*iter)->GetClientListenerVersion() >= 5)
   203 		{
   204 			(*iter)->OnServerActivated(clientMax);
   205 		}
   206 	}
   207 
   208 	g_OnMapStarted = true;
   209 
   210 	SM_ExecuteAllConfigs();
   211 }
   212 
   213 bool PlayerManager::IsServerActivated()
   214 {
   215 	return m_FirstPass;
   216 }
   217 
   218 bool PlayerManager::CheckSetAdmin(int index, CPlayer *pPlayer, AdminId id)
   219 {
   220 	const char *password = g_Admins.GetAdminPassword(id);
   221 	if (password != NULL)
   222 	{
   223 		if (m_PassInfoVar.size() < 1)
   224 		{
   225 			return false;
   226 		}
   227 
   228 		/* Whoa... the user needs a password! */
   229 		const char *given = engine->GetClientConVarValue(index, m_PassInfoVar.c_str());
   230 		if (!given || strcmp(given, password) != 0)
   231 		{
   232 			return false;
   233 		}
   234 	}
   235 
   236 	pPlayer->SetAdminId(id, false);
   237 
   238 	return true;
   239 }
   240 
   241 bool PlayerManager::CheckSetAdminName(int index, CPlayer *pPlayer, AdminId id)
   242 {
   243 	const char *password = g_Admins.GetAdminPassword(id);
   244 	if (password == NULL)
   245 	{
   246 		return false;
   247 	}
   248 
   249 	if (m_PassInfoVar.size() < 1)
   250 	{
   251 		return false;
   252 	}
   253 
   254 	/* Whoa... the user needs a password! */
   255 	const char *given = engine->GetClientConVarValue(index, m_PassInfoVar.c_str());
   256 	if (!given || strcmp(given, password) != 0)
   257 	{
   258 		return false;
   259 	}
   260 
   261 	pPlayer->SetAdminId(id, false);
   262 
   263 	return true;
   264 }
   265 
   266 void PlayerManager::RunAuthChecks()
   267 {
   268 	CPlayer *pPlayer;
   269 	const char *authstr;
   270 	unsigned int removed = 0;
   271 	for (unsigned int i=1; i<=m_AuthQueue[0]; i++)
   272 	{
   273 		pPlayer = GetPlayerByIndex(m_AuthQueue[i]);
   274 		authstr = engine->GetPlayerNetworkIDString(pPlayer->m_pEdict);
   275 		if (authstr && authstr[0] != '\0'
   276 			&& (strcmp(authstr, "STEAM_ID_PENDING") != 0))
   277 		{
   278 			/* Set authorization */
   279 			pPlayer->Authorize(authstr);
   280 
   281 			/* Mark as removed from queue */
   282 			unsigned int client = m_AuthQueue[i];
   283 			m_AuthQueue[i] = 0;
   284 			removed++;
   285 
   286 			/* Send to extensions */
   287 			List<IClientListener *>::iterator iter;
   288 			IClientListener *pListener;
   289 			for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   290 			{
   291 				pListener = (*iter);
   292 				pListener->OnClientAuthorized(client, authstr);
   293 				if (!pPlayer->IsConnected())
   294 				{
   295 					break;
   296 				}
   297 			}
   298 
   299 			/* Send to plugins if player is still connected */
   300 			if (pPlayer->IsConnected() && m_clauth->GetFunctionCount())
   301 			{
   302 				/* :TODO: handle the case of a player disconnecting in the middle */
   303 				m_clauth->PushCell(client);
   304 				m_clauth->PushString(authstr);
   305 				m_clauth->Execute(NULL);
   306 			}
   307 
   308 			if (pPlayer->IsConnected())
   309 			{
   310 				pPlayer->Authorize_Post();
   311 			}
   312 		}
   313 	}
   314 
   315 	/* Clean up the queue */
   316 	if (removed)
   317 	{
   318 		/* We don't have to compact the list if it's empty */
   319 		if (removed != m_AuthQueue[0])
   320 		{
   321 			unsigned int diff = 0;
   322 			for (unsigned int i=1; i<=m_AuthQueue[0]; i++)
   323 			{
   324 				/* If this member is removed... */
   325 				if (m_AuthQueue[i] == 0)
   326 				{
   327 					/* Increase the differential */
   328 					diff++;
   329 				} else {
   330 					/* diff cannot increase faster than i+1 */
   331 					assert(i > diff);
   332 					assert(i - diff >= 1);
   333 					/* move this index down */
   334 					m_AuthQueue[i - diff] = m_AuthQueue[i];
   335 				}
   336 			}
   337 			m_AuthQueue[0] -= removed;
   338 		} else {
   339 			m_AuthQueue[0] = 0;
   340 		}
   341 	}
   342 }
   343 
   344 bool PlayerManager::OnClientConnect(edict_t *pEntity, const char *pszName, const char *pszAddress, char *reject, int maxrejectlen)
   345 {
   346 	int client = engine->IndexOfEdict(pEntity);
   347 
   348 	List<IClientListener *>::iterator iter;
   349 	IClientListener *pListener = NULL;
   350 	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   351 	{
   352 		pListener = (*iter);
   353 		if (!pListener->InterceptClientConnect(client, reject, maxrejectlen))
   354 		{
   355 			return false;
   356 		}
   357 	}
   358 
   359 	cell_t res = 1;
   360 
   361 	m_Players[client].Initialize(pszName, pszAddress, pEntity);
   362 	m_clconnect->PushCell(client);
   363 	m_clconnect->PushStringEx(reject, maxrejectlen, SM_PARAM_STRING_UTF8, SM_PARAM_COPYBACK);
   364 	m_clconnect->PushCell(maxrejectlen);
   365 	m_clconnect->Execute(&res, NULL);
   366 
   367 	if (res)
   368 	{
   369 		if (!m_Players[client].IsAuthorized())
   370 		{
   371 			m_AuthQueue[++m_AuthQueue[0]] = client;
   372 		}
   373 	} else {
   374 		RETURN_META_VALUE(MRES_SUPERCEDE, false);
   375 	}
   376 
   377 	m_UserIdLookUp[engine->GetPlayerUserId(pEntity)] = client;
   378 
   379 	return true;
   380 }
   381 
   382 bool PlayerManager::OnClientConnect_Post(edict_t *pEntity, const char *pszName, const char *pszAddress, char *reject, int maxrejectlen)
   383 {
   384 	int client = engine->IndexOfEdict(pEntity);
   385 	bool orig_value = META_RESULT_ORIG_RET(bool);
   386 	CPlayer *pPlayer = GetPlayerByIndex(client);
   387 
   388 	if (orig_value)
   389 	{
   390 		List<IClientListener *>::iterator iter;
   391 		IClientListener *pListener = NULL;
   392 		for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   393 		{
   394 			pListener = (*iter);
   395 			pListener->OnClientConnected(client);
   396 			if (!pPlayer->IsConnected())
   397 			{
   398 				break;
   399 			}
   400 		}
   401 	}
   402 
   403 	return true;
   404 }
   405 
   406 void PlayerManager::OnClientPutInServer(edict_t *pEntity, const char *playername)
   407 {
   408 	cell_t res;
   409 	int client = engine->IndexOfEdict(pEntity);
   410 
   411 	CPlayer *pPlayer = GetPlayerByIndex(client);
   412 	/* If they're not connected, they're a bot */
   413 	if (!pPlayer->IsConnected())
   414 	{
   415 		/* Run manual connection routines */
   416 		char error[255];
   417 		const char *authid = engine->GetPlayerNetworkIDString(pEntity);
   418 		pPlayer->Authorize(authid);
   419 		if (!OnClientConnect(pEntity, playername, "127.0.0.1", error, sizeof(error)))
   420 		{
   421 			/* :TODO: kick the bot if it's rejected */
   422 			return;
   423 		}
   424 		List<IClientListener *>::iterator iter;
   425 		IClientListener *pListener = NULL;
   426 		for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   427 		{
   428 			pListener = (*iter);
   429 			pListener->OnClientConnected(client);
   430 			/* See if bot was kicked */
   431 			if (!pPlayer->IsConnected())
   432 			{
   433 				return;
   434 			}
   435 		}
   436 		/* Now do authorization */
   437 		for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   438 		{
   439 			pListener = (*iter);
   440 			pListener->OnClientAuthorized(client, authid);
   441 		}
   442 		/* Finally, tell plugins */
   443 		if (m_clauth->GetFunctionCount())
   444 		{
   445 			m_clauth->PushCell(client);
   446 			m_clauth->PushString(authid);
   447 			m_clauth->Execute(NULL);
   448 		}
   449 		pPlayer->Authorize_Post();
   450 	}
   451 
   452 	if (playerinfo)
   453 	{
   454 		pPlayer->m_Info = playerinfo->GetPlayerInfo(pEntity);
   455 	}
   456 
   457 	/* Get the client's language */
   458 	if (m_QueryLang)
   459 	{
   460 		const char *name;
   461 		if (!pPlayer->IsFakeClient() && (name=engine->GetClientConVarValue(client, "cl_language")))
   462 		{
   463 			unsigned int langid;
   464 			pPlayer->m_LangId = (g_Translator.GetLanguageByName(name, &langid)) ? langid : g_Translator.GetServerLanguage();
   465 		} else {
   466 			pPlayer->m_LangId = g_Translator.GetServerLanguage();
   467 		}
   468 	}
   469 
   470 	m_Players[client].Connect();
   471 	m_PlayerCount++;
   472 
   473 	List<IClientListener *>::iterator iter;
   474 	IClientListener *pListener = NULL;
   475 	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   476 	{
   477 		pListener = (*iter);
   478 		pListener->OnClientPutInServer(client);
   479 	}
   480 
   481 	m_clputinserver->PushCell(client);
   482 	m_clputinserver->Execute(&res, NULL);
   483 }
   484 
   485 void PlayerManager::OnSourceModLevelEnd()
   486 {
   487 	/* Disconnect all bots still in game */
   488 	for (int i=1; i<=m_maxClients; i++)
   489 	{
   490 		if (m_Players[i].IsConnected())
   491 		{
   492 			OnClientDisconnect(m_Players[i].GetEdict());
   493 		}
   494 	}
   495 	m_PlayerCount = 0;
   496 }
   497 
   498 void PlayerManager::OnClientDisconnect(edict_t *pEntity)
   499 {
   500 	cell_t res;
   501 	int client = engine->IndexOfEdict(pEntity);
   502 
   503 	if (m_Players[client].IsConnected())
   504 	{
   505 		m_cldisconnect->PushCell(client);
   506 		m_cldisconnect->Execute(&res, NULL);
   507 	} else {
   508 		/* We don't care, prevent a double call */
   509 		return;
   510 	}
   511 
   512 	if (m_Players[client].WasCountedAsInGame())
   513 	{
   514 		m_PlayerCount--;
   515 	}
   516 
   517 	List<IClientListener *>::iterator iter;
   518 	IClientListener *pListener = NULL;
   519 	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   520 	{
   521 		pListener = (*iter);
   522 		pListener->OnClientDisconnecting(client);
   523 	}
   524 
   525 	/**
   526 	 * Remove client from auth queue if necessary
   527 	 */
   528 	if (!m_Players[client].IsAuthorized())
   529 	{
   530 		for (unsigned int i=1; i<=m_AuthQueue[0]; i++)
   531 		{
   532 			if (m_AuthQueue[i] == (unsigned)client)
   533 			{
   534 				/* Move everything ahead of us back by one */
   535 				for (unsigned int j=i+1; j<=m_AuthQueue[0]; j++)
   536 				{
   537 					m_AuthQueue[j-1] = m_AuthQueue[j];
   538 				}
   539 				/* Remove us and break */
   540 				m_AuthQueue[0]--;
   541 				break;
   542 			}
   543 		}
   544 	}
   545 
   546 	m_Players[client].Disconnect();
   547 	m_UserIdLookUp[engine->GetPlayerUserId(pEntity)] = 0;
   548 }
   549 
   550 void PlayerManager::OnClientDisconnect_Post(edict_t *pEntity)
   551 {
   552 	cell_t res;
   553 	int client = engine->IndexOfEdict(pEntity);
   554 
   555 	m_cldisconnect_post->PushCell(client);
   556 	m_cldisconnect_post->Execute(&res, NULL);
   557 
   558 	List<IClientListener *>::iterator iter;
   559 	IClientListener *pListener = NULL;
   560 	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   561 	{
   562 		pListener = (*iter);
   563 		pListener->OnClientDisconnected(client);
   564 	}
   565 }
   566 
   567 void PlayerManager::OnClientCommand(edict_t *pEntity)
   568 {
   569 	int client = engine->IndexOfEdict(pEntity);
   570 	cell_t res = Pl_Continue;
   571 
   572 	CPlayer *pPlayer = GetPlayerByIndex(client);
   573 	if (!pPlayer || !pPlayer->IsConnected())
   574 	{
   575 		return;
   576 	}
   577 
   578 	/**
   579 	 * We cache this because the engine is not re-entrant.
   580 	 */
   581 	char cmd[300];
   582 	int args = engine->Cmd_Argc() - 1;
   583 	strncopy(cmd, engine->Cmd_Argv(0), sizeof(cmd));
   584 
   585 	bool result = g_ValveMenuStyle.OnClientCommand(client, cmd);
   586 	if (result)
   587 	{
   588 		res = Pl_Handled;
   589 	} else {
   590 		result = g_RadioMenuStyle.OnClientCommand(client, cmd);
   591 		if (result)
   592 		{
   593 			res = Pl_Handled;
   594 		}
   595 	}
   596 
   597 	cell_t res2 = Pl_Continue;
   598 	m_clcommand->PushCell(client);
   599 	m_clcommand->PushCell(args);
   600 	m_clcommand->Execute(&res2, NULL);
   601 
   602 	if (res2 > res)
   603 	{
   604 		res = res2;
   605 	}
   606 
   607 	if (res >= Pl_Stop)
   608 	{
   609 		RETURN_META(MRES_SUPERCEDE);
   610 	}
   611 
   612 	res = g_ConCmds.DispatchClientCommand(client, cmd, args, (ResultType)res);
   613 
   614 	if (res >= Pl_Handled)
   615 	{
   616 		RETURN_META(MRES_SUPERCEDE);
   617 	}
   618 }
   619 
   620 void PlayerManager::OnClientSettingsChanged(edict_t *pEntity)
   621 {
   622 	cell_t res;
   623 	int client = engine->IndexOfEdict(pEntity);
   624 
   625 	if (!m_Players[client].IsConnected())
   626 	{
   627 		return;
   628 	}
   629 
   630 	m_clinfochanged->PushCell(engine->IndexOfEdict(pEntity));
   631 	m_clinfochanged->Execute(&res, NULL);
   632 
   633 	IPlayerInfo *info = m_Players[client].GetPlayerInfo();
   634 	const char *new_name = info ? info->GetName() : engine->GetClientConVarValue(client, "name");
   635 	const char *old_name = m_Players[client].m_Name.c_str();
   636 
   637 	if (strcmp(old_name, new_name) != 0)
   638 	{
   639 		AdminId id = g_Admins.FindAdminByIdentity("name", new_name);
   640 		if (id != INVALID_ADMIN_ID && m_Players[client].GetAdminId() != id)
   641 		{
   642 			if (!CheckSetAdminName(client, &m_Players[client], id))
   643 			{
   644 				m_Players[client].Kick("Your name is reserved by SourceMod; set your password to use it.");
   645 				RETURN_META(MRES_IGNORED);
   646 			}
   647 		} else if ((id = g_Admins.FindAdminByIdentity("name", old_name)) != INVALID_ADMIN_ID) {
   648 			if (id == m_Players[client].GetAdminId())
   649 			{
   650 				/* This player is changing their name; force them to drop admin privileges! */
   651 				m_Players[client].SetAdminId(INVALID_ADMIN_ID, false);
   652 			}
   653 		}
   654 		m_Players[client].SetName(new_name);
   655 	}
   656 	
   657 	if (m_PassInfoVar.size() > 0)
   658 	{
   659 		/* Try for a password change */
   660 		const char *old_pass = m_Players[client].m_LastPassword.c_str();
   661 		const char *new_pass = engine->GetClientConVarValue(client, m_PassInfoVar.c_str());
   662 		if (strcmp(old_pass, new_pass) != 0)
   663 		{
   664 			m_Players[client].m_LastPassword.assign(new_pass);
   665 			if (m_Players[client].IsInGame() && m_Players[client].IsAuthorized())
   666 			{
   667 				/* If there is already an admin id assigned, this will just bail out. */
   668 				m_Players[client].DoBasicAdminChecks();
   669 			}
   670 		}
   671 	}
   672 }
   673 
   674 int PlayerManager::GetMaxClients()
   675 {
   676 	return m_maxClients;
   677 }
   678 
   679 CPlayer *PlayerManager::GetPlayerByIndex(int client) const
   680 {
   681 	if (client > m_maxClients || client < 1)
   682 	{
   683 		return NULL;
   684 	}
   685 	return &m_Players[client];
   686 }
   687 
   688 int PlayerManager::GetNumPlayers()
   689 {
   690 	return m_PlayerCount;
   691 }
   692 
   693 int PlayerManager::GetClientOfUserId(int userid)
   694 {
   695 	if (userid < 0 || userid > USHRT_MAX)
   696 	{
   697 		return 0;
   698 	}
   699 	
   700 	int client = m_UserIdLookUp[userid];
   701 
   702 	/* Verify the userid.  The cache can get messed up with older
   703 	 * Valve engines.  :TODO: If this gets fixed, do an old engine 
   704 	 * check before invoking this backwards compat code.
   705 	 */
   706 	if (client)
   707 	{
   708 		CPlayer *player = GetPlayerByIndex(client);
   709 		if (player && player->IsConnected())
   710 		{
   711 			int realUserId = engine->GetPlayerUserId(player->GetEdict());
   712 			if (realUserId == userid)
   713 			{
   714 				return client;
   715 			}
   716 		}
   717 	}
   718 
   719 	/* If we can't verify the userid, we have to do a manual loop */
   720 	CPlayer *player;
   721 	for (int i = 1; i <= m_maxClients; i++)
   722 	{
   723 		player = GetPlayerByIndex(i);
   724 		if (!player || !player->IsConnected())
   725 		{
   726 			continue;
   727 		}
   728 		if (engine->GetPlayerUserId(player->GetEdict()) == userid)
   729 		{
   730 			m_UserIdLookUp[userid] = i;
   731 			return i;
   732 		}
   733 	}
   734 
   735 	return 0;
   736 }
   737 
   738 void PlayerManager::AddClientListener(IClientListener *listener)
   739 {
   740 	m_hooks.push_back(listener);
   741 }
   742 
   743 void PlayerManager::RemoveClientListener(IClientListener *listener)
   744 {
   745 	m_hooks.remove(listener);
   746 }
   747 
   748 IGamePlayer *PlayerManager::GetGamePlayer(edict_t *pEdict)
   749 {
   750 	int index = engine->IndexOfEdict(pEdict);
   751 	return GetGamePlayer(index);
   752 }
   753 
   754 IGamePlayer *PlayerManager::GetGamePlayer(int client)
   755 {
   756 	return GetPlayerByIndex(client);
   757 }
   758 
   759 void PlayerManager::ClearAdminId(AdminId id)
   760 {
   761 	for (int i=1; i<=m_maxClients; i++)
   762 	{
   763 		if (m_Players[i].m_Admin == id)
   764 		{
   765 			m_Players[i].DumpAdmin(true);
   766 		}
   767 	}
   768 }
   769 
   770 void PlayerManager::ClearAllAdmins()
   771 {
   772 	for (int i=1; i<=m_maxClients; i++)
   773 	{
   774 		m_Players[i].DumpAdmin(true);
   775 	}
   776 }
   777 
   778 const char *PlayerManager::GetPassInfoVar()
   779 {
   780 	return m_PassInfoVar.c_str();
   781 }
   782 
   783 void PlayerManager::RecheckAnyAdmins()
   784 {
   785 	for (int i=1; i<=m_maxClients; i++)
   786 	{
   787 		if (m_Players[i].IsInGame() && m_Players[i].IsAuthorized())
   788 		{
   789 			m_Players[i].DoBasicAdminChecks();
   790 		}
   791 	}
   792 }
   793 
   794 /*******************
   795  *** PLAYER CODE ***
   796  *******************/
   797 
   798 CPlayer::CPlayer()
   799 {
   800 	m_IsConnected = false;
   801 	m_IsInGame = false;
   802 	m_IsAuthorized = false;
   803 	m_pEdict = NULL;
   804 	m_Admin = INVALID_ADMIN_ID;
   805 	m_TempAdmin = false;
   806 	m_Info = NULL;
   807 	m_bAdminCheckSignalled = false;
   808 	m_bIsInKickQueue = false;
   809 	m_LastPassword.clear();
   810 	m_LangId = LANGUAGE_ENGLISH;
   811 }
   812 
   813 void CPlayer::Initialize(const char *name, const char *ip, edict_t *pEntity)
   814 {
   815 	m_IsConnected = true;
   816 	m_Name.assign(name);
   817 	m_Ip.assign(ip);
   818 	m_pEdict = pEntity;
   819 	m_iIndex = engine->IndexOfEdict(pEntity);
   820 	m_LangId = g_Translator.GetServerLanguage();
   821 
   822 	char ip2[24], *ptr;
   823 	strncopy(ip2, ip, sizeof(ip2));
   824 	if ((ptr = strchr(ip2, ':')) != NULL)
   825 	{
   826 		*ptr = '\0';
   827 	}
   828 	m_IpNoPort.assign(ip2);
   829 }
   830 
   831 void CPlayer::Connect()
   832 {
   833 	if (m_IsInGame)
   834 	{
   835 		return;
   836 	}
   837 
   838 	m_IsInGame = true;
   839 
   840 	const char *var = g_Players.GetPassInfoVar();
   841 	int client = engine->IndexOfEdict(m_pEdict);
   842 	if (var[0] != '\0')
   843 	{
   844 		const char *pass = engine->GetClientConVarValue(client, var);
   845 		m_LastPassword.assign(pass ? pass : "");
   846 	} else {
   847 		m_LastPassword.assign("");
   848 	}
   849 
   850 	if (m_IsAuthorized)
   851 	{
   852 		DoPostConnectAuthorization();
   853 	}
   854 }
   855 
   856 void CPlayer::Authorize(const char *steamid)
   857 {
   858 	if (m_IsAuthorized)
   859 	{
   860 		return;
   861 	}
   862 
   863 	m_IsAuthorized = true;
   864 	m_AuthID.assign(steamid);
   865 }
   866 
   867 void CPlayer::Disconnect()
   868 {
   869 	DumpAdmin(false);
   870 	m_IsConnected = false;
   871 	m_IsInGame = false;
   872 	m_IsAuthorized = false;
   873 	m_Name.clear();
   874 	m_Ip.clear();
   875 	m_AuthID.clear();
   876 	m_pEdict = NULL;
   877 	m_Info = NULL;
   878 	m_bAdminCheckSignalled = false;
   879 	m_UserId = -1;
   880 	m_bIsInKickQueue = false;
   881 }
   882 
   883 void CPlayer::SetName(const char *name)
   884 {
   885 	m_Name.assign(name);
   886 }
   887 
   888 const char *CPlayer::GetName()
   889 {
   890 	if (m_Info && m_pEdict->GetUnknown())
   891 	{
   892 		return m_Info->GetName();
   893 	}
   894 	
   895 	return m_Name.c_str();
   896 }
   897 
   898 const char *CPlayer::GetIPAddress()
   899 {
   900 	return m_Ip.c_str();
   901 }
   902 
   903 const char *CPlayer::GetAuthString()
   904 {
   905 	return m_AuthID.c_str();
   906 }
   907 
   908 edict_t *CPlayer::GetEdict()
   909 {
   910 	return m_pEdict;
   911 }
   912 
   913 bool CPlayer::IsInGame()
   914 {
   915 	return m_IsInGame && (m_pEdict->GetUnknown() != NULL);
   916 }
   917 
   918 bool CPlayer::WasCountedAsInGame()
   919 {
   920 	return m_IsInGame;
   921 }
   922 
   923 bool CPlayer::IsConnected()
   924 {
   925 	return m_IsConnected;
   926 }
   927 
   928 bool CPlayer::IsAuthorized()
   929 {
   930 	return m_IsAuthorized;
   931 }
   932 
   933 IPlayerInfo *CPlayer::GetPlayerInfo()
   934 {
   935 	if (m_pEdict->GetUnknown())
   936 	{
   937 		return m_Info;
   938 	}
   939 
   940 	return NULL;
   941 }
   942 
   943 bool CPlayer::IsFakeClient()
   944 {
   945 	return (strcmp(m_AuthID.c_str(), "BOT") == 0);
   946 }
   947 
   948 void CPlayer::SetAdminId(AdminId id, bool temporary)
   949 {
   950 	if (!m_IsConnected)
   951 	{
   952 		return;
   953 	}
   954 
   955 	DumpAdmin(false);
   956 
   957 	m_Admin = id;
   958 	m_TempAdmin = temporary;
   959 }
   960 
   961 AdminId CPlayer::GetAdminId()
   962 {
   963 	return m_Admin;
   964 }
   965 
   966 void CPlayer::DumpAdmin(bool deleting)
   967 {
   968 	if (m_Admin != INVALID_ADMIN_ID)
   969 	{
   970 		if (m_TempAdmin && !deleting)
   971 		{
   972 			g_Admins.InvalidateAdmin(m_Admin);
   973 		}
   974 		m_Admin = INVALID_ADMIN_ID;
   975 		m_TempAdmin = false;
   976 	}
   977 }
   978 
   979 void CPlayer::Kick(const char *str)
   980 {
   981 	INetChannel *pNetChan = static_cast<INetChannel *>(engine->GetPlayerNetInfo(m_iIndex));
   982 	IClient *pClient = static_cast<IClient *>(pNetChan->GetMsgHandler());
   983 	pClient->Disconnect("%s", str);
   984 }
   985 
   986 void CPlayer::Authorize_Post()
   987 {
   988 	if (m_IsInGame)
   989 	{
   990 		DoPostConnectAuthorization();
   991 	}
   992 }
   993 
   994 void CPlayer::DoPostConnectAuthorization()
   995 {
   996 	cell_t result = 0;
   997 	PreAdminCheck->PushCell(m_iIndex);
   998 	PreAdminCheck->Execute(&result);
   999 
  1000 	/* Defer, for better or worse */
  1001 	if ((ResultType)result >= Pl_Handled)
  1002 	{
  1003 		return;
  1004 	}
  1005 
  1006 	/* Sanity check */
  1007 	if (!IsConnected())
  1008 	{
  1009 		return;
  1010 	}
  1011 
  1012 	/* Otherwise, go ahead and do admin checks */
  1013 	DoBasicAdminChecks();
  1014 
  1015 	/* Send the notification out */
  1016 	NotifyPostAdminChecks();
  1017 }
  1018 
  1019 void CPlayer::NotifyPostAdminChecks()
  1020 {
  1021 	if (m_bAdminCheckSignalled)
  1022 	{
  1023 		return;
  1024 	}
  1025 
  1026 	/* Block beforehand so they can't double-call */
  1027 	m_bAdminCheckSignalled = true;
  1028 
  1029 	PostAdminCheck->PushCell(m_iIndex);
  1030 	PostAdminCheck->Execute(NULL);
  1031 }
  1032 
  1033 void CPlayer::DoBasicAdminChecks()
  1034 {
  1035 	if (GetAdminId() != INVALID_ADMIN_ID)
  1036 	{
  1037 		return;
  1038 	}
  1039 
  1040 	/* First check the name */
  1041 	AdminId id;
  1042 	int client = engine->IndexOfEdict(m_pEdict);
  1043 
  1044 	if ((id = g_Admins.FindAdminByIdentity("name", GetName())) != INVALID_ADMIN_ID)
  1045 	{
  1046 		if (!g_Players.CheckSetAdminName(client, this, id))
  1047 		{
  1048 			int userid = engine->GetPlayerUserId(m_pEdict);
  1049 			g_Timers.CreateTimer(&s_KickPlayerTimer, 0.1f, (void *)userid, 0);
  1050 		}
  1051 		return;
  1052 	}
  1053 	
  1054 	/* Check IP */
  1055 	if ((id = g_Admins.FindAdminByIdentity("ip", m_IpNoPort.c_str())) != INVALID_ADMIN_ID)
  1056 	{
  1057 		if (g_Players.CheckSetAdmin(client, this, id))
  1058 		{
  1059 			return;
  1060 		}
  1061 	}
  1062 
  1063 	/* Check IP address */
  1064 	if ((id = g_Admins.FindAdminByIdentity("steam", m_AuthID.c_str())) != INVALID_ADMIN_ID)
  1065 	{
  1066 		if (g_Players.CheckSetAdmin(client, this, id))
  1067 		{
  1068 			return;
  1069 		}
  1070 	}
  1071 }
  1072 
  1073 unsigned int CPlayer::GetLanguageId()
  1074 {
  1075 	return m_LangId;
  1076 }
  1077 
  1078 int CPlayer::GetUserId()
  1079 {
  1080 	if (m_UserId == -1)
  1081 	{
  1082 		m_UserId = engine->GetPlayerUserId(GetEdict());
  1083 	}
  1084 
  1085 	return m_UserId;
  1086 }
  1087 
  1088 bool CPlayer::IsInKickQueue()
  1089 {
  1090 	return m_bIsInKickQueue;
  1091 }
  1092 
  1093 void CPlayer::MarkAsBeingKicked()
  1094 {
  1095 	m_bIsInKickQueue = true;
  1096 }