469 lines
16 KiB
C++
469 lines
16 KiB
C++
//===== Copyright © 1996-2009, Valve Corporation, All rights reserved. ======//
|
|
//
|
|
// Purpose:
|
|
//
|
|
//===========================================================================//
|
|
|
|
#include "mm_framework.h"
|
|
#include "filesystem.h"
|
|
|
|
// memdbgon must be the last include file in a .cpp file!!!
|
|
#include "tier0/memdbgon.h"
|
|
|
|
#ifdef _X360
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: Adjust our rate based on our quality of service
|
|
//-----------------------------------------------------------------------------
|
|
static ConVar mm_clientrateupdate_enabled( "mm_clientrateupdate_enabled", "1", 0, "Automatically update the client rate based on Xbox LIVE QoS" );
|
|
static ConVar mm_clientrateupdate_adjust( "mm_clientrateupdate_adjust", "0.6", 0, "Downstream rate adjustment" );
|
|
static ConVar mm_clientrateupdate_minimum( "mm_clientrateupdate_minimum", "20000", 0, "Minimum supported rate, Xbox TCR requires 40kbps" );
|
|
static ConVar mm_clientrateupdate_maximum( "mm_clientrateupdate_maximum", "30000", 0, "Maximum supported rate" );
|
|
static ConVar mm_clientrateupdate_qos_timeout( "mm_clientrateupdate_qos_timeout", "20", 0, "How long to wait for QOS to be determined" );
|
|
|
|
static void AdjustClientRateBasedOnQoS( DWORD dwDnBitsPerSec )
|
|
{
|
|
if ( !mm_clientrateupdate_enabled.GetBool() )
|
|
return;
|
|
|
|
static ConVarRef cl_rate( "rate" );
|
|
|
|
int desiredRate = (int)( ( dwDnBitsPerSec / 8.0f ) * mm_clientrateupdate_adjust.GetFloat() );
|
|
|
|
desiredRate = clamp( desiredRate, mm_clientrateupdate_minimum.GetInt(), mm_clientrateupdate_maximum.GetInt() );
|
|
|
|
// Update the client rate
|
|
ConColorMsg( Color( 255, 0, 255, 255 ), "[QoS] Bandwidth %d bps, Updating client rate to %d\n", dwDnBitsPerSec, desiredRate );
|
|
|
|
cl_rate.SetValue( desiredRate );
|
|
}
|
|
|
|
struct RateAdjustmentAsyncCall
|
|
{
|
|
// X360 peer
|
|
XNADDR apxna;
|
|
XNADDR const *papxna;
|
|
XNKID apxnkid;
|
|
XNKID const *papxnkid;
|
|
XNKEY apxnkey;
|
|
XNKEY const *papxnkey;
|
|
|
|
// XLSP server
|
|
IN_ADDR ina;
|
|
DWORD dwServiceId;
|
|
|
|
// QOS handle
|
|
XNQOS *pQOS;
|
|
|
|
// Time when QOS probe started
|
|
float flTimeStarted;
|
|
}
|
|
*g_pRateAdjustmentAsyncCall = NULL;
|
|
|
|
void MatchSession_RateAdjustmentUpdate_Release()
|
|
{
|
|
if ( !g_pRateAdjustmentAsyncCall )
|
|
return;
|
|
|
|
if ( g_pRateAdjustmentAsyncCall->pQOS )
|
|
g_pMatchExtensions->GetIXOnline()->XNetQosRelease( g_pRateAdjustmentAsyncCall->pQOS );
|
|
|
|
delete g_pRateAdjustmentAsyncCall;
|
|
g_pRateAdjustmentAsyncCall = NULL;
|
|
}
|
|
|
|
// Keeps adjusting client side rate setting based on QOS with server
|
|
void MatchSession_RateAdjustmentUpdate()
|
|
{
|
|
if ( !g_pRateAdjustmentAsyncCall )
|
|
return;
|
|
|
|
if ( g_pRateAdjustmentAsyncCall->pQOS->cxnqosPending &&
|
|
Plat_FloatTime() < g_pRateAdjustmentAsyncCall->flTimeStarted + mm_clientrateupdate_qos_timeout.GetFloat() )
|
|
return;
|
|
|
|
ConColorMsg( Color( 255, 0, 255, 255 ), "[QoS] Rate adjustment query %s\n", g_pRateAdjustmentAsyncCall->pQOS->cxnqosPending ? "timed out" : "completed" );
|
|
|
|
// QOS finished or timed out
|
|
XNQOSINFO &xni = g_pRateAdjustmentAsyncCall->pQOS->axnqosinfo[0];
|
|
AdjustClientRateBasedOnQoS( xni.dwDnBitsPerSec );
|
|
MatchSession_RateAdjustmentUpdate_Release();
|
|
}
|
|
|
|
void MatchSession_RateAdjustmentUpdate_Start( IN_ADDR const &ina )
|
|
{
|
|
MatchSession_RateAdjustmentUpdate_Release();
|
|
|
|
g_pRateAdjustmentAsyncCall = new RateAdjustmentAsyncCall;
|
|
ZeroMemory( g_pRateAdjustmentAsyncCall, sizeof( *g_pRateAdjustmentAsyncCall ) );
|
|
|
|
g_pRateAdjustmentAsyncCall->ina = ina;
|
|
g_pRateAdjustmentAsyncCall->dwServiceId = g_pMatchFramework->GetMatchTitle()->GetTitleServiceID();
|
|
g_pRateAdjustmentAsyncCall->flTimeStarted = Plat_FloatTime();
|
|
|
|
ConColorMsg( Color( 255, 0, 255, 255 ), "[QoS] Rate adjustment query scheduled for XLSP server: %08X\n", ina.s_addr );
|
|
|
|
INT ret = g_pMatchExtensions->GetIXOnline()->XNetQosLookup(
|
|
0, NULL, NULL, NULL,
|
|
1, &g_pRateAdjustmentAsyncCall->ina, &g_pRateAdjustmentAsyncCall->dwServiceId,
|
|
2, 0, 0, NULL, &g_pRateAdjustmentAsyncCall->pQOS );
|
|
if ( ret != ERROR_SUCCESS )
|
|
{
|
|
g_pRateAdjustmentAsyncCall->flTimeStarted = 0.0f;
|
|
}
|
|
}
|
|
|
|
void MatchSession_RateAdjustmentUpdate_Start( XSESSION_INFO const &xsi )
|
|
{
|
|
MatchSession_RateAdjustmentUpdate_Release();
|
|
|
|
g_pRateAdjustmentAsyncCall = new RateAdjustmentAsyncCall;
|
|
ZeroMemory( g_pRateAdjustmentAsyncCall, sizeof( *g_pRateAdjustmentAsyncCall ) );
|
|
|
|
g_pRateAdjustmentAsyncCall->apxna = xsi.hostAddress;
|
|
g_pRateAdjustmentAsyncCall->papxna = &g_pRateAdjustmentAsyncCall->apxna;
|
|
|
|
g_pRateAdjustmentAsyncCall->apxnkid = xsi.sessionID;
|
|
g_pRateAdjustmentAsyncCall->papxnkid = &g_pRateAdjustmentAsyncCall->apxnkid;
|
|
|
|
g_pRateAdjustmentAsyncCall->apxnkey = xsi.keyExchangeKey;
|
|
g_pRateAdjustmentAsyncCall->papxnkey = &g_pRateAdjustmentAsyncCall->apxnkey;
|
|
|
|
g_pRateAdjustmentAsyncCall->flTimeStarted = Plat_FloatTime();
|
|
|
|
ConColorMsg( Color( 255, 0, 255, 255 ), "[QoS] Rate adjustment query scheduled for Xbox 360 peer: %08X/%08X\n", xsi.hostAddress.ina.s_addr, xsi.hostAddress.inaOnline.s_addr );
|
|
|
|
INT ret = g_pMatchExtensions->GetIXOnline()->XNetQosLookup(
|
|
1, &g_pRateAdjustmentAsyncCall->papxna, &g_pRateAdjustmentAsyncCall->papxnkid, &g_pRateAdjustmentAsyncCall->papxnkey,
|
|
0, NULL, NULL,
|
|
2, 0, 0, NULL, &g_pRateAdjustmentAsyncCall->pQOS );
|
|
if ( ret != ERROR_SUCCESS )
|
|
{
|
|
g_pRateAdjustmentAsyncCall->flTimeStarted = 0.0f;
|
|
}
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
void MatchSession_BroadcastSessionSettingsUpdate( KeyValues *pUpdateDeletePackage )
|
|
{
|
|
KeyValues *notify = new KeyValues( "OnMatchSessionUpdate" );
|
|
notify->SetString( "state", "updated" );
|
|
|
|
if ( KeyValues *kvUpdate = pUpdateDeletePackage->FindKey( "update" ) )
|
|
notify->AddSubKey( kvUpdate->MakeCopy() );
|
|
if ( KeyValues *kvDelete = pUpdateDeletePackage->FindKey( "delete" ) )
|
|
notify->AddSubKey( kvDelete->MakeCopy() );
|
|
|
|
g_pMatchEventsSubscription->BroadcastEvent( notify );
|
|
}
|
|
|
|
|
|
ConVar cl_session( "cl_session", "", FCVAR_USERINFO | FCVAR_HIDDEN | FCVAR_SERVER_CAN_EXECUTE | FCVAR_DEVELOPMENTONLY );
|
|
|
|
void MatchSession_PrepareClientForConnect( KeyValues *pSettings, uint64 uiReservationCookieOverride )
|
|
{
|
|
char chSession[64];
|
|
sprintf( chSession, "$%llx", uiReservationCookieOverride ? uiReservationCookieOverride :
|
|
g_pMatchFramework->GetMatchSession()->GetSessionSystemData()->
|
|
GetUint64( "xuidReserve", 0ull ) );
|
|
cl_session.SetValue( chSession );
|
|
|
|
g_pMatchFramework->GetMatchTitle()->PrepareClientForConnect( pSettings );
|
|
}
|
|
|
|
static bool MatchSession_ResolveServerInfo_Helper_DsResult( KeyValues *pSettings, CSysSessionBase *pSysSession,
|
|
MatchSessionServerInfo_t &info, uint uiResolveFlags, uint64 ullCrypt )
|
|
{
|
|
#ifdef _X360
|
|
// On dedicated servers host should have given us an insecure
|
|
// address representing our Title Server
|
|
char const *szInsecureServerAddr = pSettings->GetString( "server/adrInsecure" );
|
|
netadr_t inetInsecure;
|
|
inetInsecure.SetFromString( szInsecureServerAddr );
|
|
|
|
IN_ADDR inaddrInsecure;
|
|
inaddrInsecure.s_addr = inetInsecure.GetIPNetworkByteOrder();
|
|
|
|
if ( ( uiResolveFlags & ( info.RESOLVE_DSRESULT | info.RESOLVE_QOS_RATE_PROBE ) ) == info.RESOLVE_QOS_RATE_PROBE )
|
|
{
|
|
// We are not required to resolve the DSRESULT, just submit the QOS rate probe
|
|
MatchSession_RateAdjustmentUpdate_Start( inaddrInsecure );
|
|
return true;
|
|
}
|
|
|
|
char const *szServerType = pSettings->GetString( "server/server", "listen" );
|
|
if ( !Q_stricmp( szServerType, "listen" ) )
|
|
{
|
|
info.m_dsResult.m_bDedicated = false;
|
|
return true;
|
|
}
|
|
if ( !Q_stricmp( szServerType, "externalpeer" ) )
|
|
{
|
|
info.m_dsResult.m_bDedicated = false;
|
|
return true;
|
|
}
|
|
|
|
Q_strncpy( info.m_dsResult.m_szInsecureSendableServerAddress,
|
|
szInsecureServerAddr,
|
|
ARRAYSIZE( info.m_dsResult.m_szInsecureSendableServerAddress ) );
|
|
|
|
// Map it to a secure address
|
|
IN_ADDR inaddrSecure;
|
|
DWORD ret = ERROR_FUNCTION_FAILED;
|
|
if ( CommandLine()->FindParm( "-xlsp_fake_gateway" ) )
|
|
{
|
|
inaddrSecure = inaddrInsecure;
|
|
ret = ERROR_SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
ret = g_pMatchExtensions->GetIXOnline()->XNetServerToInAddr( inaddrInsecure, g_pMatchFramework->GetMatchTitle()->GetTitleServiceID(), &inaddrSecure );
|
|
}
|
|
if ( ret != ERROR_SUCCESS )
|
|
{
|
|
DevWarning( "Failed to resolve XLSP secure address (code = 0x%08X, insecure = %s/%s)!\n",
|
|
ret, inetInsecure.ToString(), szInsecureServerAddr );
|
|
return false;
|
|
}
|
|
else
|
|
{
|
|
netadr_t inetSecure = inetInsecure;
|
|
inetSecure.SetIP( inaddrSecure.s_addr );
|
|
DevMsg( "Resolved XLSP secure address %s, insecure address was %s.\n",
|
|
inetSecure.ToString(), szInsecureServerAddr );
|
|
|
|
Q_strncpy( info.m_dsResult.m_szConnectionString,
|
|
inetSecure.ToString(),
|
|
ARRAYSIZE( info.m_dsResult.m_szConnectionString ) );
|
|
|
|
info.m_dsResult.m_bDedicated = true;
|
|
|
|
// Start QOS rate calculation for the dedicated XLSP server
|
|
MatchSession_RateAdjustmentUpdate_Start( inaddrInsecure );
|
|
}
|
|
#elif !defined( NO_STEAM )
|
|
|
|
char const *szAddress = pSettings->GetString( "server/adronline", "0.0.0.0" );
|
|
if ( char const *szDecrypted = MatchSession_DecryptAddressString( szAddress, ullCrypt ) )
|
|
szAddress = szDecrypted;
|
|
Q_strncpy( info.m_dsResult.m_szPublicConnectionString, szAddress,
|
|
ARRAYSIZE( info.m_dsResult.m_szPublicConnectionString ) );
|
|
|
|
szAddress = pSettings->GetString( "server/adrlocal", "0.0.0.0" );
|
|
if ( char const *szDecrypted = MatchSession_DecryptAddressString( szAddress, ullCrypt ) )
|
|
szAddress = szDecrypted;
|
|
Q_strncpy( info.m_dsResult.m_szPrivateConnectionString, szAddress,
|
|
ARRAYSIZE( info.m_dsResult.m_szPrivateConnectionString ) );
|
|
#endif
|
|
|
|
return true;
|
|
}
|
|
|
|
static bool MatchSession_ResolveServerInfo_Helper_ConnectString( KeyValues *pSettings, CSysSessionBase *pSysSession, MatchSessionServerInfo_t &info, uint uiResolveFlags )
|
|
{
|
|
//
|
|
// Prepare the connect command
|
|
//
|
|
#ifdef _X360
|
|
char const *szServerType = pSettings->GetString( "server/server", "listen" );
|
|
if ( !Q_stricmp( "externalpeer", szServerType ) && !( uiResolveFlags & info.RESOLVE_ALLOW_EXTPEER ) )
|
|
pSysSession = NULL;
|
|
|
|
char const *szConnectionString = info.m_dsResult.m_szConnectionString;
|
|
if ( info.m_dsResult.m_bDedicated )
|
|
{
|
|
info.m_szSecureServerAddress = info.m_dsResult.m_szConnectionString;
|
|
}
|
|
else if ( CSysSessionClient *pSysSessionClient = dynamic_cast< CSysSessionClient * >( pSysSession ) )
|
|
{
|
|
XSESSION_INFO xsi = {0};
|
|
szConnectionString = pSysSessionClient->GetHostNetworkAddress( xsi );
|
|
if ( !szConnectionString )
|
|
{
|
|
DevWarning( "MatchSession_ResolveServerInfo_Helper_ConnectString::GetHostNetworkAddress failed!\n" );
|
|
return false;
|
|
}
|
|
|
|
// Start QOS rate calculation for our session host X360 xnaddr
|
|
MatchSession_RateAdjustmentUpdate_Start( xsi );
|
|
}
|
|
else if ( char const *szSessionInfo = pSettings->GetString( "server/sessioninfo", NULL ) )
|
|
{
|
|
// We don't have a dedicated server and don't allow to use external peer directly,
|
|
// register security keys
|
|
XSESSION_INFO xsi = {0};
|
|
MMX360_SessionInfoFromString( xsi, szSessionInfo );
|
|
|
|
// Resolve XNADDR
|
|
IN_ADDR inaddrRemote;
|
|
g_pMatchExtensions->GetIXOnline()->XNetRegisterKey( &xsi.sessionID, &xsi.keyExchangeKey );
|
|
if ( int err = g_pMatchExtensions->GetIXOnline()->XNetXnAddrToInAddr( &xsi.hostAddress, &xsi.sessionID, &inaddrRemote ) )
|
|
{
|
|
DevWarning( "MatchSession_ResolveServerInfo_Helper_ConnectString::XNetXnAddrToInAddr"
|
|
" failed to resolve XNADDR ( code 0x%08X, sessioninfo = %s )\n",
|
|
err, szSessionInfo );
|
|
|
|
g_pMatchExtensions->GetIXOnline()->XNetUnregisterKey( &xsi.sessionID );
|
|
return false;
|
|
}
|
|
|
|
// Initiate secure connection and key exchange
|
|
if ( int err = g_pMatchExtensions->GetIXOnline()->XNetConnect( inaddrRemote ) )
|
|
{
|
|
DevWarning( "MatchSession_ResolveServerInfo_Helper_ConnectString::XNetConnect"
|
|
" failed to start key exchange ( code 0x%08X, sessioninfo = %s )\n",
|
|
err, szSessionInfo );
|
|
|
|
// Secure IN_ADDR associations are removed implicitly when their key gets unregistered
|
|
g_pMatchExtensions->GetIXOnline()->XNetUnregisterKey( &xsi.sessionID );
|
|
return false;
|
|
}
|
|
|
|
//
|
|
// Prepare connection string
|
|
//
|
|
netadr_t inetAddr;
|
|
inetAddr.SetType( NA_IP );
|
|
inetAddr.SetIPAndPort( inaddrRemote.s_addr, 0 );
|
|
|
|
// Now we know the address for the game to connect
|
|
Q_strncpy( info.m_dsResult.m_szConnectionString, inetAddr.ToString( true ), ARRAYSIZE( info.m_dsResult.m_szConnectionString ) );
|
|
|
|
//
|
|
// Remember all the settings needed to deallocate the secure association
|
|
//
|
|
info.m_szSecureServerAddress = info.m_dsResult.m_szInsecureSendableServerAddress;
|
|
Q_snprintf( info.m_dsResult.m_szInsecureSendableServerAddress,
|
|
ARRAYSIZE( info.m_dsResult.m_szInsecureSendableServerAddress ),
|
|
"SESSIONINFO %s", szSessionInfo );
|
|
|
|
// Start QOS rate calculation for opponents session host X360 remote xnaddr
|
|
MatchSession_RateAdjustmentUpdate_Start( xsi );
|
|
}
|
|
else
|
|
return false;
|
|
|
|
Q_snprintf( info.m_szConnectCmd, sizeof( info.m_szConnectCmd ),
|
|
"connect_splitscreen %s %s %d\n",
|
|
szConnectionString,
|
|
szConnectionString,
|
|
XBX_GetNumGameUsers() );
|
|
#elif !defined( NO_STEAM )
|
|
Q_snprintf( info.m_szConnectCmd, sizeof( info.m_szConnectCmd ),
|
|
"connect %s %s\n",
|
|
info.m_dsResult.m_szPublicConnectionString,
|
|
info.m_dsResult.m_szPrivateConnectionString );
|
|
#endif
|
|
|
|
info.m_xuidJingle = pSettings->GetUint64( "server/xuid", 0ull );
|
|
|
|
if ( uint64 uiReservationCookieOverride = pSettings->GetUint64( "server/reservationid", 0ull ) )
|
|
info.m_uiReservationCookie = uiReservationCookieOverride;
|
|
else if ( pSysSession )
|
|
info.m_uiReservationCookie = pSysSession->GetReservationCookie();
|
|
else
|
|
info.m_uiReservationCookie = 0ull;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool MatchSession_ResolveServerInfo( KeyValues *pSettings, CSysSessionBase *pSysSession, MatchSessionServerInfo_t &info, uint uiResolveFlags, uint64 ullCrypt )
|
|
{
|
|
if ( ( uiResolveFlags & ( info.RESOLVE_DSRESULT | info.RESOLVE_QOS_RATE_PROBE ) ) &&
|
|
!MatchSession_ResolveServerInfo_Helper_DsResult( pSettings, pSysSession, info, uiResolveFlags, ullCrypt ) )
|
|
return false;
|
|
|
|
if ( ( uiResolveFlags & info.RESOLVE_CONNECTSTRING ) &&
|
|
!MatchSession_ResolveServerInfo_Helper_ConnectString( pSettings, pSysSession, info, uiResolveFlags ) )
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
ConVar mm_tu_string( "mm_tu_string", "00000000" );
|
|
|
|
uint64 MatchSession_GetMachineFlags()
|
|
{
|
|
uint64 uiFlags = 0;
|
|
if ( IsPS3() )
|
|
uiFlags |= MACHINE_PLATFORM_PS3;
|
|
return uiFlags;
|
|
}
|
|
|
|
char const * MatchSession_GetTuInstalledString()
|
|
{
|
|
return mm_tu_string.GetString();
|
|
}
|
|
|
|
char const * MatchSession_EncryptAddressString( char const *szAddress, uint64 ullCrypt )
|
|
{
|
|
if ( !szAddress || !*szAddress )
|
|
return NULL;
|
|
if ( !ullCrypt )
|
|
return NULL;
|
|
if ( szAddress[0] == ':' )
|
|
return NULL;
|
|
if ( szAddress[ 0 ] == '$' )
|
|
return NULL;
|
|
|
|
static unsigned char s_chData[256];
|
|
int nLen = Q_strlen( szAddress );
|
|
if ( nLen >= ARRAYSIZE( s_chData )/2 - 1 )
|
|
return NULL;
|
|
|
|
// Copy the address
|
|
s_chData[0] = '$';
|
|
for ( int j = 0; j < nLen; ++ j )
|
|
{
|
|
uint8 uiVal = uint8( szAddress[j] ) ^ uint8( reinterpret_cast< uint8 * >(&ullCrypt)[ j % sizeof( uint64 ) ] );
|
|
Q_snprintf( (char*)( s_chData + 1 + 2*j ), 3, "%02X", ( uint32 ) uiVal );
|
|
}
|
|
return (char*) s_chData;
|
|
}
|
|
|
|
char const * MatchSession_DecryptAddressString( char const *szAddress, uint64 ullCrypt )
|
|
{
|
|
if ( !szAddress || !*szAddress )
|
|
return NULL;
|
|
if ( !ullCrypt )
|
|
return NULL;
|
|
if ( szAddress[ 0 ] != '$' )
|
|
return NULL;
|
|
|
|
static unsigned char s_chData[ 256 ];
|
|
int nLen = Q_strlen( szAddress );
|
|
if ( nLen*2 + 2 >= ARRAYSIZE( s_chData ) )
|
|
return NULL;
|
|
|
|
// Copy the address
|
|
for ( int j = 0; j < nLen/2; ++j )
|
|
{
|
|
uint32 uiVal;
|
|
if ( !sscanf( szAddress + 1 + 2*j, "%02X", &uiVal ) )
|
|
return NULL;
|
|
if ( uiVal > 0xFF )
|
|
return NULL;
|
|
uiVal = uint8( uiVal ) ^ uint8( reinterpret_cast< uint8 * >(&ullCrypt)[ j % sizeof( uint64 ) ] );
|
|
if ( !uiVal )
|
|
return NULL;
|
|
s_chData[j] = uiVal;
|
|
}
|
|
s_chData[nLen/2] = 0;
|
|
return (char*) s_chData;
|
|
}
|
|
|
|
|
|
CON_COMMAND( mm_debugprint, "Show debug information about current matchmaking session" )
|
|
{
|
|
if ( IMatchSession *pIMatchSession = g_pMMF->GetMatchSession() )
|
|
{
|
|
( ( IMatchSessionInternal * ) pIMatchSession )->DebugPrint();
|
|
}
|
|
else
|
|
{
|
|
DevMsg( "No match session.\n" );
|
|
}
|
|
}
|