// Copyright (c) 2005-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 <centralrepository.h>

#include <e32test.h>

#include <f32file.h>

#include <badesca.h>

#include "srvrepos_noc.h"

#include "srvres.h"

#include "cachemgr.h"

#include "backup.h"

RTest TheTest(_L("Central Repository Backup & Restore Delete Tests"));

enum FileSet

	{

	EOriginals,

	EChanged,

	ERemoved,

	EDEF058823L

	};

_LIT_SECURE_ID( SecureId1, 0x12345 );

CRepositoryBackupClient* backupClient;

CDesC8ArraySeg* originalData1;

CDesC8ArraySeg* changedData1;

CDesC8ArraySeg* originalData2;

CDesC8ArraySeg* changedData2;

CDesC8ArraySeg* changedData3;

CActiveScheduler* globalAS;

const TUid KUidBackupTestRepository1 	= { 0x00000100 };

const TUint32 KBackupInt1 = 0x02010100;

const TUint32 KBackupInt2 = 0x03010200;

const TInt KIntChangedValue2 = 200;

const TUint32 KBackupInt3 = 0x04010300;

const TInt KIntOrigValue3 = 30;

const TUint32 KBackupString1 = 0x02010600;

_LIT8(KStringChangedValue, "test changed");

const TUint32 KBackupString2 = 0x03010600;

const TUint32 KBackupReal1 = 0x02010700;

const TReal KRealOrigValue = 1.1;

const TUint32 KBackupReal2 = 0x03010700;

const TReal KRealChangedValue = 2.5;

///////////////////////////////////////////////////////////////////////////////////////

///////////////////////////////////////////////////////////////////////////////////////

//Test macroses and functions

LOCAL_C void DeleteFilesL()

	{

	_LIT( KOldInstallFiles, "c:\\private\\102081E4\\*.*" );

	_LIT( KOldPersistFiles, "c:\\private\\102081E4\\persists\\*.*" );

	CFileMan* fm = CFileMan::NewL( TServerResources::iFs );

	CleanupStack::PushL( fm );

	TInt r = fm->Delete( KOldInstallFiles );

	if ( r != KErrNone && r != KErrNotFound && r != KErrPathNotFound )

		User::Leave(r);

	r = fm->Delete( KOldPersistFiles );

	if ( r != KErrNone && r != KErrNotFound && r != KErrPathNotFound )

		User::Leave(r);

	CleanupStack::PopAndDestroy( fm );

	}

LOCAL_C void CheckL( TInt aValue, TInt aLine )

	{

	if ( !aValue )

		{

		DeleteFilesL();

		TheTest( EFalse, aLine );

		}

	}

LOCAL_C void CheckL( TInt aValue, TInt aExpected, TInt aLine )

	{

	if ( aValue != aExpected )

		{

		DeleteFilesL();

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

		TheTest( EFalse, aLine );

		}

	}

#define TEST(arg) ::CheckL((arg), __LINE__)

#define TEST2(aValue, aExpected) ::CheckL(aValue, aExpected, __LINE__)

///////////////////////////////////////////////////////////////////////////////////////

///////////////////////////////////////////////////////////////////////////////////////

LOCAL_C void InitializationL()

    {

    originalData1 = new(ELeave)CDesC8ArraySeg( 128 );

	CleanupStack::PushL( originalData1 );

	changedData1 = new(ELeave)CDesC8ArraySeg( 128 );

	CleanupStack::PushL( changedData1 );

	originalData2 = new(ELeave)CDesC8ArraySeg( 128 );

	CleanupStack::PushL( originalData2 );

	changedData2 = new(ELeave)CDesC8ArraySeg( 128 );

	CleanupStack::PushL( changedData2 );

	changedData3 = new(ELeave)CDesC8ArraySeg( 128 );

	CleanupStack::PushL( changedData3 );

    }

LOCAL_C void Cleanup()

    {

    CleanupStack::PopAndDestroy( changedData3 );

	CleanupStack::PopAndDestroy( changedData2 );

	CleanupStack::PopAndDestroy( originalData2 );

	CleanupStack::PopAndDestroy( changedData1 );

	CleanupStack::PopAndDestroy( originalData1 );

    }

LOCAL_C void InstallFileSetL( FileSet aFileSet )

	{

	_LIT( KFileSrcPath, "\\private\\10202BE9\\" );

	_LIT( KInstallFileDestPath, "\\private\\102081E4\\" );

	_LIT( KPersistFileDestPath, "\\private\\102081E4\\persists\\" );

	_LIT( KEDEF058823LFile, "babababa.cre");

	_LIT( KDriveC, "c:" );

	_LIT( KDriveZ, "z:" );

	_LIT( KCreFiles, "*.cre" );

	_LIT( KAllFiles, "*.*" );

	_LIT( KTxtFilesForC_V1, "*.cra" );

	_LIT( KTxtFilesForC_V2, "*.crb" );

	//Use HBufC in preference to TBuf or TFileName to save stack usage

	HBufC* src_buf1 = HBufC::NewLC(KMaxFileName);

	TPtr src1(src_buf1->Des());

	HBufC* src_buf2 = HBufC::NewLC(KMaxFileName);

	TPtr src2(src_buf2->Des());

	HBufC* dest_buf1 = HBufC::NewLC(KMaxFileName);

	TPtr dest1(dest_buf1->Des());

	HBufC* dest_buf2 = HBufC::NewLC(KMaxFileName);

	TPtr dest2(dest_buf2->Des());

	TInt r;

	DeleteFilesL();

	CFileMan* fm = CFileMan::NewL( TServerResources::iFs );

	CleanupStack::PushL( fm );

	switch ( aFileSet )

		{

		case EOriginals:

			{

			dest1.Copy( KDriveC );

			dest1.Append( KInstallFileDestPath );

			dest1.Append( KCreFiles );

			r = fm->Delete( dest1 );

			if ( r != KErrNone && r != KErrNotFound && r != KErrPathNotFound )

				User::Leave( r );

			r = TServerResources::iFs.MkDirAll( dest1 );

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

				User::Leave( r );

			src1.Copy( KDriveZ );

			src1.Append( KFileSrcPath );

			src1.Append( KTxtFilesForC_V1 );

			User::LeaveIfError( fm->Copy( src1, dest1 ) );

			r = fm->Attribs( dest1, KEntryAttArchive, KEntryAttReadOnly, TTime( 0 ), CFileMan::ERecurse );

			TEST2( r, KErrNone );

			dest2.Copy( KDriveC );

			dest2.Append( KPersistFileDestPath );

			dest2.Append( KCreFiles );

			r = fm->Delete( dest2 );

			if ( r != KErrNone && r != KErrNotFound && r != KErrPathNotFound )

				User::Leave( r );

			r = TServerResources::iFs.MkDirAll( dest2 );

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

				User::Leave( r );

			src2.Copy( KDriveZ );

			src2.Append( KFileSrcPath );

			src2.Append( KTxtFilesForC_V1 );

			User::LeaveIfError( fm->Copy( src2, dest2 ) );

			r = fm->Attribs( dest2, KEntryAttArchive, KEntryAttReadOnly, TTime( 0 ), CFileMan::ERecurse );

			TEST2( r, KErrNone );

			break;

			}

		case EChanged:

			{

			dest1.Copy( KDriveC );

			dest1.Append( KInstallFileDestPath );

			dest1.Append( KCreFiles );

			r = fm->Delete( dest1 );

			if ( r != KErrNone && r != KErrNotFound && r != KErrPathNotFound )

				User::Leave( r );

			r = TServerResources::iFs.MkDirAll( dest1 );

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

				User::Leave( r );

			src1.Copy( KDriveZ );

			src1.Append( KFileSrcPath );

			src1.Append( KTxtFilesForC_V1 );

			User::LeaveIfError( fm->Copy( src1, dest1 ) );

			r = fm->Attribs( dest1, KEntryAttArchive, KEntryAttReadOnly, TTime( 0 ), CFileMan::ERecurse );

			TEST2( r, KErrNone );

			dest2.Copy( KDriveC );

			dest2.Append( KPersistFileDestPath );

			dest2.Append( KCreFiles );

			r = fm->Delete( dest2 );

			if ( r != KErrNone && r != KErrNotFound && r != KErrPathNotFound )

				User::Leave( r );

			r = TServerResources::iFs.MkDirAll( dest2 );

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

				User::Leave( r );

			src2.Copy( KDriveZ );

			src2.Append( KFileSrcPath );

			src2.Append( KTxtFilesForC_V2 );

			User::LeaveIfError( fm->Copy( src2, dest2 ) );

			r = fm->Attribs( dest2, KEntryAttArchive, KEntryAttReadOnly, TTime( 0 ), CFileMan::ERecurse );

			TEST2( r, KErrNone );

			break;

			}

		case ERemoved:

			{

			dest1.Copy( KDriveC );

			dest1.Append( KInstallFileDestPath );

			dest1.Append( KCreFiles );

			r = fm->Delete( dest1 );

			if ( r != KErrNone && r != KErrNotFound && r != KErrPathNotFound )

				User::Leave( r );

			r = TServerResources::iFs.MkDirAll( dest1 );

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

				User::Leave( r );

			src1.Copy( KDriveZ );

			src1.Append( KFileSrcPath );

			src1.Append( KTxtFilesForC_V1 );

			User::LeaveIfError( fm->Copy( src1, dest1 ) );

			r = fm->Attribs( dest1, KEntryAttArchive, KEntryAttReadOnly, TTime( 0 ), CFileMan::ERecurse );

			TEST2( r, KErrNone );

			dest2.Copy( KDriveC );

			dest2.Append( KPersistFileDestPath );

			dest2.Append( KCreFiles );

			r = fm->Delete( dest2 );

			if ( r != KErrNone && r != KErrNotFound && r != KErrPathNotFound )

				User::Leave( r );

			break;

			}

		case EDEF058823L:

		    {

			dest2.Copy( KDriveC );

			dest2.Append( KPersistFileDestPath );

			dest2.Append( KAllFiles );

			r = fm->Delete( dest2 );

			if ( r != KErrNone && r != KErrNotFound && r != KErrPathNotFound )

				User::Leave( r );

			r = TServerResources::iFs.MkDirAll( dest2 );

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

				User::Leave( r );

			src2.Copy( KDriveZ );

			src2.Append( KFileSrcPath );

			src2.Append( KEDEF058823LFile );

			User::LeaveIfError( fm->Copy( src2, dest2 ) );

			r = fm->Attribs( dest2, KEntryAttArchive, KEntryAttReadOnly, TTime( 0 ), CFileMan::ERecurse );

			TEST2( r, KErrNone );

		    break;

		    }

		default:

			RDebug::Print( _L( "Illegal parameter to function: %d\r\n" ), aFileSet );

			TheTest( EFalse, __LINE__ );

		}

	CleanupStack::PopAndDestroy( fm );

	CleanupStack::PopAndDestroy( 4 );

	}

LOCAL_C void BackupRepositoryL( CDesC8ArraySeg* aDataStream )

	{

	backupClient->CompleteOwnerIdLookupTableL();

	backupClient->InitialiseGetProxyBackupDataL( SecureId1, EDriveC );

	HBufC8* transferBuffer = HBufC8::NewL( 128 );

	CleanupStack::PushL( transferBuffer );

	TPtr8 bufferToSend = transferBuffer->Des();

	TBool finished = EFalse;

	aDataStream->Reset();

	do

		{

		bufferToSend.Zero();

		backupClient->GetBackupDataSectionL( bufferToSend, finished );

		aDataStream->AppendL( bufferToSend );

		} while ( !finished );

	CleanupStack::PopAndDestroy( transferBuffer );

	}

LOCAL_C void RestoreRepositoryL( CDesC8ArraySeg* aDataStream )

	{

	backupClient->InitialiseRestoreProxyBaseDataL( SecureId1, EDriveC );

	HBufC8* transferBuffer = HBufC8::NewL( 128 );

	CleanupStack::PushL( transferBuffer );

	TBool finished = EFalse;

	TInt count = aDataStream->Count();

	TInt element = 0;

	for ( element = 0; element < count; element++ )

		{

		TPtrC8 ptr = (*aDataStream)[ element ];

		if (ptr.Size()>transferBuffer->Des().MaxSize())

			{

			transferBuffer = transferBuffer->ReAlloc(ptr.Size());

			}

		TPtr8 bufferToSend = transferBuffer->Des();

		bufferToSend.Zero();

		bufferToSend.Append( ptr );

		if ( element + 1 == count )

			finished = ETrue;

		backupClient->RestoreBaseDataSectionL( bufferToSend, finished );

		}

	backupClient->RestoreComplete( EDriveC );

	CleanupStack::PopAndDestroy( transferBuffer );

	}

/**

@SYMTestCaseID SYSLIB-CENTRALREPOSITORY-UT-1535

@SYMTestCaseDesc Test Backup functionality with reference to INC072238: centrep restores deleted content.

@SYMTestPriority High

@SYMTestActions  Test Backup functionality.

@SYMTestExpectedResults The test must not fail.

@SYMPREQ PREQ752

*/

LOCAL_C void BackupTestsL()

	{

	TheTest.Next( _L( " @SYMTestCaseID:SYSLIB-CENTRALREPOSITORY-UT-1535 Backup repository " ) );

	// create and install the active scheduler we need

	CActiveScheduler* s = new(ELeave) CActiveScheduler;

	CleanupStack::PushL( s );

	CActiveScheduler::Replace( s );

	backupClient = CRepositoryBackupClient::NewLC( TServerResources::iFs );

	TEST( backupClient != 0 );

	// These tests don't test Backup&Restore functionality over Secure Backup Server so cache management

	// is not possible. For that reason, cache is disabled manually.

	TServerResources::iCacheManager->DisableCache();

	DeleteFilesL();

	TheTest.Next( _L( "Backup original version" ) );

	// Install known files

	InstallFileSetL( EOriginals );

	CServerRepository* aServerRepo=new (ELeave)CServerRepository();

	CleanupStack::PushL(aServerRepo);

	CSessionNotifier* aSessionNotif=new (ELeave) CSessionNotifier();

	CleanupStack::PushL(aSessionNotif);

	// Open repository to delete a few sessions settings

	aServerRepo->OpenL(KUidBackupTestRepository1,*aSessionNotif);

	TUint32 transId = aServerRepo->StartTransaction(EReadWriteTransaction) ;

	aServerRepo->TransactionDeleteL(KBackupInt1) ;

	aServerRepo->TransactionDeleteL(KBackupString1) ;

	aServerRepo->TransactionDeleteL(KBackupReal2) ;

	aServerRepo->CommitTransaction(transId);

	aServerRepo->CommitChangesL() ;

	aServerRepo->Close();

	CleanupStack::PopAndDestroy(aSessionNotif);

	CleanupStack::PopAndDestroy(aServerRepo);

	// Backup the repository

	BackupRepositoryL( originalData1 );

	CleanupStack::PopAndDestroy( backupClient );

	// Cleanup the scheduler

	CleanupStack::PopAndDestroy( s );

	}

/**

@SYMTestCaseID SYSLIB-CENTRALREPOSITORY-UT-1536

@SYMTestCaseDesc Test Restore functionality with reference to INC072238: centrep restores deleted content.

@SYMTestPriority High

@SYMTestActions  Test Restore functionality.

@SYMTestExpectedResults The test must not fail.

@SYMPREQ PREQ752

*/

LOCAL_C void RestoreTestsL()

	{

	TheTest.Next( _L( " @SYMTestCaseID:SYSLIB-CENTRALREPOSITORY-UT-1536 Restore repository " ) );

	// create and install the active scheduler we need

	CActiveScheduler* s = new(ELeave) CActiveScheduler;

	CleanupStack::PushL( s );

	CActiveScheduler::Replace( s );

	backupClient = CRepositoryBackupClient::NewLC(TServerResources::iFs);

	TEST( backupClient != 0 );

	CServerRepository* aServerRepo=new (ELeave)CServerRepository();

	CleanupStack::PushL(aServerRepo);

	CSessionNotifier* aSessionNotif=new (ELeave) CSessionNotifier();

	CleanupStack::PushL(aSessionNotif);

	TServerResources::iCacheManager->DisableCache();

	DeleteFilesL();

	TheTest.Next( _L( "Restore original version over changed version" ) );

	InstallFileSetL( EChanged );

	RestoreRepositoryL( originalData1 );

	TheTest.Next( _L( "Confirm settings restored" ) );

	// Open merged repository to check for values of individual settings

	aServerRepo->OpenL(KUidBackupTestRepository1,*aSessionNotif);

	TInt r, a;

	// This setting was DELETED before backup so it shouldn't be present

	// after restore

	r = aServerRepo->Get(KBackupInt1, a);

	TEST2(r, KErrNotFound);

	r = aServerRepo->Get(KBackupInt2, a);

	TEST2(r, KErrNone);

	// This setting isn't backed up, so it must still have changed value

	TEST(a==KIntChangedValue2);

	r = aServerRepo->Get(KBackupInt3, a);

	TEST2(r, KErrNone);

	// This setting is backed up, so it must be restored to original

	TEST(a==KIntOrigValue3);

	TBuf8<KMaxBinaryLength> str;

	// This setting was DELETED before backup so it shouldn't be present

	// after restore

	r = aServerRepo->Get(KBackupString1, str);

	TEST(r==KErrNotFound);

	r = aServerRepo->Get(KBackupString2, str);

	TEST(r==KErrNone);

	// This setting isn't backed up, so it must still have changed value

	TEST(str==KStringChangedValue);

	TReal realval;

	r = aServerRepo->Get(KBackupReal1, realval);

	TEST(r==KErrNone);

	// This setting is backed up, so it must be restored to original

	TEST(realval==KRealOrigValue);

	r = aServerRepo->Get(KBackupReal2, realval);

	TEST(r==KErrNone);

	// This setting was deleted but isn't backed up, so it should still

	// exist and have the changed value

	TEST(realval==KRealChangedValue);

	aServerRepo->Close();

	DeleteFilesL();

	CleanupStack::PopAndDestroy(aSessionNotif);

	CleanupStack::PopAndDestroy(aServerRepo);

	CleanupStack::PopAndDestroy( backupClient );

	// Cleanup the scheduler

	CActiveScheduler::Replace( globalAS );

	CleanupStack::PopAndDestroy( s );

	}

// ---------------------------------------------------

// OomTest

//

// Function to convert a test into an OOM test

/**

@SYMTestCaseID SYSLIB-CENTRALREPOSITORY-CT-0117-0001

@SYMTestCaseDesc Test functionality under OOM.

@SYMTestPriority High

@SYMTestActions  Test functionality under OOM.

@SYMTestExpectedResults The test must not fail.

@SYMPREQ PREQ752

*/

LOCAL_C void OomTest( void (*testFuncL)() )

	{

	TheTest.Next( _L( " @SYMTestCaseID:SYSLIB-CENTRALREPOSITORY-CT-0117-0001 " ) );

	TInt error;

	TInt count = 0;

	do

		{

		User::__DbgMarkStart( RHeap::EUser );

		// find out the number of open handles

		TInt startProcessHandleCount;

		TInt startThreadHandleCount;

		RThread().HandleCount(startProcessHandleCount, startThreadHandleCount);

		User::__DbgSetAllocFail( RHeap::EUser, RHeap::EFailNext, ++count );

		TRAP( error, (testFuncL)() );

		User::__DbgSetAllocFail( RHeap::EUser, RHeap::ENone, 1 );