// ffstrace.cpp
// 
// Copyright (c) 2010 Accenture. All rights reserved.
// This component and the accompanying materials are made available
// under the terms of the "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:
// Accenture - Initial contribution
//

#include <fshell/extrabtrace.h>
#include <fshell/btrace_parser.h>

#include <fshell/ioutils.h>
#include <fshell/common.mmh>

using namespace IoUtils;

class CCmdFfstrace : public CCommandBase, public MBtraceObserver
	{
public:
	static CCommandBase* NewLC();
	~CCmdFfstrace();
private:
	CCmdFfstrace();
	void HandleBtraceFrameL(const TBtraceFrame& aFrame);
private: // From CCommandBase.
	virtual const TDesC& Name() const;
	virtual void DoRunL();
	virtual void ArgumentsL(RCommandArgumentList& aArguments);
	virtual void OptionsL(RCommandOptionList& aOptions);
private:
	enum TCmd
		{
		EMonitor,
		ELoad,
		EUnload,
		};
	TCmd iCommand;
private:
	CBtraceReader* iBtraceReader;
	};

EXE_BOILER_PLATE(CCmdFfstrace)

CCommandBase* CCmdFfstrace::NewLC()
	{
	CCmdFfstrace* self = new(ELeave) CCmdFfstrace();
	CleanupStack::PushL(self);
	self->BaseConstructL();
	return self;
	}

CCmdFfstrace::~CCmdFfstrace()
	{
	delete iBtraceReader;
	}

CCmdFfstrace::CCmdFfstrace()
	: CCommandBase(EManualComplete)
	{
	}

const TDesC& CCmdFfstrace::Name() const
	{
	_LIT(KName, "ffstrace");	
	return KName;
	}

void CCmdFfstrace::ArgumentsL(RCommandArgumentList& aArguments)
	{
	aArguments.AppendEnumL((TInt&)iCommand, _L("command"));
	}

void CCmdFfstrace::OptionsL(RCommandOptionList& /*aOptions*/)
	{
	//TODO: aOptions.AppendBoolL(iOpt, _L("example_opt"));
	//TODO: Also remember to update the CIF file for any options you add.
	}

void CCmdFfstrace::DoRunL()
	{
	_LIT(KPluginDll, "ffstraceplugin.fxt");
	_LIT(KFfsTracerPluginName, "FfsTracePlugin");
	if (iCommand == EMonitor || iCommand == ELoad)
		{
		TInt err = FsL().AddPlugin(KPluginDll);
		if (err && err != KErrAlreadyExists) LeaveIfErr(err, _L("Couldn't load filesystem plugin %S"), &KPluginDll);

		err = FsL().MountPlugin(KFfsTracerPluginName);
		//TODO handle this being called repeatedly
		LeaveIfErr(err, _L("Couldn't mount filesystem plugin %S"), &KFfsTracerPluginName);
		}
	else if (iCommand == EUnload)
		{
		FsL().DismountPlugin(KFfsTracerPluginName);
		LeaveIfErr(FsL().RemovePlugin(KFfsTracerPluginName), _L("No plugin to unload"));
		}

	if (iCommand == EMonitor)
		{
		iBtraceReader = CBtraceReader::NewL(CBtraceReader::EFlushOnBtraceThreshold, 1024*1024, 512*1024);
		iBtraceReader->AddObserverL(BTrace::EFfsTrace, *this);
		iBtraceReader->SetMultipartReassemblyL(10);
		TBtraceTickCount now;
		now.SetToNow();
		iBtraceReader->Start(now, 1000000);
		}
	else
		{
		Complete(KErrNone);
		}
	}

void CCmdFfstrace::HandleBtraceFrameL(const TBtraceFrame& aFrame)
	{
	if (aFrame.iCategory == BTrace::EFfsTrace)
		{
		TInt fn = aFrame.iSubCategory & ~BTrace::EFfsPost;
		TBool post = aFrame.iSubCategory & BTrace::EFfsPost;
		TUint threadId = *(TUint*)aFrame.iData.Ptr();
		TPtrC name((TUint16*)(aFrame.iData.Ptr()+4), (aFrame.iData.Size()-4)/2);
		TPtrC newName;
		// Not everything gets as far as post - if an error occurs during the pre checking for eg
		if (post) Printf(_L("-"));
		else Printf(_L("+"));

		TBool twoNames = (fn == BTrace::EFfsRename || fn == BTrace::EFfsFileRename);
		if (twoNames)
			{
			TInt separator = name.Locate(0);
			if (separator != KErrNotFound)
				{
				newName.Set(name.Mid(separator+1));
				name.Set(name.Left(separator));
				}
			}

		switch (fn)
			{
			case BTrace::EFfsDelete:
				Printf(_L("Delete %S"), &name);
				break;
			case BTrace::EFfsFileOpen:
				Printf(_L("Open %S"), &name);
				break;
			case BTrace::EFfsFileCreate:
				Printf(_L("Create %S"), &name);
				break;
			case BTrace::EFfsFileSubClose:
				Printf(_L("Close %S"), &name);
				break;
			case BTrace::EFfsFileReplace:
				Printf(_L("Replace %S"), &name);
				break;
			case BTrace::EFfsFileTemp:
				Printf(_L("Open temp %S"), &name);
				break;
			case BTrace::EFfsRename:
			case BTrace::EFfsFileRename:
				Printf(_L("Rename from %S to %S"), &name, &newName);
				break;
			case BTrace::EFfsEntry:
				Printf(_L("RFs::Entry %S"), &name);
				break;
			default:
				Printf(_L("Event %d"), fn);
				break;
			}
		Printf(_L(" from thread %u"), threadId);
		RThread thread;
		TInt err = thread.Open(threadId);
		if (err == KErrNone)
			{
			Printf(_L(" "));
			TFullName name = thread.FullName();
			Write(name);
			thread.Close();
			}
		Printf(_L("\r\n"));
		}
	}