// Copyright (c) 2003-2009 Nokia Corporation and/or its subsidiary(-ies).

// All rights reserved.

// This component and the accompanying materials are made available

// under the terms of "Eclipse Public License v1.0"

// which accompanies this distribution, and is available

// at the URL "http://www.eclipse.org/legal/epl-v10.html".

//

// Initial Contributors:

// Nokia Corporation - initial contribution.

//

// Contributors:

//

// Description:

// CTcpServer

// System Include

// 

//

#include <stdio.h>

#include <stdlib.h>

#include <assert.h>

/***************************************************************************************

 *

 * Local Include

 *

 **************************************************************************************/

#include "CTcpServer.h"

/***************************************************************************************

 *

 * PUBLIC METHOD: Constructor

 *

 **************************************************************************************/

CTcpServer::CTcpServer()

{

}

/***************************************************************************************

 *

` * PUBLIC METHOD: Destructor

 *

 **************************************************************************************/

CTcpServer::~CTcpServer()

{

	disconnect();

	release();

}	

/***************************************************************************************

 *

 * PUBLIC METHOD: StartServer

 *

 **************************************************************************************/

int CTcpServer::StartServer( char* aLocalInterface, char* aLocalPort )

{	

	int r;

	r =initialise();

	if(r != ITS_OK)

		return r;

	r = setArguments( aLocalInterface, aLocalPort );

	if(r != ITS_OK)

		return r;

	r = makeConnection(iLocal_Sock, iRemote_Sock, iRemote_Addr);

	return r;

}

/***************************************************************************************

 *

 * SECTION: CUSTOMISATION METHODS

 *

 **************************************************************************************/

/***************************************************************************************

 *

 * PRIVATE METHOD: ProcessBuffer

 *

 **************************************************************************************/

void CTcpServer::ProcessBuffer( char* aBuffer, int aLen )

{

	int i;

	for( i = 0; i < aLen; i++ ) {

	//	fprintf( stderr, "%c (0x%x)\n", aBuffer[i], aBuffer[i] );

		fprintf( stderr, "%c", aBuffer[i] );

	}

}

/***************************************************************************************

 *

 * PRIVATE METHOD: IsBufferComplete

 *

 **************************************************************************************/

bool CTcpServer::IsBufferComplete(char* aBuffer,int aLen)

{

	return true;

}

/***************************************************************************************

 *

 * SECTION: PRIVATE METHODS

 *

 **************************************************************************************/

//

//	Method binds the sock to the address. 

//

int CTcpServer::connect(SOCKET &aSocket, SOCKADDR_IN &aAddr)

{

	int err;

	//Bind the local address with a socket, so it listens to that port.

	if (bind(aSocket,(struct sockaddr*)&aAddr, sizeof(SOCKADDR_IN)) != 0)

	{	

		//return, error has occured

		err = WSAGetLastError();

		printf ("Error binding socket: %d", err);

		return E_SOCKETBIND;

	}

	return ITS_OK;

}

//

//  Disoconnect

//

int CTcpServer::disconnect()

{

	shutdown(iLocal_Sock, SD_SEND);

	shutdown(iLocal_Sock, SD_RECEIVE);

	//do we close remote sock?

	if (closesocket(iLocal_Sock) == SOCKET_ERROR)

		return E_SOCKETCLOSE;

	return ITS_OK;

}

//

// Method to check if we have IP adress or hostname, and return IP address

//

SOCKADDR_IN CTcpServer::getSockAddr( const char* aIp, const char* aPort)

{

	SOCKADDR_IN sockAddr;

	// check params

	assert( aIp != NULL );

	// check for null port

	if( aPort == NULL ) {

		aPort = "0";

	}

	// set the socket

	sockAddr.sin_family = AF_INET;		

	sockAddr.sin_port = htons( atoi(aPort) );

	sockAddr.sin_addr.S_un.S_addr = inet_addr( aIp );

	return sockAddr;

}

//

// Method to check if the adress is IP format or is a hostname string.

//

bool CTcpServer::isIPAdress(const char* aAddr)

{

	int c =0;

	char* q=(char *)aAddr;

	char* tokenPtr = strtok(q,".");

	while(tokenPtr != NULL)

	{

		tokenPtr = strtok(NULL,".");	

		c++;

	}

	if (c==4)

		return true; //We have IP adrress, nothing to do

	else 

		return false;

}

//

// Makes a connection, listens, and prints any incoming data.

//

int CTcpServer::makeConnection(SOCKET &aLocalSocket, SOCKET &aRemoteSocket, SOCKADDR_IN & aRemoteAddr)

{

	int remote_addr_len;

	int err = listen( aLocalSocket, 1 );

	if( err == SOCKET_ERROR ) 

	{

		err = WSAGetLastError ();

		printf ("Error listening to socket: %d", err);

		return E_LISTENING;

	}

	//wait for connection 

	while( 1 ) 

		{

			// Wait for the next connection

			printf( "\nWaiting for connection...\n" ); 

			remote_addr_len = sizeof(aRemoteAddr);

			aRemoteSocket = accept( aLocalSocket, (struct sockaddr*)&aRemoteAddr, &remote_addr_len );

			if( aRemoteSocket == INVALID_SOCKET ) 

				{

				err = WSAGetLastError();

				printf ("Accept failed. Error: %d",err);

				return E_ACCEPT;

				}

			// Connection received 

			printf( "Connection received from %d.%d.%d.%d : %d\n",

				aRemoteAddr.sin_addr.S_un.S_un_b.s_b1,

				aRemoteAddr.sin_addr.S_un.S_un_b.s_b2,

				aRemoteAddr.sin_addr.S_un.S_un_b.s_b3,

				aRemoteAddr.sin_addr.S_un.S_un_b.s_b4,

				aRemoteAddr.sin_port );

			// Read until the socket is closed and print out the info received.

			err = 1;

			while( err > 0 ) 

				{

				err = recv(aRemoteSocket, iPacket, RECVBUFFSIZE, 0 );

				if (err == SOCKET_ERROR)

					 return E_RECEIVING;

				ProcessBuffer(iPacket, err);

				err = send(aRemoteSocket, iPacket, err, 0 );

				if (err == SOCKET_ERROR)

					return E_SENDING;

				}

			// Close the socket and go again

			closesocket(aRemoteSocket);

	}

	return ITS_OK;

}

//

// Returns a a TCP/IP socket

//

SOCKET CTcpServer::getSocket()

{

	SOCKET sock;

	int err;

	// create a socket 

	sock = socket( AF_INET, SOCK_STREAM, 0 ) ;

	if( sock == INVALID_SOCKET)

	{

		//return, error has occured

		err = WSAGetLastError ();

		printf ("Allocating socket failed. Error: %d",err);

		return NULL;

	}

	return sock;

}

//

// Initialise use of the dll.

//

int CTcpServer::initialise()

{

	WORD version;

	WSADATA wsaData;

	// 1. start the windows iSocket server

	version = MAKEWORD( 2, 2 );

	if (WSAStartup( version, &wsaData ))

		return E_SOCKETSTARTUP;		

	return ITS_OK;		

}

//

// Release.

//

int CTcpServer::release()

{

	int err =WSACleanup();

	if ( err == SOCKET_ERROR)

		return E_SOCKETSHUTDOWN;

	return ITS_OK;

}

//

// Set the arguments

//

int CTcpServer::setArguments(char* li, char* lp)

{

	if(!lp)

	{

		printf("Error: Missing argument - lp argument required.\n");

		return E_INCORRECTARG;

	}

	iLocal_Sock = getSocket();

	iLocal_Addr = getSockAddr(li,lp );	

	connect(iLocal_Sock,iLocal_Addr);	//bind

	return ITS_OK;

}