// Copyright (c) 2003-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:

//

#include <e32std.h>

#include <e32std_private.h>

#include <f32file.h>

#include <d32locd.h>

#include <e32cons.h>

#include "crashflash.h"

#include <partitions.h>

#include <ftlcontrolio.h>

#ifdef _DEBUG

#define TRACE(a) RDebug::Print(a); PrintLine(a)

#define TRACE1(a,b) RDebug::Print(a,b); PrintLine(a,b)

#define TRACE2(a,b,c) RDebug::Print(a,b,c); PrintLine(a,b,c)

#define TRACE5(a,b,c,d,e,f) RDebug::Print(a,b,c,d,e,f); PrintLine(a,b,c,d,e,f)

#else

#define TRACE(a) 

#define TRACE1(a,b) 

#define TRACE2(a,b,c) 

#define TRACE5(a,b,c,d,e,f)

#endif

#ifndef _CRASHLOG_COMPR

_LIT(KCrashLogFileName, "?:\\crashlog.txt");

#else

_LIT(KCrashLogCompFileName, "?:\\crashlog.gz");

_LIT(KCrashLogCompTruncatedFileName, "?:\\crashlog_truncated.gz");

#endif //_CRASHLOG_COMPR

_LIT8(KCrashLogSignatureStomp, "\x00\x00\x00\x00");

CConsoleBase* console = 0;

RLocalDrive gLd;

TLocalDriveCapsV4 gCaps;

TPckg<TLocalDriveCapsV4> gCapsBuf(gCaps);

#ifdef _DEBUG

LOCAL_C void CheckConsoleCreated()

	{

	if(!console)

		{

		TRAPD(r, console = Console::NewL(_L("crashread"), 

			TSize(KConsFullScreen,KConsFullScreen)));

		__ASSERT_ALWAYS(r == KErrNone, User::Panic(_L("Could not create console"), 1));

		}

	}

LOCAL_C void PrintLine(TRefByValue<const TDesC> aFmt,...)

	{

    // Print to a console screen.

    VA_LIST list;

    VA_START(list, aFmt);

    TBuf<0x100> aBuf;

    aBuf.AppendFormatList(aFmt, list);

    CheckConsoleCreated();

    console->Write(aBuf);

	console->Write(_L("\n\r"));

	}

#endif

/** Read the signature from the flash and verify it is correct.

	@return ETrue when signature found, EFalse otherwise

*/

LOCAL_C TBool SignatureExistsL()

	{

	TBuf8<KCrashLogSignatureBytes> buf(0);

	User::LeaveIfError(gLd.Read(KCrashLogSizeFieldBytes,KCrashLogSignatureBytes,buf));

	if(buf.Compare(KCrashLogSignature) == 0)

		{

		return ETrue;

		}

	return EFalse;

	}

LOCAL_C TInt LogSizeL()

	{

	TBuf8<KCrashLogSizeFieldBytes> buf(0);

	User::LeaveIfError(gLd.Read(0,KCrashLogSizeFieldBytes,buf));

	TInt size = *((TUint*)(buf.Ptr()));

	size -= (KCrashLogHeaderSize);

	return size;

	}

#ifdef _CRASHLOG_COMPR	

/** Read the log flags from the flash.  Flags located after the log size and uncompressed size

	@return The log flags byte

*/

LOCAL_C TUint32 LogFlagsL()

	{

	TBuf8<KCrashLogFlagsFieldBytes> buf(0);

	User::LeaveIfError(gLd.Read(KCrashLogSizeFieldBytes+KCrashLogUncompSizeFieldBytes+KCrashLogSignatureBytes,

						KCrashLogFlagsFieldBytes,buf));

	return *((TUint32*)buf.Ptr());

	}

#endif //_CRASHLOG_COMPR

LOCAL_C TInt InvalidateSignature()

	{

	//On Nand we erase the block.

	if(gCaps.iType == EMediaNANDFlash)

		{

		return gLd.Format(0,gCaps.iNumBytesMain * gCaps.iNumPagesPerBlock);

		}

	//On Nor we just stomp on the first 4 bytes of the signature

	return gLd.Write(KCrashLogSizeFieldBytes,KCrashLogSignatureStomp);

	}

/**

@return KErrNone if no read errors, otherwise the last read error. 

@leave if other errors occur.	

@param aFileName Where the log wll be copied to

@param aStartPosition Where to begin reads within the flash section.

@param aLogSize The total amount to read.

*/

TInt CopyToFileL(const TDesC& aFileName, const TInt aStartPosition, const TInt aLogSize)

	{

	// Connect to f32 and write out the file

	RFs fs;

	RFile file;

	User::LeaveIfError(fs.Connect());

	CleanupClosePushL(fs);

	User::LeaveIfError(file.Replace(fs, aFileName, EFileWrite));

	CleanupClosePushL(file);

	//create buffer

	const TInt KBufferSize=32*1024;

	HBufC8* buf = HBufC8::NewLC(KBufferSize);

	TPtr8 ptr = buf->Des();

	TInt readError = KErrNone;

	for(TInt offset=0; offset<aLogSize; offset+=KBufferSize)

		{

		//don't read beyond end on final iteration.

		const TInt readLength = Min(KBufferSize, aLogSize-offset);

		ptr.SetLength(0);

		TInt r = gLd.Read(aStartPosition+offset,readLength,ptr);

		// in case of error store it, but attempt to continue.

		if (r!=KErrNone)

			{

			readError=r;

			}

		User::LeaveIfError(file.Write(offset, ptr));

		}

	User::LeaveIfError(file.Flush());

	CleanupStack::PopAndDestroy(buf);

	CleanupStack::PopAndDestroy(&file);

	CleanupStack::PopAndDestroy(&fs);

	return readError;

	}

LOCAL_C TInt MainL()

	{

	// check if command line argument is 'reset'

	RBuf cl;

	cl.CreateL(User::CommandLineLength());

	cl.CleanupClosePushL();

	User::CommandLine(cl);

	TBool reset = (cl==_L("reset"));

	CleanupStack::PopAndDestroy();

	TBool changed;

	TInt r = 0;

	TInt i=0;

	// 1) Find a crash log partition.

	for(; i<KMaxLocalDrives; i++)

		{

		r = gLd.Connect(i,changed);

		if(r == KErrNone)

			{

			r = gLd.Caps(gCapsBuf);

			if(r != KErrNone)

				{

				//TRACE1(_L("Could not retrieve gCaps for drive: %d.  Skipping to next..."),i);

				continue;

				}

			if(gCaps.iPartitionType == (TUint16)KPartitionTypeSymbianCrashLog)

				{

				TRACE1(_L("Found Symbian crash log partition on drive: %d"),i);

				CleanupClosePushL(gLd);

				// 1) See if there is an existing crash log

				TBool exists = SignatureExistsL();

				if(!exists)

					{

					TRACE(_L("Did not find an existing crash log signature on this crash log partition..."));

					//There may be a second crash log partition. (nor or nand

					//depending on ordering in variantmediadef.h).  So we continue searching

					CleanupStack::PopAndDestroy(&gLd);

					continue; 

					}

				TRACE1(_L("Found a crash log signature on drive: %d."),i);

				//We've found a crash log partition with a signature on it.

				break;

				}

			else

				{

				//TRACE2(_L("Partition type on drive: %d is %d"),i,gCaps.iPartitionType);

				}

			}

		}

	if(i == KMaxLocalDrives)

		{

		TRACE(_L("No crash log partition found with valid crash log signature found.  Exiting..."));

		User::Leave(KErrNotFound);

		}

	// If we're doing a reset, don't try to read the crash log, just skip to stomping the signature

	if(!reset)

		{

		TUint8 systemDriveChar = (TUint8) RFs::GetSystemDriveChar();

#ifndef _CRASHLOG_COMPR

		// Determine size of crash log and copy to file.

		TInt logSize = LogSizeL();

		TRACE1(_L("Reading crash log of %d bytes..."), logSize);

		TBuf<sizeof(KCrashLogFileName)> crashLogFileName(KCrashLogFileName);

		crashLogFileName[0] = systemDriveChar;

		r = CopyToFileL(crashLogFileName, KCrashLogSizeFieldBytes+KCrashLogSignatureBytes, logSize);

		if (r==KErrNone)

			{

			TRACE1(_L("Crash log successfully written to: %S."), &crashLogFileName);

			}

		else

			{

			TRACE1(_L("Crash log written to %S but errors were encountered when reading, it may be incomplete or corrupt."), &crashLogFileName);

			}

#else

		// 2) 	Read crash log header to get the compressed and uncompressed size of the log

		//		also need to read the flags to determine if the log had to be truncated and

		//		if the expected log format is found

		const TUint32 logFlags = LogFlagsL();

		// Extract byte offset from the end of the header to the start of the log data

		const TInt logOff = logFlags>>KCrashLogFlagOffShift;

		// Work out if the log had to be truncated

		const TInt truncated = logFlags&KCrashLogFlagTruncated;			

		// Check the crashlog type flag is that expected - here we can only cope with GZIP compatible logs

		if ((logFlags & (0xffffffff>>(32-KCrashLogFlagTypeBits))) != KCrashLogFlagGzip)

			{// wrong log type so can't extract it

			TRACE(_L("Crash Log data is stored in an incompatible data format so can't be read"));

			}

		else

			{

			// 2) Read the log data

			const TInt logSize = LogSizeL()-logOff; // don't include any offset bytes	

			TRACE1(_L("Reading compressed crash log of %d bytes..."), logSize);

			TRACE1(_L("Writing compressed crash log to file..."), logSize);

			RBuf crashLogCompFileName;

			if (!truncated)

				{

				crashLogCompFileName.CreateL(KCrashLogCompFileName);

				}

			else

				{

				crashLogCompFileName.CreateL(KCrashLogCompTruncatedFileName);

				}

			crashLogCompFileName.CleanupClosePushL();

			crashLogCompFileName[0] = systemDriveChar;

			r = CopyToFileL(crashLogCompFileName, KCrashLogHeaderSize+logOff, logSize);

			if (r==KErrNone)

				{

				if (!truncated)

					{

					TRACE1(_L("Crash log successfully written to: %S."), &crashLogCompFileName);

					}

				else

					{

					TRACE(_L("Crash log was truncated, some log data has been lost"));

					TRACE1(_L("Crash log successfully written to: %S."), &crashLogCompFileName);

					}						

				}

			else

				{

				if(!truncated)

					{

					TRACE1(_L("Crash log written to %S but errors were encountered when reading, it may be incomplete or corrupt."), &crashLogCompFileName);

					}

				else

					{

					TRACE1(_L("Crash log written to %S but errors were encountered when reading, it may be incomplete or corrupt."), &crashLogCompFileName);

					}

				}

			CleanupStack::PopAndDestroy(&crashLogCompFileName);

			}

#endif //_CRASHLOG_COMPR			

		}

	// 5) Stomp on the signature to mark it eligible to be overwritten

	TRACE(_L("Overwriting existing signature to indicate crash log has been read..."));

	User::LeaveIfError(InvalidateSignature());

	CleanupStack::PopAndDestroy(&gLd);

	if (r==KErrNone)

		{

		TRACE(_L("Crash reader finished successfully."));

		}

	else

		{

		TRACE(_L("Crash reader finished but with errors."));

		}

	return KErrNone;

	}

GLDEF_C TInt E32Main()

	{

	__UHEAP_MARK;

	CTrapCleanup* cleanup=CTrapCleanup::New();

	TRAPD(ret, MainL());

	if(console)

		{

		console->Getch();

		delete console;

		}

	if (ret){} // stops compile warning

	delete cleanup;

	__UHEAP_MARKEND;

	return KErrNone;

	}