hl2_src-leak-2017/src/Tracker/AdminServer/serverping.cpp

145 lines
2.8 KiB
C++

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose: defines a RCon class used to send rcon commands to remote servers
//
// $NoKeywords: $
//=============================================================================
#include "ServerPing.h"
#include "Iresponse.h"
#include "ServerPingMsgHandler.h"
#include "Socket.h"
#include "proto_oob.h"
extern void v_strncpy(char *dest, const char *src, int bufsize);
CServerPing::CServerPing(IResponse *target,serveritem_t &server) {
memcpy(&m_Server, &server,sizeof(serveritem_t));
m_pResponseTarget=target;
m_bIsRefreshing=false;
int bytecode = S2A_INFO_DETAILED;
m_pQuery = new CSocket("internet server ping", -1);
m_pQuery->AddMessageHandler(new CServerPingMsgHandlerDetails(this, CMsgHandler::MSGHANDLER_ALL, &bytecode));
m_fSendTime=0;
}
CServerPing::~CServerPing() {
delete m_pQuery;
}
//-----------------------------------------------------------------------------
// Purpose: sends a status query packet to a single server
//-----------------------------------------------------------------------------
void CServerPing::Query()
{
CMsgBuffer *buffer = m_pQuery->GetSendBuffer();
assert( buffer );
if ( !buffer )
{
return;
}
m_bIsRefreshing=true;
m_bRefreshed=false;
netadr_t adr;
adr.ip[0] = m_Server.ip[0];
adr.ip[1] = m_Server.ip[1];
adr.ip[2] = m_Server.ip[2];
adr.ip[3] = m_Server.ip[3];
adr.port = (m_Server.port & 0xff) << 8 | (m_Server.port & 0xff00) >> 8;
adr.type = NA_IP;
// Create query message
buffer->Clear();
// Write control sequence
buffer->WriteLong(0xffffffff);
// Write query string
buffer->WriteString("ping");
// Sendmessage
m_pQuery->SendMessage( &adr );
m_fSendTime = CSocket::GetClock();
}
//-----------------------------------------------------------------------------
// Purpose:
//-----------------------------------------------------------------------------
void CServerPing::RunFrame()
{
/* float curtime = CSocket::GetClock();
if(m_fSendTime!=0 && (curtime-m_fSendTime)> 5.0f) // 10 seconds timeout
{
m_fSendTime = 0;
m_pResponseTarget->ServerFailedToRespond();
}
*/
if (m_pQuery)
{
m_pQuery->Frame();
}
}
void CServerPing::UpdateServer(float recvTime)
{
//A2A_ACK
int ping = (int)((recvTime - m_fSendTime) * 1000);
if (ping > 10000 || ping < 0)
{
// make sure ping is valid
ping = 1200;
}
m_Server.ping = ping;
m_bIsRefreshing=false;
m_bRefreshed=true;
m_fSendTime=0;
// notify the UI of the new server info
m_pResponseTarget->ServerResponded();
}
void CServerPing::Refresh()
{
Query();
}
bool CServerPing::IsRefreshing()
{
return m_bIsRefreshing;
}
serveritem_t &CServerPing::GetServer()
{
return m_Server;
}
bool CServerPing::Refreshed()
{
bool val = m_bRefreshed;
m_bRefreshed=false;
return val;
}