IatDirectory.h

/*
* IatDirectory.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 IATDIRECTORY_H
#define IATDIRECTORY_H

#include "PeLibInc.h"

namespace PeLib
{

        class IatDirectory
        {
                private:
                  std::vector<dword> m_vIat; 
                  
                public:
                  int read(const std::string& strFilename, unsigned int dwOffset, unsigned int dwSize);
                  unsigned int numberOfAddresses() const;
                  void addAddress(dword dwValue);
                  void removeAddress(dword dwAddrnr);
                  void removeAll();
                  // Rebuilds the IAT.
                  void rebuild(std::vector<byte>& vBuffer) const;
                  unsigned int size() const;
                  int write(const std::string& strFilename, unsigned int uiOffset) const;

                  dword getAddress(dword dwAddrnr) const;
                  void setAddress(dword dwAddrnr, dword dwValue);
        };
}

#endif

---