/*
 * Copyright 2013-2014, Haiku, Inc.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *	Alexander von Gluck IV, <kallisti5@unixzen.com>
 */
#ifndef _PE_COMMON_H
#define _PE_COMMON_H


#include <SupportDefs.h>
#include <ByteOrder.h>


// Magic strings
#define MZ_MAGIC "MZ"
#define PE_MAGIC "PE"
#define PE_OPTIONAL_MAGIC_PE32 0x010b
#define PE_OPTIONAL_MAGIC_PE32P 0x020b


typedef struct {
	uint16 magic; /* == MZ_MAGIC */
	uint16 bytesInLastBlock;
	uint16 blocksInFile;
	uint16 numRelocations;
	uint16 headerParagraphs;
	uint16 minExtraParagraphs;
	uint16 maxExtraParagraphs;
	uint16 ss;
	uint16 sp;
	uint16 checksum;
	uint16 ip;
	uint16 cs;
	uint16 relocationTableOffset;
	uint16 overlayNumber;
	uint16 reserved[4];
	uint16 oemID;
	uint16 oemInfo;
	uint16 reserved2[10];
	uint32 lfaNew;  // PE Header start addr
} MzHeader;

typedef struct {
	uint32 magic; // == PE_MAGIC */
	uint16 machine;
	uint16 numberOfSections;
	uint32 timeDateStamp;
	uint32 pointerToSymbolTable;
	uint32 numberOfSymbols;
	uint16 sizeOfOptionalHeader;
	uint16 characteristics;
} PeHeader;

typedef struct {
	uint16 magic; // == 0x010b - PE32, 0x020b - PE32+ (64 bit)
	uint8  majorLinkerVersion;
	uint8  minorLinkerVersion;
	uint32 sizeOfCode;
	uint32 sizeOfInitializedData;
	uint32 sizeOfUninitializedData;
	uint32 addressOfEntryPoint;
	uint32 baseOfCode;
	uint32 baseOfData;
	uint32 imageBase;
	uint32 sectionAlignment;
	uint32 fileAlignment;
	uint16 majorOperatingSystemVersion;
	uint16 minorOperatingSystemVersion;
	uint16 majorImageVersion;
	uint16 minorImageVersion;
	uint16 majorSubsystemVersion;
	uint16 minorSubsystemVersion;
	uint32 win32VersionValue;
	uint32 sizeOfImage;
	uint32 sizeOfHeaders;
	uint32 checksum;
	uint16 subsystem;
	uint16 llCharacteristics;
	uint32 sizeOfStackReserve;
	uint32 sizeOfStackCommit;
	uint32 sizeOfHeapReserve;
	uint32 sizeOfHeapCommit;
	uint32 loaderFlags;
	uint32 numberOfRvaAndSizes;
} Pe32OptionalHeader;

#endif /* _PE_COMMON_H */