ExportDirectory.h

/*
* ExportDirectory.h - Part of the PeLib library.
*
* Copyright (c) 2004 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.
*/

#ifndef EXPORTDIRECTORY_H
#define EXPORTDIRECTORY_H
#include "PeHeader.h"

namespace PeLib
{

//      template<int bits>
        class ExportDirectory
        {
                private:
                  PELIB_IMAGE_EXP_DIRECTORY m_ied;

                public:
                  void addFunction(const std::string& strFuncname, dword dwFuncAddr);
                  int getFunctionIndex(const std::string& strFunctionName) const;
                  int read(const std::string& strFilename, unsigned int uiOffset, unsigned int uiSize, const PeHeader& pehHeader);
                  void rebuild(std::vector<byte>& vBuffer, dword dwRva) const;
                  unsigned int size() const;
                  int write(const std::string& strFilename, unsigned int uiOffset, unsigned int uiRva) const;

                  void setNameString(const std::string& strFilename);
                  std::string getNameString() const;

                  std::string getFunctionName(dword dwIndex) const;
                  word getFunctionOrdinal(dword dwIndex) const;
                  dword getAddressOfName(dword dwIndex) const;
                  dword getAddressOfFunction(dword dwIndex) const;
                  
                  void setFunctionName(dword dwIndex, const std::string& strName);
                  void setFunctionOrdinal(dword dwIndex, word wValue);
                  void setAddressOfName(dword dwIndex, dword dwValue);
                  void setAddressOfFunction(dword dwIndex, dword dwValue);
                  
                  /*
                  word getFunctionOrdinal(std::string strFuncname) const;
                  dword getAddressOfName(std::string strFuncname) const;
                  dword getAddressOfFunction(std::string strFuncname) const;

                  void setFunctionOrdinal(std::string strFuncname, word wValue);
                  void setAddressOfName(std::string strFuncname, dword dwValue);
                  void setAddressOfFunction(std::string strFuncname, dword dwValue);
                  */

                  dword getBase() const;
                  dword getCharacteristics() const;
                  dword getTimeDateStamp() const;
                  word getMajorVersion() const;
                  word getMinorVersion() const;
                  dword getName() const;
                  dword getNumberOfFunctions() const;
                  dword getNumberOfNames() const;
                  dword getAddressOfFunctions() const;
                  dword getAddressOfNames() const;

                  dword getNumberOfNameOrdinals() const;
                  dword getNumberOfAddressOfFunctionNames() const;
                  dword getNumberOfAddressOfFunctions() const;
                  dword getAddressOfNameOrdinals() const;

                  void setBase(dword dwValue);
                  void setCharacteristics(dword dwValue);
                  void setTimeDateStamp(dword dwValue);
                  void setMajorVersion(word wValue);
                  void setMinorVersion(word wValue);
                  void setName(dword dwValue);
                  void setNumberOfFunctions(dword dwValue);
                  void setNumberOfNames(dword dwValue);
                  void setAddressOfFunctions(dword dwValue);
                  void setAddressOfNames(dword dwValue);
        };
}
#endif

---