MzHeader.h

/*
* MzHeader.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 MZHEADER_H
#define MZHEADER_H

#include "PeLibInc.h"

namespace PeLib
{

        class MzHeader
        {
                private:
                  PELIB_IMAGE_DOS_HEADER m_idhHeader; 
                  
                  void read(InputBuffer& ibBuffer);
                  
                  unsigned int originalOffset;
                  
                public:

          enum Field {e_magic, e_cblp, e_cp, e_crlc, e_cparhdr, e_minalloc, e_maxalloc,
                        e_ss, e_sp, e_csum, e_ip, e_cs, e_lfarlc, e_ovno, e_res, e_oemid,
                        e_oeminfo, e_res2, e_lfanew};

                  bool isValid() const;
          
                  bool isValid(Field) const;

                  void makeValid();

                  void makeValid(Field);

                  int read(const std::string& strFilename);

                  int read(unsigned char* pcBuffer, unsigned int uiSize, unsigned int originalOffs = 0);

                  void rebuild(std::vector<byte>& vBuffer) const;

                  unsigned int size() const;

                  int write(const std::string& strFilename, dword dwOffset) const;

                  word getMagicNumber() const;
                  word getBytesOnLastPage() const;
                  word getPagesInFile() const;
                  word getRelocations() const;
                  word getSizeOfHeader() const;
                  word getMinExtraParagraphs() const;
                  word getMaxExtraParagraphs() const;
                  word getSsValue() const;
                  word getSpValue() const;
                  word getChecksum() const;
                  word getIpValue() const;
                  word getCsValue() const;
                  word getAddrOfRelocationTable() const;
                  word getOverlayNumber() const;
                  word getOemIdentifier() const;
                  word getOemInformation() const;
                  dword getAddressOfPeHeader() const;
                  word getReservedWords1(unsigned int uiNr) const;
                  word getReservedWords2(unsigned int uiNr) const;
        
                  void setMagicNumber(word wValue);
                  void setBytesOnLastPage(word wValue);
                  void setPagesInFile(word wValue);
                  void setRelocations(word wValue);
                  void setSizeOfHeader(word wValue);
                  void setMinExtraParagraphs(word wValue);
                  void setMaxExtraParagraphs(word wValue);
                  void setSsValue(word wValue);
                  void setSpValue(word wValue);
                  void setChecksum(word wValue);
                  void setIpValue(word wValue);
                  void setCsValue(word wValue);
                  void setAddrOfRelocationTable(word wValue);
                  void setOverlayNumber(word wValue);
                  void setOemIdentifier(word wValue);
                  void setOemInformation(word wValue);
                  void setAddressOfPeHeader(dword dwValue);
                  void setReservedWords1(unsigned int uiNr, word wValue);
                  void setReservedWords2(unsigned int uiNr, word wValue);
        };
}

#endif

---