// 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:

// engine.cpp - protocol analyzer engine

//

#include <es_sock.h>

#include <e32base.h>

#include <nifmbuf.h>

#include <ip4_hdr.h>

#include <ip6_hdr.h>

#include <icmp6_hdr.h>

#include <tcp_hdr.h>

#include <udp_hdr.h>

#include <ext_hdr.h>

#include "iprotor.h"

#include "engine.h"

CRotorEngine::CRotorEngine(CRotorAppView *aAppView):CTimer(EPriorityStandard)

{

	iAppView = aAppView;

	CActiveScheduler::Add(this);

}

CRotorEngine::~CRotorEngine()

{

	if (IsActive())

		Cancel();

	if (iReceiver)

		delete iReceiver;

	if (iIPv6Receiver)

		delete iIPv6Receiver;

	if (iDumper)

		delete iDumper;

	if (iBraker)

		delete iBraker;

	iSockServ.Close();

#if EPOC_SDK >= 0x07010000

	// Here would go the functionality that you would need

	// to close the opened link or links

#else

	xnetdial.Stop();

	xnetdial.Close();

#endif

}

void CRotorEngine::ConstructL(const TPreferences& aPref)

{

	//Base class 2nd phase constructor

	CTimer::ConstructL();

	iShowPackets=EFalse;	//Tells the engine to avoid using the console

	//General Options

	iPref = aPref;

	//iProtocol=IP;	//numbered like the list in options dialog

	//iNumBlades=DEFAULT_NUM_BLADES;

	iFactor=1;

	iPacketReceived=EFalse;

//	iViewIPHdr=ETrue;

#if EPOC_SDK >= 0x07010000

	// Here would go the functionality that you would need

	// to open a link or links

#else

	if (xnetdial.Open()!=KErrNone)

	{

		iAppView->Status(R_ROTOR_NET_OPEN_ERROR);	

	}

	else

	{

		if (xnetdial.DisableTimers()!=KErrNone)

			iAppView->Status(R_ROTOR_NET_DISABLE_ERROR);

	}

#endif

	TInt err = iSockServ.Connect();	//KESockDefaultMessageSlots

	TProtocolDesc protInfo;

	if (err != KErrNone)

	{

		iAppView->Write(_L("Socket Server Error (Connect)"));

		User::Leave(err);

	}

	err = iSockServ.FindProtocol(_L("probe"), protInfo);

	if (err!=KErrNone)

		{

		iAppView->Write(_L("Protocol Probe (hook) Not available"));

		iProbeActive = EFalse;

		}

	else

		iProbeActive = ETrue;

	err = iSockServ.FindProtocol(_L("ip"),protInfo);

	if (err!=KErrNone)

	{

		iAppView->Write(_L("Protocol IPv4 Not available"));

		iIPv4Active = EFalse;

		iPref.iDumpIPv4 = EFalse;

	}

	else

	{

		//iPref.iDumpIPv4 = ETrue;

		iIPv4Active = ETrue;

	}

	err = iSockServ.FindProtocol(_L("ip6"),protInfo);

	if (err!=KErrNone)

	{

		iAppView->Write(_L("Protocol IPv6 Not available"));

		iIPv6Active = EFalse;

		iPref.iDumpIPv6 = EFalse;

	}

	else

	{

		//iPref.iDumpIPv6 = ETrue;

		iIPv6Active = ETrue;

	}

	err = iSockServ.FindProtocol(_L("secpol"),protInfo);

	if (err!=KErrNone)

	{

		iAppView->Write(_L("Protocol IPSEC Not available"));

		iIPSECActive = EFalse;

		iPref.iDumpIPSEC = EFalse;

	}

	else

	{

		//iPref.iDumpIPSEC = ETrue;

		iIPSECActive = ETrue;

	}

	//IPv4 Monitoring data

	//IP

	iMonIPv4Info.iIPVersion=ETrue;

	iMonIPv4Info.iIPHdrLen=ETrue;

	iMonIPv4Info.iIPTOS=ETrue;

	iMonIPv4Info.iIPTotalLen=ETrue;

	iMonIPv4Info.iIPId=ETrue;

	iMonIPv4Info.iIPFlags=ETrue;

	iMonIPv4Info.iIPOffset=ETrue;

	iMonIPv4Info.iIPTTL=ETrue;

	iMonIPv4Info.iIPProtocol=ETrue;

	iMonIPv4Info.iIPChksum=ETrue;

	iMonIPv4Info.iIPSrcAddr=ETrue;

	iMonIPv4Info.iIPDstAddr=ETrue;

	//ICMP

	iMonIPv4Info.iICMPType=ETrue;

	iMonIPv4Info.iICMPCode=ETrue;

	iMonIPv4Info.iICMPChksum=ETrue;

	//TCP

	iMonIPv4Info.iTCPSrcPort=ETrue;

	iMonIPv4Info.iTCPDstPort=ETrue;

	iMonIPv4Info.iTCPSeq=ETrue;

	iMonIPv4Info.iTCPAckNum=ETrue;

	iMonIPv4Info.iTCPHdrLen=ETrue;

	iMonIPv4Info.iTCPFlags=ETrue;

	iMonIPv4Info.iTCPHdrWinSize=ETrue;

	iMonIPv4Info.iTCPChksum=ETrue;

	iMonIPv4Info.iTCPHdrUrgPtr=ETrue;

	//UDP

	iMonIPv4Info.iUDPSrcPort=ETrue;

	iMonIPv4Info.iUDPDstPort=ETrue;

	iMonIPv4Info.iUDPLen=ETrue;

	iMonIPv4Info.iUDPChksum=ETrue;

	//AH

	iMonIPv4Info.iAHProtocol=ETrue;

	iMonIPv4Info.iAHHdrLen=ETrue;

	iMonIPv4Info.iAHSPI=ETrue;

	iMonIPv4Info.iAHSeq=ETrue;

	//ESP

	iMonIPv4Info.iESPSPI=ETrue;

	iMonIPv4Info.iESPSeq=ETrue;

	//IPv6 Monitoring data

	//IP

	iMonIPv6Info.iIPVersion=ETrue;

	iMonIPv6Info.iIPTraffic=ETrue;

	iMonIPv6Info.iIPFlowLabel=ETrue;

	iMonIPv6Info.iIPPayloadLen=ETrue;

	iMonIPv6Info.iIPNextHdr=ETrue;

	iMonIPv6Info.iIPHopLimit=ETrue;

	iMonIPv6Info.iIPSrcAddr=ETrue;

	iMonIPv6Info.iIPDstAddr=ETrue;

	//ICMP

	iMonIPv6Info.iICMPType=ETrue;

	iMonIPv6Info.iICMPCode=ETrue;

	iMonIPv6Info.iICMPChksum=ETrue;

	iMonIPv6Info.iICMPParameter=ETrue;

	//TCP

	iMonIPv6Info.iTCPSrcPort=ETrue;

	iMonIPv6Info.iTCPDstPort=ETrue;

	iMonIPv6Info.iTCPSeq=ETrue;

	iMonIPv6Info.iTCPAckNum=ETrue;

	iMonIPv6Info.iTCPHdrLen=ETrue;

	iMonIPv6Info.iTCPFlags=ETrue;

	iMonIPv6Info.iTCPHdrWinSize=ETrue;

	iMonIPv6Info.iTCPChksum=ETrue;

	iMonIPv6Info.iTCPHdrUrgPtr=ETrue;

	iMonIPv6Info.iTCPOptions=ETrue;

	//UDP

	iMonIPv6Info.iUDPSrcPort=ETrue;

	iMonIPv6Info.iUDPDstPort=ETrue;

	iMonIPv6Info.iUDPLen=ETrue;

	iMonIPv6Info.iUDPChksum=ETrue;

	//HopByHop Hdr

	iMonIPv6Info.iHOPNextHdr=ETrue;

	iMonIPv6Info.iHOPHdrExtLen=ETrue;

	iMonIPv6Info.iHOPOptionType=ETrue;

	iMonIPv6Info.iHOPOptionLen=ETrue;

	//Dest Opt Hdr

	iMonIPv6Info.iDSTNextHdr=ETrue;

	iMonIPv6Info.iDSTHdrExtLen=ETrue;

	iMonIPv6Info.iDSTHomeAddr=ETrue;

	iMonIPv6Info.iDSTBindingUpdate=ETrue;

	iMonIPv6Info.iDSTBindingRequest=ETrue;

	iMonIPv6Info.iDSTBindingAck=ETrue;

	iMonIPv6Info.iDSTPad=ETrue;

	iMonIPv6Info.iDSTUnknown=ETrue;

	//Routing Header

	iMonIPv6Info.iRTNextHdr=ETrue;

	iMonIPv6Info.iRTHdrExtLen=ETrue;

	iMonIPv6Info.iRTRoutingType=ETrue;

	iMonIPv6Info.iRTSegLeft=ETrue;

	iMonIPv6Info.iRTSLBitMap=ETrue;

	iMonIPv6Info.iRTAddresses=ETrue;

	//Fragment Hdr

	iMonIPv6Info.iFRAGNextHdr=ETrue;

	iMonIPv6Info.iFRAGFragOffset=ETrue;

	iMonIPv6Info.iFRAGMFlag=ETrue;

	iMonIPv6Info.iFRAGId=ETrue;

	//AH

	iMonIPv6Info.iAHProtocol=ETrue;

	iMonIPv6Info.iAHHdrLen=ETrue;

	iMonIPv6Info.iAHSPI=ETrue;

	iMonIPv6Info.iAHSeq=ETrue;

	//ESP

	iMonIPv6Info.iESPSPI=ETrue;

	iMonIPv6Info.iESPSeq=ETrue;

	//iRunning=EFalse;

}

void CRotorEngine::GetPreferences(TPreferences &aPref) const

{

	aPref = iPref;

}

void CRotorEngine::DefaultPreferences(TPreferences &aPref)

{

	aPref.iDumpIPv4 = ETrue;

	aPref.iDumpIPv6 = EFalse;

	aPref.iDumpIPSEC = EFalse;

	aPref.iProtocol = IP;	//numbered like the list in options dialog

	aPref.iPort = 0;	//Any port

	aPref.iViewIPHdr = ETrue;

	aPref.iNumBlades = DEFAULT_NUM_BLADES;

}

//Issues next RunL execution

void CRotorEngine::IssueRequest()

{

	if (iFactor<0)

		iFactor=0.5;

	//After((TInt)(iFactor*SECOND));	//Also sets the object as Active

	//TInt time=iFactor*SECOND;

	//TTimeIntervalMicroSeconds32 interval(time);

	After((TInt)iFactor*SECOND);	//Also sets the object as Active

}

/*

//Issues last RunL execution

void CPingSender::IssueLastRequest()

{

	After(iPingModel->iLastSecWait*SECOND);	//Also sets the object as Active

}

*/

void CRotorEngine::PacketReceived()

{

	iPacketReceived=ETrue;

	//iRecvPackets++;

	iStatInfo.iTotalPackets++;

	iPartialPackets++;		//Used only for speed calculation. Is reset when Rotor is not moving.

	IncreaseSpeed();

	iBraker->ReIssueRequest();

}

void CRotorEngine::FirstRun()

{

	iFactor=MAX_FACTOR;		//Factor reset

	After(1);		//Also sets the object as Active.

	if (!IsActive())

		SetActive();

	//IssueRequest();	//First RunL

}

// will send all the packets. One packet each Time

void CRotorEngine::RunL()

{

	/*

	if (!iReceiver)

		iAppView->Write(_L("-iReceiver DEAD"));

	if (!iBraker)

		iAppView->Write(_L("-iBraker DEAD"));

	if (!iBraker->IsActive())

		iAppView->Write(_L("-iBraker NOT ACTIVE"));

	if (!iBraker->IsAdded())

		iAppView->Write(_L("-iBraker NOT ADDED"));

	if (iBraker->iStatus!=KRequestPending)

		iAppView->Write(_L("-iBraker NOT Pending"));

	if (!iReceiver->IsActive())

		iAppView->Write(_L("-iReceiver NOT ACTIVE"));

	if (!iReceiver->IsAdded())

		iAppView->Write(_L("-iReceiver NOT ADDED"));

*/

	/*

	if (iReceiver->iStatus!=KRequestPending)

		iAppView->Write(_L("-iReceiver NOT Pending"));

	*/

	if (iFactor<MAX_FACTOR)

	{

		iAppView->UpdateRotor();

		UpdateSpeedL();

	}

	IssueRequest();	//First RunL

}

//Cancel Packet Sending

void CRotorEngine::DoCancel()

{

	CTimer::DoCancel();

	//Deque();	//Dequeues the active object of the Scheduler

}

void CRotorEngine::SetNetworkUp() const

{

#if EPOC_SDK >= 0x07010000

	// Here would go the functionality you would

	// need to open a link or links

#else

	TBuf<50> msg;

	TNifProgress progress;

	xnetdial.Progress(progress);

	TInt err=progress.iError;

	if (err!=KErrNone)

	{

		msg.Format(_L("Network error: "));

		iAppView->ShowError(msg, err);

		iAppView->UpdateNetwork(EFalse);

		Stop();

		return;

	}

	if (progress.iStage!=EConnectionOpen)	//If net not UP

	{

#if EPOC_SDK >= 0x06010000

		err = xnetdial.StartOutgoing();

		xnetdial.DisableTimers();

#elif EPOC_SDK

		err=xnetdial.StartDialOut();

#else

		err=xnetdial.StartDialOut();

#endif

		if ((err!=KErrNone) && (err!=KErrAlreadyExists))

		{

			msg.Format(_L("Network error: "));

			iAppView->ShowError(msg, err);

			iAppView->UpdateNetwork(EFalse);

			Stop();

			return;

		}

	}

	else

		iAppView->UpdateNetwork(ETrue);

#endif

}

CRotorReceiver *CRotorEngine::StartReceiver(const TDesC &aName)

	{

	CRotorReceiver *receiver = new CRotorReceiver(iAppView, this);	//Receives incoming packets

	if (receiver == NULL)

		return NULL;

	TRAPD(err, receiver->ConstructL(iPref.iProtocol, aName));

	if (err != KErrNone)

		{

		delete receiver;

		return NULL;

		}

	receiver->Start();

	return receiver;

	}

void CRotorEngine::StartL()

{

	iStatInfo.iTotalPackets=0;

	iStatInfo.iIPv4Packets=0;

	iStatInfo.iIPv6Packets=0;

	iStatInfo.iTCPPackets=0;

	iStatInfo.iUDPPackets=0;

	iStatInfo.iICMPPackets=0;

	iStatInfo.iExtPackets=0;

	FirstRun();

	SetNetworkUp();

	TInt err;

	if (iPref.iDumpIPv4 || iPref.iDumpIPv6)

	{

		if (iProbeActive)

			iReceiver = StartReceiver(_L("probe"));

		// If probe is not available, start ip6 and ip on raw sockets...

		if (!iReceiver)

			{

			if (iPref.iDumpIPv6)

				iIPv6Receiver = StartReceiver(_L("ip6"));

			if (iPref.iDumpIPv4)

				iReceiver = StartReceiver(_L("ip"));

			}

		if (iReceiver == NULL && iIPv6Receiver == NULL)

			{

			iPref.iDumpIPv4 = EFalse;

			iPref.iDumpIPv6 = EFalse;

			}

	}

	if (iPref.iDumpIPSEC)

	{

		iDumper= new (ELeave) CRotorReceiver(iAppView,this);	//Receives incoming packets

		TRAP(err,iDumper->ConstructL(iPref.iProtocol, _L("secpol")));

		if (err==KErrNone)

			iDumper->Start();

		else

		{

			delete iDumper;

			iDumper = NULL;

			iPref.iDumpIPSEC = EFalse;

		}

	}

	if (!iPref.iDumpIPv4 && !iPref.iDumpIPv6 && !iPref.iDumpIPSEC)

	{

		//iRunning = EFalse;

		iAppView->Write(_L("No traffic to Dump! (Use Options to Set it)"));

		User::Leave(1);

		// NOTREACHED

		return;

	}

	iBraker= new (ELeave) CRotorBraker(this);	//Brakes the rotor gradually

	iBraker->ConstructL();

	iBraker->Start();

	//iRunning=ETrue;

}

void CRotorEngine::Stop()

{

	iFactor=1;

	//iRecvPackets=0;

	iStatInfo.iTotalPackets=0;

	if (IsActive())

		Cancel();

	//Stops hearing from all sockets

	delete iReceiver;

	iReceiver=NULL;

	delete iIPv6Receiver;

	iIPv6Receiver=NULL;

	delete iDumper;

	iDumper=NULL;

	delete iBraker;

	iBraker=NULL;