/*

* Copyright (c) 2008-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: 

*

*/

/**

 @file

 @internalComponent

 @released

*/

#include "dirreader.h"

#include "e32reader.h"

#ifdef __LINUX__

#include <dirent.h>

#include <sys/stat.h>

#else

#include <io.h>

#include <direct.h>

#endif

#define MAXPATHLEN 255

/** 

Constructor.

@internalComponent

@released

*/

DirReader::DirReader(char* aDirName)

:ImageReader(aDirName)

{

}

/** 

Destructor.

@internalComponent

@released

*/

DirReader::~DirReader(void)

{

	ExeVsE32ImageMap::iterator begin = iExeVsE32ImageMap.begin();

	ExeVsE32ImageMap::iterator end = iExeVsE32ImageMap.end();

	while(begin != end)

	{

		DELETE(begin->second);

		++begin;

	}

	iExeVsE32ImageMap.clear();

}

/** 

Function to check whether the node is an executable or not.

@internalComponent

@released

@param aName - Executable name

*/

bool DirReader::IsExecutable(String aName)

{

	unsigned int strPos = aName.find_last_of('.');

	if(strPos != String::npos)

	{

		aName = aName.substr(strPos);

		if(aName.length() <= 4)

		{

			ReaderUtil::ToLower(aName);

			if (aName.find(".exe") != String::npos || aName.find(".dll") != String::npos ||

				aName.find(".prt") != String::npos || aName.find(".nif") != String::npos ||

				aName.find(".pdl") != String::npos || aName.find(".csy") != String::npos || 

				aName.find(".agt") != String::npos || aName.find(".ani") != String::npos || 

				aName.find(".loc") != String::npos || aName.find(".drv") != String::npos || 

				aName.find(".pdd") != String::npos || aName.find(".ldd") != String::npos ||

				aName.find(".tsy") != String::npos || aName.find(".fsy") != String::npos ||

				aName.find(".fxt") != String::npos)

			{

				return true;

			}

		}

	}

	return false;

}

/** 

Dummy function to be compatible with other Readers.

@internalComponent

@released

*/

void DirReader::ReadImage(void)

{

}

/** 

Function to 

1. Preserve the present working directory

2. Invoke the function which reads the directory entires recursively.

3. Go back to the original directory.

@internalComponent

@released

*/

void DirReader::ProcessImage()

{

	char* cwd = new char[MAXPATHLEN];

	char *cp = getcwd(cwd,MAXPATHLEN); (void)cp;

	ReadDir(iImgFileName);

	int i = chdir(cwd); (void)i;

	if(cwd != NULL)

		delete [] cwd;

	cwd = 0;

}

/** 

Function to 

1. Read the directory entires recursively.

2. Prepare the ExeVsE32ImageMap.

@internalComponent

@released

@param aPath - Directory name.

*/

void DirReader::ReadDir(String aPath)

{

#ifndef __LINUX__

	int handle;

	int retVal = 0;

#endif

	E32Image* e32Image = KNull;

#ifdef __LINUX__

	DIR* dirEntry = opendir( aPath.c_str());

	static struct dirent* dirPtr;

	while ((dirPtr= readdir(dirEntry)) != NULL)

	{

		if ((strcmp(dirPtr->d_name, KChildDir.c_str()) == 0) || 

			(strcmp(dirPtr->d_name, KParentDir.c_str()) == 0)) 

			continue; // current dir || parent dir

		String fullName( aPath + "/" + dirPtr->d_name );

		struct stat fileEntrybuf;

		int retVal = stat((char*)fullName.c_str(), &fileEntrybuf);

		if(retVal >= 0)

		{

			if(S_ISDIR(fileEntrybuf.st_mode)) //Is Directory?

			{

				ReadDir(fullName);

			}

			else if(S_ISREG(fileEntrybuf.st_mode)) //Is regular file?

			{

				if ((fileEntrybuf.st_blksize > 0) && IsExecutable(String(dirPtr->d_name)) && E32Image::IsE32ImageFile((char*)fullName.c_str()))

				{

					iExeAvailable = true;

					e32Image = new E32Image();

					Ifstream inputStream((char*)fullName.c_str(), Ios::binary | Ios::in);

					inputStream.seekg(0, Ios::end);

					TUint32 aSz = inputStream.tellg();

					inputStream.seekg(0, Ios::beg);

					e32Image->iFileSize=aSz;

					e32Image->Adjust(aSz);

					inputStream >> *e32Image;

					String exeName(dirPtr->d_name);

					ReaderUtil::ToLower(exeName);

					if(iExeVsE32ImageMap.find(exeName) != iExeVsE32ImageMap.end())

					{

						cout << "Warning: "<< "Duplicate entry '" << dirPtr->d_name << " '"<< endl;

						continue;

					}

					iExeVsE32ImageMap.insert(std::make_pair(exeName, e32Image));

					iExecutableList.push_back(exeName);

				}

				else

				{

					cout << "Warning: "<< dirPtr->d_name << " is not a valid E32 executable" << endl;

				}

			}

		}

	}

	closedir(dirEntry);

#else

	retVal = chdir(aPath.c_str());

	struct _finddata_t  finder;

	handle = _findfirst("*.*", &finder);

	while (retVal == 0)

	{

		if ((strcmp(finder.name, KChildDir.c_str()) == 0) || 

			(strcmp(finder.name, KParentDir.c_str()) == 0) ) // current dir || parent dir  

		{

			retVal = _findnext(handle, &finder);

			continue;

		}

		if (finder.attrib & _A_SUBDIR)

		{

			ReadDir(finder.name);

			chdir(KParentDir.c_str());

		}

		else

		{

			if ((finder.size > 0) && IsExecutable(String(finder.name)) && E32Image::IsE32ImageFile(finder.name))

			{

				e32Image = new E32Image();

				Ifstream inputStream(finder.name, Ios::binary | Ios::in);

				iExeAvailable = true;

				e32Image->iFileSize=finder.size;

				e32Image->Adjust(finder.size);

				inputStream >> *e32Image;

				String exeName(finder.name);

				ReaderUtil::ToLower(exeName);

				if(iExeVsE32ImageMap.find(exeName) != iExeVsE32ImageMap.end())

				{

					cout << "Warning: "<< "Duplicate entry '" << finder.name << " '"<< endl;

					retVal = _findnext(handle, &finder);

					continue;

				}

				iExeVsE32ImageMap.insert(std::make_pair(exeName, e32Image));

				iExecutableList.push_back(exeName);

			}

			else

			{

				cout << "Warning: "<< finder.name << " is not a valid E32 executable" << endl;

			}

		}

		retVal = _findnext(handle,&finder);

	}

#endif

}

/** 

Function to traverse through ExeVsE32ImageMap and prepare ExeVsIdData map.

@internalComponent

@released

*/

void DirReader::PrepareExeVsIdMap(void)

{

	ExeVsE32ImageMap::iterator begin = iExeVsE32ImageMap.begin();

	ExeVsE32ImageMap::iterator end = iExeVsE32ImageMap.end();

	String exeName;

	E32Image* e32Image = KNull;

	IdData* id = KNull;

	if(iExeVsIdData.size() == 0) //Is not already prepared

	{

		while(begin != end)

		{

			exeName = begin->first;

			e32Image = begin->second;

			id = new IdData;

			id->iUid = e32Image->iOrigHdr->iUid1;

			id->iDbgFlag = (e32Image->iOrigHdr->iFlags & KImageDebuggable)? true : false;

			TUint aHeaderFmt = E32ImageHeader::HdrFmtFromFlags(e32Image->iOrigHdr->iFlags);

			if (aHeaderFmt >= KImageHdrFmt_V)

			{

				E32ImageHeaderV* v = e32Image->iHdr;

				id->iSid = v->iS.iSecureId;

				id->iVid = v->iS.iVendorId;

				id->iFileOffset = 0;//Entry read from directory input, has no offset.

			}

			iExeVsIdData[exeName] = id;

			++begin;

		}

	}

	id = KNull;

}

/** 

Function to return ExeVsIdData map.

@internalComponent

@released

@return returns iExeVsIdData.

*/

const ExeVsIdDataMap& DirReader::GetExeVsIdMap() const

{

	return iExeVsIdData;

}

/** 

Function responsible to gather dependencies for all the executables using the container iExeVsE32ImageMap.

@internalComponent

@released

@return iImageVsDepList - returns all executable's dependencies

*/

ExeNamesVsDepListMap& DirReader::GatherDependencies()

{

	ExeVsE32ImageMap::iterator begin = iExeVsE32ImageMap.begin();

	ExeVsE32ImageMap::iterator end = iExeVsE32ImageMap.end();

	StringList executableList;

	while(begin != end)

	{

		PrepareExeDependencyList((*begin).second, executableList);

		iImageVsDepList.insert(std::make_pair((*begin).first, executableList));

		executableList.clear();

		++begin;

	}

	return iImageVsDepList;

}

/** 

Function responsible to prepare the dependency list.

@internalComponent

@released

@param - aE32Image, Using this, can get all the information about the executable

@param - aExecutableList, Excutables placed into this list

*/

void DirReader::PrepareExeDependencyList(E32Image* aE32Image, StringList& aExecutableList)

{

	int count = 0;

	char** nameList = aE32Image->GetImportExecutableNames(count);

	int i = 0;

	String dependency;

	for(; i < count; ++i)

	{

		dependency.assign(nameList[i]);

		aExecutableList.push_back(ReaderUtil::ToLower(dependency));

	}

	DELETE(nameList);

}

/** 

Function to identify the given path as file or directory

@internalComponent

@released

@param - aStr, path name

@return - retuns the either Directory, file or Unknown.

*/

EImageType DirReader::EntryType(char* aStr)

{

	int strLength = strlen(aStr);

	if(aStr[strLength - 1] == '\\' || aStr[strLength - 1] == '/')

	{

		aStr[strLength - 1] = KNull;

	}

	int retVal = 0;

	#ifdef __LINUX__

		struct stat fileEntrybuf;

		retVal = stat(aStr, &fileEntrybuf);

		if(retVal >= 0)

		{

			if(S_ISDIR(fileEntrybuf.st_mode))

			{

	#else

		struct _finddata_t  finder;

		retVal = _findfirst(aStr, &finder);

		if(retVal > 0) //No error

		{

			if(finder.attrib & _A_SUBDIR)

			{

	#endif

				return EE32Directoy;

			}

			else

			{

				if(E32Reader::IsE32Image(aStr) == true)

				{

					return EE32File;

				}

			}

		}

	return EE32InputNotExist;

}