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