/*

* Copyright (c) 2009-2010 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: Packet probe hook

*

*/

#include <f32file.h>

#include <e32svr.h>

_LIT(KTcpDumpFolder, "tcpdump\\");

_LIT(KProbeDumpFile, "probe.cap");

_LIT(KLogFolder,     "c:\\logs\\");

CProtocolProbe::CProtocolProbe(TUint aId) : 

iId(aId), 

iDumpCb(CActive::EPriorityStandard),

iFileServerOpen(EFalse),

iFileOpen(EFalse),

iBufCreated(EFalse)

	iTimeOrigin.UniversalTime();

	TCallBack cbFunc(DumpCb, this);

	iDumpCb.Set(cbFunc);

	LOG(Log::Printf(_L("Probe::~CProtocolProbe()\r\n"));)	

	if(iFileOpen)

	    {

		iFile.Close();

		iFileOpen = EFalse;

	    }

	if(iFileServerOpen)

		{

		iFs.Close();

		iFileServerOpen = EFalse;

		}	

    if(iBufCreated)

        {

        iBuf.Close();

        iBufCreated = EFalse;

        }

		{

		}

		{

		}

	// initialise dump file

	if(iFileOpen)

		{

		iFile.Close();

		iFileOpen = EFalse;

		}

	if(iFileServerOpen)

		{

		iFs.Close();

		iFileServerOpen = EFalse;

		}

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

    iFileServerOpen = ETrue;

    TBuf<50> filename;

    filename.Append(KLogFolder);

    filename.Append(KTcpDumpFolder);

    filename.Append(KProbeDumpFile);

	User::LeaveIfError(iFile.Replace(iFs, filename, EFileWrite));

    iFileOpen = ETrue;

    // allocate buffer

    if(!iBufCreated)

        {

		const TInt KMaxBufferSize = 65535;

        iBuf.CreateL(KMaxBufferSize);

        iBufCreated = ETrue;

        }

	LibcapDumpFileHeader();

void CProtocolProbe::Queue(RMBufChain &aPacket)

//  Takes a copy of aPacket, adds it to the queue and triggers the callback

	RMBufPacketBase copy;

	TRAPD(err, copy.CopyPackedL(aPacket));

	if (err == KErrNone)

		{

		//

		// Calculate packet time-stamp

		//

		TTime newTime;

		newTime.UniversalTime();

		TTimeIntervalMicroSeconds interval = newTime.MicroSecondsFrom(iTimeOrigin);

		const TInt KMicrosInASecond = 1000000;

		//TUint32 micros = interval.Int64().Low();

		TUint32 micros = I64LOW(interval.Int64());//.Low();  x.Low() -> I64LOW(x)

		TUint32 secs = micros / KMicrosInASecond;

		micros -= (secs * KMicrosInASecond);

		//

		// Reuse the protocol and flags fields of

		// RMBufPktInfo to store the time-stamp

		//

		RMBufPktInfo* info = RMBufPacket::PeekInfoInChain(copy);

		info->iProtocol = static_cast<TInt>(secs);

		info->iFlags = static_cast<TUint>(micros);

		iQueue.Append(copy);

		iDumpCb.CallBack();

		}

	else

		copy.Free();

//

//	CProtocolProbe::Dump

//	***********************

//	Log the packet to file

//

void CProtocolProbe::Dump(RMBufChain &aPacket)

	{

	RMBufPacket packet;

	packet.Assign(aPacket);

	RMBufPktInfo* info = packet.Unpack();

	TUint32 secs = static_cast<TUint32>(info->iProtocol);

	TUint32 micros = static_cast<TUint32>(info->iFlags);

    iBuf.SetMax();

    packet.CopyOut(iBuf, 0);

    if(iBuf.Length() != info->iLength)

        {

        iBuf.SetLength(info->iLength);

        }

	LibcapDump(iBuf, secs, micros);

	packet.Free();

	}

	Queue(aPacket);

	Queue(aPacket);

void CProtocolProbe::LibcapDumpFileHeader()

//

//  Dump file header in a format compatible with Libcap

//

//  Format is:

//  

//  struct FileHeader

//      {

//	    TUint32 magic; 

//	    TUint16 version_major;

//	    TUint16 version_minor;

//	    TUint32 thiszone;	/* gmt to local correction */

//	    TUint32 sigfigs;	/* accuracy of timestamps */

//	    TUint32 snaplen;	/* max length saved portion of each pkt */

//	    TUint32 linktype;	/* data link type (LINKTYPE_*) */

//      };

//

//  Note LINKTYPE specified in libpcap/bpf/net/bpf.h (see www.tcpdump.org)

//

	{

	TBuf8<sizeof(TUint32)*5+sizeof(TUint16)*2> fileHeader;

	*((TUint32*) &(fileHeader.Ptr()[0])) = 0xa1b2c3d4;

	*((TUint16*) &(fileHeader.Ptr()[4])) = 0x02; 

	*((TUint16*) &(fileHeader.Ptr()[6])) = 0x04; 

	*((TUint32*) &(fileHeader.Ptr()[8])) = 0x00;

	*((TUint32*) &(fileHeader.Ptr()[12])) = 0x00;

	*((TUint32*) &(fileHeader.Ptr()[16])) = 0xffff;

	*((TUint32*) &(fileHeader.Ptr()[20])) = 12;       // DLT_RAW		12	/* raw IP */

	fileHeader.SetLength(fileHeader.MaxLength());

	iFile.Write(fileHeader);

	}

void CProtocolProbe::LibcapDump(const TDesC8& aBuffer, TUint32 aTimeStampSecs, TUint32 aTimeStampMicros)

//

//  Dumps a packet in a format compatbible with Libcap

//

//  For each record the format is:

//

//  struct record 

//      {

//      TUint32 sec;	/* time stamp - secs*/

//      TUint32 usec;	/* time stamp - microsecs*/

//      TUint32 captureLen;	/* length packet captured */

//      TUint32 packetLen;	/* total length of packet*/

//      };

//

//  Byte ordering of the header is little endian

//  Byte ordering of the packet is network byte order (big endian)

//

	{

	TBuf8<sizeof(TUint32)*4> recordHeader;

	recordHeader.FillZ();

	*((TUint32*) &(recordHeader.Ptr()[0])) = aTimeStampSecs;

	*((TUint32*) &(recordHeader.Ptr()[4])) = aTimeStampMicros;

	*((TUint32*) &(recordHeader.Ptr()[8])) = aBuffer.Length();

	*((TUint32*) &(recordHeader.Ptr()[12])) = aBuffer.Length();

	recordHeader.SetLength(recordHeader.MaxLength());

	iFile.Write(recordHeader);

	iFile.Write(aBuffer);

	}

TInt CProtocolProbe::DumpCb(TAny* aThisPtr)

//

//  Callback function - calls DumpQueuedPackets()

//

	{

	CProtocolProbe* self = static_cast<CProtocolProbe*>(aThisPtr);

	self->DumpQueuedPackets();

	return KErrNone;

	}

void CProtocolProbe::DumpQueuedPackets()

//

//  Dumps all packets on the queue to the log file

//

	{

	RMBufPacketBase packet;

	while(iQueue.Remove(packet))

		{

		Dump(packet);

		}

	}