Home - sourcemod-central - file revision - core/CPlayerManager.cpp@7611078f11cd Mercurial
summary | shortlog | changelog | graph | tags | branches | files | changeset | file | latest | revisions | annotate | diff | raw | help
core/CPlayerManager.cpp
author David Anderson <dvander@alliedmods.net>
Thu Feb 08 01:11:45 2007 +0000 (2007-02-08)
changeset 453 7611078f11cd
parent 452 842dc60e31fd
child 456 b181fc6f59c7
permissions -rw-r--r--
added gameevents to the publicly available interfaces 
     1 /**
     2  * ===============================================================
     3  * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.
     4  * ===============================================================
     5  *
     6  * This file is not open source and may not be copied without explicit
     7  * written permission of AlliedModders LLC.  This file may not be redistributed 
     8  * in whole or significant part.
     9  * For information, see LICENSE.txt or http://www.sourcemod.net/license.php
    10  *
    11  * Version: $Id$
    12  */
    13 
    14 #include "CPlayerManager.h"
    15 #include "ForwardSys.h"
    16 #include "ShareSys.h"
    17 
    18 CPlayerManager g_Players;
    19 
    20 SH_DECL_HOOK5(IServerGameClients, ClientConnect, SH_NOATTRIB, 0, bool, edict_t *, const char *, const char *, char *, int);
    21 SH_DECL_HOOK2_void(IServerGameClients, ClientPutInServer, SH_NOATTRIB, 0, edict_t *, const char *);
    22 SH_DECL_HOOK1_void(IServerGameClients, ClientDisconnect, SH_NOATTRIB, 0, edict_t *);
    23 SH_DECL_HOOK1_void(IServerGameClients, ClientCommand, SH_NOATTRIB, 0, edict_t *);
    24 SH_DECL_HOOK1_void(IServerGameClients, ClientSettingsChanged, SH_NOATTRIB, 0, edict_t *);
    25 SH_DECL_HOOK3_void(IServerGameDLL, ServerActivate, SH_NOATTRIB, 0, edict_t *, int, int);
    26 
    27 CPlayerManager::CPlayerManager()
    28 {
    29 	m_AuthQueue = NULL;
    30 	m_FirstPass = true;
    31 }
    32 
    33 CPlayerManager::~CPlayerManager()
    34 {
    35 	delete [] m_AuthQueue;
    36 }
    37 
    38 void CPlayerManager::OnSourceModAllInitialized()
    39 {
    40 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientConnect, serverClients, this, &CPlayerManager::OnClientConnect, false);
    41 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientConnect, serverClients, this, &CPlayerManager::OnClientConnect_Post, true);
    42 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientPutInServer, serverClients, this, &CPlayerManager::OnClientPutInServer, true);
    43 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientDisconnect, serverClients, this, &CPlayerManager::OnClientDisconnect, false);
    44 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientDisconnect, serverClients, this, &CPlayerManager::OnClientDisconnect_Post, true);
    45 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientCommand, serverClients, this, &CPlayerManager::OnClientCommand, false);
    46 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientSettingsChanged, serverClients, this, &CPlayerManager::OnClientSettingsChanged, true);
    47 	SH_ADD_HOOK_MEMFUNC(IServerGameDLL, ServerActivate, gamedll, this, &CPlayerManager::OnServerActivate, true);
    48 
    49 	g_ShareSys.AddInterface(NULL, this);
    50 
    51 	ParamType p1[] = {Param_Cell, Param_String, Param_Cell};
    52 	ParamType p2[] = {Param_Cell};
    53 	ParamType p3[] = {Param_Cell, Param_String};
    54 
    55 	m_clconnect = g_Forwards.CreateForward("OnClientConnect", ET_Event, 3, p1);
    56 	m_clputinserver = g_Forwards.CreateForward("OnClientPutInServer", ET_Ignore, 1, p2);
    57 	m_cldisconnect = g_Forwards.CreateForward("OnClientDisconnect", ET_Ignore, 1, p2);
    58 	m_cldisconnect_post = g_Forwards.CreateForward("OnClientDisconnect_Post", ET_Ignore, 1, p2);
    59 	m_clcommand = g_Forwards.CreateForward("OnClientCommand", ET_Hook, 1, p2);
    60 	m_clinfochanged = g_Forwards.CreateForward("OnClientSettingsChanged", ET_Ignore, 1, p2);
    61 	m_clauth = g_Forwards.CreateForward("OnClientAuthorized", ET_Ignore, 2, p2);
    62 }
    63 
    64 void CPlayerManager::OnSourceModShutdown()
    65 {
    66 	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientConnect, serverClients, this, &CPlayerManager::OnClientConnect, false);
    67 	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientPutInServer, serverClients, this, &CPlayerManager::OnClientPutInServer, true);
    68 	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientDisconnect, serverClients, this, &CPlayerManager::OnClientDisconnect, false);
    69 	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientDisconnect, serverClients, this, &CPlayerManager::OnClientDisconnect_Post, true);
    70 	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientCommand, serverClients, this, &CPlayerManager::OnClientCommand, false);
    71 	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientSettingsChanged, serverClients, this, &CPlayerManager::OnClientSettingsChanged, true);
    72 	SH_REMOVE_HOOK_MEMFUNC(IServerGameDLL, ServerActivate, gamedll, this, &CPlayerManager::OnServerActivate, true);
    73 
    74 	/* Release forwards */
    75 	g_Forwards.ReleaseForward(m_clconnect);
    76 	g_Forwards.ReleaseForward(m_clputinserver);
    77 	g_Forwards.ReleaseForward(m_cldisconnect);
    78 	g_Forwards.ReleaseForward(m_cldisconnect_post);
    79 	g_Forwards.ReleaseForward(m_clcommand);
    80 	g_Forwards.ReleaseForward(m_clinfochanged);
    81 	g_Forwards.ReleaseForward(m_clauth);
    82 
    83 	delete [] m_Players;
    84 }
    85 
    86 void CPlayerManager::OnServerActivate(edict_t *pEdictList, int edictCount, int clientMax)
    87 {
    88 	if (m_FirstPass)
    89 	{
    90 		/* Initialize all players */
    91 		m_maxClients = clientMax;
    92 		m_PlayerCount = 0;
    93 		m_Players = new CPlayer[m_maxClients + 1];
    94 		m_AuthQueue = new unsigned int[m_maxClients + 1];
    95 		m_FirstPass = false;
    96 
    97 		memset(m_AuthQueue, 0, sizeof(unsigned int) * (m_maxClients + 1));
    98 	}
    99 }
   100 
   101 void CPlayerManager::RunAuthChecks()
   102 {
   103 	CPlayer *pPlayer;
   104 	const char *authstr;
   105 	unsigned int removed = 0;
   106 	for (unsigned int i=1; i<=m_AuthQueue[0]; i++)
   107 	{
   108 		pPlayer = GetPlayerByIndex(m_AuthQueue[i]);
   109 		authstr = engine->GetPlayerNetworkIDString(pPlayer->m_pEdict);
   110 		if (authstr && authstr[0] != '\0'
   111 			&& (strcmp(authstr, "STEAM_ID_PENDING") != 0))
   112 		{
   113 			/* Set authorization */
   114 			pPlayer->m_AuthID.assign(authstr);
   115 			pPlayer->m_IsAuthorized = true;
   116 
   117 			/* Send to extensions */
   118 			List<IClientListener *>::iterator iter;
   119 			IClientListener *pListener;
   120 			for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   121 			{
   122 				pListener = (*iter);
   123 				pListener->OnClientAuthorized(m_AuthQueue[i], authstr);
   124 			}
   125 
   126 			/* Send to plugins */
   127 			if (m_clauth->GetFunctionCount())
   128 			{
   129 				m_clauth->PushCell(m_AuthQueue[i]);
   130 				m_clauth->PushString(authstr);
   131 				m_clauth->Execute(NULL);
   132 			}
   133 
   134 			/* Mark as removed from queue */
   135 			m_AuthQueue[i] = 0;
   136 			removed++;
   137 		}
   138 	}
   139 
   140 	/* Clean up the queue */
   141 	if (removed)
   142 	{
   143 		/* We don't have to compcat the list if it's empty */
   144 		if (removed != m_AuthQueue[0])
   145 		{
   146 			unsigned int diff = 0;
   147 			for (unsigned int i=1; i<=m_AuthQueue[0]; i++)
   148 			{
   149 				/* If this member is removed... */
   150 				if (m_AuthQueue[i] == 0)
   151 				{
   152 					/* Increase the differential */
   153 					diff++;
   154 				} else {
   155 					/* diff cannot increase faster than i+1 */
   156 					assert(i > diff);
   157 					assert(i - diff >= 1);
   158 					/* move this index down */
   159 					m_AuthQueue[i - diff] = m_AuthQueue[i];
   160 				}
   161 			}
   162 			m_AuthQueue[0] -= removed;
   163 		} else {
   164 			m_AuthQueue[0] = 0;
   165 			g_SourceMod.SetAuthChecking(false);
   166 		}
   167 	}
   168 }
   169 
   170 bool CPlayerManager::OnClientConnect(edict_t *pEntity, const char *pszName, const char *pszAddress, char *reject, int maxrejectlen)
   171 {
   172 	int client = engine->IndexOfEdict(pEntity);
   173 
   174 	List<IClientListener *>::iterator iter;
   175 	IClientListener *pListener = NULL;
   176 	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   177 	{
   178 		pListener = (*iter);
   179 		if (!pListener->InterceptClientConnect(client, reject, maxrejectlen))
   180 		{
   181 			return false;
   182 		}
   183 	}
   184 
   185 	cell_t res = 1;
   186 
   187 	m_Players[client].Initialize(pszName, pszAddress, pEntity);
   188 	m_clconnect->PushCell(client);
   189 	m_clconnect->PushStringEx(reject, maxrejectlen, SM_PARAM_STRING_UTF8, SM_PARAM_COPYBACK);
   190 	m_clconnect->PushCell(maxrejectlen);
   191 	m_clconnect->Execute(&res, NULL);
   192 
   193 	if (res)
   194 	{
   195 		m_AuthQueue[++m_AuthQueue[0]] = client;
   196 		g_SourceMod.SetAuthChecking(true);
   197 	}
   198 
   199 	return (res) ? true : false;
   200 }
   201 
   202 bool CPlayerManager::OnClientConnect_Post(edict_t *pEntity, const char *pszName, const char *pszAddress, char *reject, int maxrejectlen)
   203 {
   204 	int client = engine->IndexOfEdict(pEntity);
   205 	bool orig_value = META_RESULT_ORIG_RET(bool);
   206 	if (orig_value)
   207 	{
   208 		List<IClientListener *>::iterator iter;
   209 		IClientListener *pListener = NULL;
   210 		for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   211 		{
   212 			pListener = (*iter);
   213 			pListener->OnClientConnected(client);
   214 		}
   215 	}
   216 
   217 	return true;
   218 }
   219 
   220 void CPlayerManager::OnClientPutInServer(edict_t *pEntity, const char *playername)
   221 {
   222 	cell_t res;
   223 	int client = engine->IndexOfEdict(pEntity);
   224 
   225 	CPlayer *pPlayer = GetPlayerByIndex(client);
   226 	if (!pPlayer->IsConnected())
   227 	{
   228 		/* Run manual connection routines */
   229 		char error[255];
   230 		if (!OnClientConnect(pEntity, playername, "127.0.0.1", error, sizeof(error)))
   231 		{
   232 			/* :TODO: kick the bot if it's rejected */
   233 			return;
   234 		}
   235 		List<IClientListener *>::iterator iter;
   236 		IClientListener *pListener = NULL;
   237 		for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   238 		{
   239 			pListener = (*iter);
   240 			pListener->OnClientConnected(client);
   241 		}
   242 	}
   243 
   244 	List<IClientListener *>::iterator iter;
   245 	IClientListener *pListener = NULL;
   246 	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   247 	{
   248 		pListener = (*iter);
   249 		pListener->OnClientPutInServer(client);
   250 	}
   251 
   252 	m_Players[client].Connect();
   253 	m_PlayerCount++;
   254 	m_clputinserver->PushCell(client);
   255 	m_clputinserver->Execute(&res, NULL);
   256 }
   257 
   258 void CPlayerManager::OnClientAuthorized()
   259 {
   260 	//:TODO:
   261 }
   262 
   263 void CPlayerManager::OnClientDisconnect(edict_t *pEntity)
   264 {
   265 	cell_t res;
   266 	int client = engine->IndexOfEdict(pEntity);
   267 
   268 	if (m_Players[client].IsConnected())
   269 	{
   270 		m_cldisconnect->PushCell(client);
   271 		m_cldisconnect->Execute(&res, NULL);
   272 	}
   273 
   274 	if (m_Players[client].IsInGame())
   275 	{
   276 		m_PlayerCount--;
   277 	}
   278 
   279 	List<IClientListener *>::iterator iter;
   280 	IClientListener *pListener = NULL;
   281 	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   282 	{
   283 		pListener = (*iter);
   284 		pListener->OnClientDisconnecting(client);
   285 	}
   286 
   287 	/**
   288 	 * Remove client from auth queue if necessary
   289 	 */
   290 	if (!m_Players[client].IsAuthorized())
   291 	{
   292 		for (unsigned int i=1; i<=m_AuthQueue[0]; i++)
   293 		{
   294 			if (m_AuthQueue[i] == client)
   295 			{
   296 				/* Move everything ahead of us back by one */
   297 				for (unsigned int j=i+1; j<=m_AuthQueue[0]; j++)
   298 				{
   299 					m_AuthQueue[j-1] = m_AuthQueue[j];
   300 				}
   301 				/* Remove us and break */
   302 				m_AuthQueue[0]--;
   303 				break;
   304 			}
   305 		}
   306 	}
   307 
   308 	m_Players[client].Disconnect();
   309 }
   310 
   311 void CPlayerManager::OnClientDisconnect_Post(edict_t *pEntity)
   312 {
   313 	cell_t res;
   314 	int client = engine->IndexOfEdict(pEntity);
   315 
   316 	m_cldisconnect_post->PushCell(client);
   317 	m_cldisconnect_post->Execute(&res, NULL);
   318 
   319 	List<IClientListener *>::iterator iter;
   320 	IClientListener *pListener = NULL;
   321 	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)
   322 	{
   323 		pListener = (*iter);
   324 		pListener->OnClientDisconnected(client);
   325 	}
   326 }
   327 
   328 void CPlayerManager::OnClientCommand(edict_t *pEntity)
   329 {
   330 	cell_t res;
   331 
   332 	m_clcommand->PushCell(engine->IndexOfEdict(pEntity));
   333 	m_clcommand->Execute(&res, NULL);
   334 }
   335 
   336 void CPlayerManager::OnClientSettingsChanged(edict_t *pEntity)
   337 {
   338 	cell_t res;
   339 	int client = engine->IndexOfEdict(pEntity);
   340 
   341 	m_clinfochanged->PushCell(engine->IndexOfEdict(pEntity));
   342 	m_clinfochanged->Execute(&res, NULL);
   343 	m_Players[client].SetName(engine->GetClientConVarValue(client, "name"));
   344 }
   345 
   346 int CPlayerManager::GetMaxClients()
   347 {
   348 	return m_maxClients;
   349 }
   350 
   351 CPlayer *CPlayerManager::GetPlayerByIndex(int client) const
   352 {
   353 	if (client > m_maxClients || client < 0)
   354 	{
   355 		return NULL;
   356 	}
   357 	return &m_Players[client];
   358 }
   359 
   360 int CPlayerManager::GetNumPlayers()
   361 {
   362 	return m_PlayerCount;
   363 }
   364 
   365 void CPlayerManager::AddClientListener(IClientListener *listener)
   366 {
   367 	m_hooks.push_back(listener);
   368 }
   369 
   370 void CPlayerManager::RemoveClientListener(IClientListener *listener)
   371 {
   372 	m_hooks.remove(listener);
   373 }
   374 
   375 IGamePlayer *CPlayerManager::GetGamePlayer(edict_t *pEdict)
   376 {
   377 	int index = engine->IndexOfEdict(pEdict);
   378 	return GetGamePlayer(index);
   379 }
   380 
   381 IGamePlayer *CPlayerManager::GetGamePlayer(int client)
   382 {
   383 	return GetPlayerByIndex(client);
   384 }
   385 
   386 /*******************
   387  *** PLAYER CODE ***
   388  *******************/
   389 
   390 CPlayer::CPlayer()
   391 {
   392 	m_IsConnected = false;
   393 	m_IsInGame = false;
   394 	m_IsAuthorized = false;
   395 	m_pEdict = NULL;
   396 }
   397 
   398 void CPlayer::Initialize(const char *name, const char *ip, edict_t *pEntity)
   399 {
   400 	m_IsConnected = true;
   401 	m_Name.assign(name);
   402 	m_Ip.assign(ip);
   403 	m_pEdict = pEntity;
   404 }
   405 
   406 void CPlayer::Connect()
   407 {
   408 	m_IsInGame = true;
   409 }
   410 
   411 void CPlayer::Authorize(const char *steamid)
   412 {
   413 	m_IsAuthorized = true;
   414 	m_AuthID.assign(steamid);
   415 }
   416 
   417 void CPlayer::Disconnect()
   418 {
   419 	m_IsConnected = false;
   420 	m_IsInGame = false;
   421 	m_IsAuthorized = false;
   422 	m_Name.clear();
   423 	m_Ip.clear();
   424 	m_AuthID.clear();
   425 	m_pEdict = NULL;
   426 }
   427 
   428 void CPlayer::SetName(const char *name)
   429 {
   430 	m_Name.assign(name);
   431 }
   432 
   433 const char *CPlayer::GetName() const
   434 {
   435 	return m_Name.c_str();
   436 }
   437 
   438 const char *CPlayer::GetIPAddress() const
   439 {
   440 	return m_Ip.c_str();
   441 }
   442 
   443 const char *CPlayer::GetAuthString() const
   444 {
   445 	return m_AuthID.c_str();
   446 }
   447 
   448 edict_t *CPlayer::GetEdict() const
   449 {
   450 	return m_pEdict;
   451 }
   452 
   453 bool CPlayer::IsInGame() const
   454 {
   455 	return m_IsInGame;
   456 }
   457 
   458 bool CPlayer::IsConnected() const
   459 {
   460 	return m_IsConnected;
   461 }
   462 
   463 bool CPlayer::IsAuthorized() const
   464 {
   465 	return m_IsAuthorized;
   466 }
   467 
   468 bool CPlayer::IsFakeClient() const
   469 {
   470 	return (strcmp(m_AuthID.c_str(), "BOT") == 0);
   471 }
sourcemod-central
Feed