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

#include <bluetooth/logger.h>

#ifdef __FLOG_ACTIVE

_LIT8(KLogComponent, LOG_COMPONENT_BT_MANAGER_SERVER);

#endif

const TInt KMaxLengthConstraintInitial = 60;	// constraint length is initially this

const TInt KMaxLengthConstraintIncrement = 60;	// - then we extend as necessary by this amount

RBTDbQuery::RBTDbQuery()

/**

	Required for non-default constructed RTextBuf

**/

	{

	LOG_FUNC

	}

void RBTDbQuery::Close()

	{

	LOG_FUNC

	iQueryBuf.Close();

	}

const TPtrC RBTDbQuery::QuoteEscapeLC(const TBTDevAddr& aBDAddr) const

	{

	LOG_FUNC

	TBuf<6> wideAddr;	// needed to stop implicit converstion back to DevAddr!

	wideAddr.Copy(aBDAddr.Des());

	return QuoteEscapeLC(wideAddr);

	}

const TPtrC RBTDbQuery::QuoteEscapeLC(const TDesC& aDes) const

/**

	Can't search for a string in SQL with a quote (') in un-escaped

	Escape by putting extra quote in for every quote in the descriptor

**/

	{

	LOG_FUNC

	const TUint16 KEscapeCode = 0x0027;	// SQL escape value

	// maximum size of new descriptor will be twice the original (if it contained all quotes!)

	HBufC* escaped = HBufC::NewMaxLC(2*aDes.Length());

	TPtr escapedPtr= escaped->Des();

	TInt i=0;

	TInt escapedIndex = 0;

	while (i<aDes.Length())

		{

		const TChar& element = aDes[i++];	// copes with narrow and wide

#pragma warning (disable : 4244)  // conversion - its ok here

		escapedPtr[escapedIndex++] = element;

#pragma warning (default : 4244)

		if (element == KEscapeCode)

			{

			escapedPtr[escapedIndex++] = KEscapeCode;

			}

		}

	escapedPtr.SetLength(escapedIndex);	// bring down from the MaxLength

	return escapedPtr;	// returns ownership of the HBuf via cleanupstack

	}

void RBTDbQuery::FindDeviceL(const TBTDevAddr& aBDAddr)

/**

	Construct a simple SQL query to see if the device is present

**/

	{

	LOG_FUNC

	// this is used to see if the device is in the registry

	TBuf<KMaxConstraintLen> constraint;

	constraint.Append(KColName_DeviceAddress);

	constraint.Append(KSQLEqual);

	constraint.Append(KSQLQuote);

	constraint.Append(QuoteEscapeLC(aBDAddr));

	constraint.Append(KSQLQuote);

	iQueryBuf.SetMaxLengthL(constraint.Length() + KSQLSelectAllConstrained().Length() + KDeviceTable().Length());

	iQueryBuf.Text().Format(KSQLSelectAllConstrained, &KDeviceTable, &constraint);

	CleanupStack::PopAndDestroy(1); // address buffer

	}

void RBTDbQuery::FindCommPortL(TUint32 aUnitNumber)

/**

	Construct a SQL query to find the virtual serial port with aUnitNumber

**/

	{

	LOG_FUNC

	TBuf<KMaxConstraintLen> constraint;

	constraint.Append(KBTCOMMColName_Port);

	constraint.Append(KSQLEqual);

	constraint.AppendNum(static_cast<TUint>(aUnitNumber), EDecimal);

	iQueryBuf.SetMaxLengthL(constraint.Length() + KSQLSelectAllConstrained().Length() + KCSYTable().Length());

	iQueryBuf.Text().Format(KSQLSelectAllConstrained, &KCSYTable, &constraint);

	}

void RBTDbQuery::MatchLinkKeyTypeL(TBTLinkKeyType aLinkKeyType)

/**

 	Construct a SQL query to find entries that have a link key of a specific type.

**/

	{

	LOG_FUNC

	TBuf<KMaxConstraintLen> constraint;

	constraint.Append(KDeviceColName_LinkKeyType);

	constraint.Append(KSQLEqual);

	constraint.AppendNum(static_cast<TUint>(aLinkKeyType), EDecimal);

	iQueryBuf.SetMaxLengthL(constraint.Length() + KSQLSelectAllConstrained().Length() + KDeviceTable().Length());

	iQueryBuf.Text().Format(KSQLSelectAllConstrained, &KDeviceTable, &constraint);

	}

const TDesC& RBTDbQuery::QueryBuf() const

	{

	LOG_FUNC

	return iQueryBuf.Text();

	}

TPtrC RBTDbQuery::ConstraintBuf() const

	{

	LOG_FUNC

	TInt index = QueryBuf().Find(KSQLWhere);

	return (index==KErrNotFound) ? TPtrC() : QueryBuf().Mid(index+KSQLWhere().Length());

	}

void RBTDbQuery::ExtendForAnotherTokenL(RTextBuf& aSQLBuf, TUint aMask)

/**

	Sees if the token 'AND' is required in a constrained search

	If so, add into the SQL query

	If the buffer is getting small, extend it

**/

	{

	LOG_FUNC

	if (aMask)

		{

		// need an AND - more constraints follow

		aSQLBuf.Text().Append(KSQLAnd);

		//Need to increase the size of the bufferno

		// RTextBuf::SetMaxLengthL is bad as it closes the textbuf

		RTextBuf copy;

		copy.SetMaxLengthL(aSQLBuf.Text().Length());

		CleanupClosePushL(copy);

		copy.Text().Append(aSQLBuf.Text());

		aSQLBuf.SetMaxLengthL(aSQLBuf.Text().Length()+KMaxLengthConstraintIncrement);

		//copy back from the copy to the aSQLBuf

		aSQLBuf.Text().Append(copy.Text());

		CleanupStack::PopAndDestroy();

		}

	// else do nothing - no further constraint to follow

	}

void RBTDbQuery::SearchL(const TBTRegistrySearch& aPattern)

/**

	Converts a search mask into SQL search tokens

**/

	{

	LOG_FUNC

	// do the catch all first - saves time

	if (aPattern.iSearchMask & TBTRegistrySearch::EAll)

		{

		iQueryBuf.SetMaxLengthL(KSQLSelectAll().Length() + KDeviceTable().Length());

		iQueryBuf.Text().Format(KSQLSelectAll, &KDeviceTable);

		return;

		}

	// we have a constrained search; formulate the SQL

	// we use a RTextBuf that can grow as needed (to reduce stack usage)

	RTextBuf constraintBuf;

	constraintBuf.SetMaxLengthL(KMaxLengthConstraintInitial);

	TPtr& constraint = constraintBuf.Text();

	TUint toSearch = aPattern.iSearchMask;

	if (toSearch & TBTRegistrySearch::EAddress)

		{

		constraint.Append(KColName_DeviceAddress);

		constraint.Append(KSQLEqual);

		constraint.Append(KSQLQuote);

		constraint.Append(QuoteEscapeLC(aPattern.iDeviceAddress));

		constraint.Append(KSQLQuote);

		CleanupStack::PopAndDestroy(1); // address buffer

		toSearch &= ~TBTRegistrySearch::EAddress;

		ExtendForAnotherTokenL(constraintBuf, toSearch);

		}

	if (toSearch & TBTRegistrySearch::ECoD)

		{

		constraint.Append(KDeviceColName_CoD);

		constraint.Append(KSQLEqual);

		TUint cod = aPattern.iDeviceClass.DeviceClass();

		constraint.AppendNum(cod, EDecimal);

		toSearch &= ~TBTRegistrySearch::ECoD;

		ExtendForAnotherTokenL(constraintBuf, toSearch);

		}

	if (toSearch & TBTRegistrySearch::ECoDMajorDev)

		{

		constraint.Append(KDeviceColName_CoD_MajorDev);

		constraint.Append(KSQLEqual);

		TUint majDev = aPattern.iDeviceClass.MajorDeviceClass();

		constraint.AppendNum(majDev, EDecimal);

		toSearch &= ~TBTRegistrySearch::ECoDMajorDev;

		ExtendForAnotherTokenL(constraintBuf, toSearch);

		}

	if (toSearch & TBTRegistrySearch::ECoDMinorDev)

		{

		constraint.Append(KDeviceColName_CoD_MinorDev);

		constraint.Append(KSQLEqual);

		TUint minDev = aPattern.iDeviceClass.MinorDeviceClass();

		constraint.AppendNum(minDev, EDecimal);

		toSearch &= ~TBTRegistrySearch::ECoDMinorDev;

		ExtendForAnotherTokenL(constraintBuf, toSearch);

		}

	if (toSearch & TBTRegistrySearch::EBTName)

		{

		constraint.Append(KDeviceColName_BluetoothName);

		constraint.Append(KSQLEqual);

		constraint.Append(KSQLQuote);

		TBuf<KMaxBluetoothNameLen> widened;

		widened.Copy(aPattern.iBluetoothName);

		constraint.Append(QuoteEscapeLC(widened));

		constraint.Append(KSQLQuote);

		CleanupStack::PopAndDestroy(1); //name

		toSearch &= ~TBTRegistrySearch::EBTName;

		ExtendForAnotherTokenL(constraintBuf, toSearch);

		}

	if (toSearch & TBTRegistrySearch::EFriendlyName)

		{

		constraint.Append(KDeviceColName_FriendlyName);

		constraint.Append(KSQLEqual);

		constraint.Append(KSQLQuote);

		constraint.Append(QuoteEscapeLC(aPattern.iFriendlyName));

		constraint.Append(KSQLQuote);

		CleanupStack::PopAndDestroy(1); // name

		toSearch &= ~TBTRegistrySearch::EFriendlyName;

		ExtendForAnotherTokenL(constraintBuf, toSearch);

		}

	if (toSearch & TBTRegistrySearch::EProcess)

		{

		constraint.Append(KColName_ProcessSID);

		constraint.Append(KSQLEqual);

		constraint.AppendNum(MAKE_TUINT64(0,aPattern.iCurrentProcessSID.iUid), EDecimal);

		toSearch &= ~TBTRegistrySearch::EProcess;

		ExtendForAnotherTokenL(constraintBuf, toSearch);

		}

	if (toSearch & TBTRegistrySearch::EBonded)

		{

		constraint.Append(KDeviceColName_LinkKey);

		constraint.Append(KSQLIsNotNull);

		toSearch &= ~TBTRegistrySearch::EBonded;

		ExtendForAnotherTokenL(constraintBuf, toSearch);

		}

	if (toSearch & TBTRegistrySearch::ETrusted)

		{

		// check global security for no authorisation AND authentication

		constraint.Append(KSQLOpenParentheses);

		constraint.Append(KDeviceColName_GlobalSecSecurity);

		constraint.Append(KSQLEqual);

		// we do this here in case TBTDeviceSecurity changes

		// trusted device = 'NoAuthentication'=No (MUST authenticate!),

		//					'NoAuthorisation'=Yes

		//					'Encryption' could be either.

		//					'Banned' =No (!)

		TBTDeviceSecurity noEncryption(EFalse, ETrue, EFalse, EFalse);

		constraint.AppendNum(static_cast<TUint>(noEncryption.SecurityValue()));

		constraint.Append(KSQLOr);

		constraint.Append(KDeviceColName_GlobalSecSecurity);

		constraint.Append(KSQLEqual);

		TBTDeviceSecurity encryption(EFalse, ETrue, ETrue, EFalse);

		constraint.AppendNum(static_cast<TUint>(encryption.SecurityValue()));

		constraint.Append(KSQLCloseParentheses);

		toSearch &= ~TBTRegistrySearch::ETrusted;

		ExtendForAnotherTokenL(constraintBuf, toSearch);

		}

	if (toSearch & TBTRegistrySearch::ELastSeen)

		{

		constraint.Append(KDeviceColName_LastSeen);

		constraint.Append(KSQLGreaterEqual);

		constraint.AppendNum(aPattern.iLastSeen.Int64());

		toSearch &= ~TBTRegistrySearch::ELastSeen;

		ExtendForAnotherTokenL(constraintBuf, toSearch);

		}

	if (toSearch & TBTRegistrySearch::ELastUsed)

		{

		constraint.Append(KDeviceColName_LastUsed);

		constraint.Append(KSQLGreaterEqual);

		constraint.AppendNum(aPattern.iLastUsed.Int64());

		toSearch &= ~TBTRegistrySearch::ELastUsed;

		ExtendForAnotherTokenL(constraintBuf, toSearch);

		}

	if (toSearch & TBTRegistrySearch::EUiCookie)

		{

		constraint.Append(KDeviceColName_UiCookie);

		constraint.Append(KSQLLike);

		constraint.Append(KSQLQuote);

		static const TUint8 KCookieBinaryRepresentationWidth = 32; // 32bits encoded as binary string.

		TBuf<KCookieBinaryRepresentationWidth> cookiePattern;

		cookiePattern.NumFixedWidth(aPattern.iUiCookie, EBinary, KCookieBinaryRepresentationWidth);

		TUint32 shiftedMask = aPattern.iUiCookieMask;

		for(TInt i=KCookieBinaryRepresentationWidth-1; i>=0; --i)

			{

			if(!(shiftedMask&1)) // check the bottom bit (we loop through them all by shifting.)

				{

				// bit "i" isn't masked, and so we should ignore it.

				cookiePattern[i] = '?';

				}

			shiftedMask >>= 1;

			}

		constraint.Append(cookiePattern);

		constraint.Append(KSQLQuote);

		toSearch &= ~TBTRegistrySearch::EUiCookie;

		ExtendForAnotherTokenL(constraintBuf, toSearch);

		}

	if(toSearch)

		{

		// There is an unhandled search attribute that has been provided by the user.

		// If left the trailing SQL "AND" would trigger the KErrArgument error anyway.

		User::Leave(KErrArgument);

		}

	iQueryBuf.SetMaxLengthL(constraint.Length() + KSQLSelectAllConstrained().Length() + KDeviceTable().Length());

	iQueryBuf.Text().Format(KSQLSelectAllConstrained, &KDeviceTable, &constraint);

	}