// Copyright (c) 1999-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 <e32std.h>

#include <e32svr.h>

#include <f32file.h>

_LIT(KBlankLine, "\r");

class TDll

	{

public:

	TDll(const TDesC& aName, TUint8* aBase, TInt aSize)

		: iName(aName), iBase((TUint)aBase), iSize((TUint)aSize)

		{}

	TFullName iName;

	TUint iBase;

	TUint iSize;

	};

class TDllList

	{

public:

	static void FindDlls();

	static void ResetAndDestroy();

	static void ListDlls(TFileText& aTextFile);

	static TInt Match(TUint anAddr, const TDll*& aDll);

private:

	static RPointerArray<TDll> iList;

	};

RPointerArray<TDll> TDllList::iList(10);

void TDllList::ResetAndDestroy()

	{

	iList.ResetAndDestroy();

	}

void TDllList::ListDlls(TFileText& aTextFile)

	{

	TBuf<0x100> line;

	_LIT(KDllFormat, "%08x-%08x %S\r");

	aTextFile.Write(KBlankLine);

	for (TInt i=0; i<iList.Count(); i++)

		{

		const TDll& dll=*(iList[i]);

		line.Format(KDllFormat, dll.iBase, dll.iBase+dll.iSize-1, &dll.iName);

		aTextFile.Write(line);

		}

	}

TInt TDllList::Match(TUint anAddr, const TDll*& aDll)

	{

	for (TInt i=0; i<iList.Count(); i++)

		{

		aDll=iList[i];

		if (aDll->iBase<=anAddr && aDll->iBase+aDll->iSize > anAddr)

			return KErrNone;

		}

	return KErrNotFound;

	}

void TDllList::FindDlls()

//

// For each library known to the system, look for a chunk of the same name

//

	{

	TFindLibrary findLib;

	TFullName libName;

	while (findLib.Next(libName)==KErrNone)

		{

		TFindChunk findChunk(libName);

		TFullName chunkName;

		if (findChunk.Next(chunkName)!=KErrNone)

			continue;

		RChunk chunk;

		if (chunk.Open(findChunk)!=KErrNone)

			continue;

		TUint8* base=chunk.Base();

		TInt size=chunk.Size();

		chunk.Close();

		TDll* dllptr=new TDll(libName,base,size);

		if (dllptr!=0)

			iList.Append(dllptr);

		}

	}

_LIT(KFormatStackInfo,"Stack %08x-%08x (? %d?), sp=%08x\r");

//_LIT(KGrabStack,"Capture stack data");

//_LIT(KNoStack,"Don't risk it");

HBufC8* GrabStack(const TDesC& /*aLine1*/, TThreadId aId, TUint aSp, TInt& aStackBase, TInt& aStackSize)

	{

	TInt err=KErrNone;

#if 0

	// Defect in E32 162 which means that RThread::GetRamSizes will always reset the machine!

	RThread thread;

	if (thread.Open(aId)!=KErrNone)

		return 0;

	TInt heapsize=0;

	err=thread.GetRamSizes(heapsize,aStackSize);

	thread.Close();

	if (err!=KErrNone)

		return 0;

	// Sanity check

	aStackBase = aSp & ~((1024*1024)-1);

	if (aStackSize<0 || aStackSize>=1024*1024 || aSp>(TUint)(aStackBase+aStackSize))

		{

		aStackBase=0;	// indicates a daft stack pointer

		return 0;

		}

#else

	// Sanity check & guess at stack size

	aStackBase = aSp & ~((1024*1024)-1);

	aStackSize = (aSp - aStackBase + 4096) & ~4095;		// round up to a multiple of the page size

	if (aStackBase+aStackSize-aSp < 200)

			aStackSize += 4096;		// seems too small - risk another page!

	if (aStackSize<0 || aStackSize>=1024*1024)

		{

		aStackBase=0;	// indicates a daft stack pointer

		return 0;

		}

	aStackSize-=4;	// a clue to the wise that this is a guess...

#endif

	// Ask the user if we want to risk grabbing the stack...

	TBuf<0x100> line2;

	line2.Format(KFormatStackInfo, aStackBase, aStackBase+aStackSize-1, aStackSize, aSp);

    //Don't ask the user just do it for WSERV

	/*RNotifier ask;

	if (ask.Connect() != KErrNone)

		return 0;

	TRequestStatus status;

	TInt buttonval=1;

	ask.Notify(aLine1,line2,KGrabStack,KNoStack,buttonval,status);

	User::WaitForRequest(status);

	ask.Close();

	if (status.Int()!=KErrNone || buttonval != 0)

		return 0;*/

	// OK - let stack grabbing commence

	HBufC8* stackbuf = HBufC8::New(aStackSize);

	if (stackbuf==0)

		return 0;

	TPtr8 stackdes(stackbuf->Des());

	err=RDebug::ReadMemory(aId,aStackBase,stackdes,aStackSize);

	if (err!=KErrNone)

		{

		delete stackbuf;

		stackbuf=0;

		}

	return stackbuf;

	}

GLDEF_C TInt E32Main()

//

// Reporting more detail of KERN-EXEC 3 errors

//

// Warning: This code uses fragile assumptions which may not be true

// in future releases of EPOC

//

// 1) EXEs are located at 0x20000000

// 2) the map file for an EXE puts the start of text at 400010

// 3) the map file for a DLL puts the start of text at 10000010

// 4) the EPOC ROM lives at address 0x50000000

// 5) EPOC stacks start at a megabyte boundary

//

	{

	TBuf<0x100> line1;

	TBuf<0x100> line2;

	SDebugInfo info;

	struct SRegisterInfo reginfo;

	TUint pc;

	TUint regs[16];

	const TDll* faultDll;

	_LIT(KInfo1, "D_EXC started");

	User::InfoPrint(KInfo1);

	TRequestStatus stat(0);

//	FOREVER

	for (TInt rep=0; rep<2; rep++)	// die after two exceptions

		{

		TInt err;

		// wait for any thread to panic...

		err=RDebug::GetException(info,stat);

		if (err!=KErrNone)

			{

			_LIT(KInfo2, "RDebug failure");

			User::Panic(KInfo2, err);

			}

		User::WaitForRequest(stat);

		_LIT(KFormatPanic, "%S panic %S %d\r");

		_LIT(KFormatException, "%S exception type %d\r");

		_LIT(KFormatBreakpoint, "%S has breakpoint %d\r");

		switch (info.iDebugType)

			{

		case EPanic:

			line1.Format(KFormatPanic,&info.iName,&info.iCategory, info.iPanicType);

			break;

		case EException:

			line1.Format(KFormatException,&info.iName,info.iPanicType);

			break;

		case EBreakPoint:

			line1.Format(KFormatBreakpoint,&info.iName,info.iPanicType);

			break;

		default:

			continue;

			}

		// assume that it's KERN-EXEC 3 and try to use the 

		// full RDebug support to locate the faulting instruction

		HBufC8* stack=0;

		TInt stackbase=0;

		TInt stacksize=0;

		const TInt KStackPointerReg = 13;

		pc = 0x00000001;	// illegal value

		err = RDebug::Open();

		if (err==KErrNone)

			{

			err = RDebug::RegisterInfo(reginfo);

			if (!err)

				{

				RDebug::GetRegister(info.iId,reginfo.iNumberOfPcRegister, pc);

				for (int i=0; i<16; i++)

					RDebug::GetRegister(info.iId, i, regs[i]);

				}

			TDllList::FindDlls();

			stack=GrabStack(line1, info.iId, regs[KStackPointerReg], stackbase, stacksize);

			//RDebug::KillThread(info.iId);

			RDebug::Close();

			}

		_LIT(KFormatEXE,	"pc=%08x, .map equivalent %08x\r");

		_LIT(KFormatROM,	"pc=%08x, in ROM\r");

		_LIT(KFormatDll,	"pc=%08x, in DLL %S, .map equivalent %08x\r");

		_LIT(KFormatOther,	"pc=%08x, iCodeAddr=%08x\r");

		_LIT(KFormatError,	"(Unable to determine pc)\r");

		if ((pc&3) == 0)

			{

			if (pc >= 0x20000000 && pc < 0x30000000)

				line2.Format(KFormatEXE, pc, pc-0x20000000+0x400010);

			else

			if (pc >= 0x50000000 && pc < 0x60000000)

				line2.Format(KFormatROM, pc);

			else

			if (TDllList::Match(pc, faultDll)==KErrNone)

				line2.Format(KFormatDll, pc, &faultDll->iName, pc-(faultDll->iBase)+0x10000010);

			else

				line2.Format(KFormatOther, pc, info.iCodeAddr);

			}

		else

			line2.Copy(KFormatError);

		RFs fs;

		err = fs.Connect();

		if (err!=KErrNone)

			break;

		_LIT(KFormatFilename,"d:\\d_exc_%d.txt");

		_LIT(KFormatStackname,"d:\\d_exc_%d.stk");

		// Report the basic information about registers, DLLs etc

		TFileName name;

		name.Format(KFormatFilename, *(TUint*)&info.iId);

		RFile file;

		err=file.Replace(fs, name, EFileWrite+EFileShareAny+EFileStreamText);

		if (err!=KErrNone)

			break;

		TFileText textfile;

		textfile.Set(file);

		textfile.Write(line1);

		textfile.Write(line2);

		textfile.Write(KBlankLine);

		_LIT(KFormatRegisters,"r%02d=%08x %08x %08x %08x\r");

		line2.Format(KFormatRegisters, 0, regs[0], regs[1], regs[2], regs[3]);

		textfile.Write(line2);

		line2.Format(KFormatRegisters, 4, regs[4], regs[5], regs[6], regs[7]);

		textfile.Write(line2);

		line2.Format(KFormatRegisters, 8, regs[8], regs[9], regs[10], regs[11]);

		textfile.Write(line2);

		line2.Format(KFormatRegisters, 12,regs[12], regs[13], regs[14], regs[15]);

		textfile.Write(line2);

		if (stackbase!=0)

			{

			line2.Format(KFormatStackInfo, stackbase, stackbase+stacksize-1, stacksize, regs[KStackPointerReg]);

			textfile.Write(line2);

			}

		TDllList::ListDlls(textfile);

		TDllList::ResetAndDestroy();

		file.Close();

		// Dump the stack as binary data

		if (stack)

			{

			name.Format(KFormatStackname, *(TUint*)&info.iId);

			err=file.Replace(fs, name, EFileWrite+EFileShareAny+EFileStreamText);

			if (err==KErrNone)

				{

				file.Write(*stack);

				file.Close();

				}

			}

		delete stack;

		fs.Close();

		}

	return stat.Int();

	}