/* Filet - Binary Format Parsing Utility
 * Copyright (C) 2001-2002  Jay Freeman (saurik)
*/

/*
 *        Redistribution and use in source and binary
 * forms, with or without modification, are permitted
 * provided that the following conditions are met:
 * 
 * 1. Redistributions of source code must retain the
 *    above copyright notice, this list of conditions
 *    and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the
 *    above copyright notice, this list of conditions
 *    and the following disclaimer in the documentation
 *    and/or other materials provided with the
 *    distribution.
 * 3. The name of the author may not be used to endorse
 *    or promote products derived from this software
 *    without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
 * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
 * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/

#ifndef FILET_HEADERS_H
#define FILET_HEADERS_H

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#include "filet/filet.h"

namespace Filet {

namespace Signatures {

static const uint16_t DOS(0x5A4D);		// MZ
static const uint16_t OS2(0x454E);		// NE
static const uint16_t VXD(0x454C);		// LE
static const uint32_t NT(0x00004550);	// PE00

}

namespace Headers {

struct Section {
	uint8_t			Name[8];

	union {
		uint32_t		PhysicalAddress;
		uint32_t		VirtualSize;
	} Misc;

	uint32_t		VirtualAddress;
	uint32_t		SizeOfRawData;
	uint32_t		PointerToRawData;
	uint32_t		PointerToRelocations;
	uint32_t		PointerToLinenumbers;
	uint16_t		NumberOfRelocations;
	uint16_t		NumberOfLinenumbers;
	uint32_t		Characteristics;
};

struct File {
	uint16_t		Machine;
	uint16_t		NumberOfSections;
	uint32_t		TimeDateStamp;
	uint32_t		PointerToSymbolTable;
	uint32_t		NumberOfSymbols;
	uint16_t		SizeOfOptionalHeader;
	uint16_t		Characteristics;
};

struct DOS {
	uint16_t		e_magic;
	uint16_t		e_cblp;
	uint16_t		e_cp;
	uint16_t		e_crlc;
	uint16_t		e_cparhdr;
	uint16_t		e_minalloc;
	uint16_t		e_maxalloc;
	uint16_t		e_ss;
	uint16_t		e_sp;
	uint16_t		e_csum;
	uint16_t		e_ip;
	uint16_t		e_cs;
	uint16_t		e_lfarlc;
	uint16_t		e_ovno;
	uint16_t		e_res[4];
	uint16_t		e_oemid;
	uint16_t		e_oeminfo;
	uint16_t		e_res2[10];
	uint16_t		e_lfanew;
};

struct DataDirectory {
	uint32_t		VirtualAddress;
	uint32_t		Size;
};

struct Optional32 {
	uint16_t		Magic;
	uint8_t			MajorLinkerVersion;
	uint8_t			MinorLinkerVersion;
	uint32_t		SizeOfCode;
	uint32_t		SizeOfInitializedData;
	uint32_t		SizeOfUninitializedData;
	uint32_t		AddressOfEntryPoint;
	uint32_t		BaseOfCode;
	uint32_t		BaseOfData;

	uint32_t		ImageBase;
	uint32_t		SectionAlignment;
	uint32_t		FileAlignment;
	uint16_t		MajorOperatingSystemVersion;
	uint16_t		MinorOperatingSystemVersion;
	uint16_t		MajorImageVersion;
	uint16_t		MinorImageVersion;
	uint16_t		MajorSubsystemVersion;
	uint16_t		MinorSubsystemVersion;
	uint32_t		Win32VersionValue;
	uint32_t		SizeOfImage;
	uint32_t		SizeOfHeaders;
	uint32_t		CheckSum;
	uint16_t		Subsystem;
	uint16_t		DllCharacteristics;
	uint32_t		SizeOfStackReserve;
	uint32_t		SizeOfStackCommit;
	uint32_t		SizeOfHeapReserve;
	uint32_t		SizeOfHeapCommit;
	uint32_t		LoaderFlags;
	uint32_t		NumberOfRvaAndSizes;
	DataDirectory	DataDirectory[16];
};

struct NT32 {
    uint32_t		Signature;
    File			FileHeader;
    Optional32		OptionalHeader;
};

struct COR20 {
	uint32_t		cb;
	uint16_t		MajorRuntimeVersion;
	uint16_t		MinorRuntimeVersion;
	DataDirectory	MetaData;
	uint32_t		Flags;
	uint32_t		EntryPointToken;
	DataDirectory	Resources;
	DataDirectory	StrongNameSignature;
	DataDirectory	CodeManagerTable;
	DataDirectory	VTableFixups;
	DataDirectory	ExportAddressTableJumps;
	DataDirectory	ManagedNativeHeader;
};

struct GUID {
	uint32_t		Data1;
	uint16_t		Data2;
	uint16_t		Data3;
	uint8_t			Data4[8];
};

}

namespace Utils {

FILET_API void *ImageRvaToVa(Headers::NT32 *nt, void *base, uint32_t rva, Headers::Section **section);

}

namespace PE {

namespace DataDirectories {
	enum DataDirectories {
		Export			= 0,
		Import			= 1,
		Resource		= 2,
		Exception		= 3,
		Security		= 4,
		BaseRelocation	= 5,
		Debug			= 6,
		Architecture	= 7,
		GlobalPointer	= 8,
		TLS				= 9,
		LoadConfig		= 10,
		BoundImport		= 11,
		ImportAddress	= 12,
		DelayLoadImport	= 13,
		COMDescriptor	= 14
	};
}

} }

#endif//FILET_HEADERS_H