sl@0: // Copyright (c) 1998-2009 Nokia Corporation and/or its subsidiary(-ies).
sl@0: // All rights reserved.
sl@0: // This component and the accompanying materials are made available
sl@0: // under the terms of "Eclipse Public License v1.0"
sl@0: // which accompanies this distribution, and is available
sl@0: // at the URL "http://www.eclipse.org/legal/epl-v10.html".
sl@0: //
sl@0: // Initial Contributors:
sl@0: // Nokia Corporation - initial contribution.
sl@0: //
sl@0: // Contributors:
sl@0: //
sl@0: // Description:
sl@0: //
sl@0: 
sl@0: #include <s32file.h>
sl@0: #include <s32crypt.h>
sl@0: #include <e32test.h>
sl@0: #include <s32mem.h>
sl@0: #include <pbedata.h>
sl@0: #include <pbe.h>
sl@0: 
sl@0: const TInt KTestCleanupStack=0x20;
sl@0: 
sl@0: // This is a path specification and should not be used as is
sl@0: _LIT(KFileLocationSpec, "Z:\\STOR-TST\\T_CRYPT.DAT");
sl@0: const TUint8* KTestData=_S8("0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ");
sl@0: const TInt KTestLength=36;
sl@0: const TInt KTestTotal=KTestLength*(KTestLength+1);
sl@0: const TPtrC8 KTestDes(KTestData,KTestLength);
sl@0: const TPBPassword KTestPassword(_L("The password"));
sl@0: 
sl@0: LOCAL_D CTrapCleanup* TheTrapCleanup;
sl@0: LOCAL_D RTest test(_L("t_storcrypt"));
sl@0: LOCAL_D RFs TheFs;
sl@0: LOCAL_D TStreamId TheTempId;
sl@0: LOCAL_D TBuf8<KTestLength+1> TheBuf;
sl@0: 
sl@0: LOCAL_D CPBEncryptSet* thePBEKey;
sl@0: 
sl@0: /**
sl@0: @SYMTestCaseID          SYSLIB-STORE-CT-1126
sl@0: @SYMTestCaseDesc	    Tests for writing and reading an encrypted data to store
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Encrypt data and write to a store.Read the encrypted data for decrypting and compare the result.
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMREQ                 REQ0000
sl@0: */
sl@0: LOCAL_C void testEncryptionDataL()
sl@0:     {
sl@0: 	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1126 "));
sl@0: 	CPBEncryptElement* elementKey = CPBEncryptElement::NewLC(KTestPassword);
sl@0: 	TParsePtrC parse(KFileLocationSpec);
sl@0: 	
sl@0: 	RFileWriteStream writeStream;
sl@0:     writeStream.PushL();
sl@0:     User::LeaveIfError(writeStream.Replace(TheFs, parse.NameAndExt(), EFileWrite));
sl@0: 
sl@0:     REncryptStream encryptStream;
sl@0:     encryptStream.OpenLC(writeStream, *elementKey);
sl@0: 
sl@0:     encryptStream << KTestDes;
sl@0:     encryptStream.CommitL();
sl@0:     writeStream.CommitL();
sl@0:     CleanupStack::PopAndDestroy(2); // encryptStream, writeStream
sl@0: 
sl@0:     const CPBEncryptionData& encryptData = elementKey->EncryptionData();
sl@0: 
sl@0:     CPBEncryptElement* newElementKey = CPBEncryptElement::NewLC(encryptData, KTestPassword);
sl@0: 
sl@0:     RFileReadStream readStream;
sl@0:     readStream.PushL();
sl@0:     User::LeaveIfError(readStream.Open(TheFs, parse.NameAndExt(), EFileWrite));
sl@0: 
sl@0:     RDecryptStream decryptStream;
sl@0:     decryptStream.OpenLC(readStream, *newElementKey);
sl@0: 
sl@0:     decryptStream >> TheBuf;
sl@0: 
sl@0:     test(KTestDes == TheBuf);
sl@0: 
sl@0: 	CleanupStack::PopAndDestroy(4, elementKey);
sl@0:     }
sl@0: 
sl@0: /**
sl@0: @SYMTestCaseID          SYSLIB-STORE-CT-1127
sl@0: @SYMTestCaseDesc	    Stream encryption test.
sl@0:                         Tests the functionality of REncryptStream and RDecryptStream classes
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Tests for writing an encrypted string to a stream,reading back the data,decrypt and
sl@0:                         check with the original data.
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMREQ                 REQ0000
sl@0: */
sl@0: LOCAL_C void testSimpleStreamEncryptionL()
sl@0:     {
sl@0: 	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1127 "));
sl@0:    	CPBEncryptElement* elementKey = CPBEncryptElement::NewL(KTestPassword);
sl@0: 	CleanupStack::PushL(elementKey);
sl@0: 	TParsePtrC parse(KFileLocationSpec);
sl@0: 	
sl@0: 	RFileWriteStream writeStream;
sl@0:     writeStream.PushL();
sl@0:     User::LeaveIfError(writeStream.Replace(TheFs, parse.NameAndExt(), EFileWrite));
sl@0: 
sl@0: 	REncryptStream encryptStream;
sl@0: 	encryptStream.OpenLC(writeStream, *elementKey);
sl@0: 
sl@0:     encryptStream << KTestDes;
sl@0:     encryptStream.CommitL();
sl@0:     writeStream.CommitL();
sl@0:     CleanupStack::PopAndDestroy(2); // encryptStream, writeStream
sl@0: 
sl@0:     RFileReadStream readStream;
sl@0:     readStream.PushL();
sl@0:     User::LeaveIfError(readStream.Open(TheFs, parse.NameAndExt(), EFileWrite));
sl@0: 
sl@0:     RDecryptStream decryptStream;
sl@0: 	decryptStream.OpenLC(readStream, *elementKey);
sl@0: 
sl@0:     decryptStream >> TheBuf;
sl@0: 
sl@0:     test(KTestDes == TheBuf);
sl@0: 
sl@0:     CleanupStack::PopAndDestroy(3, elementKey); // decryptStream, readStream
sl@0:     }
sl@0: 
sl@0: /**
sl@0: @SYMTestCaseID          PDS-STORE-CT-4042
sl@0: @SYMTestCaseDesc	    Tests for attaching write stream and read stream to an encrypted stream.
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Encrypt data and write to a store.Read the encrypted data for decrypting and compare the result.
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMDEF                 DEF135804
sl@0: */
sl@0: LOCAL_C void testEncryptionDataAttachL()
sl@0: 	{
sl@0: 	test.Next(_L(" @SYMTestCaseID:PDS-STORE-CT-4042 "));
sl@0: 	CPBEncryptElement* elementKey = CPBEncryptElement::NewLC(KTestPassword);
sl@0: 	TParsePtrC parse(KFileLocationSpec);
sl@0: 	
sl@0: 	RFileWriteStream writeStream;
sl@0: 	writeStream.PushL();
sl@0: 	User::LeaveIfError(writeStream.Replace(TheFs, parse.NameAndExt(), EFileWrite));
sl@0: 	
sl@0: 	REncryptStream encryptStream;
sl@0: 	TRAPD(ret, encryptStream.AttachL(writeStream, *elementKey));
sl@0: 	test(ret == KErrNone);
sl@0: 	encryptStream.PushL();
sl@0: 	
sl@0: 	encryptStream << KTestDes;
sl@0: 	encryptStream.CommitL();
sl@0: 	writeStream.CommitL();
sl@0: 	
sl@0: 	CleanupStack::PopAndDestroy(2); // encryptStream, writeStream
sl@0: 	
sl@0: 	const CPBEncryptionData& encryptData = elementKey->EncryptionData();
sl@0: 	
sl@0: 	CPBEncryptElement* newElementKey = CPBEncryptElement::NewLC(encryptData, KTestPassword);
sl@0: 	
sl@0: 	RFileReadStream readStream;
sl@0: 	readStream.PushL();
sl@0: 	User::LeaveIfError(readStream.Open(TheFs, parse.NameAndExt(), EFileWrite));
sl@0: 	
sl@0: 	
sl@0: 	RDecryptStream decryptStream;
sl@0: 	decryptStream.AttachL(readStream, *newElementKey);
sl@0: 	decryptStream.PushL();
sl@0: 	
sl@0: 	decryptStream >> TheBuf;
sl@0: 	
sl@0: 	test(KTestDes == TheBuf);
sl@0: 	
sl@0: 	CleanupStack::PopAndDestroy(4, elementKey);
sl@0: 	}
sl@0: 
sl@0: 
sl@0: //
sl@0: // Test writing to a store
sl@0: //
sl@0: // Writes two streams in a store - the first containing sections of
sl@0: // the test data repeated, and the the second containing the test data
sl@0: // and the id of the first stream.  Returns the id of the second
sl@0: // stream.
sl@0: //
sl@0: LOCAL_C TStreamId testWriteL(CStreamStore& aStore)
sl@0: 	{
sl@0: 	test.Next(_L("Writing..."));
sl@0: 	RStoreWriteStream out;
sl@0: 	TStreamId id=out.CreateLC(aStore);
sl@0: 	for (TInt i=0;i<=KTestLength;++i)
sl@0: 		{
sl@0: 		out.WriteL(KTestDes,i);
sl@0: 		out.WriteL(&KTestData[i],KTestLength-i);
sl@0: 		}
sl@0: 	out.CommitL();
sl@0: 	out.Close();
sl@0: 	TStreamId head=out.CreateL(aStore);
sl@0: 	out<<KTestDes;
sl@0: 	out<<id;
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 	return head;
sl@0: 	}
sl@0: 
sl@0: //
sl@0: // Test reading from a store
sl@0: //
sl@0: // Test that the data in a stream is the same as the data the
sl@0: // testWriteL() function writes to the first stream.
sl@0: //
sl@0: LOCAL_C void testReadL(RReadStream& aStream)
sl@0: 	{
sl@0: 	for (TInt i=KTestLength;i>=0;--i)
sl@0: 		{
sl@0: 		aStream.ReadL(TheBuf,i);
sl@0: 		test(TheBuf.Length()==i);
sl@0: 		TheBuf.SetMax();
sl@0: 		aStream.ReadL(&TheBuf[i],KTestLength-i);
sl@0: 		TheBuf.SetLength(KTestLength);
sl@0: 		test(TheBuf==KTestDes);
sl@0: 		}
sl@0: 	}
sl@0: 
sl@0: //
sl@0: // Test reading from a store
sl@0: //
sl@0: // Reads back and checks the data written to a store by testWriteL(),
sl@0: // given the store and the id returned.
sl@0: //
sl@0: LOCAL_C void testReadL(const CStreamStore& aStore,TStreamId anId)
sl@0: 	{
sl@0: 	test.Next(_L("Reading..."));
sl@0: 	RStoreReadStream in;
sl@0: 	in.OpenLC(aStore,anId);
sl@0: 	in>>TheBuf;
sl@0: 	TStreamId id;
sl@0: 	in>>id;
sl@0: 	in.Close();
sl@0: 	in.OpenL(aStore,id);
sl@0: 	testReadL(in);
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 	}
sl@0: 
sl@0: //
sl@0: // Test copying from one stream to another.
sl@0: //
sl@0: //
sl@0: //
sl@0: LOCAL_C void testCopyL(RWriteStream& aWriteStream,RReadStream& aReadStream)
sl@0: 	{
sl@0: 	test.Next(_L("Copying"));
sl@0: 	for (TInt i=KTestLength;i>=0;--i)
sl@0: 		{
sl@0: 		aWriteStream.WriteL(aReadStream,i);
sl@0: 		aReadStream.ReadL(aWriteStream,KTestLength-i);
sl@0: 		}
sl@0: 	}
sl@0: /**
sl@0: @SYMTestCaseID          SYSLIB-STORE-CT-1128
sl@0: @SYMTestCaseDesc	    Tests for writing to a store
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Attempt for writing to a root store and empty store.
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMREQ                 REQ0000
sl@0: */
sl@0: LOCAL_C void testWriteL()
sl@0: 	{
sl@0: 	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1128 Replacing host file "));
sl@0: 	
sl@0: 	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	
sl@0: 	TParse parse;
sl@0: 	parse.Set(drive.Name(), &KFileLocationSpec, NULL);
sl@0: 	
sl@0: 	CFileStore* file=CPermanentFileStore::ReplaceLC(TheFs,parse.NameAndExt(),EFileWrite);
sl@0: 	file->SetTypeL(file->Layout());
sl@0: //
sl@0: 	test.Next(_L("Writing root store"));
sl@0: 	RStoreWriteStream stream;
sl@0: 	TStreamId embed=stream.CreateL(*file);
sl@0: 	CEmbeddedStore* store=CEmbeddedStore::NewLC(stream);
sl@0: 
sl@0: 	CSecureStore* secure=CSecureStore::NewLC(*store, *thePBEKey);
sl@0: 
sl@0: 	store->SetRootL(testWriteL(*secure));
sl@0: 	CleanupStack::PopAndDestroy(secure);
sl@0: 	store->CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 
sl@0: 	__UHEAP_MARK;
sl@0: 	TStreamId root=stream.CreateLC(*file);
sl@0: 	REncryptStream encrypt;
sl@0: 	encrypt.AttachLC(stream,*thePBEKey);
sl@0: 
sl@0: 	encrypt<<_L8(" root")<<embed;
sl@0: 	encrypt.CommitL();
sl@0: 	CleanupStack::PopAndDestroy(2);
sl@0: 	__UHEAP_MARKEND;
sl@0: 
sl@0: 	file->SetRootL(root);
sl@0: //
sl@0: 	test.Next(_L("Writing temp store"));
sl@0: 	embed=stream.CreateL(*file);
sl@0: 	store=CEmbeddedStore::NewLC(stream);
sl@0: 	secure = CSecureStore::NewLC(*store, *thePBEKey);
sl@0: 
sl@0: 	store->SetRootL(testWriteL(*secure));
sl@0: 	CleanupStack::PopAndDestroy(secure);
sl@0: 	store->CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 
sl@0: 	__UHEAP_MARK;
sl@0: 	TheTempId=stream.CreateLC(*file);
sl@0: 	encrypt.OpenLC(stream,*thePBEKey);
sl@0: 
sl@0: 	encrypt<<_L8(" temp")<<embed;
sl@0: 	encrypt.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 	stream.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 	__UHEAP_MARKEND;
sl@0: //
sl@0: 	file->CommitL();
sl@0: 	CleanupStack::PopAndDestroy(file);
sl@0: 	}
sl@0: /**
sl@0: @SYMTestCaseID          SYSLIB-STORE-CT-1129
sl@0: @SYMTestCaseDesc	    Tests for reading from a store
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Attempt for reading from a root store and temporary store
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMREQ                 REQ0000
sl@0: */
sl@0: LOCAL_C void testReadL()
sl@0: 	{
sl@0: 	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1129 Opening host file "));
sl@0: 	
sl@0: 	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	
sl@0: 	TParse parse;
sl@0: 	parse.Set(drive.Name(), &KFileLocationSpec, NULL);
sl@0: 	
sl@0: 	CFileStore* file=CFileStore::OpenLC(TheFs,parse.NameAndExt(),EFileRead);
sl@0: //
sl@0: 	test.Next(_L("Reading from root store"));
sl@0: 	TStreamId root=file->Root();
sl@0: 	RStoreReadStream stream;
sl@0: 	stream.OpenLC(*file,root);
sl@0: 	RDecryptStream decrypt;
sl@0: 	decrypt.AttachLC(stream,*thePBEKey);
sl@0: 
sl@0: 	TStreamId embed;
sl@0: 	TBuf8<5> b;
sl@0: 	decrypt>>b>>embed;
sl@0: 	test(b==_L8(" root"));
sl@0: 	CleanupStack::PopAndDestroy(2);
sl@0: 	stream.OpenL(*file,embed);
sl@0: 	CEmbeddedStore* store=CEmbeddedStore::FromLC(stream);
sl@0: 	CSecureStore* secure=NULL;
sl@0: 	secure=CSecureStore::NewLC(*store,*thePBEKey);
sl@0: 
sl@0: 	testReadL(*secure,store->Root());
sl@0: 	CleanupStack::PopAndDestroy(2);
sl@0: //
sl@0: 	test.Next(_L("Reading from temp store"));
sl@0: 	stream.OpenLC(*file,TheTempId);
sl@0: 	decrypt.OpenLC(stream,*thePBEKey);
sl@0: 
sl@0: 	decrypt>>b>>embed;
sl@0: 	test(b==_L8(" temp"));
sl@0: 	CleanupStack::PopAndDestroy(2);
sl@0: 	stream.OpenL(*file,embed);
sl@0: 	store=CEmbeddedStore::FromLC(stream);
sl@0: 	secure=CSecureStore::NewLC(*store,*thePBEKey);
sl@0: 
sl@0: 	testReadL(*secure,store->Root());
sl@0: //
sl@0: 	CleanupStack::PopAndDestroy(3);
sl@0: 	}
sl@0: /**
sl@0: @SYMTestCaseID          SYSLIB-STORE-CT-1130
sl@0: @SYMTestCaseDesc	    Tests copying from one stream to another stream
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Attempt for copying using different transfer sizes
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMREQ                 REQ0000
sl@0: */
sl@0: LOCAL_C void testCopyL()
sl@0: 	{
sl@0: 	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1130 Opening host file "));
sl@0: 	
sl@0: 	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	
sl@0: 	TParse parse;
sl@0: 	parse.Set(drive.Name(), &KFileLocationSpec, NULL);
sl@0: 	
sl@0: 	CFileStore* file=CFileStore::OpenLC(TheFs,parse.NameAndExt(),EFileRead|EFileWrite);
sl@0: //
sl@0: 	test.Next(_L("Opening source (root) store"));
sl@0: 	TStreamId root=file->Root();
sl@0: 	RStoreReadStream src;
sl@0: 	src.OpenLC(*file,root);
sl@0: 	RDecryptStream decrypt;
sl@0: 	decrypt.OpenLC(src,*thePBEKey);
sl@0: 
sl@0: 	TBuf8<5> b;
sl@0: 	TStreamId embed;
sl@0: 	decrypt>>b>>embed;
sl@0: 	test(b==_L8(" root"));
sl@0: 	CleanupStack::PopAndDestroy(2); // src, decrypt
sl@0: 	src.OpenL(*file,embed);
sl@0: 	CEmbeddedStore* source=CEmbeddedStore::FromLC(src);
sl@0: 	CSecureStore* ssource=NULL;
sl@0: 	ssource=CSecureStore::NewLC(*source,*thePBEKey);
sl@0: 
sl@0: 	test.Next(_L("Creating target store"));
sl@0: 	RStoreWriteStream trg;
sl@0: 	trg.CreateL(*file);
sl@0: 	CEmbeddedStore* target=CEmbeddedStore::NewLC(trg);
sl@0: 	CSecureStore* starget=NULL;
sl@0: 	starget=CSecureStore::NewLC(*target,*thePBEKey);
sl@0: 
sl@0: 	test.Next(_L("Copying using small transfers"));
sl@0: 	RStoreReadStream in;
sl@0: 	in.OpenL(*ssource,source->Root());
sl@0: 	in>>TheBuf;
sl@0: 	TStreamId copyId;
sl@0: 	in>>copyId;
sl@0: 	in.Close();
sl@0: 	in.OpenL(*ssource,copyId);
sl@0: 	RStoreWriteStream out;
sl@0: 	TStreamId id=out.CreateL(*starget);
sl@0: 	testCopyL(out,in);
sl@0: 	out.CommitL();
sl@0: 	out.Close();
sl@0: 	in.Close();
sl@0: 	in.OpenL(*starget,id);
sl@0: 	testReadL(in);
sl@0: 	in.Close();
sl@0: //
sl@0: 	test.Next(_L("Copying using a single big transfer"));
sl@0: 	in.OpenL(*ssource,copyId);
sl@0: 	id=out.CreateL(*starget);
sl@0: 	in.ReadL(out,KTestTotal);
sl@0: 	out.CommitL();
sl@0: 	out.Close();
sl@0: 	in.Close();
sl@0: 	in.OpenL(*starget,id);
sl@0: 	testReadL(in);
sl@0: 	in.Close();
sl@0: 	in.OpenL(*ssource,copyId);
sl@0: 	id=out.CreateL(*starget);
sl@0: 	out.WriteL(in,KTestTotal);
sl@0: 	out.CommitL();
sl@0: 	out.Close();
sl@0: 	in.Close();
sl@0: 	in.OpenL(*starget,id);
sl@0: 	testReadL(in);
sl@0: 	in.Close();
sl@0: //
sl@0: 	CleanupStack::PopAndDestroy(5);
sl@0: 	}
sl@0: /**
sl@0: @SYMTestCaseID          SYSLIB-STORE-CT-1131
sl@0: @SYMTestCaseDesc	    Writing and reading back a long stream test
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Attempt to write long data to a stream and read back and test for the integrity of the data.
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMREQ                 REQ0000
sl@0: */
sl@0: LOCAL_C void testBugL()
sl@0: 	{
sl@0: 	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1131 Opening host file "));
sl@0: 	
sl@0: 	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	
sl@0: 	TParse parse;
sl@0: 	parse.Set(drive.Name(), &KFileLocationSpec, NULL);
sl@0: 	
sl@0: 	CFileStore* file=CDirectFileStore::ReplaceLC(TheFs,parse.NameAndExt(),EFileRead|EFileWrite);
sl@0: 	file->SetTypeL(file->Layout());
sl@0: 	CSecureStore* secure=NULL;
sl@0: 
sl@0: 		secure=CSecureStore::NewLC(*file,*thePBEKey);
sl@0: 
sl@0: 	test.Next(_L("Writing long stream"));
sl@0: 	RStoreWriteStream out;
sl@0: 	TStreamId id=out.CreateLC(*secure);
sl@0: 	TInt ii;
sl@0: 	for (ii=0;ii<400;++ii)
sl@0: 		out<<_L("a goodly length of data");
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 	file->SetRootL(out.CreateLC(*secure));
sl@0: 	out<<_L("spam")<<id;
sl@0: 	out.CommitL();
sl@0: 	CleanupStack::PopAndDestroy(2, secure);
sl@0: 	file->CommitL();
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: //
sl@0: 	test.Next(_L("Opening host file"));
sl@0: 	file=CFileStore::OpenLC(TheFs,parse.NameAndExt(),EFileRead);
sl@0: 	secure=CSecureStore::NewLC(*file,*thePBEKey);
sl@0: //
sl@0: 	TBuf<30> buf;
sl@0: 	test.Next(_L("Reading"));
sl@0: 	RStoreReadStream in;
sl@0: 	in.OpenLC(*secure,file->Root());
sl@0: 	in>>buf>>id;
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 	test(buf==_L("spam"));
sl@0: 	in.OpenLC(*secure,id);
sl@0: 	for (ii=0;ii<400;++ii)
sl@0: 		{
sl@0: 		in>>buf;
sl@0: 		test (buf==_L("a goodly length of data"));
sl@0: 		}
sl@0: 	CleanupStack::PopAndDestroy(3);
sl@0: 	}
sl@0: 
sl@0: /**
sl@0: @SYMTestCaseID          PDS-STORE-CT-4016
sl@0: @SYMTestCaseDesc	    Tests for CPBEncryptElement
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Externalizing and internalizing CPBEncryptionData. Tests for constructors.
sl@0: @SYMTestExpectedResults Externalizing must not fail. After internalization CPBEncryptionData object should
sl@0: 						be valid. Object created with all constructors should be valid.
sl@0: @SYMDEF                 DEF135804
sl@0: */
sl@0: LOCAL_C void testForgottenAPI_L()
sl@0: 	{
sl@0: 	test.Next(_L("@SYMTestCaseID PDS-STORE-CT-4016: Tests for CPBEncryptElement"));
sl@0: 	CBufFlat* buffer = CBufFlat::NewL(10*1024);
sl@0: 	CleanupStack::PushL(buffer);
sl@0: 	RBufWriteStream wstr(*buffer,0);
sl@0: 	
sl@0: 	CPBEncryptElement* elementKey = CPBEncryptElement::NewLC(KTestPassword);
sl@0:     const CPBEncryptionData& encryptData = elementKey->EncryptionData();    
sl@0:     wstr << encryptData;
sl@0:     CleanupStack::PopAndDestroy();
sl@0: 	
sl@0: 	wstr.CommitL();
sl@0: 	wstr.Close();
sl@0: 	
sl@0: 	RBufReadStream rstr(*buffer,0);
sl@0: 	CPBEncryptionData* enData = CPBEncryptionData::NewL(rstr);
sl@0: 	test(enData != NULL);
sl@0: 	delete enData;
sl@0: 	enData = NULL;
sl@0: 	rstr.Close();
sl@0: 	rstr.Open(*buffer,0);
sl@0: 	enData = CPBEncryptionData::NewLC(rstr);
sl@0: 	test(enData != NULL);
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 	enData = NULL;
sl@0: 	
sl@0: 	rstr.Close();
sl@0: 	
sl@0: 	CleanupStack::PopAndDestroy();
sl@0: 	}
sl@0: 
sl@0: /**
sl@0: @SYMTestCaseID          PDS-STORE-CT-4019
sl@0: @SYMTestCaseDesc	    Tests adding and deleting a stream from securestore
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Create a new stream and Delete it again
sl@0: @SYMTestExpectedResults Test must not fail
sl@0: @SYMDEF                 DEF135804
sl@0: */
sl@0: LOCAL_C void testExtendDeleteL()
sl@0: 	{
sl@0: 	test.Next(_L("@SYMTestCaseID PDS-STORE-CT-4019"));
sl@0: 	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	
sl@0: 	TParse parse;
sl@0: 	parse.Set(drive.Name(), &KFileLocationSpec, NULL);
sl@0: 		
sl@0: 	test.Next(_L("Testing Extend"));
sl@0: 	CFileStore* file=CPermanentFileStore::ReplaceLC(TheFs,parse.NameAndExt(),EFileRead|EFileWrite);
sl@0: 	file->SetTypeL(file->Layout());
sl@0: 	CSecureStore* secure=NULL;
sl@0: 
sl@0: 	secure=CSecureStore::NewLC(*file,*thePBEKey);
sl@0: 	TStreamId id = secure->ExtendL();
sl@0: 	test(id != NULL);
sl@0: 	test.Next(_L("Testing Delete"));		
sl@0: 	TRAPD(r, secure->DeleteL(id));
sl@0: 	test(r == KErrNone);
sl@0: 	TRAP(r, secure->DeleteL(id));
sl@0: 	test(r == KErrNotFound);
sl@0: 	
sl@0: 	CleanupStack::PopAndDestroy(2);
sl@0: 	
sl@0: 	}
sl@0: 
sl@0: /**
sl@0: @SYMTestCaseID          PDS-STORE-CT-4020
sl@0: @SYMTestCaseDesc	    Tests writing and replacing a stream from securestore
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    Create a new stream in the store and then open it for replacement
sl@0: @SYMTestExpectedResults stream is created and writtenn to successfully
sl@0: @SYMDEF                 DEF135804
sl@0: */
sl@0: LOCAL_C void testWriteReplaceL()
sl@0: 	{
sl@0: 	test.Next(_L("@SYMTestCaseID PDS-STORE-CT-4020"));
sl@0: 	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	
sl@0: 	TParse parse;
sl@0: 	parse.Set(drive.Name(), &KFileLocationSpec, NULL);
sl@0: 		
sl@0: 	test.Next(_L("Testing Extend"));
sl@0: 	CFileStore* file=CPermanentFileStore::ReplaceLC(TheFs,parse.NameAndExt(),EFileRead|EFileWrite);
sl@0: 	file->SetTypeL(file->Layout());
sl@0: 	CSecureStore* secure=NULL;
sl@0: 
sl@0: 	secure=CSecureStore::NewLC(*file,*thePBEKey);
sl@0: 	RStoreWriteStream out;
sl@0: 	TStreamId id = out.CreateL(*secure);
sl@0: 	test(id != NULL);
sl@0: 	out.Close();
sl@0: 	TRAPD(r, out.OpenL(*secure, id) );
sl@0: 	test(r == KErrNone);
sl@0: 	out.WriteL(KTestDes,8);
sl@0: 	out.CommitL();
sl@0: 	out.Close();
sl@0: 	r = secure->Commit();
sl@0: 	test(r == KErrNone);
sl@0: 	
sl@0: 	TRAP(r,out.ReplaceL(*secure, id));
sl@0: 	test(r == KErrNone);
sl@0: 	out.WriteL(KTestDes,8);
sl@0: 	out.CommitL();
sl@0: 	out.Close();
sl@0: 	secure->Revert();
sl@0: 	out.Close();
sl@0: 	
sl@0: 	CleanupStack::PopAndDestroy(2);
sl@0: 		
sl@0: 	}
sl@0: 
sl@0: 
sl@0: LOCAL_C void setupTestDirectory()
sl@0: //
sl@0: // Prepare the test directory.
sl@0: //
sl@0:     {
sl@0: 	TInt r=TheFs.Connect();
sl@0: 	test(r==KErrNone);
sl@0: //
sl@0: 	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	
sl@0: 	TParse parse;
sl@0: 	parse.Set(drive.Name(), &KFileLocationSpec, NULL);
sl@0: 	
sl@0: 	r=TheFs.MkDir(parse.DriveAndPath());
sl@0: 	test(r==KErrNone||r==KErrAlreadyExists);
sl@0: 	r=TheFs.SetSessionPath(parse.DriveAndPath());
sl@0: 	test(r==KErrNone);
sl@0: 	}
sl@0: 
sl@0: LOCAL_C void setupCleanup()
sl@0: //
sl@0: // Initialise the cleanup stack.
sl@0: //
sl@0:     {
sl@0: 	TheTrapCleanup=CTrapCleanup::New();
sl@0: 	test(TheTrapCleanup!=NULL);
sl@0: 	TRAPD(r,\
sl@0: 		{\
sl@0: 		for (TInt i=KTestCleanupStack;i>0;i--)\
sl@0: 			CleanupStack::PushL((TAny*)0);\
sl@0: 		CleanupStack::Pop(KTestCleanupStack);\
sl@0: 		});
sl@0: 	test(r==KErrNone);
sl@0: 	}
sl@0: 
sl@0: LOCAL_C void runTestsL()
sl@0:     {
sl@0: 	test.Start(_L("Test direct file store"));
sl@0: 	TInt r = KErrNone;
sl@0: 
sl@0: //	New PBE tests
sl@0: 	TRAP(r,testEncryptionDataL());
sl@0: 	test(r==KErrNone);
sl@0: 	TRAP(r,testSimpleStreamEncryptionL());
sl@0: 	test(r==KErrNone);
sl@0: 	
sl@0: 	
sl@0: 	TRAP(r,testEncryptionDataAttachL());
sl@0: 	test(r==KErrNone);
sl@0: 	
sl@0: 
sl@0: //	Old API tests with new PBE parameters
sl@0: 	thePBEKey = CPBEncryptSet::NewL(KTestPassword);
sl@0: 
sl@0: 	TRAP(r,testWriteL());
sl@0: 	test(r==KErrNone);
sl@0: 	TRAP(r,testReadL());
sl@0: 	test(r==KErrNone);
sl@0: 	TRAP(r,testCopyL());
sl@0: 	test(r==KErrNone);
sl@0: 	TRAP(r,testBugL());
sl@0: 	test(r==KErrNone);
sl@0: 	TRAP(r,testForgottenAPI_L());
sl@0: 	test(r==KErrNone);
sl@0: 	TRAP(r,testExtendDeleteL());
sl@0: 	test(r==KErrNone);
sl@0: 	TRAP(r,testWriteReplaceL());
sl@0: 		test(r==KErrNone);
sl@0: 
sl@0: 	delete thePBEKey;
sl@0: 
sl@0:     }
sl@0: 
sl@0: LOCAL_C void DeleteDataFile(const TDesC& aFullName)
sl@0: 	{
sl@0: 	RFs fsSession;
sl@0: 	TInt err = fsSession.Connect();
sl@0: 	if(err == KErrNone)
sl@0: 		{
sl@0: 		TEntry entry;
sl@0: 		if(fsSession.Entry(aFullName, entry) == KErrNone)
sl@0: 			{
sl@0: 			RDebug::Print(_L("Deleting \"%S\" file.\n"), &aFullName);
sl@0: 			err = fsSession.SetAtt(aFullName, 0, KEntryAttReadOnly);
sl@0: 			if(err != KErrNone)
sl@0: 				{
sl@0: 				RDebug::Print(_L("Error %d changing \"%S\" file attributes.\n"), err, &aFullName);
sl@0: 				}
sl@0: 			err = fsSession.Delete(aFullName);
sl@0: 			if(err != KErrNone)
sl@0: 				{
sl@0: 				RDebug::Print(_L("Error %d deleting \"%S\" file.\n"), err, &aFullName);
sl@0: 				}
sl@0: 			}
sl@0: 		fsSession.Close();
sl@0: 		}
sl@0: 	else
sl@0: 		{
sl@0: 		RDebug::Print(_L("Error %d connecting file session. File: %S.\n"), err, &aFullName);
sl@0: 		}
sl@0: 	}
sl@0: 
sl@0: GLDEF_C TInt E32Main()
sl@0: //
sl@0: // Test streaming conversions.
sl@0: //
sl@0:     {
sl@0: 	test.Title();
sl@0: 	setupTestDirectory();
sl@0: 	setupCleanup();
sl@0: 	__UHEAP_MARK;
sl@0: //
sl@0:     TRAPD(r, runTestsL());
sl@0:     test(r == KErrNone);
sl@0: 
sl@0:     //deletion of data files must be before call to .End() - DEF047652
sl@0:     TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	
sl@0: 	TParse parse;
sl@0: 	parse.Set(drive.Name(), &KFileLocationSpec, NULL);
sl@0: 	::DeleteDataFile(parse.FullName());
sl@0: 	
sl@0:     test.End();
sl@0: //
sl@0: 	__UHEAP_MARKEND;
sl@0: 
sl@0: 	delete TheTrapCleanup;
sl@0: 	TheFs.Close();
sl@0: 	test.Close();
sl@0: 	return r!=KErrNone;
sl@0:     }