// symbolics.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/bsym.h>

#include <fshell/ltkutils.h>

#include <fshell/stringhash.h>

#include <fshell/iocli.h>

#include "bsymtree.h"

using namespace LtkUtils;

using namespace IoUtils;

EXPORT_C CSymbolics::CSymbolics(RFs& aFs)

	: iFs(aFs)

	{

	}

EXPORT_C CSymbolics::~CSymbolics()

	{

	iBsyms.ResetAndDestroy();

	TStringHashIter<CMapFile*> iter(iCodeSegHash);

	while (iter.NextValue() != NULL)

		{

		CMapFile** file = iter.CurrentValue();

		delete *file;

		}

	iCodeSegHash.Close();

	delete iTabCompleteTree;

	delete iTabCompleteCodeseg;

	}

EXPORT_C void CSymbolics::AddBsymFileL(const TDesC& aFileName)

	{

	CBsymFile* file = CBsymFile::NewL(iFs, aFileName);

	TInt err = iBsyms.Append(file);

	if (err)

		{

		delete file;

		User::Leave(err);

		}

	}

EXPORT_C void CSymbolics::AddMapFileL(const TDesC& aFileName)

	{

	CMapFile* file = CMapFile::NewL(iFs, aFileName);

	CleanupStack::PushL(file);

	AddMapFileL(file);

	CleanupStack::Pop(file);

	}

EXPORT_C void CSymbolics::AddMapFileL(CMapFile* aMapFile)

	{

	aMapFile->GetFileNameL(iTempString);

	TParsePtrC parse(iTempString);

	TPtrC name = parse.Name(); // Don't use NameAndExt, that will include the .map!

	TPtr namePtr((TUint16*)name.Ptr(), name.Length(), name.Length()); // Nasty but TParsePtr won't give us a non-const Name().

	namePtr.LowerCase();

	User::LeaveIfError(iCodeSegHash.Insert(namePtr, aMapFile));

	}

EXPORT_C void CSymbolics::SetFallbackMapFileDirL(const TDesC& aDir)

	{

	iFallbackMapFileDir = aDir;

	}

EXPORT_C TPtrC CSymbolics::LookupL(TUint32 aRomAddress)

	{

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

		{

		TPtrC res = iBsyms[i]->LookupL(aRomAddress);

		if (res.Length()) return res;

		}

	return TPtrC();

	}

EXPORT_C TPtrC CSymbolics::LookupL(const TDesC& aCodeseg, TUint32 aOffset)

	{

	CMapFile* mapFile = FindOrLoadMapFileL(aCodeseg);

	iTempString.Zero();

	if (mapFile)

		{

		mapFile->Lookup(aOffset, iTempString);

		if (iTempString.Length())

			{

			iTempString.AppendFormat(_L(" (%S)"), &aCodeseg);

			}

		}

	else

		{

		// Try BSYMs in case it's in ROFS

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

			{

			TPtrC res = iBsyms[i]->LookupL(aCodeseg, aOffset);

			}

		}

	return TPtrC(iTempString);

	}

EXPORT_C void CSymbolics::CompleteL(const TDesC& aCodeseg, TDes& aSymbolName, CDesC16Array& aSuggestions)

	{

	RNode* tree = TreeForCodesegL(aCodeseg);

	if (tree)

		{

		tree->CompleteL(aSymbolName, aSuggestions);

		}

	}

RNode* CSymbolics::TreeForCodesegL(const TDesC& aCodeseg)

	{

	iTempString.Copy(aCodeseg);

	iTempString.LowerCase();

	if (iTabCompleteCodeseg && *iTabCompleteCodeseg != iTempString)

		{

		// If iTabCompleteCodeseg is set it means that iTabCompleteTree (if it's non-null) matches.

		delete iTabCompleteCodeseg;

		iTabCompleteCodeseg = NULL;

		delete iTabCompleteTree;

		iTabCompleteTree = NULL;

		}

	if (!iTabCompleteCodeseg) iTabCompleteCodeseg = iTempString.AllocL();

	if (iTabCompleteTree == NULL)

		{

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

			{

			RNode* tree = iBsyms[i]->CreateCompletionTreeL(*iTabCompleteCodeseg);

			if (tree)

				{

				iTabCompleteTree = tree;

				break;

				}

			}

		}

	// Try the map files now

	if (iTabCompleteTree == NULL)

		{

		CMapFile* mapFile = FindOrLoadMapFileL(aCodeseg);

		if (mapFile)

			{

			iTabCompleteTree = mapFile->CreateCompletionTreeL();

			}

		}

	return iTabCompleteTree;

	}

EXPORT_C TUint32 CSymbolics::CodesegOffsetFromSymbolNameL(const TDesC& aCodeseg, const TDesC& aSymbolName)

	{

	RNode* tree = TreeForCodesegL(aCodeseg);

	if (!tree) User::Leave(KErrNotFound);

	return tree->ValueForStringL(aSymbolName);

	}

CMapFile* CSymbolics::FindOrLoadMapFileL(const TDesC& aCodeseg)

	{

	iTempString.Copy(aCodeseg);

	iTempString.LowerCase();

	// If it's already loaded, return it

	CMapFile* mapFile = iCodeSegHash.FindPtr(iTempString);

	if (mapFile) return mapFile;

	if (iFallbackMapFileDir.Length() == 0) return NULL;

	// Try loading the mapfile

	TFileName2 fn = iFallbackMapFileDir;

	fn.AppendComponentL(iTempString);

	fn.Append(_L(".map"));

	TEntry e;

	if (iFs.Entry(fn, e) == KErrNone)

		{

		TRAPD(err, AddMapFileL(fn));

		StaticLeaveIfErr(err, _L("Could not load map file %S"), &fn);

		// AddMapFileL overwrites iTempString, so restore it (nasty but I object to wasting buffers on the stack...)

		iTempString.Copy(aCodeseg);

		iTempString.LowerCase();

		return iCodeSegHash.FindPtr(iTempString);

		}

	return NULL;

	}