//======== (C) Copyright 1999, 2000 Valve, L.L.C. All rights reserved. ========
//
// The copyright to the contents herein is the property of Valve, L.L.C.
// The contents may be used and/or copied only with the written permission of
// Valve, L.L.C., or in accordance with the terms and conditions stipulated in
// the agreement/contract under which the contents have been supplied.
//
// Purpose: 
//
// $Workfile:     $
// $Date:         $
// $NoKeywords: $
//=============================================================================
#include "quakedef.h"
#include "server.h"
#include "master.h"
#include "proto_oob.h"
#include "vstdlib/ICommandLine.h"



//-----------------------------------------------------------------------------
// Purpose: List of master servers and some state info about them
//-----------------------------------------------------------------------------
typedef struct adrlist_s
{
	// Next master in chain
	struct adrlist_s	*next;
	// Challenge request sent to master
	qboolean			heartbeatwaiting;     
	// Challenge request send time
	float				heartbeatwaitingtime; 
	// Last one is Main master
	int					heartbeatchallenge;
	// Time we sent last heartbeat
	double				last_heartbeat;      
	// Master server address
	netadr_t			adr;
} adrlist_t;

//-----------------------------------------------------------------------------
// Purpose: Implements the master server interface
//-----------------------------------------------------------------------------
class CMaster : public IMaster
{
public:
	CMaster( void );
	virtual ~CMaster( void );

	// Heartbeat functions.
	void Init( void );
	void Shutdown( void );
	// Sets up master address
	void InitConnection(void);		  
	void ShutdownConnection(void);
	void SendHeartbeat( struct adrlist_s *p );
	void AddServer( struct netadr_s *adr );
	void UseDefault ( void );
	void CheckHeartbeat (void);
	void RespondToHeartbeatChallenge(void);

	void SetMaster_f( void );
	void Heartbeat_f( void );
private:
	// List of known master servers
	adrlist_t *m_pMasterAddresses;
	// If nomaster is true, the server will not send heartbeats to the master server
	bool	m_bNoMasters;
};

static CMaster g_MasterServer;
IMaster *master = (IMaster *)&g_MasterServer;

#define	HEARTBEAT_SECONDS	300
#define MASTER_PARSE_FILE "woncomm.lst"

//-----------------------------------------------------------------------------
// Purpose: Constructor
//-----------------------------------------------------------------------------
CMaster::CMaster( void )
{
	m_pMasterAddresses	= NULL;
	m_bNoMasters		= false;
}

CMaster::~CMaster( void )
{
}

//-----------------------------------------------------------------------------
// Purpose: Sends a heartbeat to the master server
// Input  : *p - 
//-----------------------------------------------------------------------------
void CMaster::SendHeartbeat ( adrlist_t *p )
{


	static char	string[2048];    // Buffer for sending heartbeat
	int			active;          // Number of active client connections
	char        szGD[ MAX_OSPATH ];


//	if ( gfUseLANAuthentication )  // No HB on lan games
	//	return;

	if ( !p )
		return;

	// Still waiting on challenge response?
	if ( p->heartbeatwaiting )
		return;

	// Waited too long
	if ( (realtime - p->heartbeatwaitingtime ) >= HB_TIMEOUT )
		return;

	InitConnection();

	//
	// count active users
	//
	SV_CountPlayers( &active );

	memset(string, 0, 2048);
	
	COM_FileBase( com_gamedir, szGD );

	// Send to master
	Q_snprintf (string, sizeof( string ), "%c\n%i\n%i\n%i\n%i\n%s\n", S2M_HEARTBEAT, p->heartbeatchallenge,
		1, active, PROTOCOL_VERSION, szGD );

	NET_SendPacket (NS_SERVER, strlen(string), string,  p->adr );

}

//-----------------------------------------------------------------------------
// Purpose: Requests a challenge so we can then send a heartbeat
//-----------------------------------------------------------------------------
void CMaster::CheckHeartbeat (void)
{


	unsigned char c;    // Buffer for sending heartbeat
	adrlist_t *p;

/*	if ( m_bNoMasters ||      // We are ignoring heartbeats
	    gfUseLANAuthentication || // No HB on lan games
		sv_lan.GetInt() ||           // Lan servers don't heartbeat
		(svs.maxclients <= 1) ||  // not a multiplayer server.
		!sv.active )			  // only heartbeat if a server is running.
		return;*/

	InitConnection();

	p = m_pMasterAddresses;
	while ( p )
	{
		// Time for another try?
		if ( ( realtime - p->last_heartbeat) < HEARTBEAT_SECONDS)  // not time to send yet
		{
			p = p->next;
			continue;
		}

		// Should we resend challenge request?
		if ( p->heartbeatwaiting && 
			( ( realtime - p->heartbeatwaitingtime ) < HB_TIMEOUT ) )
		{
			p = p->next;
			continue;
		}

		p->heartbeatwaiting     = true;
		p->heartbeatwaitingtime = realtime;
	
		p->last_heartbeat       = realtime;  // Flag at start so we don't just keep trying for hb's when

		c = A2A_GETCHALLENGE;

		// Send to master asking for a challenge #
		NET_SendPacket (NS_SERVER, 1, "q", p->adr);
	}

}

//-----------------------------------------------------------------------------
// Purpose: Server is shutting down, unload master servers list, tell masters that we are closing the server
//-----------------------------------------------------------------------------
void CMaster::ShutdownConnection( void )
{
	char		string[2048];
	adrlist_t *p;

	if ( !host_initialized )
		return;

	/*if ( m_bNoMasters ||      // We are ignoring heartbeats
	    gfUseLANAuthentication || // No HB on lan games
		sv_lan.GetInt() ||           // Lan servers don't heartbeat
		(svs.maxclients <= 1) )   // not a multiplayer server.
		return;*/

	InitConnection();

	Q_snprintf (string, sizeof( string ), "%c\n", S2M_SHUTDOWN);
	
	p = m_pMasterAddresses;
	while ( p )
	{
		NET_SendPacket (NS_SERVER, strlen(string), string, p->adr);
		p = p->next;
	}
}

//-----------------------------------------------------------------------------
// Purpose: Add server to the master list
// Input  : *adr - 
//-----------------------------------------------------------------------------
void CMaster::AddServer( netadr_t *adr )
{
	adrlist_t *n;

	InitConnection();

	// See if it's there
	n = m_pMasterAddresses;
	while ( n )
	{
		if ( NET_CompareAdr( n->adr, *adr ) )
			break;
		n = n->next;
	}

	// Found it in the list.
	if ( n )
		return;

	n = ( adrlist_t * ) malloc ( sizeof( adrlist_t ) );
	if ( !n )
		Sys_Error( "Error allocating %i bytes for master address.", sizeof( adrlist_t ) );

	memset( n, 0, sizeof( adrlist_t ) );

	n->adr = *adr;

	// Queue up a full heartbeat to all master servers.
	n->last_heartbeat = -99999;

	// Link it in.
	n->next = m_pMasterAddresses;
	m_pMasterAddresses = n;
}

//-----------------------------------------------------------------------------
// Purpose: Add built-in default master if woncomm.lst doesn't parse
//-----------------------------------------------------------------------------
void CMaster::UseDefault ( void )
{
	char szValveMasterAddress[256];
	netadr_t adr;

	// Get default master
	Q_snprintf( szValveMasterAddress, sizeof( szValveMasterAddress ), VALVE_MASTER_ADDRESS );    // IP:PORT string

	// Convert to netadr_t
	if ( NET_StringToAdr ( szValveMasterAddress, &adr ) )
	{
		// Add to master list
		AddServer( &adr );
	}
}

//-----------------------------------------------------------------------------
// Purpose: Initializes the default master server address
//-----------------------------------------------------------------------------
void CMaster::InitConnection( void )
{
	char szValveMasterAddress[256];
	char szWONCommFile[MAX_OSPATH];
	char *pbuffer;
	char *pstart;
	netadr_t adr;
	char szAdr[64];
	int nPort;
	int nCount = 0;
	qboolean bIgnore;
	int nDefaultPort;
	static qboolean bInitialized = false;

/*	if ( m_bNoMasters ||      // We are ignoring heartbeats
	     gfUseLANAuthentication || // No HB on lan games
		 sv_lan.GetInt() ||           // Lan servers don't heartbeat
		 (svs.maxclients <= 1) )
	{
		return;
	}*/

	// Already able to initialize at least once?
	if ( bInitialized )
		return;

	// So we don't do this a second time.
	bInitialized = true;

	// Assume default master
	Q_snprintf( szValveMasterAddress, sizeof( szValveMasterAddress ), VALVE_MASTER_ADDRESS);    // IP:PORT string
	Q_snprintf( szWONCommFile, sizeof( szWONCommFile ), "%s", MASTER_PARSE_FILE );

	const char *pComFile = CommandLine()->ParmValue( "-comm" );
	if ( pComFile )
	{
		strcpy ( szWONCommFile, pComFile );
	}

	// Read them in from WONCOMM.LST
	pbuffer = (char*)COM_LoadFile( szWONCommFile, 5, NULL ); // Use malloc
	if ( !pbuffer )
	{
		Con_Printf( "Couldn't load woncomm.lst\nUsing default master\n" );
		UseDefault();
		return;
	}

	pstart = pbuffer;

	while ( 1 )
	{
		pstart = COM_Parse( pstart );

		if ( strlen(com_token) <= 0)
			break;

		bIgnore = true;

		nDefaultPort = PORT_MASTER;
		if ( !stricmp( com_token, "Master" ) )
		{
			bIgnore = FALSE;
		}

		// Now parse all addresses between { }
		pstart = COM_Parse( pstart );
		if ( strlen(com_token) <= 0 )
			break;

		if ( stricmp ( com_token, "{" ) )
			break;

		// Parse addresses until we get to "}"
		while ( 1 )
		{
			char base[256];

			// Now parse all addresses between { }
			pstart = COM_Parse( pstart );
			
			if (strlen(com_token) <= 0)
				break;

			if ( !stricmp ( com_token, "}" ) )
				break;
			
			Q_snprintf( base, sizeof( base ), "%s", com_token );
				
			pstart = COM_Parse( pstart );
			
			if (strlen(com_token) <= 0)
				break;

			if ( stricmp( com_token, ":" ) )
				break;

			pstart = COM_Parse( pstart );
			
			if (strlen(com_token) <= 0)
				break;

			nPort = atoi ( com_token );
			if ( !nPort )
				nPort = nDefaultPort;

			Q_snprintf( szAdr, sizeof( szAdr ), "%s:%i", base, nPort );

			// Can we resolve it any better
			if ( !NET_StringToAdr( szAdr, &adr ) )
				bIgnore = true;

			if ( !bIgnore )
			{
				Con_Printf( "Adding master server %s\n", NET_AdrToString( adr ) );
				AddServer( &adr );
				nCount++;
			}
		}
	}

	if ( !nCount )
	{
		Con_Printf( "No masters parsed from woncomm.lst\nUsing default master\n" );
		UseDefault();
	}

	free( pbuffer );
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void CMaster::RespondToHeartbeatChallenge(void)
{
	adrlist_t *p;

	// No masters, just ignore.
	/*if ( !m_pMasterAddresses )
		return;*/

	p = m_pMasterAddresses;
	while ( p )
	{
		if ( NET_CompareAdr( p->adr, net_from ) )
			break;

		p = p->next;
	}

	// Not a known master server.
	if ( !p )
		return;

	// Kill timer
	p->heartbeatwaiting   = false;
	p->heartbeatchallenge = MSG_ReadLong();

	// Send the actual heartbeat request to this master server.
	SendHeartbeat( p );
}

//-----------------------------------------------------------------------------
// Purpose: Add/remove master servers
//-----------------------------------------------------------------------------
void CMaster::SetMaster_f (void)
{
	int		i;
	char   *pMasterAddress; // IP:Port string of master
	char   *pMasterPort;    // Port of master
	char    szMaster[128];  // Full string
	int     ipPort;         // Numeric port
	char    *pszCmd;
	char    szMasterPort[20];
	netadr_t adr;
	qboolean bfirst = false;

	ipPort = PORT_MASTER; // default
	Q_snprintf( szMasterPort, sizeof( szMasterPort ), "%i", ipPort );

	master->InitConnection();

	if ( !m_pMasterAddresses )
		bfirst = true;

	i = Cmd_Argc();

	// Usage help
	if ( ( i < 2 ) ||
		 ( i > 4 ) )
	{
		adrlist_t *p;
		Con_Printf("Usage:\nSetmaster <add | remove | enable | disable> <IP:port>\n");

		p = m_pMasterAddresses;
		if ( !p )
			Con_Printf("Current:  None\n");
		else
		{
			int i = 1;
			Con_Printf("Current:\n");
			while ( p )
			{
				Con_Printf("  %i:  %s\n", i++, NET_AdrToString( p->adr ) );
				p = p->next;
			}
		}
		return;
	}

	pszCmd = Cmd_Argv(1);
	if ( !pszCmd || !pszCmd[0] )
		return;

	// Check for disabling...
	if ( !stricmp( pszCmd, "disable") )
	{
		m_bNoMasters = true;
		return;
	}
	else if (!stricmp( pszCmd, "enable") )
	{
		m_bNoMasters = false;
		return;
	}

	if ( stricmp( pszCmd, "add" ) && 
		 stricmp( pszCmd, "remove" ) )
	{
		Con_Printf( "Setmaster:  Unknown command %s\n", pszCmd );
		return;
	}

	// Get address and, if a port is specified get it, too.
	pMasterAddress = Cmd_Argv(2);
	if (i == 4)
	{
		pMasterPort      = Cmd_Argv(3);
		if ( pMasterPort && pMasterPort[0] )
		{
			ipPort = atoi(pMasterPort);
			if (ipPort == 0)   // If any problem, assuem default
				ipPort = PORT_MASTER;
		}
		else
			pMasterPort = szMasterPort;
	}
				 
	// Construct a full string
	Q_snprintf( szMaster, sizeof( szMaster ), "%s:%i", pMasterAddress, ipPort );

	// Convert to a valid address
	if ( !NET_StringToAdr ( szMaster, &adr ) )
	{
		Con_Printf(" Invalid address \"%s\", setmaster command ignored\n", szMaster );
		return;
	}

	if ( !stricmp( pszCmd, "add" ) )
	{
		master->AddServer( &adr );

		// If we get here, masters are definitely being used.
		m_bNoMasters = false;

		Con_Printf ( "Adding master at %s\n", szMaster );
	}
	else
	{
		// Find master server
		adrlist_t *p;

		if ( !m_pMasterAddresses )
		{
			Con_Printf( "Can't remove master, list is empty\n" );
			return;
		}

		p = m_pMasterAddresses;
		while ( p )
		{
			if ( NET_CompareAdr( p->adr, adr ) )
				break;
			p = p->next;
		}

		if ( !p )
		{
			Con_Printf( "Can't remove master %s, not in list\n", szMaster );
			return;
		}

		// p is pointing at the master to remove.
		if ( p == m_pMasterAddresses )
		{
			m_pMasterAddresses = m_pMasterAddresses->next;
			free( p );
		}
		else
		{		
			adrlist_t *pbefore;

			pbefore = m_pMasterAddresses;
			while ( pbefore )
			{
				if ( pbefore->next == p )
					break;
				pbefore = pbefore->next;
			}

			if ( !pbefore ) // !!!
				return;

			// Unlink p
			pbefore->next = p->next;
			free( p );
		}
	}
}

//-----------------------------------------------------------------------------
// Purpose: Send a new heartbeat to the master
//-----------------------------------------------------------------------------
void CMaster::Heartbeat_f (void)
{
	adrlist_t *p;

	p = m_pMasterAddresses;
	while ( p )
	{
		// Queue up a full hearbeat
		p->last_heartbeat = -9999;
		p = p->next;
	}
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void SetMaster_f( void )
{
	master->SetMaster_f();
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void Heartbeat_f( void )
{
	master->Heartbeat_f();
}

static ConCommand setmaster("setmaster", SetMaster_f );  
static ConCommand heartbeat("heartbeat", Heartbeat_f ); 

//-----------------------------------------------------------------------------
// Purpose: Adds master server console commands
//-----------------------------------------------------------------------------
void CMaster::Init( void )
{
}

//-----------------------------------------------------------------------------
// Purpose: 
//-----------------------------------------------------------------------------
void CMaster::Shutdown(void)
{
	adrlist_t *p, *n;

	// Free the master list now.
	p = m_pMasterAddresses;
	while ( p )
	{
		n = p->next;
		free( p );
		p = n;
	}

	m_pMasterAddresses = NULL;
}