RelocationsDirectory.h

/*
* Relocations.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 RELOCATIONSDIRECTORY_H
#define RELOCATIONSDIRECTORY_H

namespace PeLib
{

        class RelocationsDirectory
        {
                private:
                  std::vector<IMG_BASE_RELOC> m_vRelocations; 

                public:
                  unsigned int getNumberOfRelocations() const;
                  unsigned int getNumberOfRelocationData(unsigned int ulRelocation) const;

                  int read(const std::string& strFilename, unsigned int uiOffset, unsigned int uiSize);
                  void rebuild(std::vector<byte>& vBuffer) const;
                  unsigned int size() const;
                  int write(const std::string& strFilename, unsigned int dwOffset) const;

                  dword getVirtualAddress(unsigned int ulRelocation) const;
                  dword getSizeOfBlock(unsigned int ulRelocation) const;
                  word getRelocationData(unsigned int ulRelocation, unsigned int ulDataNumber) const;
                  
                  void setRelocationData(unsigned int ulRelocation, unsigned int ulDataNumber, word wData);
                  
                  void setVirtualAddress(unsigned int ulRelocation, dword dwValue);
                  void setSizeOfBlock(unsigned int ulRelocation, dword dwValue);
                  void addRelocationData(unsigned int ulRelocation, word wValue);
                  void removeRelocationData(unsigned int ulRelocation, word wValue);
        };
}

#endif

---