Backups/csgo-2018-source
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
93f753e8ef
csgo-2018-source/engine/hltvbroadcast.cpp

742 lines
26 KiB
C++
Raw Normal View History

uid issue
2021-07-25 12:11:47 +08:00
//========= Copyright <20> Valve Corporation, All rights reserved. ============//
//
// Purpose: Broadcasting HLTV demo in 3-second chunks, preparing for SteamCast
// TODO test cases: start broadcast before relay server is ready; quit the server while broadcast hasn't flushed yet, make sure broadcast is properly stopped, flush is done and sends returned at least one confirm or error before quitting.
//
//=============================================================================//
#include <tier1/strtools.h>
#include <eiface.h>
#include <bitbuf.h>
#include <time.h>
#include "hltvbroadcast.h"
#include "hltvserver.h"
#include "demo.h"
#include "host_cmd.h"
#include "demofile/demoformat.h"
#include "filesystem_engine.h"
#include "net.h"
#include "networkstringtable.h"
#include "dt_common_eng.h"
#include "host.h"
#include "server.h"
#include "sv_steamauth.h"
#ifndef DEDICATED
#include "cl_steamauth.h"
#endif
#include "networkstringtableclient.h"
#include "tier1/fmtstr.h"
// memdbgon must be the last include file in a .cpp file!!!
#include "tier0/memdbgon.h"
extern CNetworkStringTableContainer *networkStringTableContainerServer;
static ISteamHTTP *s_pSteamHTTP = NULL;
ConVar tv_broadcast_keyframe_interval( "tv_broadcast_keyframe_interval", "3", FCVAR_RELEASE, "The frequency, in seconds, of sending keyframes and delta fragments to the broadcast relay server" );
ConVar tv_broadcast_startup_resend_interval( "tv_broadcast_startup_resend_interval", "10", FCVAR_RELEASE, "The interval, in seconds, of re-sending startup data to the broadcast relay server (useful in case relay crashes, restarts or startup data http request fails)" );
ConVar tv_broadcast_max_requests( "tv_broadcast_max_requests", "20", FCVAR_RELEASE, "Max number of broadcast http requests in flight. If there is a network issue, the requests may start piling up, degrading server performance. If more than the specified number of requests are in flight, the new requests are dropped." );
ConVar tv_broadcast_drop_fragments( "tv_broadcast_drop_fragments", "0", FCVAR_RELEASE | FCVAR_HIDDEN, "Drop every Nth fragment" );
ConVar tv_broadcast_terminate( "tv_broadcast_terminate", "1", FCVAR_RELEASE | FCVAR_HIDDEN, "Terminate every broadcast with a stop command" );
ConVar tv_broadcast_origin_auth( "tv_broadcast_origin_auth", "gocastauth" /*use something secure, like hMugYm7Lv4o5*/, FCVAR_RELEASE | FCVAR_HIDDEN, "X-Origin-Auth header of the broadcast POSTs" );
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
CHLTVBroadcast::CHLTVBroadcast(CHLTVServer *pHltvServer) :m_pHltvServer( pHltvServer )
{
m_flTimeout = 30;
m_pFile = NULL;
m_bIsRecording = false;
m_nHttpRequestBacklogHighWatermark = 0;
m_nMatchFragmentCounter = 0;
m_flBroadcastKeyframeInterval = tv_broadcast_keyframe_interval.GetFloat();
}
CHLTVBroadcast::CMemoryStream::CMemoryStream() : m_Buffer( NET_MAX_PAYLOAD, NET_MAX_PAYLOAD )
{
m_nCommitted = 0;
m_pReserved = NULL;
}
CHLTVBroadcast::~CHLTVBroadcast()
{
StopRecording();
}
void CHLTVBroadcast::OnMasterStarted()
{
// we need some kind of unique id per match, when we can't have MatchID and use SteamID. This is it - it doesn't change if you restart broadcast during a match, but it changes every time a master proxy starts, which signifies a (re)start of tv server, which will always happen at the beginning of a match.
m_nMasterCookie = Plat_GetTime();
m_nMatchFragmentCounter = 0; // new match, start from the new match fragment counter
}
void CHLTVBroadcast::StartRecording( const char *pBroadcastUrl )
{
StopRecording(); // stop if we're already recording
#ifndef DEDICATED
s_pSteamHTTP = Steam3Client().SteamHTTP();
#endif
if ( !s_pSteamHTTP )
s_pSteamHTTP = Steam3Server().SteamHTTP();
ConMsg( "Recording Broadcast\n" );
m_flBroadcastKeyframeInterval = tv_broadcast_keyframe_interval.GetFloat();
m_nLastWrittenTick = 0;
m_nFrameCount = 0;
m_nKeyframeTick = 0; // when broadcasting starts, we immediately send a keyframe
//Tag it as -1 so we can really set the start tick when the first frame is written
m_nStartTick = -1;
m_nCurrentTick = 0;
m_nSignonDataAckTick = -1;
m_nSignonDataFragment = -1;
m_bIsRecording = true;
m_nDeltaTick = -1;
m_mpKeyframe.Reset();
m_mpLowLevelSend.Reset();
m_nMaxKeyframeTicks = 0;
m_nMaxLowLevelSendTicks = 0;
m_nDecayMaxKeyframeTicks = 0;
m_nKeyframeBytes = m_nDeltaFrameBytes = 0;
m_nFailedHttpRequests = 0;
m_mpFrame.Reset();
// extern ConVar sv_mmqueue_reservation;
if ( uint64 nMatchId = sv.GetMatchId() )
{
int nHltvInstance = m_pHltvServer->GetInstanceIndex();
m_Url.Format( "%s/%llui%d", pBroadcastUrl, nMatchId, nHltvInstance );
}
else
{
uint64 nSteamId = Steam3Server().GetGSSteamID().ConvertToUint64();
m_Url.Format( "%s/s%llut%llu", pBroadcastUrl, nSteamId, m_nMasterCookie );
}
}
bool CHLTVBroadcast::IsRecording()
{
return m_bIsRecording;
}
void CHLTVBroadcast::StopRecording( )
{
if ( !m_bIsRecording )
return;
DevMsg( "Stop Broadcast @frag %d\n", m_nMatchFragmentCounter );
if ( tv_broadcast_terminate.GetBool() )
m_DeltaStream.WriteCmdHeader( dem_stop, GetRecordingTick(), 0 );
FlushCollectedStreams( "&final" );
m_nMatchFragmentCounter += 2; // we need to create a hole in the broadcast in case we change our mind and start broadcast again some moments (or hours) later. This will force all clients to re-sync to the new keyframe
m_DeltaStream.Purge();
m_SignonDataStream.Reset();
m_SignonDataStream.Purge();
m_nStartTick = -1;
m_nSignonDataAckTick = -1;
for ( int i = 0; i < m_HttpRequests.Count(); ++i )
m_HttpRequests[ i ]->DetachFromParent(); // TODO: if we're quitting, make sure that all the http requests finished before destroying the broadcast object
m_HttpRequests.Purge(); // the elements will delete themselves
// Note: some data fragments will still be in flight until they finish. We may start and stop another recording, and that's fine, SteamHTTP will continue uploading those data chunks until done or failed.
m_bIsRecording = false;
}
void CHLTVBroadcast::WriteServerInfo( CMemoryStream &stream )
{
stream.WriteCmdHeader( dem_signon, GetRecordingTick(), 0 );
CInStreamMsgWithSize msg( stream, "CHLTVBroadcast::WriteServerInfo" );
// on the master demos are using sv object, on relays hltv
CBaseServer *pServer = m_pHltvServer->IsMasterProxy()?(CBaseServer*)(&sv):(CBaseServer*)(m_pHltvServer);
CSVCMsg_ServerInfo_t serverinfo;
m_pHltvServer->FillServerInfo( serverinfo ); // fill rest of info message
serverinfo.WriteToBuffer( msg );
// send first tick
CNETMsg_Tick_t signonTick( m_nSignonTick, 0, 0, 0 );
signonTick.WriteToBuffer( msg );
// Write replicated ConVars to non-listen server clients only
CNETMsg_SetConVar_t convars;
// build a list of all replicated convars
Host_BuildConVarUpdateMessage( convars.mutable_convars(), FCVAR_REPLICATED, true );
if ( m_pHltvServer->IsMasterProxy() )
{
// for SourceTV server demos write set "tv_transmitall 1" even
// if it's off for the real broadcast
convars.AddToTail( "tv_transmitall", "1" );
m_pHltvServer->FixupConvars( convars );
}
// write convars to demo
convars.WriteToBuffer( msg );
// write stringtable baselines
#ifndef SHARED_NET_STRING_TABLES
if ( m_pHltvServer->m_StringTables )
m_pHltvServer->m_StringTables->WriteBaselines( pServer->GetMapName(), msg );
#endif
// send signon state
CNETMsg_SignonState_t signonMsg( SIGNONSTATE_NEW, pServer->GetSpawnCount() );
signonMsg.WriteToBuffer( msg );
}
void CHLTVBroadcast::RecordCommand( const char *cmdstring )
{
if ( !IsRecording() )
return;
if ( !cmdstring || !cmdstring[0] )
return;
m_DeltaStream.WriteCmdHeader( dem_consolecmd, GetRecordingTick(), 0 );
CInStreamMsg msg( m_DeltaStream, "RecordCommand", 1024 );
CSVCMsg_Broadcast_Command_t cmd;
cmd.set_cmd( cmdstring );
}
void CHLTVBroadcast::RecordServerClasses( CMemoryStream &stream, ServerClass *pClasses )
{
stream.WriteCmdHeader( dem_datatables, GetRecordingTick(), 0 );
CInStreamMsg buf( stream, "CHLTVBroadcast::RecordServerClasses", DEMO_RECORD_BUFFER_SIZE );
// Send SendTable info.
DataTable_WriteSendTablesBuffer( pClasses, &buf );
// Send class descriptions.
DataTable_WriteClassInfosBuffer( pClasses, &buf );
if ( buf.GetNumBitsLeft() <= 0 )
{
Sys_Error( "unable to record server classes\n" );
}
// this was a dem_datatables in the demo file
}
void CHLTVBroadcast::RecordStringTables( CMemoryStream &stream )
{
stream.WriteCmdHeader( dem_stringtables, GetRecordingTick(), 0 );
CInStreamMsg buf( stream, "CHLTVBroadcast::RecordStringTables", DEMO_RECORD_BUFFER_SIZE );
networkStringTableContainerServer->WriteStringTables( buf );
}
void CHLTVBroadcast::WriteSignonData()
{
// on the master demos are using sv object, on relays hltv
CBaseServer *pServer = m_pHltvServer->IsMasterProxy()?(CBaseServer*)(&sv):(CBaseServer*)(m_pHltvServer);
m_nSignonTick = pServer->m_nTickCount;
m_SignonDataStream.Purge();
m_nSignonDataFragment = m_nMatchFragmentCounter;
WriteServerInfo( m_SignonDataStream );
RecordServerClasses( m_SignonDataStream, serverGameDLL->GetAllServerClasses() );
RecordStringTables( m_SignonDataStream );
{
m_SignonDataStream.WriteCmdHeader( dem_signon, GetRecordingTick(), 0 );
CInStreamMsgWithSize msg( m_SignonDataStream, "CHLTVBroadcast::WriteSignonData" );
// use your class infos, CRC is correct
// use your class infos, CRC is correct
CSVCMsg_ClassInfo_t classmsg;
classmsg.set_create_on_client( true );
classmsg.WriteToBuffer( msg );
// Write the regular signon now
msg.WriteBits( m_pHltvServer->m_Signon.GetData(), m_pHltvServer->m_Signon.GetNumBitsWritten() );
// write new state
CNETMsg_SignonState_t signonMsg1( SIGNONSTATE_PRESPAWN, pServer->GetSpawnCount() );
signonMsg1.WriteToBuffer( msg );
}
{
// Dump all accumulated avatar data messages into signon portion of the demo file
FOR_EACH_MAP_FAST( m_pHltvServer->m_mapPlayerAvatarData, iData )
{
m_SignonDataStream.WriteCmdHeader( dem_signon, GetRecordingTick(), 0 );
CInStreamMsgWithSize msg( m_SignonDataStream, "CHLTVBroadcast::WriteSignonData(avatar data)" );
CNETMsg_PlayerAvatarData_t &msgPlayerAvatarData = *m_pHltvServer->m_mapPlayerAvatarData.Element( iData );
msgPlayerAvatarData.WriteToBuffer( msg );
}
// For official tournament servers also dump all the avatars of possible players into signon portion
extern ConVar sv_mmqueue_reservation;
extern ConVar sv_reliableavatardata;
if ( ( sv_reliableavatardata.GetInt() == 2 ) && ( sv_mmqueue_reservation.GetString()[ 0 ] == 'Q' ) )
{
// CSteamID steamIdGs( SteamGameServer() ? SteamGameServer()->GetSteamID() : CSteamID() );
CSteamID steamIdGs( Steam3Server().GetGSSteamID() );
if ( !steamIdGs.IsValid() ) steamIdGs.SetEUniverse( k_EUniversePublic );
CSteamID steamIdUser( 1, steamIdGs.GetEUniverse(), k_EAccountTypeIndividual );
CUtlVector< AccountID_t > arrAvatarsToAddFromDisk;
for ( char const *pszPrev = sv_mmqueue_reservation.GetString(), *pszNext = pszPrev;
( pszNext = strchr( pszPrev, '[' ) ) != NULL; ( pszPrev = pszNext + 1 ) )
{
uint32 uiAccountId = 0;
sscanf( pszNext, "[%x]", &uiAccountId );
if ( uiAccountId && // valid account and not yet sent in previous loop
( m_pHltvServer->m_mapPlayerAvatarData.Find( uiAccountId ) == m_pHltvServer->m_mapPlayerAvatarData.InvalidIndex() ) &&
( arrAvatarsToAddFromDisk.Find( uiAccountId ) == arrAvatarsToAddFromDisk.InvalidIndex() ) )
{
arrAvatarsToAddFromDisk.AddToTail( uiAccountId );
}
}
for ( char const *pszPrev = sv_mmqueue_reservation.GetString(), *pszNext = pszPrev;
( pszNext = strchr( pszPrev, '{' ) ) != NULL; ( pszPrev = pszNext + 1 ) )
{
uint32 uiAccountId = 0;
sscanf( pszNext, "{%x}", &uiAccountId );
if ( uiAccountId && // valid account and not yet sent in previous loop
( m_pHltvServer->m_mapPlayerAvatarData.Find( uiAccountId ) == m_pHltvServer->m_mapPlayerAvatarData.InvalidIndex() ) &&
( arrAvatarsToAddFromDisk.Find( uiAccountId ) == arrAvatarsToAddFromDisk.InvalidIndex() ) )
{
arrAvatarsToAddFromDisk.AddToTail( uiAccountId );
}
}
FOR_EACH_VEC( arrAvatarsToAddFromDisk, i )
{
steamIdUser.SetAccountID( arrAvatarsToAddFromDisk[i] );
//
// Try to load the avatar data for this player
//
CUtlBuffer bufAvatarData;
CUtlBuffer bufAvatarDataDefault;
CUtlBuffer *pbufUseRgb = NULL;
if ( !pbufUseRgb &&
g_pFullFileSystem->ReadFile( CFmtStr( "avatars/%llu.rgb", steamIdUser.ConvertToUint64() ), "MOD", bufAvatarData ) &&
( bufAvatarData.TellPut() == 64 * 64 * 3 ) )
pbufUseRgb = &bufAvatarData;
if ( !pbufUseRgb &&
g_pFullFileSystem->ReadFile( "avatars/default.rgb", "MOD", bufAvatarDataDefault ) &&
( bufAvatarDataDefault.TellPut() == 64 * 64 * 3 ) )
pbufUseRgb = &bufAvatarDataDefault;
if ( pbufUseRgb )
{
CNETMsg_PlayerAvatarData_t msgPlayerAvatarData;
msgPlayerAvatarData.set_rgb( pbufUseRgb->Base(), pbufUseRgb->TellPut() );
msgPlayerAvatarData.set_accountid( steamIdUser.GetAccountID() );
m_SignonDataStream.WriteCmdHeader( dem_signon, GetRecordingTick(), 0 );
CInStreamMsgWithSize msg( m_SignonDataStream, "CHLTVBroadcast::WriteSignonData(avatar data)" );
msgPlayerAvatarData.WriteToBuffer( msg );
}
}
}
}
{
m_SignonDataStream.WriteCmdHeader( dem_signon, GetRecordingTick(), 0 );
CInStreamMsgWithSize msg( m_SignonDataStream, "CHLTVBroadcast::WriteSignonData(tail)" );
// set view entity
CSVCMsg_SetView_t viewent;
viewent.set_entity_index( m_pHltvServer->m_nViewEntity );
viewent.WriteToBuffer( msg );
// Spawned into server, not fully active, though
CNETMsg_SignonState_t signonMsg2( SIGNONSTATE_SPAWN, pServer->GetSpawnCount() );
signonMsg2.WriteToBuffer( msg );
}
}
void CHLTVBroadcast::SendSignonData()
{
Assert( !m_SignonDataStream.IsEmpty() );
Send( CFmtStr( "/%d/start?tick=%d&tps=%.1f&map=%s&protocol=%d", m_nSignonDataFragment, m_nStartTick, 1.0f / sv.GetTickInterval(), sv.GetMapName(), DEMO_PROTOCOL ), m_SignonDataStream );
m_nSignonDataAckTick = m_nCurrentTick;
}
void CHLTVBroadcast::OnHttpRequestFailed()
{
m_nFailedHttpRequests++;
}
void CHLTVBroadcast::OnHttpRequestResetContent()
{
// may need to re-send the startup data - the http response says that the server hasn't received startup data yet at the time it was responding to the request
if ( ( m_nCurrentTick - m_nSignonDataAckTick ) * sv.GetTickInterval() > tv_broadcast_startup_resend_interval.GetFloat() )
{
Msg( "Broadcast[%d] Re-sending signon data\n", m_pHltvServer->GetInstanceIndex() );
// we haven't seen an OK response (meaning the startup tick is recognized by the server) in . Try to re-send
SendSignonData();
}
}
void CHLTVBroadcast::OnHttpRequestSuccess()
{
m_nSignonDataAckTick = m_nCurrentTick;
}
void CHLTVBroadcast::Register( CHttpCallback *pCallback )
{
m_HttpRequests.AddToTail( pCallback );
m_nHttpRequestBacklogHighWatermark = Max( m_HttpRequests.Count(), m_nHttpRequestBacklogHighWatermark );
}
void CHLTVBroadcast::Unregister( CHttpCallback *pCallback )
{
m_HttpRequests.FindAndFastRemove( pCallback );
}
CON_COMMAND( tv_broadcast_resend, "resend broadcast data to broadcast relay" )
{
for ( CActiveHltvServerIterator hltv; hltv; hltv.Next() )
hltv->m_Broadcast.ResendStartup();
}
void CHLTVBroadcast::ResendStartup()
{
if ( m_nStartTick != -1 )
{
// the signon data has been sent
SendSignonData();
}
}
void CHLTVBroadcast::WriteFrame( CHLTVFrame *pFrame, bf_write *additionaldata )
{
Assert( m_pHltvServer->IsMasterProxy() ); // this works only on the master since we use sv.
m_nCurrentTick = pFrame->tick_count;
bool bKeyFrame = ( m_nCurrentTick - m_nKeyframeTick ) * sv.GetTickInterval() >= m_flBroadcastKeyframeInterval;
if ( m_nStartTick == -1 )
{
m_nStartTick = pFrame->tick_count;
WriteSignonData();
SendSignonData();
bKeyFrame = true;
}
else
{
m_DeltaStream.WriteCmdHeader( dem_packet, GetRecordingTick(), 0 );
CInStreamMsgWithSize msg( m_DeltaStream, "CHLTVBroadcast:Frame(delta)", NET_MAX_PAYLOAD );
MICRO_PROFILE( m_mpFrame );
RecordSnapshot( pFrame, additionaldata, msg, m_nDeltaTick );
// update delta tick just like fakeclients do
m_nDeltaTick = pFrame->tick_count;
m_nDeltaFrameBytes += msg.GetNumBytesWritten();
}
m_nLastWrittenTick = pFrame->tick_count;
m_nFrameCount++;
if ( bKeyFrame )
{
if ( m_HttpRequests.Count() > tv_broadcast_max_requests.GetInt() )
{
int nFragment = ++m_nMatchFragmentCounter;
Warning( "Broadcast backlog of http requests in flight is too high (%d > %d), dropping %d/full and %d/delta.\n",
m_HttpRequests.Count(), tv_broadcast_max_requests.GetInt(), m_nMatchFragmentCounter, nFragment );
}
else
{
// upload the delta frames (IMPORTANT: including the current frame that we'll re-record the keyframe for) and the TOC (the last entry of TOC becomes useful only after we upload the current delta frames)
// So, each 3-second fragment will full and delta frames has the full frame of the tick preceding the first delta frame tick
FlushCollectedStreams();
// it's been 3 seconds already, let's re-send the keyframe - starting a new fragment with a full frame update
m_nKeyframeTick = m_nCurrentTick;
int nFragment = ++m_nMatchFragmentCounter;
Assert( m_DeltaStream.IsEmpty() );
{
m_DeltaStream.WriteCmdHeader( dem_packet, GetRecordingTick(), 0 );
CInStreamMsgWithSize msg( m_DeltaStream, "CHLTVBroadcast:FullFrame", NET_MAX_PAYLOAD );
CMicroProfilerSample sample;
RecordSnapshot( pFrame, additionaldata, msg, -1 );
int64 nElapsedTicks = sample.GetElapsed();
m_mpKeyframe.Add( nElapsedTicks );
m_nMaxKeyframeTicks = Max( m_nMaxKeyframeTicks, nElapsedTicks );
m_nDecayMaxKeyframeTicks = Max( m_nDecayMaxKeyframeTicks * 933 / 1000, nElapsedTicks ); // 0.933 ^ 20 = .25 , this will decay 1/4 every minute
}
m_nKeyframeBytes += m_DeltaStream.GetCommitSize();
Send( CFmtStr( "/%d/full?tick=%d", nFragment, m_nCurrentTick ), m_DeltaStream );
}
m_DeltaStream.Reset();
}
}
void CHLTVBroadcast::DumpStats()
{
uint32 numKeyframes = m_nMatchFragmentCounter;
int64 nKeyframeBytes = m_nKeyframeBytes, nDeltaFrameBytes = m_nDeltaFrameBytes;
if ( numKeyframes > 1 )
{
// I'm missing the last keyframe, its delta portion is truncated
nKeyframeBytes /= numKeyframes ;
nDeltaFrameBytes /= numKeyframes ;
}
if ( numKeyframes )
{
Msg( "signup frag %d (%u bytes), frag counter %d.\nDelta frames: %s Bps, %.3f ms/frame.\nAvg Keyframe: %s Bytes, %.3f ms (%.1f max, %.1f recent max)\nHttp: %d failed, %d/%d max in flight\n",
m_nSignonDataFragment, m_SignonDataStream.GetCommitSize(),
m_nMatchFragmentCounter,
V_pretifynum( int( nDeltaFrameBytes / m_flBroadcastKeyframeInterval ) ),
m_mpFrame.GetAverageMilliseconds(),
V_pretifynum( nKeyframeBytes ),
m_mpKeyframe.GetAverageMilliseconds(),
CMicroProfiler::TimeBaseTicksToMilliseconds( m_nMaxKeyframeTicks ),
CMicroProfiler::TimeBaseTicksToMilliseconds( m_nDecayMaxKeyframeTicks ),
m_nFailedHttpRequests, m_HttpRequests.Count(), m_nHttpRequestBacklogHighWatermark
);
Msg( "http Send %.3f ms ave, %.3f ms max\n", m_mpLowLevelSend.GetAverageMilliseconds(), CMicroProfiler::TimeBaseTicksToMilliseconds( m_nMaxLowLevelSendTicks ) );
}
else
{
Msg( "no keyframes written\n" );
}
}
void CHLTVBroadcast::RecordSnapshot( CHLTVFrame * pFrame, bf_write * additionaldata, bf_write &msg, int nDeltaTick )
{
//first write reliable data
bf_write *data = &pFrame->m_Messages[ HLTV_BUFFER_RELIABLE ];
if ( data->GetNumBitsWritten() )
msg.WriteBits( data->GetBasePointer(), data->GetNumBitsWritten() );
//now send snapshot data
if ( host_frameendtime_computationduration > .1f )
DevMsg( "CHLTVBroadcast::RecordSnapshot tick %.1f ms\n", host_frameendtime_computationduration * 1000 );
// send tick time
CNETMsg_Tick_t tickmsg( pFrame->tick_count, host_frameendtime_computationduration, host_frametime_stddeviation, host_framestarttime_stddeviation );
tickmsg.WriteToBuffer( msg );
#ifndef SHARED_NET_STRING_TABLES
// Update shared client/server string tables. Must be done before sending entities
m_pHltvServer->m_StringTables->WriteUpdateMessage( NULL, MAX( m_nSignonTick, nDeltaTick ), msg );
#endif
// get delta frame
CClientFrame *deltaFrame = m_pHltvServer->GetClientFrame( nDeltaTick ); // NULL if delta_tick is not found or -1
// send entity update, delta compressed if deltaFrame != NULL
CSVCMsg_PacketEntities_t packetmsg;
sv.WriteDeltaEntities( m_pHltvServer->m_MasterClient, pFrame, deltaFrame, packetmsg );
packetmsg.WriteToBuffer( msg );
// send all unreliable temp ents between last and current frame
CSVCMsg_TempEntities_t tempentsmsg;
CFrameSnapshot * fromSnapshot = deltaFrame ? deltaFrame->GetSnapshot() : NULL;
sv.WriteTempEntities( m_pHltvServer->m_MasterClient, pFrame->GetSnapshot(), fromSnapshot, tempentsmsg, 255 );
if ( tempentsmsg.num_entries() )
{
tempentsmsg.WriteToBuffer( msg );
}
// write sound data
data = &pFrame->m_Messages[ HLTV_BUFFER_SOUNDS ];
if ( data->GetNumBitsWritten() )
msg.WriteBits( data->GetBasePointer(), data->GetNumBitsWritten() );
// write voice data
data = &pFrame->m_Messages[ HLTV_BUFFER_VOICE ];
if ( data->GetNumBitsWritten() )
msg.WriteBits( data->GetBasePointer(), data->GetNumBitsWritten() );
// last write unreliable data
data = &pFrame->m_Messages[ HLTV_BUFFER_UNRELIABLE ];
if ( data->GetNumBitsWritten() )
msg.WriteBits( data->GetBasePointer(), data->GetNumBitsWritten() );
if ( additionaldata && additionaldata->GetNumBitsWritten() )
{
msg.WriteBits( additionaldata->GetBasePointer(), additionaldata->GetNumBitsWritten() );
}
}
void * CHLTVBroadcast::CMemoryStream::Reserve( uint nReserveBytes )
{
Assert( !m_pReserved );
m_Buffer.EnsureCapacity( m_nCommitted + nReserveBytes );
m_pReserved = m_Buffer.Base() + m_nCommitted;
return m_pReserved;
}
void CHLTVBroadcast::CMemoryStream::Commit( uint nCommitBytes )
{
Assert( m_pReserved && m_nCommitted + nCommitBytes <= uint( m_Buffer.NumAllocated() ) );
m_nCommitted += nCommitBytes;
m_pReserved = NULL;
}
void CHLTVBroadcast::CMemoryStream::Purge()
{
Assert( !m_nCommitted && !m_pReserved ); // we should not have anything waiting to be committed
m_Buffer.Purge();
m_nCommitted = 0;
m_pReserved = NULL;
}
void CHLTVBroadcast::CMemoryStream::WriteCmdHeader( unsigned char cmd, int tick, int nPlayerSlot )
{
unsigned char *pCmdHeader = ( unsigned char* ) Reserve( 6 );
pCmdHeader[ 0 ] = cmd;
*( int* )( pCmdHeader + 1 ) = tick;
pCmdHeader[ 5 ] = nPlayerSlot;
Commit( 6 );
}
void CHLTVBroadcast::FlushCollectedStreams( const char *pExtraParams )
{
// send the collected delta stream payload
if ( !m_DeltaStream.IsEmpty() )
{
//int nBaseTick = m_Keyframes.IsEmpty() ? -1 : m_Keyframes.Tail().nKeyframeTick;
Send( CFmtStr( "/%d/delta?endtick=%d%s", m_nMatchFragmentCounter, m_nCurrentTick, pExtraParams ), m_DeltaStream );
m_DeltaStream.Reset();
}
}
// protocol is very simple: a=<account/match id> & t= <type, i=initial/startup, k=keyframe/full frame, d=delta frames> &
CHLTVBroadcast::CHttpCallback * CHLTVBroadcast::Send( const char* pPath, CMemoryStream &stream )
{
return Send( pPath, stream.Base(), stream.GetCommitSize() );
}
CHLTVBroadcast::CHttpCallback * CHLTVBroadcast::Send( const char* pPath, const void *pBase, uint nSize )
{
if ( !s_pSteamHTTP )
{
Warning( "HLTV Broadcast cannot send data because steam http is not available. Are you logged into Steam?\n" );
return NULL;
}
if ( tv_broadcast_drop_fragments.GetInt() > 0 && ( rand() % tv_broadcast_drop_fragments.GetInt() ) == 0 )
{
Msg( "Dropping %d bytes to %s%s\n", nSize, GetUrl(), pPath );
return NULL;
}
return LowLevelSend( m_Url + pPath, pBase, nSize );
}
CHLTVBroadcast::CHttpCallback * CHLTVBroadcast::LowLevelSend( const CUtlString &path, const void *pBase, uint nSize )
{
CMicroProfilerGuard mpg( &m_mpLowLevelSend );
HTTPRequestHandle hRequest = s_pSteamHTTP->CreateHTTPRequest( k_EHTTPMethodPOST, path.Get() );
if ( !hRequest )
{
Warning( "Cannot create http put: %s, %u bytes lost\n", path.Get(), nSize );
return NULL;
}
s_pSteamHTTP->SetHTTPRequestNetworkActivityTimeout( hRequest, m_flTimeout );
const char *pOriginAuth = tv_broadcast_origin_auth.GetString();
if ( pOriginAuth && *pOriginAuth )
{
if ( !s_pSteamHTTP->SetHTTPRequestHeaderValue( hRequest, "X-Origin-Auth", tv_broadcast_origin_auth.GetString() ) )
{
Warning( "Cannot set http X-Origin-Auth\n" );
}
}
if ( !s_pSteamHTTP->SetHTTPRequestRawPostBody( hRequest, "application/octet-stream", ( uint8* )pBase, nSize ) )
{
Warning( "Cannot set http post body for %s, %u bytes\n", path.Get(), nSize );
s_pSteamHTTP->ReleaseHTTPRequest( hRequest );
return NULL;
}
SteamAPICall_t hCall;
CHttpCallback *pResult = NULL;
if ( s_pSteamHTTP->SendHTTPRequest( hRequest, &hCall ) && hCall )
{
pResult = new CHttpCallback( this, hRequest, path.Get() );
SteamAPI_RegisterCallResult( pResult, hCall );
}
else
{
s_pSteamHTTP->ReleaseHTTPRequest( hRequest );
}
m_nMaxLowLevelSendTicks = Max( m_nMaxLowLevelSendTicks, mpg.GetElapsed() );
return pResult;
}
CHLTVBroadcast::CHttpCallback::CHttpCallback( CHLTVBroadcast *pParent, HTTPRequestHandle hRequest, const char *pResource ) :
m_pParent( pParent ), m_hRequest( hRequest ), m_Resource( pResource )
{
m_iCallback = HTTPRequestCompleted_t::k_iCallback;
pParent->Register( this );
}
CHLTVBroadcast::CHttpCallback::~CHttpCallback()
{
s_pSteamHTTP->ReleaseHTTPRequest( m_hRequest );
if ( m_pParent )
m_pParent->Unregister( this );
}
void CHLTVBroadcast::CHttpCallback::Run( void *pvParam )
{ // success!
Run( pvParam, false, 0 );
}
void CHLTVBroadcast::CHttpCallback::Run( void *pvParam, bool bIOFailure, SteamAPICall_t hSteamAPICall )
{
EHTTPStatusCode nStatus = ( ( HTTPRequestCompleted_t * )pvParam )->m_eStatusCode;
if ( bIOFailure )
{
Msg( "Broadcast[%d] IO Failure, Http code %d on %s\n", m_pParent ? m_pParent->m_pHltvServer->GetInstanceIndex() : -1, int( nStatus ), m_Resource.Get() );
if ( m_pParent )
m_pParent->OnHttpRequestFailed();
}
else
{
if ( nStatus == k_EHTTPStatusCode205ResetContent )
{
if ( m_pParent )
m_pParent->OnHttpRequestResetContent();
}
else
{
if ( nStatus != k_EHTTPStatusCode200OK ) // we should always get a 200
{
Msg( "Broadcast[%d] Relay returned Http code %d on %s\n", m_pParent ? m_pParent->m_pHltvServer->GetInstanceIndex() : -1, int( nStatus ), m_Resource.Get() );
}
}
}
delete this;
}
Reference in New Issue Copy Permalink