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

// DBMS security policy - testing new APIs - utility functions

// 

//

#include <e32test.h>

#include "t_dbplatsecutl.h"

//Supply your a pointer to your cleanup function or NULL.

TDBSCUtils::TDBSCUtils(RTest& aTest, TCleanupFunc aFunc) :

	iTest(aTest),

	iFunc(aFunc)

	{

	}

//Create a CDbColSet instance. The caller is responsible for deleting it.

CDbColSet* TDBSCUtils::CreateColSetLC(const TColDef* aColDef)

	{

	CDbColSet* set = CDbColSet::NewLC();

	for(;aColDef->iName;++aColDef)

		{

		TDbCol col(TPtrC(aColDef->iName), aColDef->iType);

		col.iAttributes = aColDef->iAttributes;

		set->AddL(col);

		}

	return set;

	}

//Create a CDbKey instance. The caller is responsible for deleting it.

CDbKey* TDBSCUtils::CreateKeyLC(const TDesC& aColName)

	{

	CDbKey* key = CDbKey::NewLC();

	key->AddL(aColName);

	return key;

	}

//Create a DBMS table.

void TDBSCUtils::CreateTableL(RDbNamedDatabase& aDb, const TDesC& aTblName, const TColDef* aColDef)

	{

	CDbColSet* colset = TDBSCUtils::CreateColSetLC(aColDef);

	TInt err = aDb.CreateTable(aTblName, *colset);

	TEST2(err, KErrNone);

	CleanupStack::PopAndDestroy(colset);

	}

// Create secure shared database. The caller is responsible for closing it.

RDbNamedDatabase TDBSCUtils::CreateDatabase(RDbs& aDbs, TUid aPolicyUid, const TDesC& aDbName)

	{

	_LIT(KSecure, "SECURE");

	TBuf<32> format;

	format.Copy(KSecure);

	format.Append(aPolicyUid.Name());

	RDbNamedDatabase db;

	TInt err = db.Create(aDbs, aDbName, format);

	TEST2(err, KErrNone);

	return db;

	}

// Delete secure shared database.

TInt TDBSCUtils::DeleteDatabase(RDbs& aDbs, TUid aPolicyUid, const TDesC& aDbName)

	{

	RDebug::Print(_L("Deleting %X \"%S\" database.\n"), aPolicyUid.iUid, &aDbName);

	TInt err = aDbs.DeleteDatabase(aDbName, aPolicyUid);

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

		{

		RDebug::Print(_L("Error %d deleting \"%S\" database.\n"), err, &aDbName);

		}

	return err;

	}

//Check if a secure shared database with "aDatabaseName" name exists.

//aDatabaseName format: <name>.<ext>

TBool TDBSCUtils::IsDatabaseThereL(RDbs& aDbs, TUid aPolicyUid, TDriveNumber aDriveNumber, const TDesC& aDatabaseName)

	{

	CDbDatabaseNames* dbNames = aDbs.DatabaseNamesL(aDriveNumber, aPolicyUid);

	TEST(dbNames != NULL);

	TBool found = EFalse;

	for(TInt i=0;i<dbNames->Count();++i)

		{

		const TDesC& dbName = (*dbNames)[i];

		if(aDatabaseName.CompareF(dbName) == 0)

			{

			found = ETrue;

			break;

			}

		}

	delete dbNames;

	return found;

	}

//Check if the supplied aPolicy parameter has aCapability capability.

TBool TDBSCUtils::HasCapability(const TCompiledSecurityPolicy& aPolicy, TCapability aCapability)

	{

	TInt maxCount = 0;

	if(aPolicy.Type() == TSecurityPolicy::ETypeS3 || 

	   aPolicy.Type() == TSecurityPolicy::ETypeV3 ||

	   aPolicy.Type() == TSecurityPolicy::ETypeC3)

		{

		maxCount = 3;	

		}

	else if(aPolicy.Type() == TSecurityPolicy::ETypeC7)

		{

		maxCount = 7;	

		}

	else

		{

		TEST(0);

		}

	for(TInt i=0;i<maxCount;++i)

		{

		if(aPolicy.Capability(i) == aCapability)

			{

			return ETrue;	

			}

		}

	return EFalse;

	}

//If aValue is 0 then the method prints an error message and terminates the application.

//It will do a test cleanup before the termination, if iFunc data member is not null.

void TDBSCUtils::Check(TInt aValue, TInt aLine)

	{

	if(!aValue)

		{

		if(TheDbscUtils.iFunc)

			{

			TheDbscUtils.iFunc();

			}

		TheDbscUtils.iTest(EFalse, aLine);

		}

	}

//If aValue != aExpected then the method prints an error message and terminates the application.

//It will do a test cleanup before the termination, if iFunc data member is not null.

void TDBSCUtils::Check(TInt aValue, TInt aExpected, TInt aLine)

	{

	if(aValue != aExpected)

		{

		RDebug::Print(_L("*** Expected error: %d, got: %d\r\n"), aExpected, aValue);

		if(TheDbscUtils.iFunc)

			{

			TheDbscUtils.iFunc();

			}

		TheDbscUtils.iTest(EFalse, aLine);

		}

	}