// Copyright (c) 2002-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:

// Contain the implementation of the class for this null agent test

// 

//

#include "NullAgentTestSteps.h"

#include "dummynifvar.h"

#include "in_sock.h"

CTestStepNullAgtReconnect::CTestStepNullAgtReconnect(TPtrC aName)

{

	iTestStepName=aName;

}

enum TVerdict CTestStepNullAgtReconnect::doTestStepL(void)

{	

	__UHEAP_MARK;

	TRequestStatus status; // status of asynchronous ops

	TInt r;                // the result of various operations

	RSocketServ server;    // connection paraphanelia

	RConnection connection;

	RSocket commandSocket, dataSocket;

	TInetAddr commandDest;

	commandDest.SetAddress(KDummyNifLocalAddressBase + 4);

	commandDest.SetPort(KDummyNifCmdPort);

	TInetAddr dataDest;

	dataDest.SetAddress(KDummyNifLocalAddressBase + 4);

	dataDest.SetPort(KPortNo);

	TBuf8<KBufferLength> buffer;

	/* START THE CONNECTION WE WILL USE */

	r = server.Connect();

	TESTEL(r == KErrNone, r);

	CleanupClosePushL(server);

	r = connection.Open(server, KAfInet);

	TESTEL(r == KErrNone, r);

	CleanupClosePushL(connection);

	connection.Start(status);

	User::WaitForRequest(status);

	TESTEL(status.Int() == KErrNone, status.Int());

	/* OPEN 2 SOCKETS - ONE FOR COMMANDS AND ONE FOR TRAFFIC */

	// to send commands to the dummy nif we need to send pseudo-traffic which 

	// it will interpret as a command... hence all the socket stuff that follows

	// open a udp socket to send the command to the dummy nif

	r = commandSocket.Open(server, KAfInet, KSockDatagram, KProtocolInetUdp);

	TESTEL(r == KErrNone, r);

	CleanupClosePushL(commandSocket);

	TESTL(commandSocket.SetOpt(KSoReuseAddr, KSolInetIp, 1)==KErrNone);

	// set the source port number - otherwise will panic cos it's zero

	r = commandSocket.SetLocalPort(KDummyNifCmdPort);

	TESTEL(r == KErrNone, r);

	// open another socket to send data later on

	r = dataSocket.Open(server, KAfInet, KSockDatagram, KProtocolInetUdp);

	TESTEL(r == KErrNone, r);

	CleanupClosePushL(dataSocket);

	TESTL(dataSocket.SetOpt(KSoReuseAddr, KSolInetIp, 1)==KErrNone);

	r = dataSocket.SetLocalPort(KPortNo);

	TESTEL(r == KErrNone, r);

	/* SEND SOME LOOPBACK TRAFFIC */

	buffer.SetMax();

	buffer.FillZ();

	buffer[0] = (TUint8) 0x8;		// ICMP type = 8

	buffer[1] = (TUint8) 0x0;		// ICMP code = 0

	buffer[2] = (TUint8) 0xF7;		// ICMP checksum high byte

	buffer[3] = (TUint8) 0xFF;		// ICMP checksum low byte

	dataSocket.SendTo(buffer, dataDest, 0, status);

	User::WaitForRequest(status);

	TESTEL(status.Int() == KErrNone, status.Int());

	buffer.Zero();

  	dataSocket.RecvFrom(buffer, dataDest, 0, status);

  	User::WaitForRequest(status);

  	TESTEL(status.Int() == KErrNone, status.Int());

	/* SEND THE SPECIAL COMMAND PACKET TO THE DUMMY NIF TO FORCE A RECONNECT */

	buffer.SetMax();

	buffer.FillZ();

	buffer[0] = (TUint8) KForceReconnect;

	commandSocket.SendTo(buffer, commandDest, 0, status);

	User::WaitForRequest(status);

	TESTEL(status.Int() == KErrNone, status.Int());

	/* WAIT A WHILE TO LET THE RECONNECT HAPPEN (OR NOT IF DIALOG APPEARS AND USER DOESN'T RESPOND!) */

	User::After(10000000);

	/* SEND MORE LOOPBACK TRAFFIC ON THE DATA SOCKET */

	// the idea is that reconnect does not lead to an erroring of the sockets - the stack should 

	// not see that the rug was pulled out.

	buffer.SetMax();

	buffer.FillZ();

	buffer[0] = (TUint8) 0x8;		// ICMP type = 8

	buffer[1] = (TUint8) 0x0;		// ICMP code = 0

	buffer[2] = (TUint8) 0xF7;		// ICMP checksum high byte

	buffer[3] = (TUint8) 0xFF;		// ICMP checksum low byte

	dataSocket.SendTo(buffer, dataDest, 0, status);

	User::WaitForRequest(status);

	TESTEL(status.Int() == KErrNone, status.Int());

	buffer.Zero();

  	dataSocket.RecvFrom(buffer, dataDest, 0, status);

  	User::WaitForRequest(status);

  	TESTEL(status.Int() == KErrNone, status.Int());

	/* CLOSE EVERYTHING DOWN */

 	// close the data socket

	dataSocket.Shutdown(RSocket::ENormal, status);

	User::WaitForRequest(status);

	TESTEL(status.Int() == KErrNone, status.Int());

	CleanupStack::Pop(&dataSocket);

	// close the command socket

	commandSocket.Shutdown(RSocket::ENormal, status);

	User::WaitForRequest(status);

	TESTEL(status.Int() == KErrNone, status.Int());

	CleanupStack::Pop(&commandSocket);

	// force the destruction of the connection - if the reconnect hasn't completed yet it will be cancelled

	r = connection.Stop();

	TESTEL(r == KErrNone, r);

	CleanupStack::Pop(&connection);

	// close the socket server

	server.Close();

	CleanupStack::Pop(&server);

	__UHEAP_MARKEND;

	return iTestStepResult;

}