/*



   Copyright [2008] [Trevor Hogan]



   Licensed under the Apache License, Version 2.0 (the "License");

   you may not use this file except in compliance with the License.

   You may obtain a copy of the License at



       http://www.apache.org/licenses/LICENSE-2.0



   Unless required by applicable law or agreed to in writing, software

   distributed under the License is distributed on an "AS IS" BASIS,

   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

   See the License for the specific language governing permissions and

   limitations under the License.



   CODE PORTED FROM THE ORIGINAL GHOST PROJECT: http://ghost.pwner.org/



*/



#include "ghost.h"

#include "util.h"

#include "socket.h"

#include "commandpacket.h"

#include "bnlsprotocol.h"

#include "bnlsclient.h"

#include "ui/forward.h"



//

// CBNLSClient

//



CBNLSClient :: CBNLSClient( string nServer, uint16_t nPort, uint32_t nWardenCookie )

{

	m_Socket = new CTCPClient( );

	m_Protocol = new CBNLSProtocol( );

	m_WasConnected = false;

	m_Server = nServer;

	m_Port = nPort;

	m_LastNullTime = 0;

	m_WardenCookie = nWardenCookie;

	m_TotalWardenIn = 0;

	m_TotalWardenOut = 0;

}



CBNLSClient :: ~CBNLSClient( )

{

	delete m_Socket;

	delete m_Protocol;



	while( !m_Packets.empty( ) )

	{

		delete m_Packets.front( );

		m_Packets.pop( );

	}

}



BYTEARRAY CBNLSClient :: GetWardenResponse( )

{

	BYTEARRAY WardenResponse;



	if( !m_WardenResponses.empty( ) )

	{

		WardenResponse = m_WardenResponses.front( );

		m_WardenResponses.pop( );

		m_TotalWardenOut++;

	}



	return WardenResponse;

}



unsigned int CBNLSClient :: SetFD( void *fd, void *send_fd, int *nfds )

{

	if( !m_Socket->HasError( ) && m_Socket->GetConnected( ) )

	{

		m_Socket->SetFD( (fd_set *)fd, (fd_set *)send_fd, nfds );

		return 1;

	}



	return 0;

}



bool CBNLSClient :: Update( void *fd, void *send_fd )

{

	if( m_Socket->HasError( ) )

	{

		CONSOLE_Print( "[BNLSC: " + m_Server + ":" + UTIL_ToString( m_Port ) + ":C" + UTIL_ToString( m_WardenCookie ) + "] disconnected from BNLS server due to socket error" );

		forward(new CFwdData(FWD_GENERAL, "BNLSC: " + m_Server + ":" + UTIL_ToString( m_Port ) + ":C" + UTIL_ToString( m_WardenCookie ) + " - Disconnected from BNLS server due to socket error", 6, 0));

		return true;

	}



	if( !m_Socket->GetConnecting( ) && !m_Socket->GetConnected( ) && m_WasConnected )

	{

		CONSOLE_Print( "[BNLSC: " + m_Server + ":" + UTIL_ToString( m_Port ) + ":C" + UTIL_ToString( m_WardenCookie ) + "] disconnected from BNLS server" );

		forward(new CFwdData(FWD_GENERAL, "BNLSC: " + m_Server + ":" + UTIL_ToString( m_Port ) + ":C" + UTIL_ToString( m_WardenCookie ) + " - Disconnected from BNLS server", 6, 0));

		return true;

	}



	if( m_Socket->GetConnected( ) )

	{

		m_Socket->DoRecv( (fd_set *)fd );

		ExtractPackets( );

		ProcessPackets( );



		if( GetTime( ) - m_LastNullTime >= 50 )

		{

			m_Socket->PutBytes( m_Protocol->SEND_BNLS_NULL( ) );

			m_LastNullTime = GetTime( );

		}



		while( !m_OutPackets.empty( ) )

		{

			m_Socket->PutBytes( m_OutPackets.front( ) );

			m_OutPackets.pop( );

		}



		m_Socket->DoSend( (fd_set *)send_fd );

		return false;

	}



	if( m_Socket->GetConnecting( ) && m_Socket->CheckConnect( ) )

	{

		CONSOLE_Print( "[BNLSC: " + m_Server + ":" + UTIL_ToString( m_Port ) + ":C" + UTIL_ToString( m_WardenCookie ) + "] connected" );

		forward(new CFwdData(FWD_GENERAL, "BNLSC: " + m_Server + ":" + UTIL_ToString( m_Port ) + ":C" + UTIL_ToString( m_WardenCookie ) + " - Connected", 0, 0));

		m_WasConnected = true;

		m_LastNullTime = GetTime( );

		return false;

	}



	if( !m_Socket->GetConnecting( ) && !m_Socket->GetConnected( ) && !m_WasConnected )

	{

		CONSOLE_Print( "[BNLSC: " + m_Server + ":" + UTIL_ToString( m_Port ) + ":C" + UTIL_ToString( m_WardenCookie ) + "] connecting to server [" + m_Server + "] on port " + UTIL_ToString( m_Port ) );

		forward(new CFwdData(FWD_GENERAL, "BNLSC: " + m_Server + ":" + UTIL_ToString( m_Port ) + ":C" + UTIL_ToString( m_WardenCookie ) + " - Connecting to server [" + m_Server + "] on port " + UTIL_ToString( m_Port ), 0, 0));

		m_Socket->Connect( string( ), m_Server, m_Port );

		return false;

	}



	return false;

}



void CBNLSClient :: ExtractPackets( )

{

	string *RecvBuffer = m_Socket->GetBytes( );

	BYTEARRAY Bytes = UTIL_CreateByteArray( (unsigned char *)RecvBuffer->c_str( ), RecvBuffer->size( ) );



	while( Bytes.size( ) >= 3 )

	{

		uint16_t Length = UTIL_ByteArrayToUInt16( Bytes, false );



		if( Length >= 3 )

		{

			if( Bytes.size( ) >= Length )

			{

				m_Packets.push( new CCommandPacket( 0, Bytes[2], BYTEARRAY( Bytes.begin( ), Bytes.begin( ) + Length ) ) );

				*RecvBuffer = RecvBuffer->substr( Length );

				Bytes = BYTEARRAY( Bytes.begin( ) + Length, Bytes.end( ) );

			}

			else

				return;

		}

		else

		{

			CONSOLE_Print( "[BNLSC: " + m_Server + ":" + UTIL_ToString( m_Port ) + ":C" + UTIL_ToString( m_WardenCookie ) + "] error - received invalid packet from BNLS server (bad length), disconnecting" );

			forward(new CFwdData(FWD_GENERAL, "BNLSC: " + m_Server + ":" + UTIL_ToString( m_Port ) + ":C" + UTIL_ToString( m_WardenCookie ) + " - Error - received invalid packet from BNLS server (bad length), disconnecting", 0, 0));

			m_Socket->Disconnect( );

			return;

		}

	}

}



void CBNLSClient :: ProcessPackets( )

{

	while( !m_Packets.empty( ) )

	{

		CCommandPacket *Packet = m_Packets.front( );

		m_Packets.pop( );



		if( Packet->GetID( ) == CBNLSProtocol :: BNLS_WARDEN )

		{

			BYTEARRAY WardenResponse = m_Protocol->RECEIVE_BNLS_WARDEN( Packet->GetData( ) );



			if( !WardenResponse.empty( ) )

				m_WardenResponses.push( WardenResponse );

		}



		delete Packet;

	}

}



void CBNLSClient :: QueueWardenSeed( uint32_t seed )

{

	m_OutPackets.push( m_Protocol->SEND_BNLS_WARDEN_SEED( m_WardenCookie, seed ) );

}



void CBNLSClient :: QueueWardenRaw( BYTEARRAY wardenRaw )

{

	m_OutPackets.push( m_Protocol->SEND_BNLS_WARDEN_RAW( m_WardenCookie, wardenRaw ) );

	m_TotalWardenIn++;

}