// Copyright (c) 1998-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 <s32file.h>

#include <s32crypt.h>

#include <e32test.h>

#include <s32mem.h>

#include <pbedata.h>

#include <pbe.h>

const TInt KTestCleanupStack=0x20;

// This is a path specification and should not be used as is

_LIT(KFileLocationSpec, "Z:\\STOR-TST\\T_CRYPT.DAT");

const TUint8* KTestData=_S8("0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ");

const TInt KTestLength=36;

const TInt KTestTotal=KTestLength*(KTestLength+1);

const TPtrC8 KTestDes(KTestData,KTestLength);

const TPBPassword KTestPassword(_L("The password"));

LOCAL_D CTrapCleanup* TheTrapCleanup;

LOCAL_D RTest test(_L("t_storcrypt"));

LOCAL_D RFs TheFs;

LOCAL_D TStreamId TheTempId;

LOCAL_D TBuf8<KTestLength+1> TheBuf;

LOCAL_D CPBEncryptSet* thePBEKey;

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1126

@SYMTestCaseDesc	    Tests for writing and reading an encrypted data to store

@SYMTestPriority 	    High

@SYMTestActions  	    Encrypt data and write to a store.Read the encrypted data for decrypting and compare the result.

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testEncryptionDataL()

    {

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1126 "));

	CPBEncryptElement* elementKey = CPBEncryptElement::NewLC(KTestPassword);

	TParsePtrC parse(KFileLocationSpec);

	RFileWriteStream writeStream;

    writeStream.PushL();

    User::LeaveIfError(writeStream.Replace(TheFs, parse.NameAndExt(), EFileWrite));

    REncryptStream encryptStream;

    encryptStream.OpenLC(writeStream, *elementKey);

    encryptStream << KTestDes;

    encryptStream.CommitL();

    writeStream.CommitL();

    CleanupStack::PopAndDestroy(2); // encryptStream, writeStream

    const CPBEncryptionData& encryptData = elementKey->EncryptionData();

    CPBEncryptElement* newElementKey = CPBEncryptElement::NewLC(encryptData, KTestPassword);

    RFileReadStream readStream;

    readStream.PushL();

    User::LeaveIfError(readStream.Open(TheFs, parse.NameAndExt(), EFileWrite));

    RDecryptStream decryptStream;

    decryptStream.OpenLC(readStream, *newElementKey);

    decryptStream >> TheBuf;

    test(KTestDes == TheBuf);

	CleanupStack::PopAndDestroy(4, elementKey);

    }

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1127

@SYMTestCaseDesc	    Stream encryption test.

                        Tests the functionality of REncryptStream and RDecryptStream classes

@SYMTestPriority 	    High

@SYMTestActions  	    Tests for writing an encrypted string to a stream,reading back the data,decrypt and

                        check with the original data.

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testSimpleStreamEncryptionL()

    {

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1127 "));

   	CPBEncryptElement* elementKey = CPBEncryptElement::NewL(KTestPassword);

	CleanupStack::PushL(elementKey);

	TParsePtrC parse(KFileLocationSpec);

	RFileWriteStream writeStream;

    writeStream.PushL();

    User::LeaveIfError(writeStream.Replace(TheFs, parse.NameAndExt(), EFileWrite));

	REncryptStream encryptStream;

	encryptStream.OpenLC(writeStream, *elementKey);

    encryptStream << KTestDes;

    encryptStream.CommitL();

    writeStream.CommitL();

    CleanupStack::PopAndDestroy(2); // encryptStream, writeStream

    RFileReadStream readStream;

    readStream.PushL();

    User::LeaveIfError(readStream.Open(TheFs, parse.NameAndExt(), EFileWrite));

    RDecryptStream decryptStream;

	decryptStream.OpenLC(readStream, *elementKey);

    decryptStream >> TheBuf;

    test(KTestDes == TheBuf);

    CleanupStack::PopAndDestroy(3, elementKey); // decryptStream, readStream

    }

/**

@SYMTestCaseID          PDS-STORE-CT-4042

@SYMTestCaseDesc	    Tests for attaching write stream and read stream to an encrypted stream.

@SYMTestPriority 	    High

@SYMTestActions  	    Encrypt data and write to a store.Read the encrypted data for decrypting and compare the result.

@SYMTestExpectedResults Test must not fail

@SYMDEF                 DEF135804

*/

LOCAL_C void testEncryptionDataAttachL()

	{

	test.Next(_L(" @SYMTestCaseID:PDS-STORE-CT-4042 "));

	CPBEncryptElement* elementKey = CPBEncryptElement::NewLC(KTestPassword);

	TParsePtrC parse(KFileLocationSpec);

	RFileWriteStream writeStream;

	writeStream.PushL();

	User::LeaveIfError(writeStream.Replace(TheFs, parse.NameAndExt(), EFileWrite));

	REncryptStream encryptStream;

	TRAPD(ret, encryptStream.AttachL(writeStream, *elementKey));

	test(ret == KErrNone);

	encryptStream.PushL();

	encryptStream << KTestDes;

	encryptStream.CommitL();

	writeStream.CommitL();

	CleanupStack::PopAndDestroy(2); // encryptStream, writeStream

	const CPBEncryptionData& encryptData = elementKey->EncryptionData();

	CPBEncryptElement* newElementKey = CPBEncryptElement::NewLC(encryptData, KTestPassword);

	RFileReadStream readStream;

	readStream.PushL();

	User::LeaveIfError(readStream.Open(TheFs, parse.NameAndExt(), EFileWrite));

	RDecryptStream decryptStream;

	decryptStream.AttachL(readStream, *newElementKey);

	decryptStream.PushL();

	decryptStream >> TheBuf;

	test(KTestDes == TheBuf);

	CleanupStack::PopAndDestroy(4, elementKey);

	}

//

// Test writing to a store

//

// Writes two streams in a store - the first containing sections of

// the test data repeated, and the the second containing the test data

// and the id of the first stream.  Returns the id of the second

// stream.

//

LOCAL_C TStreamId testWriteL(CStreamStore& aStore)

	{

	test.Next(_L("Writing..."));

	RStoreWriteStream out;

	TStreamId id=out.CreateLC(aStore);

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

		{

		out.WriteL(KTestDes,i);

		out.WriteL(&KTestData[i],KTestLength-i);

		}

	out.CommitL();

	out.Close();

	TStreamId head=out.CreateL(aStore);

	out<<KTestDes;

	out<<id;

	out.CommitL();

	CleanupStack::PopAndDestroy();

	return head;

	}

//

// Test reading from a store

//

// Test that the data in a stream is the same as the data the

// testWriteL() function writes to the first stream.

//

LOCAL_C void testReadL(RReadStream& aStream)

	{

	for (TInt i=KTestLength;i>=0;--i)

		{

		aStream.ReadL(TheBuf,i);

		test(TheBuf.Length()==i);

		TheBuf.SetMax();

		aStream.ReadL(&TheBuf[i],KTestLength-i);

		TheBuf.SetLength(KTestLength);

		test(TheBuf==KTestDes);

		}

	}

//

// Test reading from a store

//

// Reads back and checks the data written to a store by testWriteL(),

// given the store and the id returned.

//

LOCAL_C void testReadL(const CStreamStore& aStore,TStreamId anId)

	{

	test.Next(_L("Reading..."));

	RStoreReadStream in;

	in.OpenLC(aStore,anId);

	in>>TheBuf;

	TStreamId id;

	in>>id;

	in.Close();

	in.OpenL(aStore,id);

	testReadL(in);

	CleanupStack::PopAndDestroy();

	}

//

// Test copying from one stream to another.

//

//

//

LOCAL_C void testCopyL(RWriteStream& aWriteStream,RReadStream& aReadStream)

	{

	test.Next(_L("Copying"));

	for (TInt i=KTestLength;i>=0;--i)

		{

		aWriteStream.WriteL(aReadStream,i);

		aReadStream.ReadL(aWriteStream,KTestLength-i);

		}

	}

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1128

@SYMTestCaseDesc	    Tests for writing to a store

@SYMTestPriority 	    High

@SYMTestActions  	    Attempt for writing to a root store and empty store.

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testWriteL()

	{

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1128 Replacing host file "));

	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	

	TParse parse;

	parse.Set(drive.Name(), &KFileLocationSpec, NULL);

	CFileStore* file=CPermanentFileStore::ReplaceLC(TheFs,parse.NameAndExt(),EFileWrite);

	file->SetTypeL(file->Layout());

//

	test.Next(_L("Writing root store"));

	RStoreWriteStream stream;

	TStreamId embed=stream.CreateL(*file);

	CEmbeddedStore* store=CEmbeddedStore::NewLC(stream);

	CSecureStore* secure=CSecureStore::NewLC(*store, *thePBEKey);

	store->SetRootL(testWriteL(*secure));

	CleanupStack::PopAndDestroy(secure);

	store->CommitL();

	CleanupStack::PopAndDestroy();

	__UHEAP_MARK;

	TStreamId root=stream.CreateLC(*file);

	REncryptStream encrypt;

	encrypt.AttachLC(stream,*thePBEKey);

	encrypt<<_L8(" root")<<embed;

	encrypt.CommitL();

	CleanupStack::PopAndDestroy(2);

	__UHEAP_MARKEND;

	file->SetRootL(root);

//

	test.Next(_L("Writing temp store"));

	embed=stream.CreateL(*file);

	store=CEmbeddedStore::NewLC(stream);

	secure = CSecureStore::NewLC(*store, *thePBEKey);

	store->SetRootL(testWriteL(*secure));

	CleanupStack::PopAndDestroy(secure);

	store->CommitL();

	CleanupStack::PopAndDestroy();

	__UHEAP_MARK;

	TheTempId=stream.CreateLC(*file);

	encrypt.OpenLC(stream,*thePBEKey);

	encrypt<<_L8(" temp")<<embed;

	encrypt.CommitL();

	CleanupStack::PopAndDestroy();

	stream.CommitL();

	CleanupStack::PopAndDestroy();

	__UHEAP_MARKEND;

//

	file->CommitL();

	CleanupStack::PopAndDestroy(file);

	}

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1129

@SYMTestCaseDesc	    Tests for reading from a store

@SYMTestPriority 	    High

@SYMTestActions  	    Attempt for reading from a root store and temporary store

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testReadL()

	{

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1129 Opening host file "));

	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	

	TParse parse;

	parse.Set(drive.Name(), &KFileLocationSpec, NULL);

	CFileStore* file=CFileStore::OpenLC(TheFs,parse.NameAndExt(),EFileRead);

//

	test.Next(_L("Reading from root store"));

	TStreamId root=file->Root();

	RStoreReadStream stream;

	stream.OpenLC(*file,root);

	RDecryptStream decrypt;

	decrypt.AttachLC(stream,*thePBEKey);

	TStreamId embed;

	TBuf8<5> b;

	decrypt>>b>>embed;

	test(b==_L8(" root"));

	CleanupStack::PopAndDestroy(2);

	stream.OpenL(*file,embed);

	CEmbeddedStore* store=CEmbeddedStore::FromLC(stream);

	CSecureStore* secure=NULL;

	secure=CSecureStore::NewLC(*store,*thePBEKey);

	testReadL(*secure,store->Root());

	CleanupStack::PopAndDestroy(2);

//

	test.Next(_L("Reading from temp store"));

	stream.OpenLC(*file,TheTempId);

	decrypt.OpenLC(stream,*thePBEKey);

	decrypt>>b>>embed;

	test(b==_L8(" temp"));

	CleanupStack::PopAndDestroy(2);

	stream.OpenL(*file,embed);

	store=CEmbeddedStore::FromLC(stream);

	secure=CSecureStore::NewLC(*store,*thePBEKey);

	testReadL(*secure,store->Root());

//

	CleanupStack::PopAndDestroy(3);

	}

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1130

@SYMTestCaseDesc	    Tests copying from one stream to another stream

@SYMTestPriority 	    High

@SYMTestActions  	    Attempt for copying using different transfer sizes

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testCopyL()

	{

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1130 Opening host file "));

	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	

	TParse parse;

	parse.Set(drive.Name(), &KFileLocationSpec, NULL);

	CFileStore* file=CFileStore::OpenLC(TheFs,parse.NameAndExt(),EFileRead|EFileWrite);

//

	test.Next(_L("Opening source (root) store"));

	TStreamId root=file->Root();

	RStoreReadStream src;

	src.OpenLC(*file,root);

	RDecryptStream decrypt;

	decrypt.OpenLC(src,*thePBEKey);

	TBuf8<5> b;

	TStreamId embed;

	decrypt>>b>>embed;

	test(b==_L8(" root"));

	CleanupStack::PopAndDestroy(2); // src, decrypt

	src.OpenL(*file,embed);

	CEmbeddedStore* source=CEmbeddedStore::FromLC(src);

	CSecureStore* ssource=NULL;

	ssource=CSecureStore::NewLC(*source,*thePBEKey);

	test.Next(_L("Creating target store"));

	RStoreWriteStream trg;

	trg.CreateL(*file);

	CEmbeddedStore* target=CEmbeddedStore::NewLC(trg);

	CSecureStore* starget=NULL;

	starget=CSecureStore::NewLC(*target,*thePBEKey);

	test.Next(_L("Copying using small transfers"));

	RStoreReadStream in;

	in.OpenL(*ssource,source->Root());

	in>>TheBuf;

	TStreamId copyId;

	in>>copyId;

	in.Close();

	in.OpenL(*ssource,copyId);

	RStoreWriteStream out;

	TStreamId id=out.CreateL(*starget);

	testCopyL(out,in);

	out.CommitL();

	out.Close();

	in.Close();

	in.OpenL(*starget,id);

	testReadL(in);

	in.Close();

//

	test.Next(_L("Copying using a single big transfer"));

	in.OpenL(*ssource,copyId);

	id=out.CreateL(*starget);

	in.ReadL(out,KTestTotal);

	out.CommitL();

	out.Close();

	in.Close();

	in.OpenL(*starget,id);

	testReadL(in);

	in.Close();

	in.OpenL(*ssource,copyId);

	id=out.CreateL(*starget);

	out.WriteL(in,KTestTotal);

	out.CommitL();

	out.Close();

	in.Close();

	in.OpenL(*starget,id);

	testReadL(in);

	in.Close();

//

	CleanupStack::PopAndDestroy(5);

	}

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1131

@SYMTestCaseDesc	    Writing and reading back a long stream test

@SYMTestPriority 	    High

@SYMTestActions  	    Attempt to write long data to a stream and read back and test for the integrity of the data.

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testBugL()

	{

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1131 Opening host file "));

	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	

	TParse parse;

	parse.Set(drive.Name(), &KFileLocationSpec, NULL);

	CFileStore* file=CDirectFileStore::ReplaceLC(TheFs,parse.NameAndExt(),EFileRead|EFileWrite);

	file->SetTypeL(file->Layout());

	CSecureStore* secure=NULL;

		secure=CSecureStore::NewLC(*file,*thePBEKey);

	test.Next(_L("Writing long stream"));

	RStoreWriteStream out;

	TStreamId id=out.CreateLC(*secure);

	TInt ii;

	for (ii=0;ii<400;++ii)

		out<<_L("a goodly length of data");

	out.CommitL();

	CleanupStack::PopAndDestroy();

	file->SetRootL(out.CreateLC(*secure));

	out<<_L("spam")<<id;

	out.CommitL();

	CleanupStack::PopAndDestroy(2, secure);

	file->CommitL();

	CleanupStack::PopAndDestroy();

//

	test.Next(_L("Opening host file"));

	file=CFileStore::OpenLC(TheFs,parse.NameAndExt(),EFileRead);

	secure=CSecureStore::NewLC(*file,*thePBEKey);

//

	TBuf<30> buf;

	test.Next(_L("Reading"));

	RStoreReadStream in;

	in.OpenLC(*secure,file->Root());

	in>>buf>>id;

	CleanupStack::PopAndDestroy();

	test(buf==_L("spam"));

	in.OpenLC(*secure,id);

	for (ii=0;ii<400;++ii)

		{

		in>>buf;

		test (buf==_L("a goodly length of data"));

		}

	CleanupStack::PopAndDestroy(3);

	}

/**

@SYMTestCaseID          PDS-STORE-CT-4016

@SYMTestCaseDesc	    Tests for CPBEncryptElement

@SYMTestPriority 	    High

@SYMTestActions  	    Externalizing and internalizing CPBEncryptionData. Tests for constructors.

@SYMTestExpectedResults Externalizing must not fail. After internalization CPBEncryptionData object should

						be valid. Object created with all constructors should be valid.

@SYMDEF                 DEF135804

*/

LOCAL_C void testForgottenAPI_L()

	{