// 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 "PFSDUMP.H"

// entry point

int main(int argc,char *argv[])

	{

	if (TheProgram.ProcessCommandLine(argc,argv))

		{

		TheProgram.ExplainUsage();

		return 1;

		}

	TheProgram.Run();

	return 0;

	}

// Class Program

class Program TheProgram;

Program::Program()

	: iFile(NULL),iOptions(0),iBin(&iBinBuf),iErrors(0)

	{

	cout << setfill('0');

	}

void Program::Information()

	{

	if (!Option(quiet))

		cout << "\nEPOC PermanentFileStore Dump Utility   Version " << MajorVersion << '.' \

			<< setw(2) << MinorVersion << "(build " << setw(3) << setfill('0') << Build \

			<< ")\nCopyright (c) Symbian Software Limited 1997-2004. All rights reserved.\n\n" << flush;

	cout << hex;

	cerr << hex;

	}

void Program::ExplainUsage()

	{

	Information();

	cout << "Usage:  PFSDUMP [options] storefile\n" \

			" -v    verbose: dump contents\n" \

			" -f    generate a frame dump\n" \

			" -r<n> use nth previous store revision\n" \

			" -c    format output for comparison (not for frame dumps)\n"

			" -q    quiet: suppress logo\n" << flush;

	}

int Program::ProcessCommandLine(int argc,char ** argv)

	{

	if (argc<2)

		return 1;

	int options=0;

	while (--argc>0)

		{

		istrstream arg(*++argv);

		int c=arg.get();

		if (c=='/' || c=='-')

			{

			while ((c=arg.get())!=EOF)

				{

				switch (c)

					{

				case 'c': case 'C':

					options|=compare;

					break;

				case 'f': case 'F':

					options|=frame;

					break;

				case 'q': case 'Q':

					options|=quiet;

					break;

				case 'r': case 'R':

					arg >> iTocRevision;

					if (arg.fail() || iTocRevision<0)

						return 1;

					break;

				case 'v': case 'V':

					options|=verbose;

					break;

				default:			// unrecognised option

					return 1;

					}

				}

			}

		else if (iFile!=NULL)

			return 1;		// two filenames?

		else

			iFile=arg.str();

		}

	if (options&frame)

		options&=~compare;

	iOptions=options;

	return iFile==NULL;

	}

void Program::Run()

//

// The main part of the program

//

	{

	Information();

	StoreFile store(iFile);

	if (iErrors)

		cerr << endl;

	if (store.Empty())

		{

		cout << "Store is empty" << endl;

		return;

		}

	if (Option(frame))

		{

		store.EmitHeader();

		store.EmitFrames();

		}

	store.EmitToc();

	if (!Option(frame))

		store.EmitStreams();

	}

void Program::Abort(char const* aMessage)

	{

	cerr << aMessage << endl;

	exit(3);

	}

void Program::Corrupt(char const* aMessage)

//

// terminate after detecting a fatal corruption error

//

	{

	cerr << "\nfatal error: " << aMessage << "\ncannot continue\n" << flush;

	exit(2);

	}

ostream& Program::Error()

	{

	++TheProgram.iErrors;

	return cerr << "error: ";

	}

ostream& Program::Warning()

	{

	++TheProgram.iErrors;

	return cerr << "warning: ";

	}

// Bin stream buffer

binbuf::binbuf()

	: streambuf()

	{}

int binbuf::overflow(int)

	{return 0;}

int binbuf::underflow()

	{return EOF;}

// Class StoreFile

int StoreFile::OutWidth;

StoreFile::StoreFile(char const* aFile)

	{

#ifdef __MSVCDOTNET__

	iFile.open(aFile, ios::binary);

#else //!__MSVCDOTNET__

	iFile.open(aFile, ios::binary | ios::nocreate);

#endif //__MSVCDOTNET__

	if (!iFile)

		Program::Abort("Unable to open file or file does not exist");

//

	unsigned long uid;

	iFile.read((char*)&uid,sizeof(uid));

	iFile.seekg(0,ios::end);

	int size=iFile.tellg();

	if (size<FrameDes::First || uid!=PermanentFileStoreUid)

		Program::Abort("Not a permanent file store");

//

	iSize=size;

	if (TheProgram.Option(Program::compare))

		OutWidth=7;

	else

		{

		int width=0;

		while (size)

			{

			++width;

			size>>=4;

			}

		OutWidth=width;

		}

//

	cout << "Dumping " << aFile << "\n\n";

//

	iFile.seekg(Header::Offset,ios::beg);

	iFile.read((char*)&iHeader,Header::Size);

//

	if (Empty())

		return;

	LoadFrames();

	LoadToc();

	}

StoreFile::~StoreFile()

	{

	iFile.close();

	}

void StoreFile::LoadFrames()

	{

	FrameDes frame;

	int offset=FrameDes::First;

	int full=FrameDes::First+FrameDes::Interval;

	int diff=FrameDes::First;

	while (offset-FrameDes::Size<iSize)

		{

		if (offset==full)

			{

			full+=FrameDes::Interval;

			diff+=FrameDes::Size;

			}

		if (offset>iSize)

			{

			Program::Warning() << "incomplete link at " << FramePos(offset-diff) << endl;

			break;

			}

		iFile.seekg(offset-FrameDes::Size,ios::beg);

		iFile >> frame;

		iFrames.Add(FramePos(offset-diff),frame);

		int length=frame.Length();

		if (length==0)

			{

			if (full>iSize && offset>iSize)

				Program::Warning() << "incomplete frame at " << FramePos(offset-diff) << endl;

			offset=full;

			}

		else

			{

			int newoffset=offset+length+FrameDes::Size;

			if (newoffset>=full || newoffset-FrameDes::Size>iSize)

				{

				Program::Error() << "bad link at " << FramePos(offset-diff) << ", skipping to next anchor link" << endl;

				offset=full;

				}

			else

				{

				offset=newoffset;

				if (full-offset<=FrameDes::Size)

					offset=full;

				}

			}

		}

	iFrames.Complete();

	}

void StoreFile::LoadToc()

	{

	FramePos toc=iHeader.Toc();

	Stream stream(iFrames,toc);

	if (!stream.IsGood())

		{

		Program::Error() << "invalid toc address " << toc << endl;

		return;

		}

	if (stream.Type()!=FrameDes::Toc)

		{

		Program::Error() << "toc address " << toc << ": refers to non-toc frame"<< endl;

		return;

		}

// find the requested store revision

	Frames::Iterator f=stream.Frame();

	Frames::Iterator const first=iFrames.Begin();

	for (int rev=TheProgram.TocRevision();rev;--rev)

		{

		do	{

			if (--f<first)

				Program::Abort("Store revision not found");

			} while (f->iDes.Type()!=FrameDes::Toc);

		}

	iToc.Load(iFile,iFrames,f,iHeader.GetReloc());

// verify the Toc stream references

	Toc::Iterator const end=iToc.End();

	for (Toc::Iterator iter=iToc.Begin();iter<end;++iter)

		{

		if (iter->iHandle.IsNull())

			Program::Error() << "missing entry in toc-delta for index " << (1+iter-iToc.Begin()) << endl;

		else if (!iter->iHandle.Avail() && iter->Pos().Pos()>=0)

			{

			f=iFrames.Find(iter->Pos());

			if (f==NULL)

				Program::Error() << "invalid stream reference in toc entry " << iter->iHandle << endl;

			else if (iter->Pos().Pos()>=toc.Pos())

				Program::Error() << "virtual stream reference in toc entry " << iter->iHandle << endl;

			else if (f->iDes.Type()!=FrameDes::Data)

				Program::Error() << "toc entry " << iter->iHandle << ": refers to non-data frame" << endl;

			}

		}

	}

void StoreFile::EmitHeader()

	{

	cout << iHeader;

	}

void StoreFile::EmitFrames()

	{

	int verbose=TheProgram.Option(Program::verbose);

	Frames::Iterator const end=iFrames.End();

	for (Frames::Iterator iter=iFrames.Begin();iter<end;++iter)

		{

		FramePos pos=iter->iPos;

		cout << "Frame at " << pos << ": " << iter->iDes << endl;

		if (!verbose)

			continue;

	// dump contents

		int length=iter->iDes.Length();

		if (length==0)

			{	// full frame

			if (iter+1==end)	// no more

				continue;

			length=iter[1].iPos.Pos()-pos.Pos();

			}

		HexDump hex;

		iFile.seekg(FileOffset(pos).Offset(),ios::beg);

		hex.Dump(iFile,length);

		hex.Flush();

		cout << endl;

		}

	cout << endl;

	}

void StoreFile::EmitToc()

	{

	cout << iToc;

	}

void StoreFile::EmitStreams()

	{

	int verbose=TheProgram.Option(Program::verbose);

	cout << endl;

	Toc::Iterator const end=iToc.End();

	for (Toc::Iterator iter=iToc.Begin();iter<end;++iter)

		{

		if (!iter->iHandle.Avail())

			{

			cout << "Stream " << iter->iHandle;

			if (iter->Pos().Pos()==-1)

				{

				cout << " is empty\n";

				continue;

				}

			if (!TheProgram.Option(Program::compare))

				cout << " at " << iter->Pos();

			Stream stream(iFrames,iter->Pos());

			if (!stream.IsGood() || stream.Type()!=FrameDes::Data)

				{

				cout << " is invalid\n";

				continue;

				}

			int len=stream.Length();

			cout << ", " << dec << len << hex << " byte" << (len==1 ? "\n" : "s\n");

			if (!verbose)

				continue;

		// dump contents

			HexDump hex;

			Frames::Iterator f=stream.Frame();

			do

				{

				FramePos pos=f->iPos;

				int len=f++->iDes.Length();

				if (len==0)

					len=f->iPos.Pos()-pos.Pos();

				iFile.seekg(FileOffset(pos).Offset(),ios::beg);

				hex.Dump(iFile,len);

				} while (f->iDes.Type()==FrameDes::Continuation);

			hex.Flush();

			cout << endl;

			}

		}

	}

// Class HexDump

HexDump::HexDump()

	: iPos(0),iByte(0)

	{

	Reset();

	}

HexDump::~HexDump()

	{

	Flush();

	}

void HexDump::Reset()

	{

	memset(iChar,' ',HexInterval+3);

	iChar[HexInterval+3]=0;

	iPtr=&iChar[2];

	}

void HexDump::Dump(istream& aStream,int aBytes)

	{

	while (--aBytes>=0 && !aStream.eof())

		{

		if (iByte==0)

			cout << Margin << FileOffset(iPos) << ':';

		int c=aStream.get();

		cout << ' ' << setw(2) << c;

		*iPtr++=char(isprint(c) ? c : '.');

		if (++iByte==HexInterval/2)

			{

			cout << ' ';

			++iPtr;

			}

		else if (iByte==HexInterval)

			Flush();

		}

	}

void HexDump::Flush()

	{

	if (iByte)

		{

		for (int ii=iByte;ii<HexInterval;++ii)

			cout << "   ";

		if (iByte<HexInterval/2)

			cout << ' ';

		cout << iChar << endl;

		iPos+=iByte;

		iByte=0;

		Reset();

		}

	}

// Toc

Toc::Toc()

	: iPos(0), iRep(NULL), iAvail(0)

	{

	memset(&iHeader,0,sizeof(iHeader));

	}

Toc::~Toc()

	{

	free(iRep);

	}

void Toc::Base(const char* aPtr,int aCount)

	{

	Entry* e=iRep;

	for (int i=1;i<=aCount;++e,++i)

		{

		e->iHandle=i;	// set the index part

		memcpy((char*)e+Entry::BaseRedundant,aPtr,Entry::BaseSize);

		aPtr+=Entry::BaseSize;

		}

	}

void Toc::Load(istream& aStream,Frames const& aFrames,Frames::Iterator aFrame,Header::Reloc const* aReloc)

	{

	iPos = aFrame->iPos;

	Stream toc1(aFrame);

	void* toc = toc1.Load(aStream);

	const char* p = reinterpret_cast<char*>(toc);

	memcpy(&iHeader,p,Head::Size);

	p+=Head::Size;

	int n = iHeader.iCount;

	if (n < 0)

		{

		memset(&iHeader,0,Head::Size);

		Program::Error() << "corrupt toc" << endl;

		return;

		}

	iRep=static_cast<Entry*>(malloc(n*sizeof(Entry)));

	if (iRep==NULL)

		Program::Abort("Out of memory");

	if (iHeader.IsDelta())

		{

		// verify the delta header

		memcpy(&iDelta,p,DeltaHead::Size);

		p+=DeltaHead::Size;

		int dn = iDelta.iCount;

		if (toc1.Length() != Head::Size + DeltaHead::Size + dn * Entry::DeltaSize)

			{

			memset(&iHeader,0,Head::Size);

			Program::Error() << "incomplete toc" << endl;

			return;

			}

		// find the toc-base

		FramePos tocbase(iDelta.iBase + Header::tocoffset);

		if (aReloc && aReloc->iHandle.IsTocBase())

			tocbase = aReloc->iPos;

		Stream toc2(aFrames,tocbase);

		if (!toc2.IsGood())

			{

			memset(&iHeader,0,Head::Size);

			Program::Error() << "invalid toc-base address " << tocbase << endl;

			return;

			}

		if (toc2.Type()!=FrameDes::Toc)

			{