// 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:
// This contains ESock Test cases from section 16
// Testing read from socket with KSockReadPeek flag set
// 
//


#include <e32base.h>

#include "SocketTestSection16.h"


// Test step 16.1
const TDesC& CSocketTest16_1::GetTestName()
	{
	_LIT(ret,"Test16.1");
	return ret;
	}

enum TVerdict CSocketTest16_1::InternalDoTestStepL( void )
	{
	Logger().WriteFormat(_L("Testing read from socket with KSockReadPeek flag set in different situations."));
	
	RSocketServ sockServ;
	
	RSocket     sock_UdpIn;
    RSocket     sock_UdpOut;
    TInt        nRes;
	
    _LIT8(KSendData, "ABCDEFGHIJKLMNOPQRSTUVWXYZ"); // Sample testdata to send in these tests
	
    CleanupClosePushL(sockServ);
    CleanupClosePushL(sock_UdpIn);
    CleanupClosePushL(sock_UdpOut);
	
    Logger().WriteFormat(_L("connecting to the socket server..."));
    nRes = sockServ.Connect();
    TESTL(nRes == KErrNone);
	
    //-- open UDP datagram sockets 
	Logger().WriteFormat(_L("opening sockets..."));
    TESTL(sock_UdpIn.Open (sockServ, KAfInet, KSockDatagram, KProtocolInetUdp) == KErrNone);
	TESTL(sock_UdpOut.Open(sockServ, KAfInet, KSockDatagram, KProtocolInetUdp) == KErrNone);
	
    TRequestStatus  rqStat_in, rqStat_out;
	
    //-- local host address, port #7 - the data will be sent via this address
    TInetAddr       inetAddrTo(7);
    TInetAddr       inetAddrFrom;
	
    TBuf8<100>      tempBuf, tempBuf1;
    
    const  TInt     KIPHeaderSize = 20; //-- IP header size
    const  TInt     KUdpHeaderSize= 8;  //-- UDP header size
	
	
    inetAddrTo.Input(_L("127.0.0.2"));
    sock_UdpIn.Bind(inetAddrTo); //-- bind socket to address
	
	
    //###################################################################################################
    //--- Test normal situation with peeking data from the UDP socket, when the data arrived before peeking
    //###################################################################################################
	
	
    //--- 1. send the data to the udp socket
    sock_UdpOut.SendTo(KSendData, inetAddrTo, 0, rqStat_out);
	User::WaitForRequest(rqStat_out);
	Logger().WriteFormat(_L("UDP sent %d bytes of data, error code:%d\n"), KSendData().Size(), rqStat_out.Int());
    TESTL(rqStat_out.Int() == KErrNone);
	
	
    //--- 2. Peek data from UDP socket and check the size (taking into account the KIpHeaderIncluded flag)
    sock_UdpIn.RecvFrom(tempBuf, inetAddrFrom, KIpHeaderIncluded|KSockReadPeek, rqStat_in);
	User::WaitForRequest(rqStat_in);
	Logger().WriteFormat(_L("UDP peeked %d bytes of data, error code:%d\n"), tempBuf.Size(), rqStat_in.Int());
    TESTL(rqStat_in.Int() == KErrNone);
    TESTL(tempBuf.Size() == KIPHeaderSize + KUdpHeaderSize + KSendData().Size());
	
    //--- 3. Peek data from UDP socket and check the size (taking into account the KIpHeaderIncluded flag)
    sock_UdpIn.RecvFrom(tempBuf1, inetAddrFrom, KIpHeaderIncluded|KSockReadPeek, rqStat_in);
	User::WaitForRequest(rqStat_in);
	Logger().WriteFormat(_L("UDP peeked %d bytes of data, error code:%d\n"), tempBuf1.Size(), rqStat_in.Int());
    TESTL(rqStat_in.Int() == KErrNone);
    TESTL(tempBuf1.Size() == KIPHeaderSize + KUdpHeaderSize + KSendData().Size());
	
    //-- check, the data shall be the same
    TESTL( tempBuf.CompareF(tempBuf1) ==0);
	
    //--- 4. read data from socket and check the size
	
	sock_UdpIn.RecvFrom(tempBuf, inetAddrFrom, 0, rqStat_in);
	User::WaitForRequest(rqStat_in);
	Logger().WriteFormat(_L("UDP read real %d bytes data from the socket, error code:%d\n"), tempBuf.Size(), rqStat_in.Int());
    TESTL(rqStat_in.Int() == KErrNone);
    TESTL(tempBuf.Size() == KSendData().Size());
	
    //###################################################################################################
    //--- Test the situation with peeking data from the UDP socket, when the data arrived after read request.
    //###################################################################################################
	
	
    //--- 1. Issue a read request from the socket with KSockReadPeek flag set
    Logger().WriteFormat(_L("Issuing reading request with KSockReadPeek flag set"));
    sock_UdpIn.RecvFrom(tempBuf, inetAddrFrom, KIpHeaderIncluded|KSockReadPeek, rqStat_in);
	
	//--- 2. send data 
	sock_UdpOut.SendTo(KSendData, inetAddrTo, 0, rqStat_out);
	User::WaitForRequest(rqStat_out);
	Logger().WriteFormat(_L("UDP sent %d bytes of data, error code:%d\n"), KSendData().Size(), rqStat_out.Int());
	TESTL(rqStat_out.Int() == KErrNone);
	
	
    //--- 3. Wait for peeking data to complete, check the result
	User::WaitForRequest(rqStat_in);
	Logger().WriteFormat(_L("UDP peeked %d bytes of data, error code:%d\n"), tempBuf.Size(), rqStat_in.Int());
    TESTL(rqStat_in.Int() == KErrNone);
    TESTL(tempBuf.Size() == KIPHeaderSize + KUdpHeaderSize + KSendData().Size());
	
    
    sock_UdpIn.RecvFrom(tempBuf1, inetAddrFrom, KIpHeaderIncluded|KSockReadPeek, rqStat_in);
	User::WaitForRequest(rqStat_in);
	Logger().WriteFormat(_L("UDP peeked %d bytes of data, error code:%d\n"), tempBuf1.Size(), rqStat_in.Int());
    TESTL(rqStat_in.Int() == KErrNone);
    TESTL(tempBuf1.Size() == KIPHeaderSize + KUdpHeaderSize + KSendData().Size());
	
    //-- check, the data shall be the same
    TESTL( tempBuf.CompareF(tempBuf1) ==0);
	
    //--- 4. read data from socket and check the size
	
	sock_UdpIn.RecvFrom(tempBuf, inetAddrFrom, 0, rqStat_in);
	User::WaitForRequest(rqStat_in);
	Logger().WriteFormat(_L("UDP read real %d bytes data from the socket, error code:%d\n"), tempBuf.Size(), rqStat_in.Int());
    TESTL(rqStat_in.Int() == KErrNone);
    TESTL(tempBuf.Size() == KSendData().Size());
	
	
    //###################################################################################################
    //--- Test the situation with peeking data from the UDP socket cancellation
    //###################################################################################################
	
    //--- 1. Issue a read request from the socket with KSockReadPeek flag set
    Logger().WriteFormat(_L("Issuing reading request with KSockReadPeek flag set"));
    sock_UdpIn.RecvFrom(tempBuf, inetAddrFrom, KIpHeaderIncluded|KSockReadPeek, rqStat_in);
    
    Logger().WriteFormat(_L("Cancel receiving"));
    sock_UdpIn.CancelRecv();
	
	//--- 2. send data 
	sock_UdpOut.SendTo(KSendData, inetAddrTo, 0, rqStat_out);
	User::WaitForRequest(rqStat_out);
	Logger().WriteFormat(_L("UDP sent %d bytes of data, error code:%d\n"), KSendData().Size(), rqStat_out.Int());
	TESTL(rqStat_out.Int() == KErrNone);
	
	
    //--- 3. Wait for peeking data to complete, check the result
	User::WaitForRequest(rqStat_in);
	Logger().WriteFormat(_L("UDP peeked %d bytes of data, error code:%d\n"), tempBuf.Size(), rqStat_in.Int());
    TESTL((rqStat_in.Int() == KErrCancel) || (rqStat_in.Int() == KErrNone));
    
    //--- 4. Peek the data
    sock_UdpIn.RecvFrom(tempBuf, inetAddrFrom, KIpHeaderIncluded|KSockReadPeek, rqStat_in);
	User::WaitForRequest(rqStat_in);
	Logger().WriteFormat(_L("UDP peeked %d bytes of data, error code:%d\n"), tempBuf1.Size(), rqStat_in.Int());
    TESTL(rqStat_in.Int() == KErrNone);
	
	
	
	//-- close and destroy everything
	CleanupStack::PopAndDestroy(3); //sockServ, sock_UdpIn, sock_UdpOut
	
	SetTestStepResult(EPass);
	return EPass;
	}