// Copyright (c) 2004-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:
// Contains IPC Test 02
//
//

// EPOC includes
#include <e32base.h>
#include <es_sock.h>
#include <ss_std.h>

// Test system includes
#include "Test06IPCTestSelect.h"


const TDesC& CTest06IPCTestSelect::GetTestName()
	{
	_LIT(ret,"IPCTest06");
	return ret;
	}

enum TVerdict CTest06IPCTestSelect::InternalDoTestStepL(void)
	{
	_LIT8(KTestMsg, "Hello");
	
	//-------------------substep 00-----------------------------	
	Logger().Write(_L("00 Open socket server:"));
	
	TInt err;
	RSocketServ ss;
	CleanupClosePushL(ss);
	if ( (err = OptimalConnect(ss)) != KErrNone )
		{
		Logger().WriteFormat(_L("Connecting to socket server failed with %d"), err);
		User::Leave(EFail);
		}

	//-------------------substep 01-----------------------------	
	Logger().Write(_L("01 Open server socket, set local port & set to listen:"));

	RSocket aClientSock, aNewConn, aServerSock;
	CleanupClosePushL(aClientSock);
	CleanupClosePushL(aServerSock);
	CleanupClosePushL(aNewConn);
	
	TRequestStatus aAcceptStat, aConnectStat;
	TSockAddr aAddr;
	
	if ( (err=aServerSock.Open(ss, iProt)) != KErrNone)
		{
		_LIT(aLog,"Error:Could not open server socket. err = %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
	
	if ( (err=aServerSock.SetLocalPort(257)) != KErrTooBig)
		{
		_LIT(aLog,"Error:Set illegal port returned %d instead of %d"); 
		Logger().WriteFormat(aLog, err, KErrTooBig);
		User::Leave(EFail);
		}
	
	if ( (err=aServerSock.SetLocalPort(1)) != KErrNone)
		{
		_LIT(aLog,"Error:Could not set port. err = %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}

	if ( (err=aServerSock.Listen(1)) != KErrNone)
		{
		_LIT(aLog,"Error:Could not set up socket to listen. err = %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
	
	//-------------------substep 02-----------------------------	
	Logger().Write(_L("02 Set server socket to non-blocking mode:"));
	if ( (err=aServerSock.SetOpt(KSONonBlockingIO, KSOLSocket)) != KErrNone)
		{
		_LIT(aLog,"Error:Could not set socket option. err = %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
	
	//-------------------substep 03-----------------------------	
	Logger().Write(_L("03 Open blank socket, assosiate it to server socket (with accept)...:"));
	Logger().Write(_L("   ...and check that return status is WOULD BLOCK:"));
	if ( (err=aNewConn.Open(ss)) != KErrNone)
		{
		_LIT(aLog,"Error:Could not open blank socket. err = %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
	aServerSock.Accept(aNewConn, aAcceptStat);
    User::WaitForRequest(aAcceptStat);
    err = aAcceptStat.Int();
    if (err != KErrWouldBlock)
		{
		_LIT(aLog,"Error:Accept method returned wrong value %d instead of %d"); 
		Logger().WriteFormat(aLog, err, KErrWouldBlock);
		User::Leave(EFail);
		}
	
	//-------------------substep 04-----------------------------	
	Logger().Write(_L("04 Assosiate local buffer as status buffer on server socket,..."));
	Logger().Write(_L("   ...with initial value READ + WRITE + EXCEPTION:"));
	TPckgBuf<TUint> aSbuf, aCbuf;
	aSbuf()=KSockSelectRead | KSockSelectWrite | KSockSelectExcept;
	aServerSock.Ioctl(KIOctlSelect, aAcceptStat, &aSbuf, KSOLSocket);
	
    //-------------------substep 05-----------------------------	
	Logger().Write(_L("05 Associate local buffer as status buffer on client socket,..."));
	Logger().Write(_L("   ...with initial value WRITE:"));

	// Create client socket.
	if ( (err=aClientSock.Open(ss, iProt)) != KErrNone)
		{
		_LIT(aLog,"Error:Could not open socket. err = %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
	// Set client socket to non-blocking mode.
	if ( (err=aClientSock.SetOpt(KSONonBlockingIO, KSOLSocket)) != KErrNone)
		{
		_LIT(aLog,"Error:Could not set socket option. err = %d");
		Logger().WriteFormat(aLog, err);
		aNewConn.Close();
		User::Leave(EFail);
		}
	
	// Connect client socket to server socket.
	aAddr.SetPort(1);
	aClientSock.Connect(aAddr, aConnectStat);
	User::WaitForRequest(aConnectStat);
	err = aConnectStat.Int();
    if ( (err != KErrNone) && (err != KErrWouldBlock))
		{
		_LIT(aLog,"Error:Connection method returned wrong value: %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}

	aCbuf()=KSockSelectWrite;
    aClientSock.Ioctl(KIOctlSelect, aConnectStat, &aCbuf, KSOLSocket);
	User::WaitForRequest(aConnectStat);
	err = aConnectStat.Int();
    if (err != KErrNone)
		{
		_LIT(aLog,"Error:Ioctl method returned wrong status: %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}

    //-------------------substep 06-----------------------------	
	Logger().Write(_L("06 Check client status buffer to be WRITE..."));
	Logger().Write(_L("   ...and server status buffer to be READ:"));

	if ( aCbuf() != KSockSelectWrite)
		{
		_LIT(aLog,"Error:Wrong value of buffer: %d");
		Logger().WriteFormat(aLog, aCbuf());
		User::Leave(EFail);
		}

    User::WaitForRequest(aAcceptStat);
	err = aAcceptStat.Int();
    if (err != KErrNone)
		{
		_LIT(aLog,"Error:Ioctl method returned wrong status: %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}

    if (aSbuf() != KSockSelectRead)
		{
		_LIT(aLog,"Error:Wrong value of buffer: %d");
		Logger().WriteFormat(aLog, aSbuf());
		User::Leave(EFail);
		}
	
	//-------------------substep 07-----------------------------	
	Logger().Write(_L("07 Assosiate blank socket to the server socket:"));
	
	// Re-open blank client socket.
	aNewConn.Close();
	if ( (err=aNewConn.Open(ss)) != KErrNone)
		{
		_LIT(aLog,"Error:Could not open socket. err = %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
		
	aServerSock.Accept(aNewConn, aAcceptStat);
	User::WaitForRequest(aAcceptStat);
	err = aAcceptStat.Int();
    if (err != KErrNone)
		{
		_LIT(aLog,"Error:Accept method returned wrong status: %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
	
	//-------------------substep 08-----------------------------	
	Logger().Write(_L("08 Chack that blank socket is in non blocking mode as well"));
	if ( (err=aNewConn.GetOpt(KSONonBlockingIO, KSOLSocket, aSbuf)) != KErrNone)
		{
		_LIT(aLog,"Error:Could not get socket option. err = %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
    if (!aSbuf())
		{
		_LIT(aLog,"Error:Wrong value of option: %d");
		Logger().WriteFormat(aLog, aSbuf());
		User::Leave(EFail);
		}
	
	//-------------------substep 09-----------------------------	
	Logger().Write(_L("09 Try to read on server side and check that it would block"));
	TRequestStatus aReadStat, aWriteStat;
	TBuf8<5> aReadbuf;
	aReadbuf.SetMax();
	aNewConn.Read(aReadbuf, aReadStat);
	User::WaitForRequest(aReadStat);
	err = aReadStat.Int();
    if (err != KErrWouldBlock)
		{
		_LIT(aLog,"Error:Read method returned wrong status: %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
	
	//-------------------substep 10-----------------------------	
	Logger().Write(_L("10 Assosiate local buffer as status buffer of blank socket,..."));
	Logger().Write(_L("   ...with initial value: READ + EXCEPTION"));
	aSbuf()=KSockSelectRead | KSockSelectExcept;
	aNewConn.Ioctl(KIOctlSelect, aAcceptStat, &aSbuf, KSOLSocket);
	
	//-------------------substep 11-----------------------------	
	Logger().Write(_L("11 Write a message on the client side, and chack that"));
	Logger().Write(_L("   ...status of the other side of connection is now: READ"));

	aClientSock.Write(KTestMsg, aWriteStat);
	User::WaitForRequest(aWriteStat);
	err = aWriteStat.Int();
    if (err != KErrNone)
		{
		_LIT(aLog,"Error:Write method returned wrong status: %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
	
	User::WaitForRequest(aAcceptStat);
	err = aAcceptStat.Int();
    if (err != KErrNone)
		{
		_LIT(aLog,"Error:Ioctl method returned wrong status: %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}

    if (aSbuf()!=KSockSelectRead)
		{
		_LIT(aLog,"Error:Wrong value of option: %d");
		Logger().WriteFormat(aLog, aSbuf());
		User::Leave(EFail);
		}

    //-------------------substep 12-----------------------------	
	Logger().Write(_L("12 Read the message on server side and check the content:"));
	aReadbuf.SetMax();
	
	aNewConn.Read(aReadbuf, aReadStat);
	User::WaitForRequest(aReadStat);
	err = aReadStat.Int();
    if (err != KErrNone)
		{
		_LIT(aLog,"Error:Read method returned wrong status: %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}

    if (aReadbuf != KTestMsg)
		{
		_LIT(aLog,"Error:Read/Write mismatch");
		Logger().Write(aLog);
		User::Leave(EFail);
		}
	
	//-------------------substep 13-----------------------------	
	Logger().Write(_L("13 Try to read again on server side and check that ..."));
	Logger().Write(_L("   ...the status of the action is: WOULD BLOCK"));
	aReadbuf.SetMax();

	aNewConn.Read(aReadbuf, aReadStat);
	User::WaitForRequest(aReadStat);
	err = aReadStat.Int();
    if (err != KErrWouldBlock)
		{
		_LIT(aLog,"Error:Read method returned wrong status: %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
	
	//-------------------substep 14-----------------------------	
	Logger().Write(_L("14 Write the message on the client side:"));
	aClientSock.Write(KTestMsg, aWriteStat);
	User::WaitForRequest(aWriteStat);
	err = aWriteStat.Int();
    if (err != KErrNone)
		{
		_LIT(aLog,"Error:Write method returned wrong status: %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
	
	//-------------------substep 15-----------------------------	
	Logger().Write(_L("15 Get SelectPoll option of the server side and check..."));
	Logger().Write(_L("   ...it is READ + WRITE"));

	if ( (err=aNewConn.GetOpt(KSOSelectPoll, KSOLSocket, aSbuf)) != KErrNone)
		{
		_LIT(aLog,"Error:Could not get socket option. err = %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
 
	if (aSbuf()!=(KSockSelectRead | KSockSelectWrite))
		{
		_LIT(aLog,"Error:Wrong value of option: %d");
		Logger().WriteFormat(aLog, aSbuf());
		User::Leave(EFail);
		}

	//-------------------substep 16-----------------------------	
	Logger().Write(_L("16 Assosiate local buffer as status buffer on server side,..."));
	Logger().Write(_L("   ...with initial value: READ + EXCEPTION. Chack that it stays READ only"));

	aSbuf()=KSockSelectRead | KSockSelectExcept;
	aNewConn.Ioctl(KIOctlSelect, aAcceptStat, &aSbuf, KSOLSocket);
	User::WaitForRequest(aAcceptStat);
	err = aAcceptStat.Int();
    if (err != KErrNone)
		{
		_LIT(aLog,"Error:Ioctl method returned wrong status: %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}

    if (aSbuf() != KSockSelectRead)
		{
		_LIT(aLog,"Error:Wrong value of option: %d");
		Logger().WriteFormat(aLog, aSbuf());
		User::Leave(EFail);
		}
	
	//-------------------substep 17-----------------------------	
	Logger().Write(_L("17 Read the message on server side and check the content:"));

	aReadbuf.SetMax();
	aNewConn.Read(aReadbuf, aReadStat);
	User::WaitForRequest(aReadStat);
	err = aReadStat.Int();
    if (err != KErrNone)
		{
		_LIT(aLog,"Error:Read method returned wrong status: %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}

    if (aReadbuf != KTestMsg)
		{
		_LIT(aLog,"Error:Read/Write mismatch");
		Logger().Write(aLog);
		User::Leave(EFail);
		}
	
	//-------------------substep 18-----------------------------	
	Logger().Write(_L("18 Close the server side. Access server side and the check that..."));
	Logger().Write(_L("   ...status of the action is DISCONNECTED:"));
	CleanupStack::PopAndDestroy(&aNewConn);
	aCbuf()=KSockSelectRead | KSockSelectExcept;
	
	aClientSock.Ioctl(KIOctlSelect, aConnectStat, &aCbuf, KSOLSocket);
	User::WaitForRequest(aConnectStat);
	err = aConnectStat.Int();
    if (err != KErrDisconnected)
		{
		_LIT(aLog,"Error:Ioctl method returned wrong status: %d");
		Logger().WriteFormat(aLog, err);
		User::Leave(EFail);
		}
	
	//-------------------substep 19-----------------------------	
	Logger().Write(_L("19 Close all socket and socket server:"));
	CleanupStack::PopAndDestroy(3, &ss);
	return EPass;
	}