/* -------------------------------------------

	Copyright (C) 2024-2025, Amlal EL Mahrouss, all rights reserved.

	File: PE.h
	Purpose: Portable Executable for Kernel.

	Revision History:

	30/01/24: Added file (amlel)

------------------------------------------- */

#ifndef __KERNELKIT_INC_PE_H__
#define __KERNELKIT_INC_PE_H__

#include <NewKit/Defines.h>

#define kPeSignature 0x00004550

#define kPeDLLBase 0x4000000
#define kPeEXEBase 0x1000000

#define kPeMagic32 0x010b
#define kPeMagic64 0x020b

#define kPeMachineAMD64 0x8664
#define kPeMachineARM64 0xaa64

typedef struct LDR_EXEC_HEADER final
{
	Kernel::UInt32 Signature;
	Kernel::UInt16 Machine;
	Kernel::UInt16 NumberOfSections;
	Kernel::UInt32 TimeDateStamp;
	Kernel::UInt32 PointerToSymbolTable;
	Kernel::UInt32 NumberOfSymbols;
	Kernel::UInt16 SizeOfOptionalHeader;
	Kernel::UInt16 Characteristics;
} LDR_EXEC_HEADER, *LDR_EXEC_HEADER_PTR;

typedef struct LDR_OPTIONAL_HEADER final
{
	Kernel::UInt16 Magic; // 0x010b - PE32, 0x020b - PE32+ (64 bit)
	Kernel::UInt8  MajorLinkerVersion;
	Kernel::UInt8  MinorLinkerVersion;
	Kernel::UInt32 SizeOfCode;
	Kernel::UInt32 SizeOfInitializedData;
	Kernel::UInt32 SizeOfUninitializedData;
	Kernel::UInt32 AddressOfEntryPoint;
	Kernel::UInt32 BaseOfCode;
	Kernel::UInt32 BaseOfData;
	Kernel::UInt32 ImageBase;
	Kernel::UInt32 SectionAlignment;
	Kernel::UInt32 FileAlignment;
	Kernel::UInt16 MajorOperatingSystemVersion;
	Kernel::UInt16 MinorOperatingSystemVersion;
	Kernel::UInt16 MajorImageVersion;
	Kernel::UInt16 MinorImageVersion;
	Kernel::UInt16 MajorSubsystemVersion;
	Kernel::UInt16 MinorSubsystemVersion;
	Kernel::UInt32 Win32VersionValue;
	Kernel::UInt32 SizeOfImage;
	Kernel::UInt32 SizeOfHeaders;
	Kernel::UInt32 CheckSum;
	Kernel::UInt16 Subsystem;
	Kernel::UInt16 DllCharacteristics;
	Kernel::UInt32 SizeOfStackReserve;
	Kernel::UInt32 SizeOfStackCommit;
	Kernel::UInt32 SizeOfHeapReserve;
	Kernel::UInt32 SizeOfHeapCommit;
	Kernel::UInt32 LoaderFlags;
	Kernel::UInt32 NumberOfRvaAndSizes;
} LDR_OPTIONAL_HEADER, *LDR_OPTIONAL_HEADER_PTR;

typedef struct LDR_SECTION_HEADER final
{
	Kernel::Char   Name[8];
	Kernel::UInt32 VirtualSize;
	Kernel::UInt32 VirtualAddress;
	Kernel::UInt32 SizeOfRawData;
	Kernel::UInt32 PointerToRawData;
	Kernel::UInt32 PointerToRelocations;
	Kernel::UInt32 PointerToLineNumbers;
	Kernel::UInt16 NumberOfRelocations;
	Kernel::UInt16 NumberOfLinenumbers;
	Kernel::UInt32 Characteristics;
} LDR_SECTION_HEADER, *LDR_SECTION_HEADER_PTR;

enum kExecDataDirParams
{
	kExecExport,
	kExecImport,
	kExecInvalid,
	kExecCount,
};

typedef struct LDR_EXPORT_DIRECTORY
{
	Kernel::UInt32 Characteristics;
	Kernel::UInt32 TimeDateStamp;
	Kernel::UInt16 MajorVersion;
	Kernel::UInt16 MinorVersion;
	Kernel::UInt32 Name;
	Kernel::UInt32 Base;
	Kernel::UInt32 NumberOfFunctions;
	Kernel::UInt32 NumberOfNames;
	Kernel::UInt32 AddressOfFunctions; // export table rva
	Kernel::UInt32 AddressOfNames;
	Kernel::UInt32 AddressOfNameOrdinal; // ordinal table rva
} LDR_EXPORT_DIRECTORY, *LDR_EXPORT_DIRECTORY_PTR;

typedef struct LDR_IMPORT_DIRECTORY
{
	union {
		Kernel::UInt32 Characteristics;
		Kernel::UInt32 OriginalFirstThunk;
	};
	Kernel::UInt32 TimeDateStamp;
	Kernel::UInt32 ForwarderChain;
	Kernel::UInt32 NameRva;
	Kernel::UInt32 ThunkTableRva;
} LDR_IMPORT_DIRECTORY, *LDR_IMPORT_DIRECTORY_PTR;

typedef struct LDR_DATA_DIRECTORY
{
	Kernel::UInt32 VirtualAddress;
	Kernel::UInt32 Size;
} LDR_DATA_DIRECTORY, *LDR_DATA_DIRECTORY_PTR;

typedef struct LDR_IMAGE_HEADER
{
	LDR_EXEC_HEADER		Header;
	LDR_OPTIONAL_HEADER OptHdr;
} LDR_IMAGE_HEADER, *LDR_IMAGE_HEADER_PTR;

enum
{
	kUserSection  = 0x00000020,
	kPEResourceId = 0xFFaadd00,
};

#endif /* ifndef __KERNELKIT_INC_PE_H__ */