using System; using System.Collections.Generic; using System.IO; using System.Text; using static BurnOutSharp.Builder.Extensions; namespace BurnOutSharp.Wrappers { public class NewExecutable : WrapperBase { #region Pass-Through Properties #region MS-DOS Stub #region Standard Fields /// public byte[] Stub_Magic => _executable.Stub.Header.Magic; /// public ushort Stub_LastPageBytes => _executable.Stub.Header.LastPageBytes; /// public ushort Stub_Pages => _executable.Stub.Header.Pages; /// public ushort Stub_RelocationItems => _executable.Stub.Header.RelocationItems; /// public ushort Stub_HeaderParagraphSize => _executable.Stub.Header.HeaderParagraphSize; /// public ushort Stub_MinimumExtraParagraphs => _executable.Stub.Header.MinimumExtraParagraphs; /// public ushort Stub_MaximumExtraParagraphs => _executable.Stub.Header.MaximumExtraParagraphs; /// public ushort Stub_InitialSSValue => _executable.Stub.Header.InitialSSValue; /// public ushort Stub_InitialSPValue => _executable.Stub.Header.InitialSPValue; /// public ushort Stub_Checksum => _executable.Stub.Header.Checksum; /// public ushort Stub_InitialIPValue => _executable.Stub.Header.InitialIPValue; /// public ushort Stub_InitialCSValue => _executable.Stub.Header.InitialCSValue; /// public ushort Stub_RelocationTableAddr => _executable.Stub.Header.RelocationTableAddr; /// public ushort Stub_OverlayNumber => _executable.Stub.Header.OverlayNumber; #endregion #region PE Extensions /// public ushort[] Stub_Reserved1 => _executable.Stub.Header.Reserved1; /// public ushort Stub_OEMIdentifier => _executable.Stub.Header.OEMIdentifier; /// public ushort Stub_OEMInformation => _executable.Stub.Header.OEMInformation; /// public ushort[] Stub_Reserved2 => _executable.Stub.Header.Reserved2; /// public uint Stub_NewExeHeaderAddr => _executable.Stub.Header.NewExeHeaderAddr; #endregion #endregion #region Header /// public byte[] Magic => _executable.Header.Magic; /// public byte LinkerVersion => _executable.Header.LinkerVersion; /// public byte LinkerRevision => _executable.Header.LinkerRevision; /// public ushort EntryTableOffset => _executable.Header.EntryTableOffset; /// public ushort EntryTableSize => _executable.Header.EntryTableSize; /// public uint CrcChecksum => _executable.Header.CrcChecksum; /// public Models.NewExecutable.HeaderFlag FlagWord => _executable.Header.FlagWord; /// public ushort AutomaticDataSegmentNumber => _executable.Header.AutomaticDataSegmentNumber; /// public ushort InitialHeapAlloc => _executable.Header.InitialHeapAlloc; /// public ushort InitialStackAlloc => _executable.Header.InitialStackAlloc; /// public uint InitialCSIPSetting => _executable.Header.InitialCSIPSetting; /// public uint InitialSSSPSetting => _executable.Header.InitialSSSPSetting; /// public ushort FileSegmentCount => _executable.Header.FileSegmentCount; /// public ushort ModuleReferenceTableSize => _executable.Header.ModuleReferenceTableSize; /// public ushort NonResidentNameTableSize => _executable.Header.NonResidentNameTableSize; /// public ushort SegmentTableOffset => _executable.Header.SegmentTableOffset; /// public ushort ResourceTableOffset => _executable.Header.ResourceTableOffset; /// public ushort ResidentNameTableOffset => _executable.Header.ResidentNameTableOffset; /// public ushort ModuleReferenceTableOffset => _executable.Header.ModuleReferenceTableOffset; /// public ushort ImportedNamesTableOffset => _executable.Header.ImportedNamesTableOffset; /// public uint NonResidentNamesTableOffset => _executable.Header.NonResidentNamesTableOffset; /// public ushort MovableEntriesCount => _executable.Header.MovableEntriesCount; /// public ushort SegmentAlignmentShiftCount => _executable.Header.SegmentAlignmentShiftCount; /// public ushort ResourceEntriesCount => _executable.Header.ResourceEntriesCount; /// public Models.NewExecutable.OperatingSystem TargetOperatingSystem => _executable.Header.TargetOperatingSystem; /// public Models.NewExecutable.OS2Flag AdditionalFlags => _executable.Header.AdditionalFlags; /// public ushort ReturnThunkOffset => _executable.Header.ReturnThunkOffset; /// public ushort SegmentReferenceThunkOffset => _executable.Header.SegmentReferenceThunkOffset; /// public ushort MinCodeSwapAreaSize => _executable.Header.MinCodeSwapAreaSize; /// public byte WindowsSDKRevision => _executable.Header.WindowsSDKRevision; /// public byte WindowsSDKVersion => _executable.Header.WindowsSDKVersion; #endregion #region Tables /// public Models.NewExecutable.SegmentTableEntry[] SegmentTable => _executable.SegmentTable; /// public Models.NewExecutable.ResourceTable ResourceTable => _executable.ResourceTable; /// public Models.NewExecutable.ResidentNameTableEntry[] ResidentNameTable => _executable.ResidentNameTable; /// public Models.NewExecutable.ModuleReferenceTableEntry[] ModuleReferenceTable => _executable.ModuleReferenceTable; /// public Dictionary ImportedNameTable => _executable.ImportedNameTable; /// public Models.NewExecutable.EntryTableBundle[] EntryTable => _executable.EntryTable; /// public Models.NewExecutable.NonResidentNameTableEntry[] NonResidentNameTable => _executable.NonResidentNameTable; #endregion #endregion #region Extension Properties // TODO: Determine what extension properties are needed #endregion #region Instance Variables /// /// Internal representation of the executable /// private Models.NewExecutable.Executable _executable; #endregion #region Constructors /// /// Private constructor /// private NewExecutable() { } /// /// Create an NE executable from a byte array and offset /// /// Byte array representing the executable /// Offset within the array to parse /// An NE executable wrapper on success, null on failure public static NewExecutable Create(byte[] data, int offset) { var executable = Builder.NewExecutable.ParseExecutable(data, offset); if (executable == null) return null; var wrapper = new NewExecutable { _executable = executable, _dataSource = DataSource.ByteArray, _byteArrayData = data, _byteArrayOffset = offset, }; return wrapper; } /// /// Create an NE executable from a Stream /// /// Stream representing the executable /// An NE executable wrapper on success, null on failure public static NewExecutable Create(Stream data) { var executable = Builder.NewExecutable.ParseExecutable(data); if (executable == null) return null; var wrapper = new NewExecutable { _executable = executable, _dataSource = DataSource.Stream, _streamData = data, }; return wrapper; } #endregion #region Printing /// public override void Print() { Console.WriteLine("New Executable Information:"); Console.WriteLine("-------------------------"); Console.WriteLine(); // Stub PrintStubHeader(); PrintStubExtendedHeader(); // Header PrintHeader(); // Tables PrintSegmentTable(); PrintResourceTable(); PrintResidentNameTable(); PrintModuleReferenceTable(); PrintImportedNameTable(); PrintEntryTable(); PrintNonresidentNameTable(); } /// /// Print stub header information /// private void PrintStubHeader() { Console.WriteLine(" MS-DOS Stub Header Information:"); Console.WriteLine(" -------------------------"); Console.WriteLine($" Magic number: {BitConverter.ToString(Stub_Magic).Replace("-", string.Empty)}"); Console.WriteLine($" Last page bytes: {Stub_LastPageBytes}"); Console.WriteLine($" Pages: {Stub_Pages}"); Console.WriteLine($" Relocation items: {Stub_RelocationItems}"); Console.WriteLine($" Header paragraph size: {Stub_HeaderParagraphSize}"); Console.WriteLine($" Minimum extra paragraphs: {Stub_MinimumExtraParagraphs}"); Console.WriteLine($" Maximum extra paragraphs: {Stub_MaximumExtraParagraphs}"); Console.WriteLine($" Initial SS value: {Stub_InitialSSValue}"); Console.WriteLine($" Initial SP value: {Stub_InitialSPValue}"); Console.WriteLine($" Checksum: {Stub_Checksum}"); Console.WriteLine($" Initial IP value: {Stub_InitialIPValue}"); Console.WriteLine($" Initial CS value: {Stub_InitialCSValue}"); Console.WriteLine($" Relocation table address: {Stub_RelocationTableAddr}"); Console.WriteLine($" Overlay number: {Stub_OverlayNumber}"); Console.WriteLine(); } /// /// Print stub extended header information /// private void PrintStubExtendedHeader() { Console.WriteLine(" MS-DOS Stub Extended Header Information:"); Console.WriteLine(" -------------------------"); Console.WriteLine($" Reserved words: {string.Join(", ", Stub_Reserved1)}"); Console.WriteLine($" OEM identifier: {Stub_OEMIdentifier}"); Console.WriteLine($" OEM information: {Stub_OEMInformation}"); Console.WriteLine($" Reserved words: {string.Join(", ", Stub_Reserved2)}"); Console.WriteLine($" New EXE header address: {Stub_NewExeHeaderAddr}"); Console.WriteLine(); } /// /// Print header information /// private void PrintHeader() { Console.WriteLine(" Header Information:"); Console.WriteLine(" -------------------------"); Console.WriteLine($" Magic number: {BitConverter.ToString(Magic).Replace("-", string.Empty)}"); Console.WriteLine($" Linker version: {LinkerVersion}"); Console.WriteLine($" Linker revision: {LinkerRevision}"); Console.WriteLine($" Entry table offset: {EntryTableOffset}"); Console.WriteLine($" Entry table size: {EntryTableSize}"); Console.WriteLine($" CRC checksum: {CrcChecksum}"); Console.WriteLine($" Flag word: {FlagWord}"); Console.WriteLine($" Automatic data segment number: {AutomaticDataSegmentNumber}"); Console.WriteLine($" Initial heap allocation: {InitialHeapAlloc}"); Console.WriteLine($" Initial stack allocation: {InitialStackAlloc}"); Console.WriteLine($" Initial CS:IP setting: {InitialCSIPSetting}"); Console.WriteLine($" Initial SS:SP setting: {InitialSSSPSetting}"); Console.WriteLine($" File segment count: {FileSegmentCount}"); Console.WriteLine($" Module reference table size: {ModuleReferenceTableSize}"); Console.WriteLine($" Non-resident name table size: {NonResidentNameTableSize}"); Console.WriteLine($" Segment table offset: {SegmentTableOffset}"); Console.WriteLine($" Resource table offset: {ResourceTableOffset}"); Console.WriteLine($" Resident name table offset: {ResidentNameTableOffset}"); Console.WriteLine($" Module reference table offset: {ModuleReferenceTableOffset}"); Console.WriteLine($" Imported names table offset: {ImportedNamesTableOffset}"); Console.WriteLine($" Non-resident name table offset: {NonResidentNamesTableOffset}"); Console.WriteLine($" Moveable entries count: {MovableEntriesCount}"); Console.WriteLine($" Segment alignment shift count: {SegmentAlignmentShiftCount}"); Console.WriteLine($" Resource entries count: {ResourceEntriesCount}"); Console.WriteLine($" Target operating system: {TargetOperatingSystem}"); Console.WriteLine($" Additional flags: {AdditionalFlags}"); Console.WriteLine($" Return thunk offset: {ReturnThunkOffset}"); Console.WriteLine($" Segment reference thunk offset: {SegmentReferenceThunkOffset}"); Console.WriteLine($" Minimum code swap area size: {MinCodeSwapAreaSize}"); Console.WriteLine($" Windows SDK revision: {WindowsSDKRevision}"); Console.WriteLine($" Windows SDK version: {WindowsSDKVersion}"); Console.WriteLine(); } /// /// Print segment table information /// private void PrintSegmentTable() { Console.WriteLine(" Segment Table Information:"); Console.WriteLine(" -------------------------"); if (FileSegmentCount == 0 || SegmentTable.Length == 0) { Console.WriteLine(" No segment table items"); } else { for (int i = 0; i < SegmentTable.Length; i++) { var entry = SegmentTable[i]; Console.WriteLine($" Segment Table Entry {i}"); Console.WriteLine($" Offset = {entry.Offset}"); Console.WriteLine($" Length = {entry.Length}"); Console.WriteLine($" Flag word = {entry.FlagWord}"); Console.WriteLine($" Minimum allocation size = {entry.MinimumAllocationSize}"); } } Console.WriteLine(); } /// /// Print resource table information /// private void PrintResourceTable() { Console.WriteLine(" Resource Table Information:"); Console.WriteLine(" -------------------------"); Console.WriteLine($" Alignment shift count: {ResourceTable.AlignmentShiftCount}"); if (ResourceEntriesCount == 0 || ResourceTable.ResourceTypes.Length == 0) { Console.WriteLine(" No resource table items"); } else { for (int i = 0; i < ResourceTable.ResourceTypes.Length; i++) { // TODO: If not integer type, print out name var entry = ResourceTable.ResourceTypes[i]; Console.WriteLine($" Resource Table Entry {i}"); Console.WriteLine($" Type ID = {entry.TypeID} (Is Integer Type: {entry.IsIntegerType()})"); Console.WriteLine($" Resource count = {entry.ResourceCount}"); Console.WriteLine($" Reserved = {entry.Reserved}"); Console.WriteLine($" Resources = "); if (entry.ResourceCount == 0 || entry.Resources.Length == 0) { Console.WriteLine(" No resource items"); } else { for (int j = 0; j < entry.Resources.Length; j++) { // TODO: If not integer type, print out name var resource = entry.Resources[j]; Console.WriteLine($" Resource Entry {i}"); Console.WriteLine($" Offset = {resource.Offset}"); Console.WriteLine($" Length = {resource.Length}"); Console.WriteLine($" Flag word = {resource.FlagWord}"); Console.WriteLine($" Resource ID = {resource.ResourceID} (Is Integer Type: {resource.IsIntegerType()})"); Console.WriteLine($" Reserved = {resource.Reserved}"); } } } } if (ResourceTable.TypeAndNameStrings.Count == 0) { Console.WriteLine(" No resource table type/name strings"); } else { foreach (var typeAndNameString in ResourceTable.TypeAndNameStrings) { Console.WriteLine($" Resource Type/Name Offset {typeAndNameString.Key}"); Console.WriteLine($" Length = {typeAndNameString.Value.Length}"); Console.WriteLine($" Text = {Encoding.ASCII.GetString(typeAndNameString.Value.Text)}"); } } Console.WriteLine(); } /// /// Print resident-name table information /// private void PrintResidentNameTable() { Console.WriteLine(" Resident-Name Table Information:"); Console.WriteLine(" -------------------------"); if (ResidentNameTableOffset == 0 || ResidentNameTable.Length == 0) { Console.WriteLine(" No resident-name table items"); } else { for (int i = 0; i < ResidentNameTable.Length; i++) { var entry = ResidentNameTable[i]; Console.WriteLine($" Resident-Name Table Entry {i}"); Console.WriteLine($" Length = {entry.Length}"); Console.WriteLine($" Name string = {(entry.NameString != null ? Encoding.ASCII.GetString(entry.NameString) : "[EMPTY]")}"); Console.WriteLine($" Ordinal number = {entry.OrdinalNumber}"); } } Console.WriteLine(); } /// /// Print module-reference table information /// private void PrintModuleReferenceTable() { Console.WriteLine(" Module-Reference Table Information:"); Console.WriteLine(" -------------------------"); if (ModuleReferenceTableSize == 0 || ModuleReferenceTable.Length == 0) { Console.WriteLine(" No module-reference table items"); } else { for (int i = 0; i < ModuleReferenceTable.Length; i++) { // TODO: Read the imported names table and print value here var entry = ModuleReferenceTable[i]; Console.WriteLine($" Module-Reference Table Entry {i}"); Console.WriteLine($" Offset = {entry.Offset} (adjusted to be {entry.Offset + Stub_NewExeHeaderAddr + ImportedNamesTableOffset})"); } } Console.WriteLine(); } /// /// Print imported-name table information /// private void PrintImportedNameTable() { Console.WriteLine(" Imported-Name Table Information:"); Console.WriteLine(" -------------------------"); if (ImportedNamesTableOffset == 0 || ImportedNameTable.Count == 0) { Console.WriteLine(" No imported-name table items"); } else { foreach (var entry in ImportedNameTable) { Console.WriteLine($" Imported-Name Table at Offset {entry.Key}"); Console.WriteLine($" Length = {entry.Value.Length}"); Console.WriteLine($" Name string = {(entry.Value.NameString != null ? Encoding.ASCII.GetString(entry.Value.NameString) : "[EMPTY]")}"); } } Console.WriteLine(); } /// /// Print entry table information /// private void PrintEntryTable() { Console.WriteLine(" Entry Table Information:"); Console.WriteLine(" -------------------------"); if (EntryTableSize == 0 || EntryTable.Length == 0) { Console.WriteLine(" No entry table items"); } else { for (int i = 0; i < EntryTable.Length; i++) { var entry = EntryTable[i]; Console.WriteLine($" Entry Table Entry {i}"); Console.WriteLine($" Entry count = {entry.EntryCount}"); Console.WriteLine($" Segment indicator = {entry.SegmentIndicator} ({entry.GetEntryType()})"); switch (entry.GetEntryType()) { case BurnOutSharp.Models.NewExecutable.SegmentEntryType.FixedSegment: Console.WriteLine($" Flag word = {entry.FixedFlagWord}"); Console.WriteLine($" Offset = {entry.FixedOffset}"); break; case BurnOutSharp.Models.NewExecutable.SegmentEntryType.MoveableSegment: Console.WriteLine($" Flag word = {entry.MoveableFlagWord}"); Console.WriteLine($" Reserved = {entry.MoveableReserved}"); Console.WriteLine($" Segment number = {entry.MoveableSegmentNumber}"); Console.WriteLine($" Offset = {entry.MoveableOffset}"); break; } } } Console.WriteLine(); } /// /// Print nonresident-name table information /// private void PrintNonresidentNameTable() { Console.WriteLine(" Nonresident-Name Table Information:"); Console.WriteLine(" -------------------------"); if (NonResidentNameTableSize == 0 || NonResidentNameTable.Length == 0) { Console.WriteLine(" No nonresident-name table items"); } else { for (int i = 0; i < NonResidentNameTable.Length; i++) { var entry = NonResidentNameTable[i]; Console.WriteLine($" Nonresident-Name Table Entry {i}"); Console.WriteLine($" Length = {entry.Length}"); Console.WriteLine($" Name string = {(entry.NameString != null ? Encoding.ASCII.GetString(entry.NameString) : "[EMPTY]")}"); Console.WriteLine($" Ordinal number = {entry.OrdinalNumber}"); } } Console.WriteLine(); } #endregion #region REMOVE -- DO NOT USE /// /// Read an arbitrary range from the source /// /// The start of where to read data from, -1 means start of source /// How many bytes to read, -1 means read until end /// Byte array representing the range, null on error [Obsolete] public byte[] ReadArbitraryRange(int rangeStart = -1, int length = -1) { // If we have an unset range start, read from the start of the source if (rangeStart == -1) rangeStart = 0; // If we have an unset length, read the whole source if (length == -1) { switch (_dataSource) { case DataSource.ByteArray: length = _byteArrayData.Length - _byteArrayOffset; break; case DataSource.Stream: length = (int)_streamData.Length; break; } } return ReadFromDataSource(rangeStart, length); } #endregion } }