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

//

#include <e32base.h>

#include <es_sock.h> // for BigEndian

#include "dumper.h"

_LIT(KUnixTimeBaseDes, "19700101:");

//1st phase of construction

CPktLogger::CPktLogger():iTimeOrigin(KUnixTimeBaseDes)

	{

	}

void CPktLogger::ConstructL()

/**

2nd stage construction of packet logger.

Leaves with an error if the logging file could not be created.

*/

	{

	User::LeaveIfError(iFileSvrSession.Connect());

	_LIT(KHookDumpFile, "C:\\logs\\HookLogs\\iphookslog.tcpdump");		

	TInt err = iFileObj.Open( iFileSvrSession, KHookDumpFile, EFileWrite|EFileShareAny );

	if ( err != KErrNone )

		{

		err = iFileObj.Create( iFileSvrSession, KHookDumpFile, EFileWrite|EFileShareAny );    

		if( err != KErrNone ) 

			{

			// We aren't going to do any logging, so close down the File Server session.

			iFileSvrSession.Close();

			User::Leave(err);

			}

		}

	DumpFileHeader();

	}

CPktLogger::~CPktLogger()

	{

	iFileObj.Close();

	iFileSvrSession.Close();

	}

CPktLogger* CPktLogger::NewL()

	{

	CPktLogger* loggerObj = new( ELeave ) CPktLogger;

	CleanupStack::PushL( loggerObj ); //A might-leave operation is to be followed - so give a chance to the system to do the cleanup for you.

	loggerObj->ConstructL();

	CleanupStack::Pop(); //loggerObj

	return loggerObj;

	}

CPktLogger* CPktLogger::NewLC()

	{

	CPktLogger* loggerObj = new( ELeave ) CPktLogger;

	CleanupStack::PushL( loggerObj ); //A might-leave operation is to be followed - so give a chance to the system to do the cleanup for you.

	loggerObj->ConstructL();

	return loggerObj;

	}

void CPktLogger::DumpFileHeader()

	{

	TUint8 buffer[ KDumpFileHeaderMaxSizeAllowed ];

	TUint8* ptr = 0;

	//Magic number = 4 bytes

	ptr = (TUint8*)&KTcpDumpMagicNumber;

	buffer[ 0 ] = ptr[ 0 ] ;

	buffer[ 1 ] = ptr[ 1 ] ;

	buffer[ 2 ] = ptr[ 2 ] ;

	buffer[ 3 ] = ptr[ 3 ] ;

	//Major version = 2 bytes

	ptr = (TUint8*)&KMajorVersion;

	buffer[ 4 ] = ptr[ 0 ] ;

	buffer[ 5 ] = ptr[ 1 ] ;

	//Minor version = 2 bytes

	ptr = (TUint8*)&KMinorVersion;

	buffer[ 6 ] = ptr[ 0 ] ;

	buffer[ 7 ] = ptr[ 1 ] ;

	//Time Zone = 4 bytes

	ptr = (TUint8*)&KThisZone;

	buffer[ 8 ] = ptr[ 0 ] ;

	buffer[ 9 ] = ptr[ 1 ] ;

	buffer[ 10 ] = ptr[ 2 ] ;

	buffer[ 11 ] = ptr[ 3 ] ;

	//File Length/ sigfigs??  = 4 bytes

	ptr = (TUint8*)&KSigfgs;

	buffer[ 12 ] = ptr[ 0 ] ;

	buffer[ 13 ] = ptr[ 1 ] ;

	buffer[ 14 ] = ptr[ 2 ] ;

	buffer[ 15 ] = ptr[ 3 ] ;

	//snaplen  = 4 bytes

	ptr = (TUint8*)&KSnapLen;

	buffer[ 16 ] = ptr[ 0 ] ;

	buffer[ 17 ] = ptr[ 1 ] ;

	buffer[ 18 ] = ptr[ 2 ] ;

	buffer[ 19 ] = ptr[ 3 ] ;

	// linktype = 4 bytes

	ptr = (TUint8*)&KLinkType;

	buffer[ 20 ] = ptr[ 0 ] ;

	buffer[ 21 ] = ptr[ 1 ] ;

	buffer[ 22 ] = ptr[ 2 ] ;

	buffer[ 23 ] = ptr[ 3 ] ;

	TPtr8 fileHdr( buffer, KDumpFileHeaderMaxSizeAllowed, KDumpFileHeaderMaxSizeAllowed );

	iFileObj.Write( fileHdr );

	}

void CPktLogger::DumpLibPCapRecordHeader( TUint32 aPktLenCaptured, TUint32 aActualPktLen )

	{

	union tag

		{

		TUint8 pktHdrBuffer[ KDumpFileRecordHeaderMaxSizeAllowed ];

		TUint32 iDummyByte; // To force alignment on 4-bytes boundary - imp on arm but not on x86!

		} buffer;

	TUint8* ptr = NULL;

	TTime timeNow;

	timeNow.UniversalTime();

	TTimeIntervalSeconds sec;

	timeNow.SecondsFrom(iTimeOrigin, sec);

	LittleEndian::Put32(&buffer.pktHdrBuffer[ 0 ], sec.Int());

	TUint captureTimeMicroSecs;

	captureTimeMicroSecs = I64LOW(timeNow.Int64());

	captureTimeMicroSecs = captureTimeMicroSecs % 1000000; // get microseconds component

	LittleEndian::Put32(&buffer.pktHdrBuffer[ 4 ], captureTimeMicroSecs);

	//a 32-bit value giving the number of bytes of packet data that were captured;

	ptr = ( ( TUint8* )( TAny* )&aPktLenCaptured );

	buffer.pktHdrBuffer[ 8 ]  = ptr[ 0 ];

	buffer.pktHdrBuffer[ 9 ]  = ptr[ 1 ];

	buffer.pktHdrBuffer[ 10 ] = ptr[ 2 ];

	buffer.pktHdrBuffer[ 11 ] = ptr[ 3 ];

	//a 32-bit value giving the actual length of the packet, in bytes

	//(which may be greater than the previous number, if you're not

	//saving the entire packet).

	ptr = ( ( TUint8* )( TAny* )&aActualPktLen ) ;

	buffer.pktHdrBuffer[ 12 ] = ptr[ 0 ];

	buffer.pktHdrBuffer[ 13 ] = ptr[ 1 ];

	buffer.pktHdrBuffer[ 14 ] = ptr[ 2 ];

	buffer.pktHdrBuffer[ 15 ] = ptr[ 3 ];

	TPtrC8 recordHdr(buffer.pktHdrBuffer, KDumpFileRecordHeaderMaxSizeAllowed);

	iFileObj.Write( recordHdr );

	return;

	}

void CPktLogger::DumpLibPCapRecordData( const TDesC8& aPacket )

	{

	iFileObj.Write( aPacket );

	}