Delete the svn:executable property and set svn:eol-style to native for all those text files.

git-svn-id: svn://svn.rockbox.org/rockbox/trunk@18012 a1c6a512-1295-4272-9138-f99709370657

1 files changed, 692 insertions, 692 deletions

diff --git a/utils/zenutils/libraries/pelib-0.9/pelib/ExportDirectory.cpp b/utils/zenutils/libraries/pelib-0.9/pelib/ExportDirectory.cpp
index aa9c28a..83c4b4e 100755..100644
--- a/utils/zenutils/libraries/pelib-0.9/pelib/ExportDirectory.cpp
+++ b/utils/zenutils/libraries/pelib-0.9/pelib/ExportDirectory.cpp
@@ -1,692 +1,692 @@
-/*

-* ExportDirectory.cpp - Part of the PeLib library.

-*

-* Copyright (c) 2004 - 2005 Sebastian Porst (webmaster@the-interweb.com)

-* All rights reserved.

-*

-* This software is licensed under the zlib/libpng License.

-* For more details see http://www.opensource.org/licenses/zlib-license.php

-* or the license information file (license.htm) in the root directory 

-* of PeLib.

-*/

-

-//#ifdef false

-

-#include "PeLibInc.h"

-#include "ExportDirectory.h"

-

-namespace PeLib

-{

-	/**

-	* @param strFuncname Name of the function.

-	* @param dwFuncAddr RVA of the function.

-	**/

-	void ExportDirectory::addFunction(const std::string& strFuncname, dword dwFuncAddr)

-	{

-		PELIB_EXP_FUNC_INFORMATION efiCurr;

-		efiCurr.funcname = strFuncname;

-		efiCurr.addroffunc = dwFuncAddr;

-		m_ied.functions.push_back(efiCurr);

-	}

-	

-	void ExportDirectory::removeFunction(unsigned int index)

-	{

-		m_ied.functions.erase(m_ied.functions.begin() + index);

-	}

-	

-	void ExportDirectory::clear()

-	{

-		m_ied.functions.clear();

-	}

-		

-	unsigned int ExportDirectory::calcNumberOfFunctions() const

-	{

-		return static_cast<unsigned int>(m_ied.functions.size());

-	}

-

-	/**

-	* Identifies an exported function through it's name.

-	* @param strFunctionName Name of the function

-	* @return Number which identifies the functions.

-	**/

-	int ExportDirectory::getFunctionIndex(const std::string& strFunctionName) const

-	{

-		std::vector<PELIB_EXP_FUNC_INFORMATION>::const_iterator Iter = std::find_if(m_ied.functions.begin(), m_ied.functions.end(), std::bind2nd(std::mem_fun_ref(&PELIB_EXP_FUNC_INFORMATION::equal), strFunctionName));

-		

-		if (Iter == m_ied.functions.end())

-		{

-//			throw Exceptions::InvalidName(ExportDirectoryId, __LINE__);

-			return -1;

-		}

-		

-		return static_cast<int>(std::distance(m_ied.functions.begin(), Iter));

-	}

-

-	/**

-	* @param strFilename Name of the file.

-	* @param uiOffset File offset of the export directory.

-	* @param uiSize Size of the export directory.

-	* @param pehHeader A valid PE header which is necessary because some RVA calculations need to be done.

-    * \todo: Proper use of InputBuffer

-	**/

-	int ExportDirectory::read(const std::string& strFilename, unsigned int uiOffset, unsigned int uiSize, const PeHeader& pehHeader)

-	{

-		std::ifstream ifFile(strFilename.c_str(), std::ios::binary);

-		

-		if (!ifFile)

-		{

-			return ERROR_OPENING_FILE;

-		}

-		

-		unsigned int filesize = fileSize(ifFile);

-		

-		if (filesize < uiOffset + uiSize)

-		{

-			return ERROR_INVALID_FILE;

-		}

-

-		ifFile.seekg(uiOffset, std::ios::beg);

-

-		PELIB_IMAGE_EXP_DIRECTORY iedCurr;

-

-		std::vector<unsigned char> vExportdirectory(uiSize);

-		ifFile.read(reinterpret_cast<char*>(&vExportdirectory[0]), uiSize);

-

-		InputBuffer inpBuffer(vExportdirectory);

-

-		inpBuffer >> iedCurr.ied.Characteristics;

-		inpBuffer >> iedCurr.ied.TimeDateStamp;

-		inpBuffer >> iedCurr.ied.MajorVersion;

-		inpBuffer >> iedCurr.ied.MinorVersion;

-		inpBuffer >> iedCurr.ied.Name;

-		inpBuffer >> iedCurr.ied.Base;

-		inpBuffer >> iedCurr.ied.NumberOfFunctions;

-		inpBuffer >> iedCurr.ied.NumberOfNames;

-		inpBuffer >> iedCurr.ied.AddressOfFunctions;

-		inpBuffer >> iedCurr.ied.AddressOfNames;

-		inpBuffer >> iedCurr.ied.AddressOfNameOrdinals;

-

-		if (const PeHeader32* p32 = dynamic_cast<const PeHeader32*>(&pehHeader))

-		{

-			unsigned int offset = p32->rvaToOffset(iedCurr.ied.Name);

-			

-			if (offset >= filesize)

-				return ERROR_INVALID_FILE;

-			

-			ifFile.seekg(offset, std::ios::beg);

-		}

-		else if (const PeHeader64* p64 = dynamic_cast<const PeHeader64*>(&pehHeader))

-		{

-			// XXX: Files might be > 4 GB

-			unsigned int offset = static_cast<unsigned int>(p64->rvaToOffset(iedCurr.ied.Name));

-			

-			if (offset >= filesize)

-				return ERROR_INVALID_FILE;

-			

-			ifFile.seekg(offset, std::ios::beg);

-		}

-		

-		char c = 0;

-		std::string strFname = "";

-		do

-		{

-			ifFile.read(reinterpret_cast<char*>(&c), sizeof(c));

-			if (!ifFile) return ERROR_INVALID_FILE;

-			if (c) strFname += c;

-		}

-		while (c != 0);

-		iedCurr.name = strFname;

-

-		if (const PeHeader32* p32 = dynamic_cast<const PeHeader32*>(&pehHeader))

-		{

-			unsigned int offset = p32->rvaToOffset(iedCurr.ied.AddressOfFunctions);

-			

-			if (offset >= filesize)

-				return ERROR_INVALID_FILE;

-			

-			ifFile.seekg(offset, std::ios::beg);

-		}

-		else if (const PeHeader64* p64 = dynamic_cast<const PeHeader64*>(&pehHeader))

-		{

-			// XXX: File might be > 4 GB

-			unsigned int offset = static_cast<unsigned int>(p64->rvaToOffset(iedCurr.ied.AddressOfFunctions));

-			

-			if (offset >= filesize)

-				return ERROR_INVALID_FILE;

-				

-			ifFile.seekg(offset, std::ios::beg);

-		}

-

-		PELIB_EXP_FUNC_INFORMATION efiCurr;

-		efiCurr.ordinal = 0; efiCurr.addroffunc = 0; efiCurr.addrofname = 0;

-

-		for (unsigned int i=0;i<iedCurr.ied.NumberOfFunctions;i++)

-		{

-			if (const PeHeader32* p32 = dynamic_cast<const PeHeader32*>(&pehHeader))

-			{

-				unsigned int offset = p32->rvaToOffset(iedCurr.ied.AddressOfFunctions) + i*sizeof(efiCurr.addroffunc);

-				

-				if (offset >= filesize)

-					return ERROR_INVALID_FILE;

-					

-				ifFile.seekg(offset, std::ios::beg);

-			}

-			else if (const PeHeader64* p64 = dynamic_cast<const PeHeader64*>(&pehHeader))

-			{

-				// XXX: File might be > 4GB

-				unsigned int offset = static_cast<unsigned int>(p64->rvaToOffset(iedCurr.ied.AddressOfFunctions)) + i*sizeof(efiCurr.addroffunc);

-				

-				if (offset >= filesize)

-					return ERROR_INVALID_FILE;

-					

-				ifFile.seekg(offset, std::ios::beg);

-			}

-			

-			ifFile.read(reinterpret_cast<char*>(&efiCurr.addroffunc), sizeof(efiCurr.addroffunc));

-			

-			if (!ifFile)

-				return ERROR_INVALID_FILE;

-				

-			efiCurr.ordinal = i;

-			iedCurr.functions.push_back(efiCurr);

-		}

-

-		for (unsigned int i=0;i<iedCurr.ied.NumberOfNames;i++)

-		{

-			if (const PeHeader32* p32 = dynamic_cast<const PeHeader32*>(&pehHeader))

-			{

-				unsigned int offset = p32->rvaToOffset(iedCurr.ied.AddressOfNameOrdinals) + i*sizeof(efiCurr.ordinal);

-				

-				if (offset >= filesize)

-					return ERROR_INVALID_FILE;

-				

-				ifFile.seekg(offset, std::ios::beg);

-			}

-			else if (const PeHeader64* p64 = dynamic_cast<const PeHeader64*>(&pehHeader))

-			{

-				// XXX: File might be > 4 GB

-				unsigned int offset = static_cast<unsigned int>(p64->rvaToOffset(iedCurr.ied.AddressOfNameOrdinals)) + i*sizeof(efiCurr.ordinal);

-				

-				if (offset >= filesize)

-					return ERROR_INVALID_FILE;

-				

-				ifFile.seekg(offset, std::ios::beg);

-			}

-

-			word ordinal;

-			ifFile.read(reinterpret_cast<char*>(&ordinal), sizeof(ordinal));

-			

-			if (!ifFile)

-				return ERROR_INVALID_FILE;

-			

-			iedCurr.functions[ordinal].ordinal = ordinal;

-

-			if (const PeHeader32* p32 = dynamic_cast<const PeHeader32*>(&pehHeader))

-			{

-				unsigned int offset = p32->rvaToOffset(iedCurr.ied.AddressOfNames) + i*sizeof(efiCurr.addrofname);

-				

-				if (offset >= filesize)

-					return ERROR_INVALID_FILE;

-				

-				ifFile.seekg(offset, std::ios::beg);

-			}

-			else if (const PeHeader64* p64 = dynamic_cast<const PeHeader64*>(&pehHeader))

-			{

-				// XXX: File might be > 4 GB.

-				unsigned int offset = static_cast<unsigned int>(p64->rvaToOffset(iedCurr.ied.AddressOfNames)) + i*sizeof(efiCurr.addrofname);

-				

-				if (offset >= filesize)

-					return ERROR_INVALID_FILE;

-				

-				ifFile.seekg(offset, std::ios::beg);

-			}

-			

-			ifFile.read(reinterpret_cast<char*>(&iedCurr.functions[ordinal].addrofname), sizeof(iedCurr.functions[ordinal].addrofname));

-			

-			if (!ifFile)

-				return ERROR_INVALID_FILE;

-

-			if (const PeHeader32* p32 = dynamic_cast<const PeHeader32*>(&pehHeader))

-			{

-				unsigned int offset = p32->rvaToOffset(iedCurr.functions[ordinal].addrofname);

-

-				if (offset >= filesize)

-					return ERROR_INVALID_FILE;

-

-				ifFile.seekg(offset, std::ios::beg);

-			}

-			else if (const PeHeader64* p64 = dynamic_cast<const PeHeader64*>(&pehHeader))

-			{

-				// XXX: File might be > 4 GB.

-				unsigned int offset = static_cast<unsigned int>(p64->rvaToOffset(iedCurr.functions[ordinal].addrofname));

-

-				if (offset >= filesize)

-					return ERROR_INVALID_FILE;

-

-				ifFile.seekg(static_cast<unsigned int>(p64->rvaToOffset(iedCurr.functions[ordinal].addrofname)), std::ios::beg);

-			}

-

-			char c = 0;

-			std::string strFname = "";

-			do

-			{

-				ifFile.read(reinterpret_cast<char*>(&c), sizeof(c));

-				

-				if (!ifFile)

-					return ERROR_INVALID_FILE;

-					

-				if (c) strFname += c;

-			}

-			while (c != 0);

-

-			iedCurr.functions[ordinal].funcname = strFname;

-		}

-

-		std::swap(m_ied, iedCurr);

-		

-		return NO_ERROR;

-	}

-

-	/**

-	* @param vBuffer Buffer where the rebuilt export directory is written to.

-	* @param dwRva RVA of the export directory.

-	* \todo fValid flag

-	**/

-	void ExportDirectory::rebuild(std::vector<byte>& vBuffer, dword dwRva) const

-	{

-		unsigned int uiSizeDirectory = sizeof(PELIB_IMAGE_EXPORT_DIRECTORY);

-

-		unsigned int uiSizeNames = 0;

-		unsigned int uiSizeAddrFuncs = 0;

-		unsigned int uiSizeAddrNames = 0;

-		unsigned int uiSizeOrdinals = 0;

-

-		for (unsigned int i=0;i<m_ied.functions.size();i++)

-		{

-			uiSizeNames += (m_ied.functions[i].funcname.empty()) ? 0 : static_cast<unsigned int>(m_ied.functions[i].funcname.size()) + 1;

-			uiSizeAddrFuncs += sizeof(m_ied.functions[i].addroffunc);

-			uiSizeAddrNames += (m_ied.functions[i].funcname.empty()) ? 0 : sizeof(m_ied.functions[i].addrofname);

-			uiSizeOrdinals += (m_ied.functions[i].funcname.empty()) ? 0 : sizeof(m_ied.functions[i].ordinal);

-		}

-		

-		unsigned int uiFilenameSize = static_cast<unsigned int>(m_ied.name.size()) + 1;

-		

-		OutputBuffer obBuffer(vBuffer);

-

-		obBuffer << m_ied.ied.Characteristics;

-		obBuffer << m_ied.ied.TimeDateStamp;

-		obBuffer << m_ied.ied.MajorVersion;

-		obBuffer << m_ied.ied.MinorVersion;

-		obBuffer << dwRva + uiSizeDirectory;

-		obBuffer << m_ied.ied.Base;

-		obBuffer << static_cast<unsigned int>(m_ied.functions.size());

-

-		// TODO: Not correct but sufficient for now. (Update: I forgot what this comment refers to, but I'll leave it in)

-		obBuffer << static_cast<unsigned int>(m_ied.functions.size());

-		obBuffer << dwRva + uiSizeDirectory + uiFilenameSize;

-		obBuffer << dwRva + uiSizeDirectory + uiFilenameSize + uiSizeAddrFuncs; 

-		obBuffer << dwRva + uiSizeDirectory + uiFilenameSize + uiSizeAddrFuncs + uiSizeAddrNames;

-

-		obBuffer.add(m_ied.name.c_str(), static_cast<unsigned int>(m_ied.name.size())+1);

-

-		for (unsigned int i=0;i<m_ied.functions.size();i++)

-		{

-			obBuffer << m_ied.functions[i].addroffunc;

-		}

-

-		unsigned int ulFuncCounter = dwRva + uiSizeDirectory + uiFilenameSize + uiSizeAddrFuncs + uiSizeAddrNames + uiSizeOrdinals;

-		

-		for (unsigned int i=0;i<m_ied.functions.size();i++)

-		{

-			if (!m_ied.functions[i].funcname.empty())

-			{

-				obBuffer << ulFuncCounter;

-				ulFuncCounter += static_cast<unsigned int>(m_ied.functions[i].funcname.size()) + 1;

-			}

-		}

-

-		for (unsigned int i=0;i<m_ied.functions.size();i++)

-		{

-			if (!m_ied.functions[i].funcname.empty())

-			{

-				obBuffer <<  m_ied.functions[i].ordinal;

-			}

-		}

-

-		for (unsigned int i=0;i<m_ied.functions.size();i++)

-		{

-			if (m_ied.functions[i].funcname.empty() && m_ied.functions[i].addroffunc)

-			{

-				obBuffer <<  m_ied.functions[i].ordinal;

-			}

-		}

-

-		for (unsigned int i=0;i<m_ied.functions.size();i++)

-		{

-			if (!m_ied.functions[i].funcname.empty())

-			{

-				obBuffer.add(m_ied.functions[i].funcname.c_str(), static_cast<unsigned int>(m_ied.functions[i].funcname.size()) + 1);

-			}

-		}

-	}

-

-	/**

-	* @return Size of the current export directory.

-	**/

-	unsigned int ExportDirectory::size() const

-	{

-		return m_ied.size();

-	}

-

-	/**

-	* @param strFilename Name of the file.

-	* @param uiOffset File offset the export directory will be written to.

-	* @param uiRva RVA of the export directory.

-	* \todo Check if ofFile.write succeeded.

-	**/

-	int ExportDirectory::write(const std::string& strFilename, unsigned int uiOffset, unsigned int uiRva) const

-	{

-		std::fstream ofFile(strFilename.c_str(), std::ios_base::in);

-		

-		if (!ofFile)

-		{

-			ofFile.clear();

-			ofFile.open(strFilename.c_str(), std::ios_base::out | std::ios_base::binary);

-		}

-		else

-		{

-			ofFile.close();

-			ofFile.open(strFilename.c_str(), std::ios_base::in | std::ios_base::out | std::ios_base::binary);

-		}

-		

-		if (!ofFile)

-		{

-			return ERROR_OPENING_FILE;

-		}

-

-		ofFile.seekp(uiOffset, std::ios::beg);

-

-		std::vector<unsigned char> vBuffer;

-		rebuild(vBuffer, uiRva);

-

-		ofFile.write(reinterpret_cast<const char*>(&vBuffer[0]), static_cast<unsigned int>(vBuffer.size()));

-

-		ofFile.close();

-		

-		return NO_ERROR;

-	}

-

-	/**

-	* Changes the filename according to the export directory.

-	* @param strFilename New filename.

-	**/

-	void ExportDirectory::setNameString(const std::string& strFilename)

-	{

-		m_ied.name = strFilename;

-	}

-

-	std::string ExportDirectory::getNameString() const

-	{

-		return m_ied.name;

-	}

-

-	/**

-	* @param dwIndex Number which identifies an exported function.

-	* @return The name of that function.

-	**/

-	std::string ExportDirectory::getFunctionName(dword dwIndex) const

-	{

-		return m_ied.functions[dwIndex].funcname;

-	}

-

-	/**

-	* @param dwIndex Number which identifies an exported function.

-	* @return The ordinal of that function.

-	**/

-	word ExportDirectory::getFunctionOrdinal(dword dwIndex) const

-	{

-		return m_ied.functions[dwIndex].ordinal;

-	}

-

-	/**

-	* @param dwIndex Number which identifies an exported function.

-	* @return The RVA of the name string of that function.

-	**/

-	dword ExportDirectory::getAddressOfName(dword dwIndex) const

-	{

-		return m_ied.functions[dwIndex].addrofname;

-	}

-

-	/**

-	* @param dwIndex Number which identifies an exported function.

-	* @return The RVA of that function.

-	**/

-	dword ExportDirectory::getAddressOfFunction(dword dwIndex) const

-	{

-		return m_ied.functions[dwIndex].addroffunc;

-	}

-	

-	/**

-	* @param dwIndex Number which identifies an exported function.

-	* @param strName The name of that function.

-	**/

-	void ExportDirectory::setFunctionName(dword dwIndex, const std::string& strName)

-	{

-		m_ied.functions[dwIndex].funcname = strName;

-	}

-

-	/**

-	* @param dwIndex Number which identifies an exported function.

-	* @param wValue The ordinal of that function.

-	**/

-	void ExportDirectory::setFunctionOrdinal(dword dwIndex, word wValue)

-	{

-		m_ied.functions[dwIndex].ordinal = wValue;

-	}

-

-	/**

-	* @param dwIndex Number which identifies an exported function.

-	* @param dwValue The RVA of the name string of that function.

-	**/

-	void ExportDirectory::setAddressOfName(dword dwIndex, dword dwValue)

-	{

-		m_ied.functions[dwIndex].addrofname = dwValue;

-	}

-

-	/**

-	* @param dwIndex Number which identifies an exported function.

-	* @param dwValue The RVA of that function.

-	**/

-	void ExportDirectory::setAddressOfFunction(dword dwIndex, dword dwValue)

-	{

-		m_ied.functions[dwIndex].addroffunc = dwValue;

-	}

-

-	/**

-	* @return The ordinal base of the export directory.

-	**/

-	dword ExportDirectory::getBase() const

-	{

-		return m_ied.ied.Base;

-	}

-

-	/**

-	* @return The characteristics of the export directory.

-	**/

-	dword ExportDirectory::getCharacteristics() const

-	{

-		return m_ied.ied.Characteristics;

-	}

-

-	/**

-	* @return The time/date stamp of the export directory.

-	**/

-	dword ExportDirectory::getTimeDateStamp() const

-	{

-		return m_ied.ied.TimeDateStamp;

-	}

-

-	/**

-	* @return The MajorVersion of the export directory.

-	**/

-	word ExportDirectory::getMajorVersion() const

-	{

-		return m_ied.ied.MajorVersion;

-	}

-

-	/**

-	* @return The MinorVersion of the export directory.

-	**/

-	word ExportDirectory::getMinorVersion() const

-	{

-		return m_ied.ied.MinorVersion;

-	}

-

-	/**

-	* @return The RVA of the name of the file.

-	**/

-	dword ExportDirectory::getName() const

-	{

-		return m_ied.ied.Name;

-	}

-

-	/**

-	* @return The NumberOfFunctions of the export directory.

-	**/

-	dword ExportDirectory::getNumberOfFunctions() const

-	{

-		return m_ied.ied.NumberOfFunctions;

-	}

-

-	/**

-	* @return The NumberOfNames of the export directory.

-	**/

-	dword ExportDirectory::getNumberOfNames() const

-	{

-		return m_ied.ied.NumberOfNames;

-	}

-

-	/**

-	* @return The AddressOfFunctions of the export directory.

-	**/

-	dword ExportDirectory::getAddressOfFunctions() const

-	{

-		return m_ied.ied.AddressOfFunctions;

-	}

-

-	/**

-	* @return The AddressOfNames of the export directory.

-	**/

-	dword ExportDirectory::getAddressOfNames() const

-	{

-		return m_ied.ied.AddressOfNames;

-	}

-

-/*	dword ExportDirectory::getNumberOfNameOrdinals() const

-	{

-		return static_cast<dword>(m_ied.functions.size());

-	}

-

-	dword ExportDirectory::getNumberOfAddressOfFunctionNames() const

-	{

-		return static_cast<dword>(m_ied.functions.size());

-	}

-

-	dword ExportDirectory::getNumberOfAddressOfFunctions() const

-	{

-		return static_cast<dword>(m_ied.functions.size());

-	}

-*/

-	/**

-	* @return The AddressOfNameOrdinals of the export directory.

-	**/

-	dword ExportDirectory::getAddressOfNameOrdinals() const

-	{

-		return m_ied.ied.AddressOfNameOrdinals;

-	}

-

-	/**

-	* @param dwValue The ordinal base of the export directory.

-	**/

-	void ExportDirectory::setBase(dword dwValue)

-	{

-		m_ied.ied.Base = dwValue;

-	}

-

-	/**

-	* @param dwValue The Characteristics of the export directory.

-	**/

-	void ExportDirectory::setCharacteristics(dword dwValue)

-	{

-		m_ied.ied.Characteristics = dwValue;

-	}

-

-	/**

-	* @param dwValue The TimeDateStamp of the export directory.

-	**/

-	void ExportDirectory::setTimeDateStamp(dword dwValue)

-	{

-		m_ied.ied.TimeDateStamp = dwValue;

-	}

-

-	/**

-	* @param wValue The MajorVersion of the export directory.

-	**/

-	void ExportDirectory::setMajorVersion(word wValue)

-	{

-		m_ied.ied.MajorVersion = wValue;

-	}

-

-	/**

-	* @param wValue The MinorVersion of the export directory.

-	**/

-	void ExportDirectory::setMinorVersion(word wValue)

-	{

-		m_ied.ied.MinorVersion = wValue;

-	}

-

-	/**

-	* @param dwValue The Name of the export directory.

-	**/

-	void ExportDirectory::setName(dword dwValue)

-	{

-		m_ied.ied.Name = dwValue;

-	}

-

-	/**

-	* @param dwValue The NumberOfFunctions of the export directory.

-	**/

-	void ExportDirectory::setNumberOfFunctions(dword dwValue)

-	{

-		m_ied.ied.NumberOfFunctions = dwValue;

-	}

-

-	/**

-	* @param dwValue The NumberOfNames of the export directory.

-	**/

-	void ExportDirectory::setNumberOfNames(dword dwValue)

-	{

-		m_ied.ied.NumberOfNames = dwValue;

-	}

-

-	/**

-	* @param dwValue The AddressOfFunctions of the export directory.

-	**/

-	void ExportDirectory::setAddressOfFunctions(dword dwValue)

-	{

-		m_ied.ied.AddressOfFunctions = dwValue;

-	}

-

-	/**

-	* @param dwValue The AddressOfNames of the export directory.

-	**/

-	void ExportDirectory::setAddressOfNames(dword dwValue)

-	{

-		m_ied.ied.AddressOfNames = dwValue;

-	}

-	

-	void ExportDirectory::setAddressOfNameOrdinals(dword value)

-	{

-		m_ied.ied.AddressOfNameOrdinals = value;

-	}

-}

+/*
+* ExportDirectory.cpp - Part of the PeLib library.
+*
+* Copyright (c) 2004 - 2005 Sebastian Porst (webmaster@the-interweb.com)
+* All rights reserved.
+*
+* This software is licensed under the zlib/libpng License.
+* For more details see http://www.opensource.org/licenses/zlib-license.php
+* or the license information file (license.htm) in the root directory 
+* of PeLib.
+*/
+
+//#ifdef false
+
+#include "PeLibInc.h"
+#include "ExportDirectory.h"
+
+namespace PeLib
+{
+	/**
+	* @param strFuncname Name of the function.
+	* @param dwFuncAddr RVA of the function.
+	**/
+	void ExportDirectory::addFunction(const std::string& strFuncname, dword dwFuncAddr)
+	{
+		PELIB_EXP_FUNC_INFORMATION efiCurr;
+		efiCurr.funcname = strFuncname;
+		efiCurr.addroffunc = dwFuncAddr;
+		m_ied.functions.push_back(efiCurr);
+	}
+	
+	void ExportDirectory::removeFunction(unsigned int index)
+	{
+		m_ied.functions.erase(m_ied.functions.begin() + index);
+	}
+	
+	void ExportDirectory::clear()
+	{
+		m_ied.functions.clear();
+	}
+		
+	unsigned int ExportDirectory::calcNumberOfFunctions() const
+	{
+		return static_cast<unsigned int>(m_ied.functions.size());
+	}
+
+	/**
+	* Identifies an exported function through it's name.
+	* @param strFunctionName Name of the function
+	* @return Number which identifies the functions.
+	**/
+	int ExportDirectory::getFunctionIndex(const std::string& strFunctionName) const
+	{
+		std::vector<PELIB_EXP_FUNC_INFORMATION>::const_iterator Iter = std::find_if(m_ied.functions.begin(), m_ied.functions.end(), std::bind2nd(std::mem_fun_ref(&PELIB_EXP_FUNC_INFORMATION::equal), strFunctionName));
+		
+		if (Iter == m_ied.functions.end())
+		{
+//			throw Exceptions::InvalidName(ExportDirectoryId, __LINE__);
+			return -1;
+		}
+		
+		return static_cast<int>(std::distance(m_ied.functions.begin(), Iter));
+	}
+
+	/**
+	* @param strFilename Name of the file.
+	* @param uiOffset File offset of the export directory.
+	* @param uiSize Size of the export directory.
+	* @param pehHeader A valid PE header which is necessary because some RVA calculations need to be done.
+    * \todo: Proper use of InputBuffer
+	**/
+	int ExportDirectory::read(const std::string& strFilename, unsigned int uiOffset, unsigned int uiSize, const PeHeader& pehHeader)
+	{
+		std::ifstream ifFile(strFilename.c_str(), std::ios::binary);
+		
+		if (!ifFile)
+		{
+			return ERROR_OPENING_FILE;
+		}
+		
+		unsigned int filesize = fileSize(ifFile);
+		
+		if (filesize < uiOffset + uiSize)
+		{
+			return ERROR_INVALID_FILE;
+		}
+
+		ifFile.seekg(uiOffset, std::ios::beg);
+
+		PELIB_IMAGE_EXP_DIRECTORY iedCurr;
+
+		std::vector<unsigned char> vExportdirectory(uiSize);
+		ifFile.read(reinterpret_cast<char*>(&vExportdirectory[0]), uiSize);
+
+		InputBuffer inpBuffer(vExportdirectory);
+
+		inpBuffer >> iedCurr.ied.Characteristics;
+		inpBuffer >> iedCurr.ied.TimeDateStamp;
+		inpBuffer >> iedCurr.ied.MajorVersion;
+		inpBuffer >> iedCurr.ied.MinorVersion;
+		inpBuffer >> iedCurr.ied.Name;
+		inpBuffer >> iedCurr.ied.Base;
+		inpBuffer >> iedCurr.ied.NumberOfFunctions;
+		inpBuffer >> iedCurr.ied.NumberOfNames;
+		inpBuffer >> iedCurr.ied.AddressOfFunctions;
+		inpBuffer >> iedCurr.ied.AddressOfNames;
+		inpBuffer >> iedCurr.ied.AddressOfNameOrdinals;
+
+		if (const PeHeader32* p32 = dynamic_cast<const PeHeader32*>(&pehHeader))
+		{
+			unsigned int offset = p32->rvaToOffset(iedCurr.ied.Name);
+			
+			if (offset >= filesize)
+				return ERROR_INVALID_FILE;
+			
+			ifFile.seekg(offset, std::ios::beg);
+		}
+		else if (const PeHeader64* p64 = dynamic_cast<const PeHeader64*>(&pehHeader))
+		{
+			// XXX: Files might be > 4 GB
+			unsigned int offset = static_cast<unsigned int>(p64->rvaToOffset(iedCurr.ied.Name));
+			
+			if (offset >= filesize)
+				return ERROR_INVALID_FILE;
+			
+			ifFile.seekg(offset, std::ios::beg);
+		}
+		
+		char c = 0;
+		std::string strFname = "";
+		do
+		{
+			ifFile.read(reinterpret_cast<char*>(&c), sizeof(c));
+			if (!ifFile) return ERROR_INVALID_FILE;
+			if (c) strFname += c;
+		}
+		while (c != 0);
+		iedCurr.name = strFname;
+
+		if (const PeHeader32* p32 = dynamic_cast<const PeHeader32*>(&pehHeader))
+		{
+			unsigned int offset = p32->rvaToOffset(iedCurr.ied.AddressOfFunctions);
+			
+			if (offset >= filesize)
+				return ERROR_INVALID_FILE;
+			
+			ifFile.seekg(offset, std::ios::beg);
+		}
+		else if (const PeHeader64* p64 = dynamic_cast<const PeHeader64*>(&pehHeader))
+		{
+			// XXX: File might be > 4 GB
+			unsigned int offset = static_cast<unsigned int>(p64->rvaToOffset(iedCurr.ied.AddressOfFunctions));
+			
+			if (offset >= filesize)
+				return ERROR_INVALID_FILE;
+				
+			ifFile.seekg(offset, std::ios::beg);
+		}
+
+		PELIB_EXP_FUNC_INFORMATION efiCurr;
+		efiCurr.ordinal = 0; efiCurr.addroffunc = 0; efiCurr.addrofname = 0;
+
+		for (unsigned int i=0;i<iedCurr.ied.NumberOfFunctions;i++)
+		{
+			if (const PeHeader32* p32 = dynamic_cast<const PeHeader32*>(&pehHeader))
+			{
+				unsigned int offset = p32->rvaToOffset(iedCurr.ied.AddressOfFunctions) + i*sizeof(efiCurr.addroffunc);
+				
+				if (offset >= filesize)
+					return ERROR_INVALID_FILE;
+					
+				ifFile.seekg(offset, std::ios::beg);
+			}
+			else if (const PeHeader64* p64 = dynamic_cast<const PeHeader64*>(&pehHeader))
+			{
+				// XXX: File might be > 4GB
+				unsigned int offset = static_cast<unsigned int>(p64->rvaToOffset(iedCurr.ied.AddressOfFunctions)) + i*sizeof(efiCurr.addroffunc);
+				
+				if (offset >= filesize)
+					return ERROR_INVALID_FILE;
+					
+				ifFile.seekg(offset, std::ios::beg);
+			}
+			
+			ifFile.read(reinterpret_cast<char*>(&efiCurr.addroffunc), sizeof(efiCurr.addroffunc));
+			
+			if (!ifFile)
+				return ERROR_INVALID_FILE;
+				
+			efiCurr.ordinal = i;
+			iedCurr.functions.push_back(efiCurr);
+		}
+
+		for (unsigned int i=0;i<iedCurr.ied.NumberOfNames;i++)
+		{
+			if (const PeHeader32* p32 = dynamic_cast<const PeHeader32*>(&pehHeader))
+			{
+				unsigned int offset = p32->rvaToOffset(iedCurr.ied.AddressOfNameOrdinals) + i*sizeof(efiCurr.ordinal);
+				
+				if (offset >= filesize)
+					return ERROR_INVALID_FILE;
+				
+				ifFile.seekg(offset, std::ios::beg);
+			}
+			else if (const PeHeader64* p64 = dynamic_cast<const PeHeader64*>(&pehHeader))
+			{
+				// XXX: File might be > 4 GB
+				unsigned int offset = static_cast<unsigned int>(p64->rvaToOffset(iedCurr.ied.AddressOfNameOrdinals)) + i*sizeof(efiCurr.ordinal);
+				
+				if (offset >= filesize)
+					return ERROR_INVALID_FILE;
+				
+				ifFile.seekg(offset, std::ios::beg);
+			}
+
+			word ordinal;
+			ifFile.read(reinterpret_cast<char*>(&ordinal), sizeof(ordinal));
+			
+			if (!ifFile)
+				return ERROR_INVALID_FILE;
+			
+			iedCurr.functions[ordinal].ordinal = ordinal;
+
+			if (const PeHeader32* p32 = dynamic_cast<const PeHeader32*>(&pehHeader))
+			{
+				unsigned int offset = p32->rvaToOffset(iedCurr.ied.AddressOfNames) + i*sizeof(efiCurr.addrofname);
+				
+				if (offset >= filesize)
+					return ERROR_INVALID_FILE;
+				
+				ifFile.seekg(offset, std::ios::beg);
+			}
+			else if (const PeHeader64* p64 = dynamic_cast<const PeHeader64*>(&pehHeader))
+			{
+				// XXX: File might be > 4 GB.
+				unsigned int offset = static_cast<unsigned int>(p64->rvaToOffset(iedCurr.ied.AddressOfNames)) + i*sizeof(efiCurr.addrofname);
+				
+				if (offset >= filesize)
+					return ERROR_INVALID_FILE;
+				
+				ifFile.seekg(offset, std::ios::beg);
+			}
+			
+			ifFile.read(reinterpret_cast<char*>(&iedCurr.functions[ordinal].addrofname), sizeof(iedCurr.functions[ordinal].addrofname));
+			
+			if (!ifFile)
+				return ERROR_INVALID_FILE;
+
+			if (const PeHeader32* p32 = dynamic_cast<const PeHeader32*>(&pehHeader))
+			{
+				unsigned int offset = p32->rvaToOffset(iedCurr.functions[ordinal].addrofname);
+
+				if (offset >= filesize)
+					return ERROR_INVALID_FILE;
+
+				ifFile.seekg(offset, std::ios::beg);
+			}
+			else if (const PeHeader64* p64 = dynamic_cast<const PeHeader64*>(&pehHeader))
+			{
+				// XXX: File might be > 4 GB.
+				unsigned int offset = static_cast<unsigned int>(p64->rvaToOffset(iedCurr.functions[ordinal].addrofname));
+
+				if (offset >= filesize)
+					return ERROR_INVALID_FILE;
+
+				ifFile.seekg(static_cast<unsigned int>(p64->rvaToOffset(iedCurr.functions[ordinal].addrofname)), std::ios::beg);
+			}
+
+			char c = 0;
+			std::string strFname = "";
+			do
+			{
+				ifFile.read(reinterpret_cast<char*>(&c), sizeof(c));
+				
+				if (!ifFile)
+					return ERROR_INVALID_FILE;
+					
+				if (c) strFname += c;
+			}
+			while (c != 0);
+
+			iedCurr.functions[ordinal].funcname = strFname;
+		}
+
+		std::swap(m_ied, iedCurr);
+		
+		return NO_ERROR;
+	}
+
+	/**
+	* @param vBuffer Buffer where the rebuilt export directory is written to.
+	* @param dwRva RVA of the export directory.
+	* \todo fValid flag
+	**/
+	void ExportDirectory::rebuild(std::vector<byte>& vBuffer, dword dwRva) const
+	{
+		unsigned int uiSizeDirectory = sizeof(PELIB_IMAGE_EXPORT_DIRECTORY);
+
+		unsigned int uiSizeNames = 0;
+		unsigned int uiSizeAddrFuncs = 0;
+		unsigned int uiSizeAddrNames = 0;
+		unsigned int uiSizeOrdinals = 0;
+
+		for (unsigned int i=0;i<m_ied.functions.size();i++)
+		{
+			uiSizeNames += (m_ied.functions[i].funcname.empty()) ? 0 : static_cast<unsigned int>(m_ied.functions[i].funcname.size()) + 1;
+			uiSizeAddrFuncs += sizeof(m_ied.functions[i].addroffunc);
+			uiSizeAddrNames += (m_ied.functions[i].funcname.empty()) ? 0 : sizeof(m_ied.functions[i].addrofname);
+			uiSizeOrdinals += (m_ied.functions[i].funcname.empty()) ? 0 : sizeof(m_ied.functions[i].ordinal);
+		}
+		
+		unsigned int uiFilenameSize = static_cast<unsigned int>(m_ied.name.size()) + 1;
+		
+		OutputBuffer obBuffer(vBuffer);
+
+		obBuffer << m_ied.ied.Characteristics;
+		obBuffer << m_ied.ied.TimeDateStamp;
+		obBuffer << m_ied.ied.MajorVersion;
+		obBuffer << m_ied.ied.MinorVersion;
+		obBuffer << dwRva + uiSizeDirectory;
+		obBuffer << m_ied.ied.Base;
+		obBuffer << static_cast<unsigned int>(m_ied.functions.size());
+
+		// TODO: Not correct but sufficient for now. (Update: I forgot what this comment refers to, but I'll leave it in)
+		obBuffer << static_cast<unsigned int>(m_ied.functions.size());
+		obBuffer << dwRva + uiSizeDirectory + uiFilenameSize;
+		obBuffer << dwRva + uiSizeDirectory + uiFilenameSize + uiSizeAddrFuncs; 
+		obBuffer << dwRva + uiSizeDirectory + uiFilenameSize + uiSizeAddrFuncs + uiSizeAddrNames;
+
+		obBuffer.add(m_ied.name.c_str(), static_cast<unsigned int>(m_ied.name.size())+1);
+
+		for (unsigned int i=0;i<m_ied.functions.size();i++)
+		{
+			obBuffer << m_ied.functions[i].addroffunc;
+		}
+
+		unsigned int ulFuncCounter = dwRva + uiSizeDirectory + uiFilenameSize + uiSizeAddrFuncs + uiSizeAddrNames + uiSizeOrdinals;
+		
+		for (unsigned int i=0;i<m_ied.functions.size();i++)
+		{
+			if (!m_ied.functions[i].funcname.empty())
+			{
+				obBuffer << ulFuncCounter;
+				ulFuncCounter += static_cast<unsigned int>(m_ied.functions[i].funcname.size()) + 1;
+			}
+		}
+
+		for (unsigned int i=0;i<m_ied.functions.size();i++)
+		{
+			if (!m_ied.functions[i].funcname.empty())
+			{
+				obBuffer <<  m_ied.functions[i].ordinal;
+			}
+		}
+
+		for (unsigned int i=0;i<m_ied.functions.size();i++)
+		{
+			if (m_ied.functions[i].funcname.empty() && m_ied.functions[i].addroffunc)
+			{
+				obBuffer <<  m_ied.functions[i].ordinal;
+			}
+		}
+
+		for (unsigned int i=0;i<m_ied.functions.size();i++)
+		{
+			if (!m_ied.functions[i].funcname.empty())
+			{
+				obBuffer.add(m_ied.functions[i].funcname.c_str(), static_cast<unsigned int>(m_ied.functions[i].funcname.size()) + 1);
+			}
+		}
+	}
+
+	/**
+	* @return Size of the current export directory.
+	**/
+	unsigned int ExportDirectory::size() const
+	{
+		return m_ied.size();
+	}
+
+	/**
+	* @param strFilename Name of the file.
+	* @param uiOffset File offset the export directory will be written to.
+	* @param uiRva RVA of the export directory.
+	* \todo Check if ofFile.write succeeded.
+	**/
+	int ExportDirectory::write(const std::string& strFilename, unsigned int uiOffset, unsigned int uiRva) const
+	{
+		std::fstream ofFile(strFilename.c_str(), std::ios_base::in);
+		
+		if (!ofFile)
+		{
+			ofFile.clear();
+			ofFile.open(strFilename.c_str(), std::ios_base::out | std::ios_base::binary);
+		}
+		else
+		{
+			ofFile.close();
+			ofFile.open(strFilename.c_str(), std::ios_base::in | std::ios_base::out | std::ios_base::binary);
+		}
+		
+		if (!ofFile)
+		{
+			return ERROR_OPENING_FILE;
+		}
+
+		ofFile.seekp(uiOffset, std::ios::beg);
+
+		std::vector<unsigned char> vBuffer;
+		rebuild(vBuffer, uiRva);
+
+		ofFile.write(reinterpret_cast<const char*>(&vBuffer[0]), static_cast<unsigned int>(vBuffer.size()));
+
+		ofFile.close();
+		
+		return NO_ERROR;
+	}
+
+	/**
+	* Changes the filename according to the export directory.
+	* @param strFilename New filename.
+	**/
+	void ExportDirectory::setNameString(const std::string& strFilename)
+	{
+		m_ied.name = strFilename;
+	}
+
+	std::string ExportDirectory::getNameString() const
+	{
+		return m_ied.name;
+	}
+
+	/**
+	* @param dwIndex Number which identifies an exported function.
+	* @return The name of that function.
+	**/
+	std::string ExportDirectory::getFunctionName(dword dwIndex) const
+	{
+		return m_ied.functions[dwIndex].funcname;
+	}
+
+	/**
+	* @param dwIndex Number which identifies an exported function.
+	* @return The ordinal of that function.
+	**/
+	word ExportDirectory::getFunctionOrdinal(dword dwIndex) const
+	{
+		return m_ied.functions[dwIndex].ordinal;
+	}
+
+	/**
+	* @param dwIndex Number which identifies an exported function.
+	* @return The RVA of the name string of that function.
+	**/
+	dword ExportDirectory::getAddressOfName(dword dwIndex) const
+	{
+		return m_ied.functions[dwIndex].addrofname;
+	}
+
+	/**
+	* @param dwIndex Number which identifies an exported function.
+	* @return The RVA of that function.
+	**/
+	dword ExportDirectory::getAddressOfFunction(dword dwIndex) const
+	{
+		return m_ied.functions[dwIndex].addroffunc;
+	}
+	
+	/**
+	* @param dwIndex Number which identifies an exported function.
+	* @param strName The name of that function.
+	**/
+	void ExportDirectory::setFunctionName(dword dwIndex, const std::string& strName)
+	{
+		m_ied.functions[dwIndex].funcname = strName;
+	}
+
+	/**
+	* @param dwIndex Number which identifies an exported function.
+	* @param wValue The ordinal of that function.
+	**/
+	void ExportDirectory::setFunctionOrdinal(dword dwIndex, word wValue)
+	{
+		m_ied.functions[dwIndex].ordinal = wValue;
+	}
+
+	/**
+	* @param dwIndex Number which identifies an exported function.
+	* @param dwValue The RVA of the name string of that function.
+	**/
+	void ExportDirectory::setAddressOfName(dword dwIndex, dword dwValue)
+	{
+		m_ied.functions[dwIndex].addrofname = dwValue;
+	}
+
+	/**
+	* @param dwIndex Number which identifies an exported function.
+	* @param dwValue The RVA of that function.
+	**/
+	void ExportDirectory::setAddressOfFunction(dword dwIndex, dword dwValue)
+	{
+		m_ied.functions[dwIndex].addroffunc = dwValue;
+	}
+
+	/**
+	* @return The ordinal base of the export directory.
+	**/
+	dword ExportDirectory::getBase() const
+	{
+		return m_ied.ied.Base;
+	}
+
+	/**
+	* @return The characteristics of the export directory.
+	**/
+	dword ExportDirectory::getCharacteristics() const
+	{
+		return m_ied.ied.Characteristics;
+	}
+
+	/**
+	* @return The time/date stamp of the export directory.
+	**/
+	dword ExportDirectory::getTimeDateStamp() const
+	{
+		return m_ied.ied.TimeDateStamp;
+	}
+
+	/**
+	* @return The MajorVersion of the export directory.
+	**/
+	word ExportDirectory::getMajorVersion() const
+	{
+		return m_ied.ied.MajorVersion;
+	}
+
+	/**
+	* @return The MinorVersion of the export directory.
+	**/
+	word ExportDirectory::getMinorVersion() const
+	{
+		return m_ied.ied.MinorVersion;
+	}
+
+	/**
+	* @return The RVA of the name of the file.
+	**/
+	dword ExportDirectory::getName() const
+	{
+		return m_ied.ied.Name;
+	}
+
+	/**
+	* @return The NumberOfFunctions of the export directory.
+	**/
+	dword ExportDirectory::getNumberOfFunctions() const
+	{
+		return m_ied.ied.NumberOfFunctions;
+	}
+
+	/**
+	* @return The NumberOfNames of the export directory.
+	**/
+	dword ExportDirectory::getNumberOfNames() const
+	{
+		return m_ied.ied.NumberOfNames;
+	}
+
+	/**
+	* @return The AddressOfFunctions of the export directory.
+	**/
+	dword ExportDirectory::getAddressOfFunctions() const
+	{
+		return m_ied.ied.AddressOfFunctions;
+	}
+
+	/**
+	* @return The AddressOfNames of the export directory.
+	**/
+	dword ExportDirectory::getAddressOfNames() const
+	{
+		return m_ied.ied.AddressOfNames;
+	}
+
+/*	dword ExportDirectory::getNumberOfNameOrdinals() const
+	{
+		return static_cast<dword>(m_ied.functions.size());
+	}
+
+	dword ExportDirectory::getNumberOfAddressOfFunctionNames() const
+	{
+		return static_cast<dword>(m_ied.functions.size());
+	}
+
+	dword ExportDirectory::getNumberOfAddressOfFunctions() const
+	{
+		return static_cast<dword>(m_ied.functions.size());
+	}
+*/
+	/**
+	* @return The AddressOfNameOrdinals of the export directory.
+	**/
+	dword ExportDirectory::getAddressOfNameOrdinals() const
+	{
+		return m_ied.ied.AddressOfNameOrdinals;
+	}
+
+	/**
+	* @param dwValue The ordinal base of the export directory.
+	**/
+	void ExportDirectory::setBase(dword dwValue)
+	{
+		m_ied.ied.Base = dwValue;
+	}
+
+	/**
+	* @param dwValue The Characteristics of the export directory.
+	**/
+	void ExportDirectory::setCharacteristics(dword dwValue)
+	{
+		m_ied.ied.Characteristics = dwValue;
+	}
+
+	/**
+	* @param dwValue The TimeDateStamp of the export directory.
+	**/
+	void ExportDirectory::setTimeDateStamp(dword dwValue)
+	{
+		m_ied.ied.TimeDateStamp = dwValue;
+	}
+
+	/**
+	* @param wValue The MajorVersion of the export directory.
+	**/
+	void ExportDirectory::setMajorVersion(word wValue)
+	{
+		m_ied.ied.MajorVersion = wValue;
+	}
+
+	/**
+	* @param wValue The MinorVersion of the export directory.
+	**/
+	void ExportDirectory::setMinorVersion(word wValue)
+	{
+		m_ied.ied.MinorVersion = wValue;
+	}
+
+	/**
+	* @param dwValue The Name of the export directory.
+	**/
+	void ExportDirectory::setName(dword dwValue)
+	{
+		m_ied.ied.Name = dwValue;
+	}
+
+	/**
+	* @param dwValue The NumberOfFunctions of the export directory.
+	**/
+	void ExportDirectory::setNumberOfFunctions(dword dwValue)
+	{
+		m_ied.ied.NumberOfFunctions = dwValue;
+	}
+
+	/**
+	* @param dwValue The NumberOfNames of the export directory.
+	**/
+	void ExportDirectory::setNumberOfNames(dword dwValue)
+	{
+		m_ied.ied.NumberOfNames = dwValue;
+	}
+
+	/**
+	* @param dwValue The AddressOfFunctions of the export directory.
+	**/
+	void ExportDirectory::setAddressOfFunctions(dword dwValue)
+	{
+		m_ied.ied.AddressOfFunctions = dwValue;
+	}
+
+	/**
+	* @param dwValue The AddressOfNames of the export directory.
+	**/
+	void ExportDirectory::setAddressOfNames(dword dwValue)
+	{
+		m_ied.ied.AddressOfNames = dwValue;
+	}
+	
+	void ExportDirectory::setAddressOfNameOrdinals(dword value)
+	{
+		m_ied.ied.AddressOfNameOrdinals = value;
+	}
+}