hl2_src-leak-2017/src/game/server/tf/tf_gc_api.cpp

270 lines
9.9 KiB
C++

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
// $NoKeywords: $
//=============================================================================//
#include "cbase.h"
#include "tf_gc_api.h"
#include <checksum_md5.h>
#include "econ_game_account_server.h"
#include "econ_gcmessages.h"
#include "tf_gcmessages.h"
#include "tf_gamerules.h"
#include "gc_clientsystem.h"
//-----------------------------------------------------------------------------
static ConVar tf_server_identity_token( "tf_server_identity_token", "", FCVAR_ARCHIVE | FCVAR_PROTECTED, "Server identity token, used to authenticate with the TF2 Game Coordinator." );
static ConVar tf_server_identity_account_id( "tf_server_identity_account_id", "0", FCVAR_ARCHIVE, "Server identity account id, used to authenticate with the TF2 Game Coordinator." );
static ConVar tf_server_identity_disable_quickplay( "tf_server_identity_disable_quickplay", "0", FCVAR_ARCHIVE | FCVAR_NOTIFY, "Disable this server from being chosen by the quickplay matchmaking." );
static ConVar sv_registration_successful( "sv_registration_successful", "0", FCVAR_DONTRECORD | FCVAR_HIDDEN | FCVAR_NOTIFY, "Nonzero if we were able to login OK" );
static ConVar sv_registration_message( "sv_registration_message", "No account specified", FCVAR_DONTRECORD | FCVAR_HIDDEN | FCVAR_NOTIFY, "Error message of other status text" );
static bool BSendMessage( const GCSDK::CProtoBufMsgBase& msg )
{
return GCClientSystem()->BSendMessage( msg );
}
//-----------------------------------------------------------------------------
// Public API
//-----------------------------------------------------------------------------
void GameCoordinator_NotifyGameState()
{
if ( TFGameRules() )
{
GCSDK::CProtoBufMsg< CMsgGC_GameServer_LevelInfo > msg( k_EMsgGC_GameServer_LevelInfo );
msg.Body().set_level_loaded( true );
msg.Body().set_level_name( gpGlobals->mapname.ToCStr() );
BSendMessage( msg );
}
else
{
GameCoordinator_NotifyLevelShutdown();
}
}
//-----------------------------------------------------------------------------
void GameCoordinator_NotifyLevelShutdown()
{
GCSDK::CProtoBufMsg< CMsgGC_GameServer_LevelInfo> msg( k_EMsgGC_GameServer_LevelInfo );
msg.Body().set_level_loaded( false );
BSendMessage( msg );
}
//-----------------------------------------------------------------------------
const char *GameCoordinator_GetRegistrationString()
{
return sv_registration_message.GetString();
}
//-----------------------------------------------------------------------------
/**
* Game Coordinator wants to know if we have a token
*/
class CGC_GameServer_AuthChallenge : public GCSDK::CGCClientJob
{
public:
CGC_GameServer_AuthChallenge( GCSDK::CGCClient *pClient ) : GCSDK::CGCClientJob( pClient ) {}
virtual bool BYieldingRunGCJob( GCSDK::IMsgNetPacket *pNetPacket )
{
GCSDK::CProtoBufMsg< CMsgGC_GameServer_AuthChallenge > msg( pNetPacket );
// send information on what level is loaded
GameCoordinator_NotifyGameState();
const uint32 unGameServerAccountID = tf_server_identity_account_id.GetInt();
if ( unGameServerAccountID == 0 )
{
Msg( "%s not set; not logging into registered account\n", tf_server_identity_account_id.GetName() );
UTIL_LogPrintf( "%s not set; not logging into registered account\n", tf_server_identity_account_id.GetName() );
return true;
}
CUtlString challenge = msg.Body().challenge_string().c_str();
if ( challenge.IsEmpty() )
{
Warning( "Received CGC_GameServer_AuthChallenge with invalid challenge from GC!\n" );
UTIL_LogPrintf( "Received CGC_GameServer_AuthChallenge with invalid challenge from GC!\n" );
Assert( false );
return true;
}
char szKeyBuffer[16];
int nKeyLength = Q_snprintf( szKeyBuffer, sizeof( szKeyBuffer ), "%s", tf_server_identity_token.GetString() );
if ( nKeyLength <= 0 || nKeyLength >= ARRAYSIZE( szKeyBuffer ) )
{
Warning( "%s is not valid, or not set! Not signing into gameserver account\n", tf_server_identity_token.GetName() );
UTIL_LogPrintf( "%s is not valid, or not set! Not signing into gameserver account\n", tf_server_identity_token.GetName() );
return true;
}
MD5Context_t ctx;
unsigned char digest[16]; // The MD5 Hash
memset( &ctx, 0, sizeof( ctx ) );
memset( digest, 0, sizeof( digest ) );
// hash together the identity token and the challenge
MD5Init( &ctx );
MD5Update( &ctx, (unsigned char*)szKeyBuffer, nKeyLength );
MD5Update( &ctx, (unsigned char*)challenge.Get(), challenge.Length() );
MD5Final( digest, &ctx );
GCSDK::CProtoBufMsg< CMsgGC_GameServer_AuthChallengeResponse > msg_response( k_EMsgGC_GameServer_AuthChallengeResponse );
msg_response.Body().set_game_server_account_id( unGameServerAccountID );
msg_response.Body().set_hashed_challenge_string( digest, sizeof( digest ) );
BSendMessage( msg_response );
Msg( "Received auth challenge; signing into gameserver account...\n" );
UTIL_LogPrintf( "Received auth challenge; signing into gameserver account...\n" );
return true;
}
};
GC_REG_JOB( GCSDK::CGCClient, CGC_GameServer_AuthChallenge, "CGC_GameServer_AuthChallenge", k_EMsgGC_GameServer_AuthChallenge, GCSDK::k_EServerTypeGCClient );
/**
* Game Coordinator tells game server whether authentication was successful or not
*/
class CGC_GameServer_AuthResult : public GCSDK::CGCClientJob
{
public:
CGC_GameServer_AuthResult( GCSDK::CGCClient *pClient ) : GCSDK::CGCClientJob( pClient ) {}
virtual bool BYieldingRunGCJob( GCSDK::IMsgNetPacket *pNetPacket )
{
GCSDK::CProtoBufMsg< CMsgGC_GameServer_AuthResult > msg( pNetPacket );
if ( msg.Body().is_valve_server() )
{
// Note: I put in this cryptic message, in the hopes that if a server operator
// actually sees it (when they are not supposed to), they will contact us,
// as opposed to just secretly enjoying the benefits of their server being
// treated as a valve server and us not knowing something is wrong.
Msg( "WARNING: Game server status 'Gordon'.\n" );
engine->LogPrint( "WARNING: Game server status 'Gordon'.\n" );
}
if ( msg.Body().authenticated() )
{
const char *pStanding = GameServerAccount_GetStandingString( (eGameServerScoreStanding)msg.Body().game_server_standing() );
const char *pStandingTrend = GameServerAccount_GetStandingTrendString( (eGameServerScoreStandingTrend)msg.Body().game_server_standing_trend() );
Msg( "Game server authentication: SUCCESS! Standing: %s. Trend: %s\n", pStanding, pStandingTrend );
UTIL_LogPrintf( "Game server authentication: SUCCESS! Standing: %s. Trend: %s\n", pStanding, pStandingTrend );
if ( !msg.Body().message().empty() )
{
Msg( " %s\n", msg.Body().message().c_str() );
UTIL_LogPrintf( " %s\n", msg.Body().message().c_str() );
}
// What else could we save here?
if ( msg.Body().is_valve_server() )
{
sv_registration_message.SetValue( "Status 'Gordon'" );
}
else
{
sv_registration_message.SetValue( "" );
}
}
else
{
Warning( "Game server authentication: FAILURE!\n" );
UTIL_LogPrintf( "Game server authentication: FAILURE!\n" );
if ( !msg.Body().message().empty() )
{
Warning( " %s\n", msg.Body().message().c_str() );
UTIL_LogPrintf( " %s\n", msg.Body().message().c_str() );
sv_registration_message.SetValue( msg.Body().message().c_str() );
}
else
{
sv_registration_message.SetValue( "failed" );
}
}
// Update the convar
sv_registration_successful.SetValue( msg.Body().authenticated() );
// !FIXME! How to force server to recalculate tags??
return true;
}
};
GC_REG_JOB( GCSDK::CGCClient, CGC_GameServer_AuthResult, "CGC_GameServer_AuthResult", k_EMsgGC_GameServer_AuthResult, GCSDK::k_EServerTypeGCClient );
//-----------------------------------------------------------------------------
// GC telling us that a player is joining via Quickplay
class CGCTFQuickplay_PlayerJoining : public GCSDK::CGCClientJob
{
public:
CGCTFQuickplay_PlayerJoining( GCSDK::CGCClient *pClient ) : GCSDK::CGCClientJob( pClient ) {}
virtual bool BYieldingRunGCJob( GCSDK::IMsgNetPacket *pNetPacket )
{
GCSDK::CProtoBufMsg<CMsgTFQuickplay_PlayerJoining> msg( pNetPacket );
Log("(Quickplay) Incoming player (%d):\n", msg.Body().account_id() );
return true;
}
};
GC_REG_JOB( GCSDK::CGCClient, CGCTFQuickplay_PlayerJoining, "CGCTFQuickplay_PlayerJoining", k_EMsgGC_QP_PlayerJoining, GCSDK::k_EServerTypeGCClient );
//-----------------------------------------------------------------------------
// GC telling us that a player has moved an item into his or her backpack for the first time
class CGCTFItemAcknowledged : public GCSDK::CGCClientJob
{
public:
CGCTFItemAcknowledged( GCSDK::CGCClient *pClient ) : GCSDK::CGCClientJob( pClient ) {}
virtual bool BYieldingRunGCJob( GCSDK::IMsgNetPacket *pNetPacket )
{
GCSDK::CProtoBufMsg<CMsgItemAcknowledged> msg( pNetPacket );
CSteamID steamID;
CTFPlayer *pFoundPlayer = NULL;
uint32 unAccountID = msg.Body().account_id();
for ( int i = 1; i <= gpGlobals->maxClients; i++ )
{
CTFPlayer *pPlayer = ToTFPlayer( UTIL_PlayerByIndex( i ) );
if ( pPlayer == NULL )
continue;
if ( pPlayer->GetSteamID( &steamID ) == false )
continue;
if ( steamID.GetAccountID() == unAccountID )
{
pFoundPlayer = pPlayer;
break;
}
}
if ( pFoundPlayer )
{
// Crack open some attributes
IGameEvent *event = gameeventmanager->CreateEvent( "item_found" );
if ( event )
{
event->SetInt( "player", pFoundPlayer->entindex() );
event->SetInt( "quality", msg.Body().quality() );
event->SetInt( "method", GetUnacknowledgedReason( msg.Body().inventory() ) - 1 );
event->SetInt( "itemdef", msg.Body().def_index() );
event->SetInt( "isstrange", msg.Body().is_strange() );
event->SetInt( "isunusual", msg.Body().is_unusual() );
event->SetFloat( "wear", msg.Body().wear() );
gameeventmanager->FireEvent( event );
}
}
return true;
}
};
GC_REG_JOB( GCSDK::CGCClient, CGCTFItemAcknowledged, "CGCTFItemAcknowledged", k_EMsgGCItemAcknowledged, GCSDK::k_EServerTypeGCClient );