// 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 <s32mem.h>

#include <s32file.h>

#include <e32test.h>

const TInt KTestCleanupStack=0x20;

const TInt KTestExpandSize=0x20;

const TInt KTestGranularity=0x02;

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

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

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

LOCAL_D CTrapCleanup* TheTrapCleanup;

LOCAL_D RFs TheFs;

LOCAL_D CFileStore* TheStore;

LOCAL_D CBufStore* TheBufStore;

LOCAL_D RStoreWriteStream TheSink;

LOCAL_D RStoreReadStream TheSource;

class THexEncodeFilter : public TStreamFilter

	{

public:

	void Set(MStreamBuf* aBuf,TInt aMode);

private:

	TInt Capacity(TInt aMaxLength);

	TInt FilterL(TAny* aPtr,TInt aMaxLength,const TUint8*& aFrom,const TUint8* anEnd);

	void DoSynchL();

//

	static TText8 HexDigit(TUint aNibble);

	static TText8 HexMSN(TUint aByte);

	static TText8 HexLSN(TUint aByte);

private:

	TChar iSpare;

	};

class RHexDumpStream : public RWriteStream

	{

public:

	RHexDumpStream() {}

	RHexDumpStream(RWriteStream& aStream);

	void Open(RWriteStream& aStream);

private:

	THexEncodeFilter iFilter;

	};

class RHexLoadStream : public RReadStream

	{

public:

	RHexLoadStream() {}

	RHexLoadStream(RReadStream& aStream);

	void Open(RReadStream& aStream);

private:

	THexEncodeFilter iFilter;

	};

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1205

@SYMTestCaseDesc	    Streaming in and out tests

@SYMTestPriority 	    High

@SYMTestActions  	    Tests for copying and filtering

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testFixL(CArrayFix< TBuf<16> >& aFix)

	{

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1205 Test streaming in and out "));

	TBuf<16> aa(_L("aa"));

	aFix.AppendL(aa);

	TBuf<16> bb(_L("bbbb"));

	aFix.AppendL(bb);

	TStreamId id=TheSink.CreateL(*TheStore);

	TheSink<<aFix;

	TBuf<16> cc(_L("cccccc"));

	aFix.AppendL(cc);

	TheSink.Close();

	id=TheSink.CreateL(*TheStore);

	TheSink<<aFix;

	TheSink.Close();

	TheStore->SetRootL(id);

	TheStore->CommitL();

	aFix.Reset();

	TheSource.OpenL(*TheStore,TheStore->Root());

	TheSource>>aFix;

	TheSource.Close();

	test(aFix[0]==aa&&aFix[1]==bb&&aFix[2]==cc);

	test.Printf(_L("%S %S %S\n"),&aa,&bb,&cc);

	test.Printf(_L("%S %S %S\n"),&aFix[0],&aFix[1],&aFix[2]);

//

// Test copying and filtering.

//

	CBufSeg* bufp=CBufSeg::NewL(KTestExpandSize);

	CleanupStack::PushL(bufp);

//

	TheSource.OpenL(*TheStore,id);

	id=TheSink.CreateL(*TheStore);

//

	RBufWriteStream bout(*bufp);

	RHexDumpStream hout(bout);

	hout.WriteL(TheSource,10);

	RHexLoadStream hin(TheSource);

	hin.ReadL(bout,10);

	bout.CommitL();

//

	RBufReadStream bin(*bufp);

	bin.ReadL(TheSink);

//

	TheSink.Close();

	TheSource.Close();

//

	CleanupStack::PopAndDestroy();

	}

/**

@SYMTestCaseID          PDS-STORE-CT-4018

@SYMTestCaseDesc	    Testing RStoreWriteStream::Append API

@SYMTestPriority 	    High

@SYMTestActions  	    Open file for append, add data to Store

@SYMTestExpectedResults Data is succesfully added to store

@SYMDEF                 DEF135804

*/

LOCAL_C void testAppendL()

	{

	test.Next(_L("PDS-STORE-CT-4018: Testing Append API"));

	TheBufStore=CBufStore::NewLC(KTestExpandSize);

	//TStreamId id=TheSink.CreateL(*TheBufStore);

	TStreamId id = TheBufStore->ExtendL();

	TRAPD(r, TheSink.AppendL(*TheBufStore, id));

	test(r == KErrNone);

	TBuf<16> inbuf(_L("appendL"));

	TheSink<<inbuf;

	TheSink.CommitL();

	TheSink.Close();

	TBuf<16> outbuf;

	TheSource.OpenL(*TheBufStore,id);

	TheSource>>outbuf;

	TheSource.Close();

	test(inbuf==outbuf);

	test.Printf(_L("Write Buf: %S\n"),&inbuf);

	test.Printf(_L("Read Buf: %S\n"),&outbuf);

	CleanupStack::PopAndDestroy(TheBufStore);

	}

RHexDumpStream::RHexDumpStream(RWriteStream& aStream)

	{

	Open(aStream);

	}

void RHexDumpStream::Open(RWriteStream& aStream)

	{

	iFilter.Set(aStream.Sink(),THexEncodeFilter::EWrite);

	RWriteStream::Attach(&iFilter);

	}

RHexLoadStream::RHexLoadStream(RReadStream& aStream)

	{

	Open(aStream);

	}

void RHexLoadStream::Open(RReadStream& aStream)

	{

	iFilter.Set(aStream.Source(),THexEncodeFilter::ERead);

	RReadStream::Attach(&iFilter);

	}

void THexEncodeFilter::Set(MStreamBuf* aBuf,TInt aMode)

	{

	TStreamFilter::Set(aBuf,aMode);

	iSpare=TChar(0);

	}

TInt THexEncodeFilter::Capacity(TInt aMaxLength)

	{

	__ASSERT_DEBUG(aMaxLength>0,User::Panic(_L("aargh!"),0));

	return (iSpare.Eos()?aMaxLength+1:aMaxLength)>>1;

	}

TInt THexEncodeFilter::FilterL(TAny* aPtr,TInt aMaxLength,const TUint8*& aFrom,const TUint8* anEnd)

	{

	__ASSERT_DEBUG(aMaxLength>0,User::Panic(_L("aargh!"),0));

	TPtr8 hex((TText8*)aPtr,aMaxLength);

	const TUint8* bin=aFrom;

	if (!iSpare.Eos())

		{

		hex.Append(iSpare);

		aMaxLength--;

		}

	TInt n=Min(aMaxLength>>1,anEnd-bin);

	for (const TUint8* end=bin+n;bin<end;bin++)

		{

		TUint8 byte=*bin;

		hex.Append(HexMSN(byte));

		hex.Append(HexLSN(byte));

		}

	if (aMaxLength>(n<<1)&&bin<anEnd)

		{

		TUint8 byte=*bin++;

		hex.Append(HexMSN(byte));

		iSpare=HexLSN(byte);

		}

	else

		iSpare=TChar(0);

	__ASSERT_DEBUG(hex.Length()==hex.MaxLength()||bin==anEnd,User::Panic(_L("aargh!"),0));

	aFrom=bin;

	return hex.Length();

	}

void THexEncodeFilter::DoSynchL()

	{

	if (IsCommitted())

		return;

//

	if (!iSpare.Eos())

		{

		TText8 c=TText8(iSpare);

		iSpare=TChar(0);

		EmitL(&c,1);

		}

	TStreamFilter::DoSynchL();

	}

TText8 THexEncodeFilter::HexDigit(TUint aNibble)

	{

	return TText8(aNibble+(aNibble<10?'0':('a'-10)));

	}

TText8 THexEncodeFilter::HexMSN(TUint aByte)

	{

	return HexDigit(aByte>>4);

	}

TText8 THexEncodeFilter::HexLSN(TUint aByte)

	{

	return HexDigit(aByte&0x0f);

	}

LOCAL_C void doMainL()

	{

	test.Start(_L("Creating Store"));

	TParsePtrC parse(KFileLocationSpec);

//	TheStore=CBufStore::NewLC(KTestExpandSize);

	TheStore=CDirectFileStore::ReplaceLC(TheFs,parse.NameAndExt(),EFileRead|EFileWrite);

	TheStore->SetTypeL(TUidType(KDirectFileStoreLayoutUid,TUid::Uid('H'|('E'<<8)|('X'<<16))));

	test.Next(_L("Creating CArrayFixFlat"));

	CArrayFixFlat< TBuf<16> >* pFixFlat=new(ELeave) CArrayFixFlat< TBuf<16> >(KTestGranularity);

	CleanupStack::PushL(pFixFlat);

	testFixL(*pFixFlat);

	testAppendL();

	CleanupStack::PopAndDestroy(2,TheStore);

	}

//

// Prepare the test directory.

//

LOCAL_C void setupTestDirectory()

    {

	TInt r=TheFs.Connect();

	test(r==KErrNone);

//

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

	TParse parse;

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

	r=TheFs.MkDir(parse.DriveAndPath());

	test(r==KErrNone||r==KErrAlreadyExists);

	r=TheFs.SetSessionPath(parse.DriveAndPath());

	test(r==KErrNone);

	}

//

// Initialise the cleanup stack.

//

LOCAL_C void setupCleanup()

    {

	TheTrapCleanup=CTrapCleanup::New();

	test(TheTrapCleanup!=NULL);

	TRAPD(r,\

		{\

		for (TInt i=KTestCleanupStack;i>0;i--)\

			CleanupStack::PushL((TAny*)1);\

		test(r==KErrNone);\

		CleanupStack::Pop(KTestCleanupStack);\

		});

	test(r==KErrNone);

	}

LOCAL_C void DeleteDataFile(const TDesC& aFullName)

	{

	RFs fsSession;

	TInt err = fsSession.Connect();

	if(err == KErrNone)

		{

		TEntry entry;

		if(fsSession.Entry(aFullName, entry) == KErrNone)

			{

			RDebug::Print(_L("Deleting \"%S\" file.\n"), &aFullName);

			err = fsSession.SetAtt(aFullName, 0, KEntryAttReadOnly);

			if(err != KErrNone)

				{

				RDebug::Print(_L("Error %d changing \"%S\" file attributes.\n"), err, &aFullName);

				}

			err = fsSession.Delete(aFullName);

			if(err != KErrNone)

				{

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

				}

			}

		fsSession.Close();

		}

	else

		{

		RDebug::Print(_L("Error %d connecting file session. File: %S.\n"), err, &aFullName);

		}

	}

//

// Test the streaming framework.

//

GLDEF_C TInt E32Main()

    {

	test.Title();

	setupTestDirectory();

	setupCleanup();

	__UHEAP_MARK;

//

	TRAPD(r,doMainL());

	test(r==KErrNone);

	//deletion of data files must be before call to .End() - DEF047652

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

	TParse parse;

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

	::DeleteDataFile(parse.FullName());

	test.End();

//

	__UHEAP_MARKEND;

	delete TheTrapCleanup;

	TheFs.Close();

	test.Close();

	return 0;

    }