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

const TUint KMaxPort = 65535;

//

// TQoSSelector

//

/**

Constructor. 

Initialises all variables to specified, but null, values. 

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

*/

//lint -e{1927} would like some of the following to be in initializer list!

EXPORT_C TQoSSelector::TQoSSelector()

	{

	iSrc.SetAddress(KInet6AddrNone);

	iDst.SetAddress(KInet6AddrNone);

	iSrcMask.SetAddress(KInet6AddrNone);

	iDstMask.SetAddress(KInet6AddrNone);

	iProtocol = 0;

	iSrcPortMax = 0;

	iDstPortMax = 0;

	iIapId = 0;

	}

/** 

Compares if the selectors are equal.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aSelector TQoSSelector object that is compared with this object.

@return ETrue, if selectors are equal to this object; otherwise, EFalse. 

*/

EXPORT_C TInt TQoSSelector::operator==(const TQoSSelector& aSelector) const

	{

	if (iProtocol == aSelector.Protocol() && iSrcPortMax == aSelector.MaxPortSrc() &&

		iDstPortMax == aSelector.MaxPortDst() && iIapId == (TUint)aSelector.IapId() &&

		iSrc.CmpAddr(aSelector.GetSrc()) && iDst.CmpAddr(aSelector.GetDst()))

		return TRUE;

	else

		return FALSE;

	}

/** 

Compares if the selector matches aSocket. 

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aSocket RSocket that is compared with this object.

@return ETrue, if selector created from aSocket is equal to this object; 

otherwise, EFalse. 

*/

EXPORT_C TBool TQoSSelector::Match(RSocket& aSocket) const

	{

	TQoSSelector sel;

	TInt ret = sel.SetAddr(aSocket);

	if (ret!=KErrNone)

		return EFalse;

	return (*this == sel);

	}

/**

Sets the addresses for selector.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aSrcAddr Source address. Port must have value <= 65535 

(0 is used as uspecified value for a port).

@param aSrcAddrMask Source address mask. Port must have value <= 65535 

(0 is used as uspecified value for a port).

@param aDstAddr Destination address. Port must have value <= 65535 

(0 is used as uspecified value for a port).

@param aDstAddrMask Destination address mask. Port must have value <= 65535

(0 is used as uspecified value for a port).

@param aProtocol Protocol ID.

@param aSrcPortMax Maximum source port. Must have value <= 65535 

(0 is used as uspecified value for a port).

@param aDstPortMax Maximum destination port. Must have value <= 65535 

(0 is used as uspecified value for a port).

@return KErrNone, if parameters have valid values; otherwise, KErrArgument.

*/

EXPORT_C TInt TQoSSelector::SetAddr(const TInetAddr& aSrcAddr, 

									const TInetAddr& aSrcAddrMask, 

									const TInetAddr& aDstAddr, 

									const TInetAddr& aDstAddrMask, 

									TUint aProtocol, 

									TUint aSrcPortMax, 

									TUint aDstPortMax)

	{

	if (aSrcAddr.Port() > KMaxPort || aDstAddr.Port() > KMaxPort || 

		aSrcPortMax > KMaxPort || aDstPortMax > KMaxPort)

		return KErrArgument;

	iSrc = aSrcAddr;

	iDst = aDstAddr;

	iSrcMask = aSrcAddrMask;

	iDstMask = aDstAddrMask;

	iProtocol = aProtocol;

	iSrcPortMax = aSrcPortMax;

	iDstPortMax = aDstPortMax;

	iIapId = 0;

	return KErrNone;

	}

//lint -e{708} does not like union initializers

const TIp6Addr KInet6AddrMask = {{{0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,

								  0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff}}};

//

// RSocket must be connected before this!!

//

/**

Sets the addresses for selector. 

RSocket is used to fetch addresses and ports for a selector. 

The resulting selector will match only for one socket. 

Note that RSocket::Connect() must be called before calling this function.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aSocket RSocket object that is used to set the selector variables. 

Note: RSocket must be connected

@return KErrNone, if parameters have valid values; otherwise, KErrArgument.

*/

EXPORT_C TInt TQoSSelector::SetAddr(RSocket& aSocket)

	{

	TSockAddr aRemoteAddr, aLocalAddr;

	TProtocolDesc aDesc;

	aSocket.LocalName(aLocalAddr);

	aSocket.RemoteName(aRemoteAddr);

	if (aLocalAddr.Port() < 1 || aLocalAddr.Port() > KMaxPort)

		return KErrArgument;

	if (aRemoteAddr.Port() < 1 || aRemoteAddr.Port() > KMaxPort)

		return KErrArgument;

	if (iDst.Family() == KAFUnspec)

		return KErrArgument;

	iSrc = TInetAddr(aLocalAddr);

	iDst = TInetAddr(aRemoteAddr);

	// Srcaddr is not used when a selector is made from RSocket: 

	// srcaddr might not be set in Connect()

	iSrc.SetFamily(KAFUnspec);

	iSrc.SetAddress(KInet6AddrNone);

	if (iDst.Family() == KAfInet)

		iDst.ConvertToV4Mapped();

	//

	// Set prefixes.

	//

	iSrcMask.SetAddress(KInet6AddrMask);

	iDstMask.SetAddress(KInet6AddrMask);

	iSrcPortMax = iSrc.Port();

	iDstPortMax = iDst.Port();

	aSocket.Info(aDesc);

	iProtocol = aDesc.iProtocol;

	return KErrNone;

	}

/**

Sets the source address for selector.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aAddr Source address. Port must have value <= 65535 (0 is used as 

uspecified value for a port).

@return KErrNone, if parameters have valid values; otherwise, KErrArgument.

*/

EXPORT_C TInt TQoSSelector::SetSrc(const TInetAddr& aAddr)

	{

	if (aAddr.Port() > KMaxPort)

		return KErrArgument;

	iSrc = aAddr;

	return KErrNone;

	}

/**

Sets the destination address for selector.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aAddr Destination address. Port must have value <= 65535 (0 is used 

as uspecified value for a port).

@return KErrNone, if parameters have valid values; otherwise, KErrArgument.

*/

EXPORT_C TInt TQoSSelector::SetDst(const TInetAddr& aAddr)

	{

	if (aAddr.Port() > KMaxPort)

		return KErrArgument;

	iDst = aAddr;

	return KErrNone;

	}

/**

Gets the current source address.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@return Source address.

*/

EXPORT_C TInetAddr TQoSSelector::GetSrc() const

	{

	return iSrc;

	}

/**

Gets the current destination address.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@return Destination address.

*/

EXPORT_C TInetAddr TQoSSelector::GetDst() const

	{

	return iDst;

	}

/**

Sets the source address mask.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aMask Source address mask. Port must have value <= 65535 (0 is used 

as uspecified value for a port).

@return KErrNone, if parameters have valid values; otherwise, KErrArgument.

*/

EXPORT_C TInt TQoSSelector::SetSrcMask(const TInetAddr& aMask)

	{

	if (aMask.Port() > KMaxPort)

		return KErrArgument;

	iSrcMask = aMask;

	return KErrNone;

	}

/**

Gets the current source address mask.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@return Source address mask.

*/

EXPORT_C TInetAddr TQoSSelector::GetSrcMask() const

	{

	return iSrcMask;

	}

/**

Sets the destination address mask.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aMask Destination address mask. Port must have value <= 65535 (0 is 

used as uspecified value for a port).

@return KErrNone, if parameters have valid values; otherwise, KErrArgument.

*/

EXPORT_C TInt TQoSSelector::SetDstMask(const TInetAddr& aMask)

	{

	if (aMask.Port() > KMaxPort)

		return KErrArgument;

	iDstMask = aMask;

	return KErrNone;

	}

/**

Gets the current destination address mask.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@return Destination address mask.

*/

EXPORT_C TInetAddr TQoSSelector::GetDstMask() const

	{

	return iDstMask;

	}

/**

Sets the Internet access point identifier. 

0 is used as unspecified value.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aIapId Value to which to set the IapId.

*/

EXPORT_C void TQoSSelector::SetIapId(TInt aIapId)

	{

	iIapId = aIapId;

	}

/**

Gets the current Internet access point identifier.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@return Internet access point identifier.

*/

EXPORT_C TInt TQoSSelector::IapId() const

	{

	return iIapId;

	}

/**

Sets the protocol identifier. 

0 is used as unspecified value.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aProtocol Value to which to set the protocol ID.

*/

EXPORT_C void TQoSSelector::SetProtocol(TUint aProtocol)

	{

	iProtocol = aProtocol;

	}

/**

Gets the current protocol identifier.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@return Protocol identifier.

*/

EXPORT_C TUint TQoSSelector::Protocol() const

	{

	return iProtocol;

	}

/**

Sets the maximum source port. 

Port must have value <= 65535 (0 is used as unspecified value).

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aMaxPort Value to which to set maximum source port.

@return KErrNone, if aMaxPort has valid values; otherwise, KErrArgument.

*/

EXPORT_C TInt TQoSSelector::SetMaxPortSrc(TUint aMaxPort)

	{

	if (aMaxPort > KMaxPort)

		return KErrArgument;

	iSrcPortMax = aMaxPort;

	return KErrNone;

	}

/**

Sets the maximum destination port. 

Port must have value <= 65535 (0 is used as unspecified value).

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@param aMaxPort Value to which to set maximum destination port.

@return KErrNone, if aMaxPort has valid values; otherwise, KErrArgument.

*/

EXPORT_C TInt TQoSSelector::SetMaxPortDst(TUint aMaxPort)

	{

	if (aMaxPort > KMaxPort)

		return KErrArgument;

	iDstPortMax = aMaxPort;

	return KErrNone;

	}

/**

Gets the maximum source port.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@return Maximum source port.

*/

EXPORT_C TUint TQoSSelector::MaxPortSrc() const

	{

	return iSrcPortMax;

	}

/**

Gets the maximum destination port.

@publishedPartner

@released

@capability NetworkControl Restrict QoS policy operations because they may 

affect several data flows.

@return Maximum destination port.

*/

EXPORT_C TUint TQoSSelector::MaxPortDst() const

	{

	return iDstPortMax;

	}