Win32/Sojaner.MemoryScanner/PEReader.cs

#region Logging Defines
// include this any class or method that required logging, and comment-out what is not needed

#region Enabled logging levels
#define LOGGING_ENABLE_INFO
#define LOGGING_ENABLE_WARN
#define LOGGING_ENABLE_DEBUG
//#define LOGGING_ENABLE_VERBOSEDEBUG
#define LOGGING_ENABLE_ERROR
#define LOGGING_ENABLE_VERBOSEERROR
#define LOGGING_ENABLE_PROFILER
#endregion
#endregion
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.IO;
using RomCheater.Logging;
using System.Runtime.InteropServices;
using System.Diagnostics;
using System.ComponentModel;
using RomCheater.PluginFramework.Events;
using Sojaner.MemoryScanner;
using Sojaner.MemoryScanner.MemoryProviers;
using RomCheater.PluginFramework.Interfaces;


namespace Sojaner.MemoryScanner
{

    #region Structs
    #region helpers
    [Flags]
    public enum MachineTypeFlags
    {
        x86 = 0x14C,
        Alpha = 0x184,
        ARM = 0x1C0,
        MIPS16R3000 = 0x162,
        MIPS16R4000 = 0x166,
        MIPS16R10000 = 0x168,
        PowerPCLE = 0x1F0,
        PowerPCBE = 0x1F2,
        Itanium = 0x200,
        MIPS16 = 0x266,
        Alpha64 = 0x284,
        MIPSFPU = 0x366,
        MIPSFPU16 = 0x466,
        x64 = 0x8664,
    }
    public enum MagicType : ushort
    {
        NT_OPTIONAL_HEADER_NOT_PRESENT, // 0     
        NT_OPTIONAL_HEADER_32 = 0x10b,
        NT_OPTIONAL_HEADER_64 = 0x20b
    }
    public enum SubSystemType : ushort
    {
        IMAGE_SUBSYSTEM_UNKNOWN = 0,
        IMAGE_SUBSYSTEM_NATIVE = 1,
        IMAGE_SUBSYSTEM_WINDOWS_GUI = 2,
        IMAGE_SUBSYSTEM_WINDOWS_CUI = 3,
        IMAGE_SUBSYSTEM_POSIX_CUI = 7,
        IMAGE_SUBSYSTEM_WINDOWS_CE_GUI = 9,
        IMAGE_SUBSYSTEM_EFI_APPLICATION = 10,
        IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER = 11,
        IMAGE_SUBSYSTEM_EFI_RUNTIME_DRIVER = 12,
        IMAGE_SUBSYSTEM_EFI_ROM = 13,
        IMAGE_SUBSYSTEM_XBOX = 14

    }
    public enum DllCharacteristicsType : ushort
    {
        RES_0 = 0x0001,
        RES_1 = 0x0002,
        RES_2 = 0x0004,
        RES_3 = 0x0008,
        IMAGE_DLL_CHARACTERISTICS_DYNAMIC_BASE = 0x0040,
        IMAGE_DLL_CHARACTERISTICS_FORCE_INTEGRITY = 0x0080,
        IMAGE_DLL_CHARACTERISTICS_NX_COMPAT = 0x0100,
        IMAGE_DLLCHARACTERISTICS_NO_ISOLATION = 0x0200,
        IMAGE_DLLCHARACTERISTICS_NO_SEH = 0x0400,
        IMAGE_DLLCHARACTERISTICS_NO_BIND = 0x0800,
        RES_4 = 0x1000,
        IMAGE_DLLCHARACTERISTICS_WDM_DRIVER = 0x2000,
        IMAGE_DLLCHARACTERISTICS_TERMINAL_SERVER_AWARE = 0x8000
    }
    [Flags]
    public enum FileCharacteristicType : ushort
    {
        RelocationInformationStrippedFromFile = 0x1,
        Executable = 0x2,
        LineNumbersStripped = 0x4,
        SymbolTableStripped = 0x8,
        AggresiveTrimWorkingSet = 0x10,
        LargeAddressAware = 0x20,
        Supports16Bit = 0x40,
        ReservedBytesWo = 0x80,
        Supports32Bit = 0x100,
        DebugInfoStripped = 0x200,
        RunFromSwapIfInRemovableMedia = 0x400,
        RunFromSwapIfInNetworkMedia = 0x800,
        IsSytemFile = 0x1000,
        IsDLL = 0x2000,
        IsOnlyForSingleCoreProcessor = 0x4000,
        BytesOfWordReserved = 0x8000,
    }
    #region DataSectionFlags
    [Flags]
    public enum DataSectionFlags : uint
    {
        /// <summary>
        /// Reserved for future use.
        /// </summary>
        TypeReg = 0x00000000,
        /// <summary>
        /// Reserved for future use.
        /// </summary>
        TypeDsect = 0x00000001,
        /// <summary>
        /// Reserved for future use.
        /// </summary>
        TypeNoLoad = 0x00000002,
        /// <summary>
        /// Reserved for future use.
        /// </summary>
        TypeGroup = 0x00000004,
        /// <summary>
        /// The section should not be padded to the next boundary. This flag is obsolete and is replaced by IMAGE_SCN_ALIGN_1BYTES. This is valid only for object files.
        /// </summary>
        TypeNoPadded = 0x00000008,
        /// <summary>
        /// Reserved for future use.
        /// </summary>
        TypeCopy = 0x00000010,
        /// <summary>
        /// The section contains executable code.
        /// </summary>
        ContentCode = 0x00000020,
        /// <summary>
        /// The section contains initialized data.
        /// </summary>
        ContentInitializedData = 0x00000040,
        /// <summary>
        /// The section contains uninitialized data.
        /// </summary>
        ContentUninitializedData = 0x00000080,
        /// <summary>
        /// Reserved for future use.
        /// </summary>
        LinkOther = 0x00000100,
        /// <summary>
        /// The section contains comments or other information. The .drectve section has this type. This is valid for object files only.
        /// </summary>
        LinkInfo = 0x00000200,
        /// <summary>
        /// Reserved for future use.
        /// </summary>
        TypeOver = 0x00000400,
        /// <summary>
        /// The section will not become part of the image. This is valid only for object files.
        /// </summary>
        LinkRemove = 0x00000800,
        /// <summary>
        /// The section contains COMDAT data. For more information, see section 5.5.6, COMDAT Sections (Object Only). This is valid only for object files.
        /// </summary>
        LinkComDat = 0x00001000,
        /// <summary>
        /// Reset speculative exceptions handling bits in the TLB entries for this section.
        /// </summary>
        NoDeferSpecExceptions = 0x00004000,
        /// <summary>
        /// The section contains data referenced through the global pointer (GP).
        /// </summary>
        RelativeGP = 0x00008000,
        /// <summary>
        /// Reserved for future use.
        /// </summary>
        MemPurgeable = 0x00020000,
        /// <summary>
        /// Reserved for future use.
        /// </summary>
        Memory16Bit = 0x00020000,
        /// <summary>
        /// Reserved for future use.
        /// </summary>
        MemoryLocked = 0x00040000,
        /// <summary>
        /// Reserved for future use.
        /// </summary>
        MemoryPreload = 0x00080000,
        /// <summary>
        /// Align data on a 1-byte boundary. Valid only for object files.
        /// </summary>
        Align1Bytes = 0x00100000,
        /// <summary>
        /// Align data on a 2-byte boundary. Valid only for object files.
        /// </summary>
        Align2Bytes = 0x00200000,
        /// <summary>
        /// Align data on a 4-byte boundary. Valid only for object files.
        /// </summary>
        Align4Bytes = 0x00300000,
        /// <summary>
        /// Align data on an 8-byte boundary. Valid only for object files.
        /// </summary>
        Align8Bytes = 0x00400000,
        /// <summary>
        /// Align data on a 16-byte boundary. Valid only for object files.
        /// </summary>
        Align16Bytes = 0x00500000,
        /// <summary>
        /// Align data on a 32-byte boundary. Valid only for object files.
        /// </summary>
        Align32Bytes = 0x00600000,
        /// <summary>
        /// Align data on a 64-byte boundary. Valid only for object files.
        /// </summary>
        Align64Bytes = 0x00700000,
        /// <summary>
        /// Align data on a 128-byte boundary. Valid only for object files.
        /// </summary>
        Align128Bytes = 0x00800000,
        /// <summary>
        /// Align data on a 256-byte boundary. Valid only for object files.
        /// </summary>
        Align256Bytes = 0x00900000,
        /// <summary>
        /// Align data on a 512-byte boundary. Valid only for object files.
        /// </summary>
        Align512Bytes = 0x00A00000,
        /// <summary>
        /// Align data on a 1024-byte boundary. Valid only for object files.
        /// </summary>
        Align1024Bytes = 0x00B00000,
        /// <summary>
        /// Align data on a 2048-byte boundary. Valid only for object files.
        /// </summary>
        Align2048Bytes = 0x00C00000,
        /// <summary>
        /// Align data on a 4096-byte boundary. Valid only for object files.
        /// </summary>
        Align4096Bytes = 0x00D00000,
        /// <summary>
        /// Align data on an 8192-byte boundary. Valid only for object files.
        /// </summary>
        Align8192Bytes = 0x00E00000,
        /// <summary>
        /// The section contains extended relocations.
        /// </summary>
        LinkExtendedRelocationOverflow = 0x01000000,
        /// <summary>
        /// The section can be discarded as needed.
        /// </summary>
        MemoryDiscardable = 0x02000000,
        /// <summary>
        /// The section cannot be cached.
        /// </summary>
        MemoryNotCached = 0x04000000,
        /// <summary>
        /// The section is not pageable.
        /// </summary>
        MemoryNotPaged = 0x08000000,
        /// <summary>
        /// The section can be shared in memory.
        /// </summary>
        MemoryShared = 0x10000000,
        /// <summary>
        /// The section can be executed as code.
        /// </summary>
        MemoryExecute = 0x20000000,
        /// <summary>
        /// The section can be read.
        /// </summary>
        MemoryRead = 0x40000000,
        /// <summary>
        /// The section can be written to.
        /// </summary>
        MemoryWrite = 0x80000000
    }
    #endregion
    #endregion
    [TypeConverter(typeof(ExpandableObjectConverter))]
    [StructLayout(LayoutKind.Sequential)]
    public struct IMAGE_DOS_HEADER
    {
        public UInt16 _e_magic;
        public UInt16 _e_cblp;
        public UInt16 _e_cp;
        public UInt16 _e_crlc;
        public UInt16 _e_cparhdr;
        public UInt16 _e_minalloc;
        public UInt16 _e_maxalloc;
        public UInt16 _e_ss;
        public UInt16 _e_sp;
        public UInt16 _e_csum;
        public UInt16 _e_ip;
        public UInt16 _e_cs;
        public UInt16 _e_lfarlc;
        public UInt16 _e_ovno;
        [MarshalAs(UnmanagedType.ByValArray, SizeConst = 4)]
        public UInt16[] _e_res1;
        public UInt16 _e_oemid;
        public UInt16 _e_oeminfo;
        [MarshalAs(UnmanagedType.ByValArray, SizeConst = 10)]
        public UInt16[] _e_res2;
        public UInt32 _e_lfanew;

        public string e_magic { get { return string.Format("0x{0:x4}", _e_magic); } }
        public string e_cblp { get { return string.Format("0x{0:x4}", _e_cblp); } }
        public string e_cp { get { return string.Format("0x{0:x4}", _e_cp); } }
        public string e_crlc { get { return string.Format("0x{0:x4}", _e_crlc); } }
        public string e_cparhdr { get { return string.Format("0x{0:x4}", _e_cparhdr); } }
        public string e_minalloc { get { return string.Format("0x{0:x4}", _e_minalloc); } }
        public string e_maxalloc { get { return string.Format("0x{0:x4}", _e_maxalloc); } }
        public string e_ss { get { return string.Format("0x{0:x4}", _e_ss); } }
        public string e_sp { get { return string.Format("0x{0:x4}", _e_sp); } }
        public string e_csum { get { return string.Format("0x{0:x4}", _e_csum); } }
        public string e_ip { get { return string.Format("0x{0:x4}", _e_ip); } }
        public string e_cs { get { return string.Format("0x{0:x4}", _e_cs); } }
        public string e_lfarlc { get { return string.Format("0x{0:x4}", _e_lfarlc); } }
        public string e_ovno { get { return string.Format("0x{0:x4}", _e_ovno); } }
        public ushort[] e_res1 { get { return _e_res1; } }
        public string e_oemid { get { return string.Format("0x{0:x4}", _e_oemid); } }
        public string e_oeminfo { get { return string.Format("0x{0:x4}", _e_oeminfo); } }
        public ushort[] e_res2 { get { return _e_res2; } }
        public string e_lfanew { get { return string.Format("0x{0:x8}", _e_lfanew); } }

        public override string ToString()
        {
            return Encoding.UTF8.GetString(BitConverter.GetBytes(_e_magic));
        }
    }
    [TypeConverter(typeof(ExpandableObjectConverter))]
    [StructLayout(LayoutKind.Sequential)]
    public struct IMAGE_NT_HEADERS
    {
        public UInt32 _Signature;
        public IMAGE_FILE_HEADER _FileHeader;
        public IMAGE_OPTIONAL_HEADER32 _OptionalHeader32;
        public IMAGE_OPTIONAL_HEADER64 _OptionalHeader64;

        public string Signature { get { return string.Format("0x{0:x8}", _Signature); } }
        public IMAGE_FILE_HEADER FileHeader { get { return _FileHeader; } }
        public IMAGE_OPTIONAL_HEADER32 OptionalHeader32 { get { return _OptionalHeader32; } }
        public IMAGE_OPTIONAL_HEADER64 OptionalHeader64 { get { return _OptionalHeader64; } }

        public override string ToString()
        {
            return Encoding.UTF8.GetString(BitConverter.GetBytes(_Signature));
        }
    }
    [TypeConverter(typeof(ExpandableObjectConverter))]
    [StructLayout(LayoutKind.Sequential)]
    public struct IMAGE_FILE_HEADER
    {
        public UInt16 _MachineType;
        public UInt16 _NumberOfSections;
        public Int32 _LinkerTime; // The time that the linker (or compiler for an OBJ file) produced this file. This field holds the number of seconds since December 31st, 1969, at 4:00 P.M.
        public UInt32 _PointerToSymbolTable;
        public UInt32 _NumberOfSymbols;
        public UInt16 _SizeOfOptionalHeader;
        public FileCharacteristicType _Characteristics;


        public string MachineType { get { return ((MachineTypeFlags)_MachineType).ToString(); } }
        public string NumberOfSections { get { return string.Format("0x{0:x4}", _NumberOfSections); } }
        public string LinkerTime { get { return string.Format("{0} (0x{1:x8})", GetDateTimeFromLinkerTime(_LinkerTime).ToString(), _LinkerTime); } }
        public string PointerToSymbolTable { get { return string.Format("0x{0:x8}", _PointerToSymbolTable); } }
        public string NumberOfSymbols { get { return string.Format("0x{0:x8}", _NumberOfSymbols); } }
        public string SizeOfOptionalHeader { get { return string.Format("0x{0:x4}", _SizeOfOptionalHeader); } }
        public FileCharacteristicType Characteristics { get { return _Characteristics; } }
        public override string ToString()
        {
            return MachineType;
        }

        private DateTime GetDateTimeFromLinkerTime(Int32 linker_timer)
        {
            // http://msdn.microsoft.com/en-us/library/ms809762.aspx
            // This field holds the number of seconds since December 31st, 1969, at 4:00 P.M.
            DateTime t = new DateTime(1969, 12, 31, 16, 0, 0);
            t = t.AddSeconds(linker_timer);
            return t;
        }

    }
    [TypeConverter(typeof(ExpandableObjectConverter))]
    [StructLayout(LayoutKind.Sequential)]
    public struct IMAGE_OPTIONAL_HEADER32
    {
        public UInt16 _Magic;
        public Byte _MajorLinkerVersion;
        public Byte _MinorLinkerVersion;
        public UInt32 _SizeOfCode;
        public UInt32 _SizeOfInitializedData;
        public UInt32 _SizeOfUninitializedData;
        public UInt32 _AddressOfEntryPoint;
        public UInt32 _BaseOfCode;
        public UInt32 _BaseOfData; // 32-but specific
        public UInt32 _ImageBase;
        public UInt32 _SectionAlignment;
        public UInt32 _FileAlignment;
        public UInt16 _MajorOperatingSystemVersion;
        public UInt16 _MinorOperatingSystemVersion;
        public UInt16 _MajorImageVersion;
        public UInt16 _MinorImageVersion;
        public UInt16 _MajorSubsystemVersion;
        public UInt16 _MinorSubsystemVersion;
        public UInt32 _Win32VersionValue;
        public UInt32 _SizeOfImage;
        public UInt32 _SizeOfHeaders;
        public UInt32 _CheckSum;
        public UInt16 _Subsystem;
        public UInt16 _DllCharacteristics;
        public UInt32 _SizeOfStackReserve;
        public UInt32 _SizeOfStackCommit;
        public UInt32 _SizeOfHeapReserve;
        public UInt32 _SizeOfHeapCommit;
        public UInt32 _LoaderFlags;
        public UInt32 _NumberOfRvaAndSizes;
        [MarshalAs(UnmanagedType.ByValArray, SizeConst = 16)]
        public IMAGE_DATA_DIRECTORY[] _DataDirectory;


        public string Magic { get { return ((MagicType)_Magic).ToString(); } }
        public string MajorLinkerVersion { get { return string.Format("0x{0:x2}", _MajorLinkerVersion); } }
        public string MinorLinkerVersion { get { return string.Format("0x{0:x2}", _MajorLinkerVersion); } }

        public string SizeOfCode { get { return string.Format("0x{0:x2}", _SizeOfCode); } }
        public string SizeOfInitializedData { get { return string.Format("0x{0:x8}", _SizeOfInitializedData); } }
        public string SizeOfUninitializedData { get { return string.Format("0x{0:x8}", _SizeOfUninitializedData); } }
        public string AddressOfEntryPoint { get { return string.Format("0x{0:x8}", _AddressOfEntryPoint); } }
        public string BaseOfCode { get { return string.Format("0x{0:x8}", _BaseOfCode); } }
        public string BaseOfData { get { return string.Format("0x{0:x8}", _BaseOfData); } }
        public string ImageBase { get { return string.Format("0x{0:x16}", _ImageBase); } }

        public string SectionAlignment { get { return string.Format("0x{0:x8}", _SectionAlignment); } }
        public string FileAlignment { get { return string.Format("0x{0:x8}", _FileAlignment); } }

        public string MajorOperatingSystemVersion { get { return string.Format("0x{0:x4}", _MajorOperatingSystemVersion); } }
        public string MinorOperatingSystemVersion { get { return string.Format("0x{0:x4}", _MinorOperatingSystemVersion); } }
        public string MajorImageVersion { get { return string.Format("0x{0:x4}", _MajorImageVersion); } }
        public string MinorImageVersion { get { return string.Format("0x{0:x4}", _MinorImageVersion); } }
        public string MajorSubsystemVersion { get { return string.Format("0x{0:x4}", _MajorSubsystemVersion); } }
        public string MinorSubsystemVersion { get { return string.Format("0x{0:x4}", _MinorSubsystemVersion); } }

        public string Win32VersionValue { get { return string.Format("0x{0:x8}", _Win32VersionValue); } }
        public string SizeOfImage { get { return string.Format("0x{0:x8}", _SizeOfImage); } }
        public string SizeOfHeaders { get { return string.Format("0x{0:x8}", _SizeOfHeaders); } }
        public string CheckSum { get { return string.Format("0x{0:x8}", _CheckSum); } }

        public string Subsystem { get { return ((SubSystemType)_Subsystem).ToString(); } }
        public string DllCharacteristics { get { return string.Format("0x{0:x4}", _DllCharacteristics); } }

        public string SizeOfStackReserve { get { return string.Format("0x{0:x16}", _SizeOfStackReserve); } }
        public string SizeOfStackCommit { get { return string.Format("0x{0:x16}", _SizeOfStackCommit); } }
        public string SizeOfHeapReserve { get { return string.Format("0x{0:x16}", _SizeOfHeapReserve); } }
        public string SizeOfHeapCommit { get { return string.Format("0x{0:x16}", _SizeOfHeapCommit); } }

        public string LoaderFlags { get { return string.Format("0x{0:x8}", _LoaderFlags); } }
        public string NumberOfRvaAndSizes { get { return string.Format("0x{0:x8}", _NumberOfRvaAndSizes); } }
        public override string ToString()
        {
            return Magic;
        }
    }
    [TypeConverter(typeof(ExpandableObjectConverter))]
    [StructLayout(LayoutKind.Sequential)]
    public struct IMAGE_OPTIONAL_HEADER64
    {
        public UInt16 _Magic;
        public Byte _MajorLinkerVersion;
        public Byte _MinorLinkerVersion;
        public UInt32 _SizeOfCode;
        public UInt32 _SizeOfInitializedData;
        public UInt32 _SizeOfUninitializedData;
        public UInt32 _AddressOfEntryPoint;
        public UInt32 _BaseOfCode;
        public UInt64 _ImageBase;
        public UInt32 _SectionAlignment;
        public UInt32 _FileAlignment;
        public UInt16 _MajorOperatingSystemVersion;
        public UInt16 _MinorOperatingSystemVersion;
        public UInt16 _MajorImageVersion;
        public UInt16 _MinorImageVersion;
        public UInt16 _MajorSubsystemVersion;
        public UInt16 _MinorSubsystemVersion;
        public UInt32 _Win32VersionValue;
        public UInt32 _SizeOfImage;
        public UInt32 _SizeOfHeaders;
        public UInt32 _CheckSum;
        public UInt16 _Subsystem;
        public UInt16 _DllCharacteristics;
        public UInt64 _SizeOfStackReserve;
        public UInt64 _SizeOfStackCommit;
        public UInt64 _SizeOfHeapReserve;
        public UInt64 _SizeOfHeapCommit;
        public UInt32 _LoaderFlags;
        public UInt32 _NumberOfRvaAndSizes;
        [MarshalAs(UnmanagedType.ByValArray, SizeConst = 16)]
        public IMAGE_DATA_DIRECTORY[] _DataDirectory;


        public string Magic { get { return ((MagicType)_Magic).ToString(); } }
        public string MajorLinkerVersion { get { return string.Format("0x{0:x2}", _MajorLinkerVersion); } }
        public string MinorLinkerVersion { get { return string.Format("0x{0:x2}", _MajorLinkerVersion); } }

        public string SizeOfCode { get { return string.Format("0x{0:x2}", _SizeOfCode); } }
        public string SizeOfInitializedData { get { return string.Format("0x{0:x8}", _SizeOfInitializedData); } }
        public string SizeOfUninitializedData { get { return string.Format("0x{0:x8}", _SizeOfUninitializedData); } }
        public string AddressOfEntryPoint { get { return string.Format("0x{0:x8}", _AddressOfEntryPoint); } }
        public string BaseOfCode { get { return string.Format("0x{0:x8}", _BaseOfCode); } }

        public string ImageBase { get { return string.Format("0x{0:x16}", _ImageBase); } }

        public string SectionAlignment { get { return string.Format("0x{0:x8}", _SectionAlignment); } }
        public string FileAlignment { get { return string.Format("0x{0:x8}", _FileAlignment); } }

        public string MajorOperatingSystemVersion { get { return string.Format("0x{0:x4}", _MajorOperatingSystemVersion); } }
        public string MinorOperatingSystemVersion { get { return string.Format("0x{0:x4}", _MinorOperatingSystemVersion); } }
        public string MajorImageVersion { get { return string.Format("0x{0:x4}", _MajorImageVersion); } }
        public string MinorImageVersion { get { return string.Format("0x{0:x4}", _MinorImageVersion); } }
        public string MajorSubsystemVersion { get { return string.Format("0x{0:x4}", _MajorSubsystemVersion); } }
        public string MinorSubsystemVersion { get { return string.Format("0x{0:x4}", _MinorSubsystemVersion); } }

        public string Win32VersionValue { get { return string.Format("0x{0:x8}", _Win32VersionValue); } }
        public string SizeOfImage { get { return string.Format("0x{0:x8}", _SizeOfImage); } }
        public string SizeOfHeaders { get { return string.Format("0x{0:x8}", _SizeOfHeaders); } }
        public string CheckSum { get { return string.Format("0x{0:x8}", _CheckSum); } }

        public string Subsystem { get { return ((SubSystemType)_Subsystem).ToString(); } }
        public string DllCharacteristics { get { return string.Format("0x{0:x4}", _DllCharacteristics); } }

        public string SizeOfStackReserve { get { return string.Format("0x{0:x16}", _SizeOfStackReserve); } }
        public string SizeOfStackCommit { get { return string.Format("0x{0:x16}", _SizeOfStackCommit); } }
        public string SizeOfHeapReserve { get { return string.Format("0x{0:x16}", _SizeOfHeapReserve); } }
        public string SizeOfHeapCommit { get { return string.Format("0x{0:x16}", _SizeOfHeapCommit); } }

        public string LoaderFlags { get { return string.Format("0x{0:x8}", _LoaderFlags); } }
        public string NumberOfRvaAndSizes { get { return string.Format("0x{0:x8}", _NumberOfRvaAndSizes); } }

        public override string ToString()
        {
            return Magic;
        }
    }
    [TypeConverter(typeof(ExpandableObjectConverter))]
    [StructLayout(LayoutKind.Sequential)]
    public struct IMAGE_DATA_DIRECTORY
    {
        public UInt32 _VirtualAddress;
        public UInt32 _Size;

        public string VirtualAddress { get { return string.Format("0x{0:x8}", _VirtualAddress); } }
        public string Size { get { return string.Format("0x{0:x8}", _Size); } }
    }
    [TypeConverter(typeof(ExpandableObjectConverter))]
    [StructLayout(LayoutKind.Sequential)]
    public struct IMAGE_SECTION_HEADER
    {
        [MarshalAs(UnmanagedType.ByValTStr, SizeConst = 8)]
        public string _Name;
        public Misc _Misc;
        public UInt32 _VirtualAddress;
        public UInt32 _SizeOfRawData;
        public UInt32 _PointerToRawData;
        public UInt32 _PointerToRelocations;
        public UInt32 _PointerToLinenumbers;
        public UInt16 _NumberOfRelocations;
        public UInt16 _NumberOfLinenumbers;
        public DataSectionFlags _Characteristics;

        public string Name { get { return _Name; } }
        public Misc Misc { get { return _Misc; } }
        public string VirtualAddress { get { return string.Format("0x{0:x8}", _VirtualAddress); } }
        public string SizeOfRawData { get { return string.Format("0x{0:x8}", _SizeOfRawData); } }
        public string PointerToRawData { get { return string.Format("0x{0:x8}", _PointerToRawData); } }
        public string PointerToRelocations { get { return string.Format("0x{0:x8}", _PointerToRelocations); } }
        public string PointerToLinenumbers { get { return string.Format("0x{0:x8}", _PointerToLinenumbers); } }
        public string NumberOfRelocations { get { return string.Format("0x{0:x4}", _NumberOfRelocations); } }
        public string NumberOfLinenumbers { get { return string.Format("0x{0:x4}", _NumberOfLinenumbers); } }
        public string Characteristics { get { return _Characteristics.ToString(); } }
        public override string ToString()
        {
            return Name;
        }
    }
    [TypeConverter(typeof(ExpandableObjectConverter))]
    [StructLayout(LayoutKind.Explicit)]
    public struct Misc
    {
        [FieldOffset(0)]
        public UInt32 _PhysicalAddress;
        [FieldOffset(0)]
        public UInt32 _VirtualSize;

        public string PhysicalAddress { get { return string.Format("0x{0:x8}", _PhysicalAddress); } }
        public string VirtualSize { get { return string.Format("0x{0:x8}", _VirtualSize); } }

    }

    #endregion
    #region SECTION_DATA_ARRAY
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class SECTION_DATA_ARRAY
    {
        public SECTION_DATA_ARRAY() { Name = ""; Data = new byte[] { }; _Address = 0; _PointerToData = 0; }
        public SECTION_DATA_ARRAY(string name, uint address, int pointertodata, byte[] data)
        {
            Name = name;
            _Address = address;
            _PointerToData = pointertodata;
            Data = data;
        }

        public string Name { get; private set; }
        private uint _Address;
        public string Address { get { return string.Format("0x{0:x8}", _Address); } }
        public string AddressEnd { get { return string.Format("0x{0:x8}", _Address + Data.Count()); } }
        private int _PointerToData;
        public string PointerToData { get { return string.Format("0x{0:x8}", _PointerToData); } }
        public byte[] Data { get; set; }
        public string DataSize { get { return string.Format("0x{0:x8}",Data.Count()); } }
        public override string ToString()
        {
            return string.Format("{0} ({1}) ({2}) {3}", Name, Address, AddressEnd, DataSize);
        }
    }
    #endregion
    #region IMAGE_SECTION_DATA
    public class IMAGE_SECTION_DATA
    {
        private List<SECTION_DATA_ARRAY> sdas = new List<SECTION_DATA_ARRAY>();
        public IMAGE_SECTION_DATA() { }
        public void AddData(string SectionName, uint SectionAddress, int PointerToData, byte[] SectionData)
        {
            SECTION_DATA_ARRAY sda = new SECTION_DATA_ARRAY(SectionName, SectionAddress, PointerToData, SectionData);
            sdas.Add(sda);
        }
        public SECTION_DATA_ARRAY[] SectionData
        {
            get
            {
                return sdas.ToArray();
            }
        }
    }
    #endregion


    #region PEData
    public interface IPEDData
    {
        IMAGE_DOS_HEADER DosHeader { get; }
        IMAGE_NT_HEADERS NTHeader { get; }
        IMAGE_SECTION_HEADER[] SectionHeaders { get; }
        SECTION_DATA_ARRAY[] SectionData { get; }
        bool Is32bitAssembly();
    }
    public class PEDataWrapper
    {
        #region static members
        public static IPEDData GetPEData(IAcceptsProcessAndConfig p) { return GetPEData((IAcceptsProcess)p); }
        public static IPEDData GetPEData(IAcceptsProcess p) { return GetPEData(p.AcceptedProcess); }
        public static IPEDData GetPEData(Process p)
        {
            if (p == null)
                return PEData.Empty;
            return new PEData(p);
        }
        public static IPEDData GetPEData()
        {
            return PEData.Empty;
        }        
        #endregion
    }
    internal class PEData : IPEDData
    {
        //public PEData() : this(new IMAGE_DOS_HEADER(), new IMAGE_NT_HEADERS(), new IMAGE_SECTION_HEADER[] { }, new IMAGE_SECTION_DATA()) { }
        //public PEData(IMAGE_DOS_HEADER DosHeader, IMAGE_NT_HEADERS NTHeader, IMAGE_SECTION_HEADER[] SectionHeaders, IMAGE_SECTION_DATA isd)
        //{
        //    this.DosHeader = DosHeader;
        //    this.NTHeader = NTHeader;
        //    this.SectionHeaders = SectionHeaders;
        //    this.isd = isd;
        //}
        public static readonly IPEDData Empty = new PEData();
        public PEData()
        {
            SourceProcess = null;
            DosHeader = new IMAGE_DOS_HEADER();
            NTHeader = new IMAGE_NT_HEADERS();
            SectionHeaders = new IMAGE_SECTION_HEADER[] { };
            isd = new IMAGE_SECTION_DATA();
        }
        public PEData(IAcceptsProcessAndConfig p) : this((IAcceptsProcess)p.AcceptedProcess) { }
        public PEData(IAcceptsProcess p) : this(p.AcceptedProcess) { }
        public PEData(Process p)
        {
            SourceProcess = p;
            if (SourceProcess == null)
                throw new NullReferenceException("Supplied process cannot be null");
            PEReader reader = new PEReader(SourceProcess);
            this.DosHeader = reader._dosHeader;
            this.NTHeader = reader._ntHeaders;
            this.SectionHeaders = reader._sectionHeaders.ToArray();
            isd = reader._SectionData;
        }
        private Process SourceProcess { get; set; }
        #region IPEDData members

        private IMAGE_DOS_HEADER _DosHeader;
        private IMAGE_NT_HEADERS _NTHeader;
        private IMAGE_SECTION_HEADER[] _SectionHeaders;

        public IMAGE_DOS_HEADER DosHeader { get { return _DosHeader; } private set { _DosHeader = value; } }
        public IMAGE_NT_HEADERS NTHeader { get { return _NTHeader; } private set { _NTHeader = value; } }
        public IMAGE_SECTION_HEADER[] SectionHeaders { get { return _SectionHeaders; } private set { _SectionHeaders = value; } }
        private IMAGE_SECTION_DATA isd;
        public SECTION_DATA_ARRAY[] SectionData { get { return isd.SectionData; } }
        public bool Is32bitAssembly()
        {
            return NTHeader.FileHeader._Characteristics.HasFlag(FileCharacteristicType.Supports32Bit);
        }
        #endregion      
    }
    #endregion

    internal class PEReader
    {
        // Code (C) Sergey utilized from: http://www.sergeyakopov.com/2010/11/03/reading-pe-format-using-data-marshalling-in-net/

        #region Fields

        internal readonly IMAGE_DOS_HEADER _dosHeader;
        internal IMAGE_NT_HEADERS _ntHeaders;
        internal readonly IList<IMAGE_SECTION_HEADER> _sectionHeaders = new List<IMAGE_SECTION_HEADER>();
        internal readonly IMAGE_SECTION_DATA _SectionData = new IMAGE_SECTION_DATA();
        #endregion


        public PEReader(Process p)
        {
            string filename = p.MainModule.FileName;
            Exception ErrorInfo = null;
            _SectionData = new IMAGE_SECTION_DATA();
            using (FileStream fs = new FileStream(filename, FileMode.Open, FileAccess.Read))
            {
                try
                {
                    logger.VerboseDebug.WriteLine("Reading PE Format from: {0}", filename);
                    BinaryReader reader = new BinaryReader(fs);
                    // Reset reader position, just in case
                    reader.BaseStream.Seek(0, SeekOrigin.Begin);

                    // Read MS-DOS header section
                    _dosHeader = MarshalBytesTo<IMAGE_DOS_HEADER>(reader);
                    // MS-DOS magic number should read 'MZ'
                    if (_dosHeader._e_magic != 0x5a4d)
                    {
                        throw new InvalidOperationException("File is not a portable executable.");
                    }

                    // Skip MS-DOS stub and seek reader to NT Headers
                    reader.BaseStream.Seek(_dosHeader._e_lfanew, SeekOrigin.Begin);


                    // Read NT Headers
                    _ntHeaders._Signature = MarshalBytesTo<UInt32>(reader);

                    // Make sure we have 'PE' in the pe signature
                    if (_ntHeaders._Signature != 0x4550)
                    {
                        throw new InvalidOperationException("Invalid portable executable signature in NT header.");
                    }

                    _ntHeaders._FileHeader = MarshalBytesTo<IMAGE_FILE_HEADER>(reader);
                    // Read optional headers
                    if (Is32bitAssembly())
                    {
                        logger.VerboseDebug.WriteLine("\tDetected a 32Bit PE Executable");
                        Load32bitOptionalHeaders(reader);
                    }
                    else
                    {
                        logger.VerboseDebug.WriteLine("\tDetected a 64Bit PE Executable");
                        Load64bitOptionalHeaders(reader);
                    }

                    // Read section data
                    logger.VerboseDebug.WriteLine("\tTotal Section Headers: {0}", _sectionHeaders.Count);
                    ulong image_base = 0;
                    ulong p_image_base = (ulong)p.MainModule.BaseAddress.ToInt64();
                    if (Is32bitAssembly())
                    {
                        image_base = (ulong)_ntHeaders.OptionalHeader32._ImageBase;
                        if (image_base != p_image_base)
                        {
                            image_base = p_image_base;
                            _ntHeaders._OptionalHeader32._ImageBase = (uint)image_base;
                        }
                    }
                    else
                    {
                        image_base = _ntHeaders.OptionalHeader64._ImageBase;
                        if (image_base != p_image_base)
                        {
                            image_base = p_image_base;
                            _ntHeaders._OptionalHeader64._ImageBase = (ulong)image_base;
                        }
                    }                   
                    foreach (IMAGE_SECTION_HEADER header in _sectionHeaders)
                    {
                        int section_index = _sectionHeaders.IndexOf(header) + 1;
                        logger.VerboseDebug.WriteLine("\tSection Header: {0} of {1}", section_index, _sectionHeaders.Count);
                        logger.VerboseDebug.WriteLine("\t\tName: {0}", header.Name);
                        logger.VerboseDebug.WriteLine("\t\tVirtual Address: 0x{0:x8}", header.VirtualAddress);
                        logger.VerboseDebug.WriteLine("\t\tPhysical Address: 0x{0:x8}", header.Misc.PhysicalAddress);
                        logger.VerboseDebug.WriteLine("\t\tVirtual Size: 0x{0:x8}", header.Misc.VirtualSize);
                        logger.VerboseDebug.WriteLine("\t\tRaw Data Size: 0x{0:x8}", header.SizeOfRawData);
                        logger.VerboseDebug.WriteLine("\t\tPointer To Raw Data: 0x{0:x8}", header.PointerToRawData);

                        
                        // Skip to beginning of a section
                        reader.BaseStream.Seek(header._PointerToRawData, SeekOrigin.Begin);
                        // Read section data... and do something with it
                        byte[] sectiondata = reader.ReadBytes((int)header._SizeOfRawData);
                        _SectionData.AddData(header.Name, header._VirtualAddress + (uint)image_base, (int)header._PointerToRawData, sectiondata);
                    }
                    reader.Close();
                }
                catch (Exception ex)
                {
                    ErrorInfo = ex;
                    throw ErrorInfo;
                }
            }
            if (ErrorInfo != null)
            {
                logger.VerboseDebug.WriteLine("Error Reading PE Format from: {0}", filename);
                logger.VerboseDebug.WriteLine(ErrorInfo.ToString());
            }
        }

        public IMAGE_DOS_HEADER GetDOSHeader()
        {
            return _dosHeader;
        }

        public UInt32 GetPESignature()
        {
            return _ntHeaders._Signature;
        }

        public IMAGE_FILE_HEADER GetFileHeader()
        {
            return _ntHeaders.FileHeader;
        }

        public IMAGE_OPTIONAL_HEADER32 GetOptionalHeaders32()
        {
            return _ntHeaders.OptionalHeader32;
        }

        public IMAGE_OPTIONAL_HEADER64 GetOptionalHeaders64()
        {
            return _ntHeaders.OptionalHeader64;
        }

        public IList<IMAGE_SECTION_HEADER> GetSectionHeaders()
        {
            return _sectionHeaders;
        }

        public bool Is32bitAssembly()
        {
            return _ntHeaders.FileHeader._Characteristics.HasFlag(FileCharacteristicType.Supports32Bit);
        }

        private void Load64bitOptionalHeaders(BinaryReader reader)
        {
            _ntHeaders._OptionalHeader64 = MarshalBytesTo<IMAGE_OPTIONAL_HEADER64>(reader);

            // Should have 10 data directories
            if (_ntHeaders.OptionalHeader64._NumberOfRvaAndSizes != 0x10)
            {
                throw new InvalidOperationException("Invalid number of data directories in NT header");
            }

            // Scan data directories and load section headers
            for (int i = 0; i < _ntHeaders.OptionalHeader64._NumberOfRvaAndSizes; i++)
            {
                if (_ntHeaders._OptionalHeader64._DataDirectory[i]._Size > 0)
                {
                    IMAGE_SECTION_HEADER ish = MarshalBytesTo<IMAGE_SECTION_HEADER>(reader);
                    if (ish._SizeOfRawData != 0)
                        _sectionHeaders.Add(ish);
                }
            }
        }

        private void Load32bitOptionalHeaders(BinaryReader reader)
        {
            _ntHeaders._OptionalHeader32 = MarshalBytesTo<IMAGE_OPTIONAL_HEADER32>(reader);

            // Should have 10 data directories
            if (_ntHeaders.OptionalHeader32._NumberOfRvaAndSizes != 0x10)
            {
                throw new InvalidOperationException("Invalid number of data directories in NT header");
            }

            // Scan data directories and load section headers
            for (int i = 0; i < _ntHeaders.OptionalHeader32._NumberOfRvaAndSizes; i++)
            {
                if (_ntHeaders._OptionalHeader32._DataDirectory[i]._Size > 0)
                {
                    IMAGE_SECTION_HEADER ish = MarshalBytesTo<IMAGE_SECTION_HEADER>(reader);
                    if (ish._SizeOfRawData != 0)
                        _sectionHeaders.Add(ish);
                }
            }
        }

        private static T MarshalBytesTo<T>(BinaryReader reader)
        {
            // Unmanaged data
            byte[] bytes = reader.ReadBytes(Marshal.SizeOf(typeof(T)));

            // Create a pointer to the unmanaged data pinned in memory to be accessed by unmanaged code
            GCHandle handle = GCHandle.Alloc(bytes, GCHandleType.Pinned);

            // Use our previously created pointer to unmanaged data and marshal to the specified type
            T theStructure = (T)Marshal.PtrToStructure(handle.AddrOfPinnedObject(), typeof(T));

            // Deallocate pointer
            handle.Free();

            return theStructure;
        }
    }
}
Revision:	415
Committed:	Fri Jun 22 08:33:51 2012 UTC (11 years, 5 months ago) by william
File size:	41592 byte(s)
Log Message: