// Copyright (c) 2003-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 "qoslib.h"

#include "qoslib_glob.h"

#include "pfqos_stream.h"

#ifdef SYMBIAN_ENABLE_SPLIT_HEADERS

#include <networking/qoslib_internal.h>

#endif

/**

Default constructor. 

RQoSPolicy::Open() must always be called before any other 

methods in the RQoSPolicy can be called. 

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

*/

EXPORT_C RQoSPolicy::RQoSPolicy() : iPolicy(NULL)

	{

	}

/**

Destructor. 

Closes any open policy. 

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

*/

EXPORT_C RQoSPolicy::~RQoSPolicy()

	{

	if (iPolicy)

		iPolicy->Close();

	}

/**

This must always be called before any other function can be used. 

It specifies the selector for a QoS policy. 

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aSelector Selector for the QoS policy.

@return KErrNone, if successful; otherwise, an error code if the QoS channel 

cannot be opened. 

*/

EXPORT_C TInt RQoSPolicy::Open(const TQoSSelector& aSelector)

	{

	if (iPolicy)

		return KErrAlreadyExists;

	CQoSMan* manager = NULL;

	TRAPD(err,

		manager = CQoSMan::NewL();

		// There is manager->Open() in above NewL.

		// How and when is that cancelled, if leave happens below? --msa

		iPolicy = manager->OpenQoSPolicyL(aSelector)

		);

	return err;

	}

/**

Sets the QoS parameters for the QoS policy. 

A CQoSAddEvent event is received asynchronously to indicate the success or 

failure of the request. 

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aPolicy QoS parameters.

@return KErrNone, if successful; otherwise, an error code. 

*/

EXPORT_C TInt RQoSPolicy::SetQoS(CQoSParameters& aPolicy)

	{

	if (!iPolicy)

		return KErrNotReady;

	TRAPD(err, iPolicy->SetQoSL(aPolicy));

	return err;

	}

/**

Gets the QoS policy from QoS policy database. 

A CQoSGetEvent event is received asynchronously to indicate the success or 

failure of the request.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@return KErrNone, if successful; otherwise, an error code. 

*/

EXPORT_C TInt RQoSPolicy::GetQoS()

	{

	if (!iPolicy)

		return KErrNotReady;

	TRAPD(err, iPolicy->GetQoSL());

	return err;

	}

/**

Deletes the QoS policy.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@return KErrNone, if successful; otherwise, an error code (e.g. if 

RQoSPolicy::Open() was not called).

*/

EXPORT_C TInt RQoSPolicy::Close()

	{

	if (!iPolicy)

		return KErrNotReady;

	TRAPD(err, iPolicy->DeleteL());

	if (err == KErrNone)

		{

		iPolicy->Close();

		iPolicy = NULL;

		}

	return err;

	}

/**

Registers an event observer to catch QoS events.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aObserver Event observer.

@param aMask An event mask. An application can specify a set of QoS events 

that it wants to receive. By default all events are notified to the 

application. See TQoSEvent enumerations.

@return KErrNone, if successful; otherwise, an error code. 

*/

EXPORT_C TInt RQoSPolicy::NotifyEvent(MQoSObserver& aObserver, TUint aMask)

	{

	if (!iPolicy)

		return KErrNotReady;

	return iPolicy->NotifyEvent(aObserver, aMask);

	}

/**

Deregisters an event observer to catch QoS events. 

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aObserver Event observer.

@return KErrNone if successful, otherwise an error code. 

*/

EXPORT_C TInt RQoSPolicy::CancelNotifyEvent(MQoSObserver& aObserver)

	{

	if (!iPolicy)

		return KErrNotReady;

	return iPolicy->CancelNotifyEvent(aObserver);

	}

/**

Loads a QoS policy file into the QoS policy database. 

A TQoSEvent event (EQoSEventLoadPolicyFile) is received asynchronously to 

indicate the success or failure of the request. 

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aName Name of the QoS policy file to be loaded.

@return KErrNone, if successful; otherwise, an error code. 

*/

EXPORT_C TInt RQoSPolicy::LoadPolicyFile(const TDesC& aName)

	{

	if (!iPolicy)

		return KErrNotReady;

	TRAPD(err, iPolicy->LoadFileL(aName));

	return err;

	}

/**

Unloads a QoS policy file from the QoS policy database. 

A TQoSEvent event (EQoSEventUnloadPolicyFile) is received asynchronously to 

indicate the success or failure of the request. 

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aName Name of the QoS policy file to be unloaded.

@return KErrNone, if successful; otherwise, an error code. 

*/

EXPORT_C TInt RQoSPolicy::UnloadPolicyFile(const TDesC& aName)

	{

	if (!iPolicy)

		return KErrNotReady;

	TRAPD(err, iPolicy->UnloadFileL(aName));

	return err;

	}

//

CQoSRequestBase::~CQoSRequestBase()

	{

	iManager->ClearPendingRequest(this);

	iManager->Close();

	}

void CQoSRequestBase::ParseExtensions(TPfqosMessage& aMsg, CQoSParameters& aPolicy)

	{

	aMsg.SetQoSParameters(aPolicy.iQoS);

	TSglQueIter<CPfqosPolicyData> iter(aMsg.iExtensions);

	CPfqosPolicyData *data;

	while ((data = iter++) != NULL)

		{

		TInt extensionType;

		TInt ret = GetExtensionType(data->Data(), extensionType);

		if (ret == KErrNone)

			{

			CExtensionBase *ext = aPolicy.FindExtension(extensionType);

			if (ext)

				ext->ParseMessage(data->Data());

			}

		}

	}

TInt CQoSRequestBase::GetExtensionType(const TDesC8& aData, TInt& aType)

	{

	const TUint8 *p = aData.Ptr();

	TInt length = aData.Length();

	//lint -e{826} thinks this cast is suspicious -- is not!

	struct pfqos_configure *ext = (struct pfqos_configure *) p;

	if (length < (TInt)sizeof(pfqos_configure))

		return KErrGeneral;	 // EMSGSIZE (impossible message size)

	if (ext->pfqos_configure_len * 8 != length)

		return KErrGeneral;	 // EMSGSIZE (incorrect message length)

	if (ext->pfqos_ext_type != EPfqosExtExtension)

		return KErrGeneral;

	p += sizeof(struct pfqos_configure);

	//lint -e{826} thinks this cast is suspicious -- is not!

	pfqos_extension *extensionType = (pfqos_extension *) p;

	aType = extensionType->pfqos_extension_type;

	return KErrNone;

	}

TInt CQoSRequestBase::NotifyEvent(MQoSObserver& aObserver, TUint aMask)

	{

	if (aMask & ~KQoSEventAll)

		return KErrNotSupported;

	iEventMask = aMask;

	iObserver = &aObserver;

	return KErrNone;

	}

TInt CQoSRequestBase::CancelNotifyEvent(MQoSObserver& aObserver)

	{

	if (iObserver != &aObserver)

		return KErrNotFound;

	iObserver = NULL;

	return KErrNone;

	}

//

CPolicy* CPolicy::NewL(CQoSMan* aManager, const TQoSSelector& aSelector)

	{

	CPolicy* policy = new (ELeave) CPolicy(aManager, aSelector);

	return policy;

	}

CPolicy::CPolicy(CQoSMan* aManager, const TQoSSelector& aSelector)

	{

	iManager = aManager;

	iSelector = aSelector;

	iPending = ENone;

	iPolicyCreated = EFalse;

	}

CPolicy::~CPolicy()

	{

	iManager->RemoveQoSPolicy(this);

	}

void CPolicy::Close()

	{

	delete this;

	}

TBool CPolicy::Match(const TQoSSelector& aSelector)

	{

	return (iSelector == aSelector);

	}

void CPolicy::SetQoSL(CQoSParameters& aPolicy)

	{

	if (iPending != ENone)

		User::Leave(KErrInUse);

	iPolicy.CopyL(aPolicy);

	CRequest* request = CRequest::NewL(this, KQoSDefaultBufSize);

	if (iPolicyCreated)

		{

		request->iMsg->Init(EPfqosUpdate);

		iPending = EPendingUpdate;

		}

	else

		{

		request->iMsg->Init(EPfqosAdd);

		iPending = EPendingAdd;

		}

	request->iMsg->AddSelector((TUint8)iSelector.Protocol(), 

							   iManager->Uid().UidType(), 

							   EPfqosFlowspecPolicy, 

							   iSelector.IapId(), 

							   EPfqosApplicationPriority, 

							   TPtr(0,0));

	request->iMsg->AddSrcAddress(iSelector.GetSrc(), 

								 iSelector.GetSrcMask(), 

								 (TUint16)iSelector.MaxPortSrc()); 

	request->iMsg->AddDstAddress(iSelector.GetDst(), 

								 iSelector.GetDstMask(), 

								 (TUint16)iSelector.MaxPortDst()); 

	request->iMsg->AddQoSParameters(iPolicy.iQoS);

	TQoSExtensionQueueIter iter(iPolicy.Extensions());

	CExtensionBase *extension;

	while ((extension=iter++) != NULL)

		request->iMsg->AddExtensionPolicy(extension->Data());

	iManager->Send(request);

	}

void CPolicy::GetQoSL()

	{

	if (iPending != ENone)

		User::Leave(KErrInUse);

	CRequest* request = CRequest::NewL(this, KQoSDefaultBufSize);

	request->iMsg->Init(EPfqosGet);

	iPending = EPendingGet;

	request->iMsg->AddSelector((TUint8)iSelector.Protocol(), 

							   iManager->Uid().UidType(), 

							   EPfqosFlowspecPolicy, 

							   iSelector.IapId(), 

							   EPfqosApplicationPriority, 

							   TPtr(0,0));

	request->iMsg->AddSrcAddress(iSelector.GetSrc(), 

								 iSelector.GetSrcMask(), 

								 (TUint16)iSelector.MaxPortSrc()); 

	request->iMsg->AddDstAddress(iSelector.GetDst(), 

								 iSelector.GetDstMask(), 

								 (TUint16)iSelector.MaxPortDst()); 

	TQoSExtensionQueueIter iter(iPolicy.Extensions());

	CExtensionBase *extension;

	while ((extension=iter++) != NULL)

		request->iMsg->AddExtensionPolicy(extension->Data());

	iManager->Send(request);

	}

void CPolicy::DeleteL()

	{

	CRequest* request = CRequest::NewL(this,KQoSDefaultBufSize);

	request->iMsg->Init(EPfqosDelete);

	iPending = EPendingDelete;

	request->iMsg->AddSelector((TUint8)iSelector.Protocol(), 

							   iManager->Uid().UidType(), 

							   EPfqosFlowspecPolicy, 

							   iSelector.IapId(), 

							   EPfqosApplicationPriority, 

							   TPtr(0,0));

	request->iMsg->AddSrcAddress(iSelector.GetSrc(), 

								 iSelector.GetSrcMask(), 

								 (TUint16)iSelector.MaxPortSrc()); 

	request->iMsg->AddDstAddress(iSelector.GetDst(), 

								 iSelector.GetDstMask(), 

								 (TUint16)iSelector.MaxPortDst()); 

	TQoSExtensionQueueIter iter(iPolicy.Extensions());

	CExtensionBase *item;

	while ((item = iter++) != NULL)

		request->iMsg->AddExtensionHeader((TUint16)item->Type());

	iManager->Send(request);

	}

void CPolicy::ProcessReply(TPfqosMessage& aMsg)

	{

	TInt aErrorCode = aMsg.iBase.iMsg->pfqos_msg_errno;

	TPendingStatus status = iPending;

	if (aErrorCode)

		{

		if (status == EPendingLoadFile || status == EPendingUnloadFile)

			{

			iPending = ENone;

#if _UNICODE

			TPtrC8 tmp((TUint8*)aMsg.iConfigFile.iExt->filename);

			TFileName filename;

			filename.Copy(tmp);

#else

			TPtrC8 filename((TUint8*)aMsg.iConfigFile.iExt->filename);

#endif

			TQoSEvent event_type;

			if (status == EPendingLoadFile)

				event_type = EQoSEventLoadPolicyFile;

			else

				event_type = EQoSEventUnloadPolicyFile;

			CQoSLoadEvent event(event_type, aErrorCode, filename);

			if (iObserver && iEventMask & EQoSEventLoadPolicyFile)

				iObserver->Event(event);

			}

		else

			{

			NotifyError(aErrorCode);

			}

		}

	else

		{

		iPending = ENone;

		switch (status)

			{

		case EPendingAdd:

			{

			CQoSParameters policy;

			ParseExtensions(aMsg, policy);

			CQoSAddEvent event(&policy, aErrorCode);

			if (iObserver && iEventMask & EQoSEventAddPolicy)

				iObserver->Event(event);

			iPolicyCreated = ETrue;

			}

			break;

		case EPendingUpdate:

			{

			CQoSParameters policy;

			ParseExtensions(aMsg, policy);

			CQoSAddEvent event(&policy, aErrorCode);

			if (iObserver && iEventMask & EQoSEventAddPolicy)

				iObserver->Event(event);

			}

			break;

		case EPendingDelete:

			{

			iPending = ENone;

			CQoSDeleteEvent event(aErrorCode);

			if (iObserver && iEventMask & EQoSEventDeletePolicy)

				iObserver->Event(event);

			}

			break;

		case EPendingGet:

			{

			CQoSParameters policy;

			ParseExtensions(aMsg, policy);

			CQoSGetEvent event(&policy, aErrorCode);

			if (iObserver && iEventMask & EQoSEventGetPolicy)

				iObserver->Event(event);

			}

			break;

		case EPendingLoadFile:

		case EPendingUnloadFile:

			{

#if _UNICODE

			TPtrC8 tmp((TUint8*)aMsg.iConfigFile.iExt->filename);

			TFileName filename;

			filename.Copy(tmp);

#else

			TPtrC8 filename((TUint8*)aMsg.iConfigFile.iExt->filename);

#endif

			TQoSEvent event_type;

			if (status == EPendingLoadFile)

				event_type = EQoSEventLoadPolicyFile;

			else

				event_type = EQoSEventUnloadPolicyFile;

			CQoSLoadEvent event(event_type, KErrNone, filename);

			if (iObserver && iEventMask & EQoSEventLoadPolicyFile)

				iObserver->Event(event);

			}

			break;

		default:

			break;

			}

		}

	}

TBool CPolicy::MatchReply(const TPfqosMessage& aMsg, TUint8 aMsgType)