// t_wsp.cpp
// 
// Copyright (c) 2002 - 2010 Accenture. All rights reserved.
// This component and the accompanying materials are made available
// under the terms of the "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:
// Accenture - Initial contribution
//

#include <e32std.h>
#include <e32base.h>
#include <e32test.h>
#include <e32svr.h>
#include <e32math.h>
#include <f32file.h>
#include <es_sock.h>
#include <in_sock.h>


//
// Constants.
//

//#define _HOME
#ifdef _HOME
const TInt KWebPort = 80;
const TUint32 KSymbianWebServerIpAddress = INET_ADDR(192, 168, 0, 240);
const TInt KLocalHostPort = 9999;
const TUint32 KLocalHostIpAddress = INET_ADDR(127, 0, 0, 1);
_LIT(KSymbianWebServerName, "main");
_LIT(KAltSymbianWebServerName, "sunshare.office");
_LIT8(KSmallWebPage,"GET http://sunshare.office/index.html \r\n\r\n");
_LIT(KLogFile, "c:\\t_wsp.txt");
#else
const TInt KWebPort = 80;
const TUint32 KSymbianWebServerIpAddress = INET_ADDR(81, 89, 143, 203);
const TInt KLocalHostPort = 9999;
const TUint32 KLocalHostIpAddress = INET_ADDR(127, 0, 0, 1);
_LIT(KSymbianWebServerName, "www.symbian.com");
_LIT(KAltSymbianWebServerName, "www.symbian.com");
_LIT8(KSmallWebPage,"GET http://www.symbian.com/index.html \r\n\r\n");
_LIT(KLogFile, "c:\\t_wsp.txt");
#endif

const TInt KTestTransferDataSize = 8192;

//
// Globals.
//

RFs FileServerSession;
RFile LogFile;
RSocketServ SocketServerSession;
RTest TestConsole(_L("t_wsp"));


//
// Functions.
//

void RandomFill(TDes8& aBuf)
	{
	while (aBuf.Length() < (aBuf.MaxLength() - TInt(sizeof(TUint32))))
		{
		TUint32 rand = Math::Random();
		TPckgC<TUint32> randPckg(rand);
		aBuf.Append(randPckg);
		}
	while (aBuf.Length() < (aBuf.MaxLength()))
		{
		aBuf.Append(_L("p")); // Pack out last few bytes.
		}
	}

void ReceiveHtmlL(RSocket& aSock)
	{
	HBufC8* webPage = HBufC8::NewMaxLC(4096);

	TPtr8 des(webPage->Des());
	des = KSmallWebPage;

	TRequestStatus stat;
	aSock.Write(*webPage, stat);
	User::WaitForRequest(stat);
	User::LeaveIfError(stat.Int());

	TSockXfrLength HTTPHeaderLen;
	TInt rcount = 0;

	TBuf<128> infoPrintBuf;
	FOREVER
		{
		des.Zero();
		aSock.RecvOneOrMore(des,0,stat,HTTPHeaderLen);
		User::WaitForRequest(stat);
		if (stat != KErrNone)
			break;
		
		LogFile.Write(des);
		LogFile.Flush();
		rcount += webPage->Length();
		infoPrintBuf.Format(_L("\rRecved:%10d Bytes"), rcount);
		TestConsole.Printf(infoPrintBuf);
		}
	if (stat != KErrEof)
		{
		User::Leave(stat.Int());
		}

	CleanupStack::PopAndDestroy(webPage);
	}

void TcpOpenAndConnectC(RSocket& aSocket)
	{
	RHostResolver resolver;
	TestConsole(resolver.Open(SocketServerSession, KAfInet, KProtocolInetTcp) == KErrNone);
	CleanupClosePushL(resolver);
	TNameEntry nameEntry;
	TestConsole(resolver.GetByName(KSymbianWebServerName, nameEntry) == KErrNone);
	TNameRecord nameRecord(nameEntry());
	TInetAddr addr(TInetAddr::Cast(nameRecord.iAddr).Address(), KWebPort);
	CleanupStack::PopAndDestroy(); // resolver.
	TestConsole(aSocket.Open(SocketServerSession, KAfInet, KSockStream, KProtocolInetTcp) == KErrNone);
	CleanupClosePushL(aSocket);
	TRequestStatus stat;
	aSocket.Connect(addr, stat);
	User::WaitForRequest(stat);
	TestConsole(stat.Int() == KErrNone);
	}

void GetByAddress()
	{
	RHostResolver resolver;
	TestConsole(resolver.Open(SocketServerSession, KAfInet, KProtocolInetTcp) == KErrNone);
	CleanupClosePushL(resolver);
	TNameEntry nameEntry;
	TInetAddr testAddr(KSymbianWebServerIpAddress, KWebPort);
	testAddr.SetFamily(KAfInet);
	TestConsole(resolver.GetByAddress(testAddr, nameEntry) == KErrNone);
	TNameRecord nameRecord(nameEntry());
	TestConsole(nameRecord.iName.CompareF(KAltSymbianWebServerName) == KErrNone);
	CleanupStack::PopAndDestroy(); // resolver.
	}

void TcpActiveOpen()
	{
	RSocket socket;
	TcpOpenAndConnectC(socket);
	TRAPD(err, ReceiveHtmlL(socket));
	TestConsole(err == KErrNone);
	CleanupStack::PopAndDestroy(); // socket.
	}

void TcpGetLocalName()
	{
	RSocket socket;
	TcpOpenAndConnectC(socket);
	TSockAddr localName;
	socket.LocalName(localName);
	CleanupStack::PopAndDestroy(); // socket.
	}

void TcpGetRemoteName()
	{
	RSocket socket;
	TcpOpenAndConnectC(socket);
	TSockAddr remoteName;
	socket.RemoteName(remoteName);
	CleanupStack::PopAndDestroy(); // socket.
	}

void TcpListenAccept()
	{
	// Create a listening server socket and a socket to accept connections.
	RSocket listeningSocket;
	TestConsole(listeningSocket.Open(SocketServerSession, KAfInet, KSockStream, KProtocolInetTcp) == KErrNone);
	TInetAddr addr(KLocalHostIpAddress, KLocalHostPort);
	TestConsole(listeningSocket.Bind(addr) == KErrNone);
	TestConsole(listeningSocket.Listen(1) == KErrNone);
	RSocket serverConnectionSocket;
	TestConsole(serverConnectionSocket.Open(SocketServerSession) == KErrNone);
	TRequestStatus acceptStatus;
	listeningSocket.Accept(serverConnectionSocket, acceptStatus);

	// Create a client socket.
	RSocket clientSocket;
	TestConsole(clientSocket.Open(SocketServerSession, KAfInet, KSockStream, KProtocolInetTcp) == KErrNone);
	TRequestStatus connectStatus;
	clientSocket.Connect(addr,connectStatus);
	User::WaitForRequest(connectStatus);
	User::WaitForRequest(acceptStatus);
	TestConsole(connectStatus.Int() == KErrNone);
	TestConsole(acceptStatus.Int() == KErrNone);

	// Write some data from the client to the server.
	HBufC8* out = HBufC8::NewLC(KTestTransferDataSize);
	TPtr8 outPtr(out->Des());
	RandomFill(outPtr);
	TRequestStatus writeStatus;
	TRequestStatus readStatus;
	HBufC8* in = HBufC8::NewLC(KTestTransferDataSize);
	TPtr8 inPtr(in->Des());
	clientSocket.Write(*out, writeStatus);
	serverConnectionSocket.Read(inPtr, readStatus);
	User::WaitForRequest(writeStatus);
	User::WaitForRequest(readStatus);
	TestConsole(writeStatus.Int() == KErrNone);
	TestConsole(readStatus.Int() == KErrNone);
	TestConsole(in->Compare(*out) == KErrNone);
	clientSocket.Close();
	serverConnectionSocket.Close();

	// Re-connect the client socket.
	TestConsole(clientSocket.Open(SocketServerSession, KAfInet, KSockStream, KProtocolInetTcp) == KErrNone);
	TestConsole(serverConnectionSocket.Open(SocketServerSession) == KErrNone);
	listeningSocket.Accept(serverConnectionSocket, acceptStatus);
	clientSocket.Connect(addr,connectStatus);
	User::WaitForRequest(connectStatus);
	User::WaitForRequest(acceptStatus);
	TestConsole(connectStatus.Int() == KErrNone);
	TestConsole(acceptStatus.Int() == KErrNone);
	listeningSocket.Close();

	// Re-sent the data.
	inPtr.Zero();
	clientSocket.Read(inPtr, readStatus);
	serverConnectionSocket.Write(*out, writeStatus);
	User::WaitForRequest(writeStatus);
	User::WaitForRequest(readStatus);
	TestConsole(writeStatus.Int() == KErrNone);
	TestConsole(readStatus.Int() == KErrNone);
	TestConsole(in->Compare(*out) == KErrNone);

	// Cleanup.
	clientSocket.Close();
	listeningSocket.Close();
	CleanupStack::PopAndDestroy(2, out);
	}

void UdpSendReceive()
	{
	// Create a socket to receive data.
	RSocket receiveSocket;
	TestConsole(receiveSocket.Open(SocketServerSession, KAfInet, KSockDatagram, KProtocolInetUdp) == KErrNone);
	TInetAddr receiveAddress(KLocalHostIpAddress, KLocalHostPort);
	TestConsole(receiveSocket.Bind(receiveAddress) == KErrNone);
	TBuf8<32> receiveBuf;
	receiveBuf.SetMax();
	receiveBuf.Fill('a');
	TRequestStatus receiveStatus;
	TInetAddr sendAddress;
	receiveSocket.RecvFrom(receiveBuf, sendAddress, 0, receiveStatus);

	// Create a socket to send data.
	RSocket sendSocket;
	TestConsole(sendSocket.Open(SocketServerSession, KAfInet, KSockDatagram, KProtocolInetUdp) == KErrNone);
	TBuf8<32> sendBuf;
	sendBuf.SetMax();
	sendBuf.Fill('b');
	TRequestStatus sendStatus;
	sendSocket.SendTo(sendBuf, receiveAddress, 0, sendStatus);

	// Wait and check.
	User::WaitForRequest(receiveStatus);
	User::WaitForRequest(sendStatus);
	TestConsole(receiveStatus.Int() == KErrNone);
	TestConsole(sendStatus.Int() == KErrNone);
	TestConsole(receiveBuf.Compare(sendBuf) == KErrNone);

	// Cleanup.
	receiveSocket.Close();
	sendSocket.Close();
	}

void GetSetOption()
	{
	RSocket socket;
	TcpOpenAndConnectC(socket);
	TInt val;
	TestConsole(socket.GetOpt(KSoTcpNoDelay, KSolInetTcp, val) == KErrNone);
	val = !val;
	TestConsole(socket.SetOpt(KSoTcpNoDelay, KSolInetTcp, val) == KErrNone);
	TInt newVal;
	TestConsole(socket.GetOpt(KSoTcpNoDelay, KSolInetTcp, newVal) == KErrNone);
	TestConsole(val == newVal);
	TRAPD(err, ReceiveHtmlL(socket));
	TestConsole(err == KErrNone);
	CleanupStack::PopAndDestroy(); // socket.
	}

TInt InitialiseLC()
	{
	User::LeaveIfError(FileServerSession.Connect());
	CleanupClosePushL(FileServerSession);
	User::LeaveIfError(LogFile.Replace(FileServerSession, KLogFile,EFileShareAny | EFileWrite | EFileStreamText));
	CleanupClosePushL(LogFile);
	User::LeaveIfError(SocketServerSession.Connect());
	CleanupClosePushL(SocketServerSession);
	return 3;
	}

GLDEF_C void RunTestsL()
	{
	TInt numItemsOnCleanupStack = InitialiseLC();
//	GetByAddress();
	TcpActiveOpen();
	TcpGetLocalName();
	TcpGetRemoteName();
	TcpListenAccept();
	UdpSendReceive();
	GetSetOption();
	CleanupStack::PopAndDestroy(numItemsOnCleanupStack);
	TestConsole.Close();
	}

GLDEF_C TInt E32Main()
	{
	__UHEAP_MARK;
	CActiveScheduler* scheduler = new CActiveScheduler;
	if (scheduler)
		{
		CActiveScheduler::Install(scheduler);
		CTrapCleanup* cleanup = CTrapCleanup::New();
		if (cleanup)
			{
			TRAPD(err, RunTestsL());
			delete cleanup;
			}
		delete scheduler;
		}
	__UHEAP_MARKEND;
	return KErrNone;
	}