// 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 the License "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:
// e32test\misc\flash.cpp
// 
//

#include <e32test.h>
#include <f32file.h>
#include "flash.h"

RTest test(_L("FLASH"));
TInt Mode;
TFileName FileName;
RFs Fs;
RFile File;
TUint32 StartAddr;
TUint32 Size;

const TUint KBufferSize=4096;
TUint8 Buffer[KBufferSize];
TUint8 Buffer2[KBufferSize];

enum TMode
	{
	EDump,
	EBlankCheck,
	EErase,
	EWrite,
	EVerify,
	};

void Usage()
	{
	test.Printf(_L("Usage:\n"));
	test.Printf(_L(" flash d <addr> <size> <file> - dump flash to file\n"));
	test.Printf(_L(" flash b <addr> <size>        - blank check flash\n"));
	test.Printf(_L(" flash e <addr> <size>        - erase flash\n"));
	test.Printf(_L(" flash w <addr> <file>        - write file to flash\n"));
	test.Printf(_L(" flash v <addr> <file>        - verify flash against file\n"));
	}

TInt ParseCommandLine()
	{
	TBuf<256> cmd;
	User::CommandLine(cmd);
	cmd.Trim();
	if (cmd.Length()==0)
		return KErrArgument;
	TLex lex(cmd);
	TPtrC mode_str=lex.NextToken();
	switch (mode_str[0])
		{
		case 'D':
		case 'd':
			Mode=EDump; break;
		case 'B':
		case 'b':
			Mode=EBlankCheck; break;
		case 'E':
		case 'e':
			Mode=EErase; break;
		case 'W':
		case 'w':
			Mode=EWrite; break;
		case 'V':
		case 'v':
			Mode=EVerify; break;
		default:
			return KErrArgument;
		}
	lex.SkipSpace();
	TInt r=lex.Val(StartAddr,EHex);
	if (r!=KErrNone)
		return r;
	if (Mode==EDump || Mode==EBlankCheck || Mode==EErase)
		{
		lex.SkipSpace();
		r=lex.Val(Size,EHex);
		if (r!=KErrNone)
			return r;
		}
	if (Mode==EDump || Mode==EWrite || Mode==EVerify)
		{
		lex.SkipSpace();
		FileName=lex.NextToken();
		if (FileName.Length()==0)
			return KErrArgument;
		if (Mode==EDump)
			{
			r=File.Replace(Fs,FileName,EFileWrite);
			if (r!=KErrNone)
				{
				test.Printf(_L("\nCould not open file %S for write\n"),&FileName);
				return KErrGeneral;
				}
			}
		else
			{
			r=File.Open(Fs,FileName,EFileRead);
			if (r!=KErrNone)
				{
				test.Printf(_L("\nCould not open file %S for read\n"),&FileName);
				return KErrGeneral;
				}
			else
				{
				r=File.Size((TInt&)Size);
				if (r!=KErrNone)
					{
					test.Printf(_L("\nError %d accessing file %S\n"),r,&FileName);
					File.Close();
					return KErrGeneral;
					}
				test.Printf(_L("File size %x\n"),Size);
				}
			}
		}
	return KErrNone;
	}

void DoCommand()
	{
	Flash* pF=Flash::New(StartAddr);
	if (!pF)
		{
		test.Printf(_L("Problem with flash device\n"));
		return;
		}
	TInt r=KErrNotSupported;
	switch (Mode)
		{
		case EDump:
			{
			while (Size)
				{
				TUint size=Size;
				if (size>KBufferSize)
					size=KBufferSize;
				r=pF->Read(StartAddr,size,Buffer);
				if (r!=KErrNone)
					break;
				r=File.Write(TPtrC8(Buffer,size));
				if (r!=KErrNone)
					break;
				StartAddr+=size;
				Size-=size;
				}
			break;
			}
		case EBlankCheck:
			{
			r=pF->BlankCheck(StartAddr,Size);
			if (r!=KErrNone)
				{
				test.Printf(_L("Error at address %08x\n"),StartAddr+r-1);
				}
			break;
			}
		case EErase:
			r=pF->Erase(StartAddr,Size);
			if (r!=KErrNone)
				{
				test.Printf(_L("Error at address %08x\n"),StartAddr+r-1);
				}
			break;
		case EWrite:
			{
			while (Size)
				{
				test.Printf(_L("%08x\n"),StartAddr);
				TUint size=Size;
				if (size>KBufferSize)
					size=KBufferSize;
				TPtr8 p8(Buffer,0,size);
				r=File.Read(p8);
				if (r!=KErrNone)
					break;
				r=pF->Write(StartAddr,size,Buffer);
				if (r!=KErrNone)
					break;
				StartAddr+=size;
				Size-=size;
				}
			break;
			}
		case EVerify:
			{
			while (Size)
				{
				TUint size=Size;
				if (size>KBufferSize)
					size=KBufferSize;
				r=pF->Read(StartAddr,size,Buffer);
				if (r!=KErrNone)
					break;
				TPtr8 p8(Buffer2,0,size);
				r=File.Read(p8);
				if (r!=KErrNone)
					break;
				TUint i;
				for (i=0; i<size; ++i)
					{
					if (Buffer2[i]!=Buffer[i])
						{
						test.Printf(_L("VERIFY ERROR: Flash addr %08x, flash %02x, file %02x\n"),
									StartAddr+i,Buffer[i],Buffer2[i]);
						return;
						}
					}
				StartAddr+=size;
				Size-=size;
				}
			break;
			}
		}
	if (r!=KErrNone)
		test.Printf(_L("Error %d occurred\n"),r);
	delete pF;
	}

TInt E32Main()
	{
	test.Title();
	test.Start(_L("Connect to F32"));
	TInt r=Fs.Connect();
	test(r==KErrNone);

	r=ParseCommandLine();
	if (r==KErrNone)
		DoCommand();
	else
		Usage();

	Fs.Close();
	test.End();
	return 0;
	}