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

// Implementation of the Class ElfExports for the elf2e32 tool

// @internalComponent

// @released

// 

//

#include "pl_elfexports.h"

#include "pl_elfexecutable.h"

#include "pl_dllsymbol.h"

#include <cstring>

using std::set_difference;

/**

Constructor for class ElfExports

@internalComponent

@released

*/

ElfExports::ElfExports() : iDllName(NULL), iSorted(false), iExportsFilteredP(false){

}

/**

Destructor for class ElfExports

@internalComponent

@released

*/

ElfExports::~ElfExports()

{

	if(iExportList.size())

	{

		ExportList::iterator aItr = iExportList.begin();

		ExportList::iterator last = iExportList.end();

		DllSymbol *temp;

		while( aItr != last)

		{

			temp = *aItr;

			aItr++;

			delete temp;

		}

	}

	iExportList.clear();

}

/**

This function validates exported symbols. The typeinfo name strings

are not valid export symbols and are discarded.

@param aExecutable - Instance of class ElfExecutable

@param aSym	- DllSymbol

@return True if symbol is valid, otherwise false

@internalComponent

@released

*/

bool ElfExports::ValidExportP(ElfExecutable * aExecutable, DllSymbol * aSym)

{

	char * aSymName = aExecutable->GetSymbolName(aSym->iSymbolIndex);

	int result = strncmp(aSymName, "_ZTS", strlen("_ZTS"));

	return ( result != 0);

}

/**

This function adds export symbols into exports list.

@param aDll - Dll name

@param aExecutable - Instance of class ElfExecutable

@param aSym - Dll symbol

@return Dll Symbol if its valid, otherwise NULL

@internalComponent

@released

*/

DllSymbol* ElfExports::Add(char *aDll, ElfExecutable * aExecutable, DllSymbol *aSym)

{

	if (ValidExportP(aExecutable, aSym))

	{

		if( !iDllName )

			iDllName = aDll;

		iExportList.push_back(aSym);

		iSorted = false;

		return aSym;

	}

	return NULL;

}

/**

Function to add elf exports

@param aDll - Dll name

@param aSym - Dll symbol

@internalComponent

@released

*/

void ElfExports::Add(char *aDll, DllSymbol *aSym)

{

	if( !iDllName )

		iDllName = aDll;

	iExportList.push_back(aSym);

	iSorted = false;

}

/**

Function to sort elf exports. Sorting may be done based on the symbol

name or on the ordinal number depending on the usecase.

@internalComponent

@released

*/

void ElfExports::Sort() 

{

	if (!iSorted) {

		if(iExportsFilteredP) {

			std::sort(iFilteredExports.begin(), iFilteredExports.end(), PtrELFExportNameCompare());

		}

		else {

			std::sort(iExportList.begin(), iExportList.end(), PtrELFExportNameCompare());

		}

		iSorted = true;

	}

}

/**

Function to get exports list.

@param aSorted - sort before returning the exports.

@return export list

@internalComponent

@released

*/

ElfExports::ExportList & ElfExports::GetExports(bool aSorted)

{

	if (aSorted) Sort();

	if(iExportsFilteredP) 

		return iFilteredExports;

	else

		return iExportList;

}

/**

Function to get exports in ordinal order

@return export list

@internalComponent

@released

*/

ElfExports::ExportList & ElfExports::GetExportsInOrdinalOrder()

{

	if (iExportsFilteredP)

	{

		std::sort(iFilteredExports.begin(), iFilteredExports.end(), PtrELFExportOrdinalCompare());

		return iFilteredExports;

	}

	else

	{

		std::sort(iExportList.begin(), iExportList.end(), PtrELFExportOrdinalCompare());

		return iExportList;

	}

}

/**

Function to process the filtered exports

@internalComponent

@released

*/

void ElfExports::FilterExports()

{

	std::sort(iExportList.begin(), iExportList.end(), PtrELFExportNameCompare());

	std::sort(iFilteredExports.begin(), iFilteredExports.end(), PtrELFExportNameCompare());

	ExportList aNewList(iExportList.size());

	ExportList::iterator aNewListBegin = aNewList.begin(); 

	ExportList::iterator aNewListEnd = set_difference(iExportList.begin(), iExportList.end(), \

		iFilteredExports.begin(), iFilteredExports.end(), aNewListBegin, PtrELFExportNameCompare());

	iFilteredExports.clear();

	while (aNewListBegin != aNewListEnd)

	{

		iFilteredExports.push_back(*aNewListBegin);

		aNewListBegin++;

	}

}

/**

Function to get number of exports

@return size of export list

@internalComponent

@released

*/

size_t ElfExports::GetNumExports()

{

	return iExportList.size();

}

/**

Function to get Dll name

@return Dll name

@internalComponent

@released

*/

char* ElfExports::DllName()

{ 

	return iDllName;

}

/**

Overloaded operator to compare ELF export names.

@return True if lhs symbol name < rhs symbol name, otherwise false

@internalComponent

@released

*/

bool ElfExports::PtrELFExportNameCompare::operator()(const Symbol * lhs, const Symbol * rhs) const

{

	return strcmp( lhs->SymbolName(), rhs->SymbolName()) < 0;

}

/**

Overloaded operator to compare ordinal numbers of symbols.

@return True if lhs symbol name < rhs symbol name, otherwise false

@internalComponent

@released

*/

bool ElfExports::PtrELFExportOrdinalCompare::operator()(const Symbol * lhs, const Symbol * rhs) const

{

	return lhs->OrdNum() < rhs->OrdNum();

}

/**

Overloaded operator to compare update symbol attributes that are 

being compared. The comparision is done on the symbol names.

@return True if lhs symbol name < rhs symbol name, otherwise false

@internalComponent

@released

*/

bool ElfExports::PtrELFExportNameCompareUpdateAttributes::operator()(const Symbol * lhs, const Symbol * rhs) const

{

	int result = strcmp(lhs->SymbolName(), rhs->SymbolName());

	if (!result) 

	{

		if (lhs->OrdNum() > 0)

			((Symbol*)rhs)->SetOrdinal( lhs->OrdNum());

		else if (rhs->OrdNum() > 0)

			((Symbol*)lhs)->SetOrdinal( rhs->OrdNum());

		if( ((Symbol*)lhs)->Absent() )

			((Symbol*)rhs)->SetAbsent(true);

		else if ( ((Symbol*)rhs)->Absent() )

			((Symbol*)lhs)->SetAbsent(true);

		if( ((Symbol*)lhs)->SymbolSize() )

			((Symbol*)rhs)->SetSymbolSize(((Symbol*)lhs)->SymbolSize());

		else if( ((Symbol*)rhs)->SymbolSize() )

			((Symbol*)lhs)->SetSymbolSize(((Symbol*)rhs)->SymbolSize());

	}

	return result < 0;

}