hl2_src-leak-2017/src/serverbrowser/DialogAddServer.cpp

393 lines
14 KiB
C++

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
// $NoKeywords: $
//=============================================================================
#include "pch_serverbrowser.h"
using namespace vgui;
//-----------------------------------------------------------------------------
// Purpose: Constructor
// Input : *gameList - game list to add specified server to
//-----------------------------------------------------------------------------
CDialogAddServer::CDialogAddServer(vgui::Panel *parent, IGameList *gameList) : Frame(parent, "DialogAddServer")
{
SetDeleteSelfOnClose(true);
m_pGameList = gameList;
SetTitle("#ServerBrowser_AddServersTitle", true);
SetSizeable( false );
m_pTabPanel = new PropertySheet(this, "GameTabs");
m_pTabPanel->SetTabWidth(72);
m_pDiscoveredGames = new ListPanel( this, "Servers" );
// Add the column headers
m_pDiscoveredGames->AddColumnHeader(0, "Password", "#ServerBrowser_Password", 16, ListPanel::COLUMN_FIXEDSIZE | ListPanel::COLUMN_IMAGE);
m_pDiscoveredGames->AddColumnHeader(1, "Bots", "#ServerBrowser_Bots", 16, ListPanel::COLUMN_FIXEDSIZE | ListPanel::COLUMN_IMAGE | ListPanel::COLUMN_HIDDEN);
m_pDiscoveredGames->AddColumnHeader(2, "Secure", "#ServerBrowser_Secure", 16, ListPanel::COLUMN_FIXEDSIZE | ListPanel::COLUMN_IMAGE);
bool bGameSupportsReplay = GameSupportsReplay();
int nReplayWidth = 16;
if ( !bGameSupportsReplay )
{
nReplayWidth = 0;
}
m_pDiscoveredGames->AddColumnHeader(3, "Replay", "#ServerBrowser_Replay", nReplayWidth, ListPanel::COLUMN_FIXEDSIZE | ListPanel::COLUMN_IMAGE);
m_pDiscoveredGames->AddColumnHeader(4, "Name", "#ServerBrowser_Servers", 20, ListPanel::COLUMN_RESIZEWITHWINDOW | ListPanel::COLUMN_UNHIDABLE);
m_pDiscoveredGames->AddColumnHeader(5, "IPAddr", "#ServerBrowser_IPAddress", 60, ListPanel::COLUMN_HIDDEN);
m_pDiscoveredGames->AddColumnHeader(6, "GameDesc", "#ServerBrowser_Game", 150);
m_pDiscoveredGames->AddColumnHeader(7, "Players", "#ServerBrowser_Players", 60);
m_pDiscoveredGames->AddColumnHeader(8, "Map", "#ServerBrowser_Map", 80);
m_pDiscoveredGames->AddColumnHeader(9, "Ping", "#ServerBrowser_Latency", 60);
m_pDiscoveredGames->SetColumnHeaderTooltip(0, "#ServerBrowser_PasswordColumn_Tooltip");
m_pDiscoveredGames->SetColumnHeaderTooltip(1, "#ServerBrowser_BotColumn_Tooltip");
m_pDiscoveredGames->SetColumnHeaderTooltip(2, "#ServerBrowser_SecureColumn_Tooltip");
if ( bGameSupportsReplay )
{
m_pDiscoveredGames->SetColumnHeaderTooltip(3, "#ServerBrowser_ReplayColumn_Tooltip");
}
// setup fast sort functions
m_pDiscoveredGames->SetSortFunc(0, PasswordCompare);
m_pDiscoveredGames->SetSortFunc(1, BotsCompare);
m_pDiscoveredGames->SetSortFunc(2, SecureCompare);
if ( bGameSupportsReplay )
{
m_pDiscoveredGames->SetSortFunc(3, ReplayCompare);
}
m_pDiscoveredGames->SetSortFunc(4, ServerNameCompare);
m_pDiscoveredGames->SetSortFunc(5, IPAddressCompare);
m_pDiscoveredGames->SetSortFunc(6, GameCompare);
m_pDiscoveredGames->SetSortFunc(7, PlayersCompare);
m_pDiscoveredGames->SetSortFunc(8, MapCompare);
m_pDiscoveredGames->SetSortFunc(9, PingCompare);
m_pDiscoveredGames->SetSortColumn(9); // sort on ping
m_pTextEntry = new vgui::TextEntry( this, "ServerNameText" );
m_pTextEntry->AddActionSignalTarget( this );
m_pTestServersButton = new vgui::Button( this, "TestServersButton", "" );
m_pAddServerButton = new vgui::Button( this, "OKButton", "" );
m_pAddSelectedServerButton = new vgui::Button( this, "SelectedOKButton", "", this, "addselected" );
m_pTabPanel->AddPage( m_pDiscoveredGames, "#ServerBrowser_Servers" );
LoadControlSettings("Servers/DialogAddServer.res");
// Setup the buttons. We leave them disabled until there is text in the textbox.
m_pAddServerButton->SetEnabled( false );
m_pTestServersButton->SetEnabled( false );
m_pAddSelectedServerButton->SetEnabled( false );
m_pAddSelectedServerButton->SetVisible( false );
m_pTabPanel->SetVisible( false );
m_pTextEntry->RequestFocus();
// Initially, we aren't high enough to show the tab panel.
int x, y;
m_pTabPanel->GetPos( x, y );
m_OriginalHeight = m_pTabPanel->GetTall() + y + 50;
SetTall( y );
}
//-----------------------------------------------------------------------------
// Purpose: Destructor
//-----------------------------------------------------------------------------
CDialogAddServer::~CDialogAddServer()
{
FOR_EACH_VEC( m_Queries, i )
{
if ( steamapicontext->SteamMatchmakingServers() )
steamapicontext->SteamMatchmakingServers()->CancelServerQuery( m_Queries[ i ] );
}
}
//-----------------------------------------------------------------------------
// Lets us know when the text entry has changed.
//-----------------------------------------------------------------------------
void CDialogAddServer::OnTextChanged()
{
bool bAnyText = (m_pTextEntry->GetTextLength() > 0);
m_pAddServerButton->SetEnabled( bAnyText );
m_pTestServersButton->SetEnabled( bAnyText );
}
//-----------------------------------------------------------------------------
// Purpose: button command handler
//-----------------------------------------------------------------------------
void CDialogAddServer::OnCommand(const char *command)
{
if ( Q_stricmp(command, "OK") == 0 )
{
OnOK();
}
else if ( Q_stricmp( command, "TestServers" ) == 0 )
{
SetTall( m_OriginalHeight );
m_pTabPanel->SetVisible( true );
m_pAddSelectedServerButton->SetVisible( true );
TestServers();
}
else if ( !Q_stricmp( command, "addselected" ) )
{
if ( m_pDiscoveredGames->GetSelectedItemsCount() )
{
// get the server
int serverID = m_pDiscoveredGames->GetItemUserData( m_pDiscoveredGames->GetSelectedItem(0) );
FinishAddServer( m_Servers[ serverID ] );
m_pDiscoveredGames->RemoveItem( m_pDiscoveredGames->GetSelectedItem(0) ); // as we add to favs remove from the list
m_pDiscoveredGames->SetEmptyListText( "" );
}
}
else
{
BaseClass::OnCommand(command);
}
}
//-----------------------------------------------------------------------------
// Purpose: Handles the OK button being pressed; adds the server to the game list
//-----------------------------------------------------------------------------
void CDialogAddServer::OnOK()
{
// try and parse out IP address
const char *address = GetControlString("ServerNameText", "");
netadr_t netaddr;
netaddr.SetFromString( address, true );
if ( !netaddr.GetPort() && !AllowInvalidIPs() )
{
// use the default port since it was not entered
netaddr.SetPort( 27015 );
}
if ( AllowInvalidIPs() || netaddr.IsValid() )
{
gameserveritem_t server;
memset(&server, 0, sizeof(server));
server.SetName( address );
// We assume here that the query and connection ports are the same. This is why it's much
// better if they click "Servers" and choose a server in there.
server.m_NetAdr.Init( netaddr.GetIPHostByteOrder(), netaddr.GetPort(), netaddr.GetPort() );
server.m_nAppID = 0;
FinishAddServer( server );
}
else
{
// could not parse the ip address, popup an error
MessageBox *dlg = new MessageBox("#ServerBrowser_AddServerErrorTitle", "#ServerBrowser_AddServerError");
dlg->DoModal();
}
// mark ourselves to be closed
PostMessage(this, new KeyValues("Close"));
}
//-----------------------------------------------------------------------------
// Purpose: Ping a particular IP for server presence
//-----------------------------------------------------------------------------
void CDialogAddServer::TestServers()
{
if ( !steamapicontext->SteamMatchmakingServers() )
return;
m_pDiscoveredGames->SetEmptyListText( "" );
m_pDiscoveredGames->RemoveAll();
// If they specified a port, then send a query to that port.
const char *address = GetControlString("ServerNameText", "");
netadr_t netaddr;
netaddr.SetFromString( address, true );
m_Servers.RemoveAll();
CUtlVector<netadr_t> vecAdress;
if ( netaddr.GetPort() == 0 )
{
// No port specified. Go to town on the ports.
CUtlVector<uint16> portsToTry;
GetMostCommonQueryPorts( portsToTry );
for ( int i=0; i < portsToTry.Count(); i++ )
{
netadr_t newAddr = netaddr;
newAddr.SetPort( portsToTry[i] );
vecAdress.AddToTail( newAddr );
}
}
else
{
vecAdress.AddToTail( netaddr );
}
// Change the text on the tab panel..
m_pTabPanel->RemoveAllPages();
wchar_t wstr[512];
if ( address[0] == 0 )
{
Q_wcsncpy( wstr, g_pVGuiLocalize->Find( "#ServerBrowser_ServersRespondingLocal"), sizeof( wstr ) );
}
else
{
wchar_t waddress[512];
Q_UTF8ToUnicode( address, waddress, sizeof( waddress ) );
g_pVGuiLocalize->ConstructString( wstr, sizeof( wstr ), g_pVGuiLocalize->Find( "#ServerBrowser_ServersResponding"), 1, waddress );
}
char str[512];
Q_UnicodeToUTF8( wstr, str, sizeof( str ) );
m_pTabPanel->AddPage( m_pDiscoveredGames, str );
m_pTabPanel->InvalidateLayout();
FOR_EACH_VEC( vecAdress, iAddress )
{
m_Queries.AddToTail( steamapicontext->SteamMatchmakingServers()->PingServer( vecAdress[ iAddress ].GetIPHostByteOrder(), vecAdress[ iAddress ].GetPort(), this ) );
}
}
//-----------------------------------------------------------------------------
// Purpose: A server answered our ping
//-----------------------------------------------------------------------------
void CDialogAddServer::ServerResponded( gameserveritem_t &server )
{
KeyValues *kv = new KeyValues( "Server" );
kv->SetString( "name", server.GetName() );
kv->SetString( "map", server.m_szMap );
kv->SetString( "GameDir", server.m_szGameDir );
kv->SetString( "GameDesc", server.m_szGameDescription );
kv->SetString( "GameTags", server.m_szGameTags );
kv->SetInt( "password", server.m_bPassword ? 1 : 0);
kv->SetInt( "bots", server.m_nBotPlayers ? 2 : 0);
kv->SetInt( "Replay", IsReplayServer( server ) ? 5 : 0 );
if ( server.m_bSecure )
{
// show the denied icon if banned from secure servers, the secure icon otherwise
kv->SetInt("secure", ServerBrowser().IsVACBannedFromGame( server.m_nAppID ) ? 4 : 3);
}
else
{
kv->SetInt("secure", 0);
}
netadr_t reportedIPAddr;
reportedIPAddr.SetIP( server.m_NetAdr.GetIP() );
reportedIPAddr.SetPort( server.m_NetAdr.GetConnectionPort() );
kv->SetString("IPAddr", reportedIPAddr.ToString() );
char buf[32];
Q_snprintf(buf, sizeof(buf), "%d / %d", server.m_nPlayers, server.m_nMaxPlayers);
kv->SetString("Players", buf);
kv->SetInt("Ping", server.m_nPing);
// new server, add to list
int iServer = m_Servers.AddToTail( server );
int iListID = m_pDiscoveredGames->AddItem(kv, iServer, false, false);
if ( m_pDiscoveredGames->GetItemCount() == 1 )
{
m_pDiscoveredGames->AddSelectedItem( iListID );
}
kv->deleteThis();
m_pDiscoveredGames->InvalidateLayout();
}
void CDialogAddServer::ServerFailedToRespond()
{
m_pDiscoveredGames->SetEmptyListText( "#ServerBrowser_ServerNotResponding" );
}
void CDialogAddServer::ApplySchemeSettings( IScheme *pScheme )
{
BaseClass::ApplySchemeSettings( pScheme );
ImageList *imageList = new ImageList(false);
imageList->AddImage(scheme()->GetImage("servers/icon_password", false));
imageList->AddImage(scheme()->GetImage("servers/icon_bots", false));
imageList->AddImage(scheme()->GetImage("servers/icon_robotron", false));
imageList->AddImage(scheme()->GetImage("servers/icon_secure_deny", false));
imageList->AddImage(scheme()->GetImage("servers/icon_replay", false));
int passwordColumnImage = imageList->AddImage(scheme()->GetImage("servers/icon_password_column", false));
int botColumnImage = imageList->AddImage(scheme()->GetImage("servers/icon_bots_column", false));
int secureColumnImage = imageList->AddImage(scheme()->GetImage("servers/icon_robotron_column", false));
int replayColumnImage = imageList->AddImage(scheme()->GetImage("servers/icon_replay_column", false));
m_pDiscoveredGames->SetImageList(imageList, true);
vgui::HFont hFont = pScheme->GetFont( "ListSmall", IsProportional() );
if ( !hFont )
hFont = pScheme->GetFont( "DefaultSmall", IsProportional() );
m_pDiscoveredGames->SetFont( hFont );
m_pDiscoveredGames->SetColumnHeaderImage(0, passwordColumnImage);
m_pDiscoveredGames->SetColumnHeaderImage(1, botColumnImage);
m_pDiscoveredGames->SetColumnHeaderImage(2, secureColumnImage);
m_pDiscoveredGames->SetColumnHeaderImage(3, replayColumnImage);
}
//-----------------------------------------------------------------------------
// Purpose: A server on the listed IP responded
//-----------------------------------------------------------------------------
void CDialogAddServer::OnItemSelected()
{
int nSelectedItem = m_pDiscoveredGames->GetSelectedItem(0);
if( nSelectedItem != -1 )
{
m_pAddSelectedServerButton->SetEnabled( true );
}
else
{
m_pAddSelectedServerButton->SetEnabled( false );
}
}
//-----------------------------------------------------------------------------
// Purpose:
//-----------------------------------------------------------------------------
void CDialogAddServer::FinishAddServer( gameserveritem_t &pServer )
{
ServerBrowserDialog().AddServerToFavorites( pServer );
}
//-----------------------------------------------------------------------------
// Purpose:
//-----------------------------------------------------------------------------
void CDialogAddBlacklistedServer::FinishAddServer( gameserveritem_t &pServer )
{
ServerBrowserDialog().AddServerToBlacklist( pServer );
ServerBrowserDialog().BlacklistsChanged();
}
//-----------------------------------------------------------------------------
// Purpose:
//-----------------------------------------------------------------------------
void CDialogAddBlacklistedServer::ApplySchemeSettings( vgui::IScheme *pScheme )
{
BaseClass::ApplySchemeSettings( pScheme );
m_pAddServerButton->SetText( "#ServerBrowser_AddAddressToBlacklist" );
m_pAddSelectedServerButton->SetText( "#ServerBrowser_AddSelectedToBlacklist" );
}