From b067b671db87333b2e504d06cf8db3cba1199b51 Mon Sep 17 00:00:00 2001
From: Matt Nadareski <mnadareski@outlook.com>
Date: Tue, 8 Nov 2022 14:04:02 -0800
Subject: [PATCH] Add PE signature parsing to builder

---
 BurnOutSharp.Builder/PortableExecutable.cs | 34 ++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/BurnOutSharp.Builder/PortableExecutable.cs b/BurnOutSharp.Builder/PortableExecutable.cs
index 677825b7..d97e7c14 100644
--- a/BurnOutSharp.Builder/PortableExecutable.cs
+++ b/BurnOutSharp.Builder/PortableExecutable.cs
@@ -30,6 +30,8 @@ namespace BurnOutSharp.Builder
             // Create a new executable to fill
             var executable = new Executable();
 
+            #region MS-DOS Stub
+
             // Parse the MS-DOS stub
             var stub = MSDOS.ParseExecutable(data, offset);
             if (stub?.Header == null || stub.Header.NewExeHeaderAddr == 0)
@@ -38,6 +40,21 @@ namespace BurnOutSharp.Builder
             // Set the MS-DOS stub
             executable.Stub = stub;
 
+            #endregion
+
+            #region Signature
+
+            offset = (int)(initialOffset + stub.Header.NewExeHeaderAddr);
+            executable.Signature = new byte[4];
+            for (int i = 0; i < executable.Signature.Length; i++)
+            {
+                executable.Signature[i] = data.ReadByte(ref offset);
+            }
+            if (executable.Signature[0] != 'P' || executable.Signature[1] != 'E' || executable.Signature[2] != '\0' || executable.Signature[3] != '\0')
+                return null;
+
+            #endregion
+
             // TODO: Implement PE parsing
             return null;
         }
@@ -79,6 +96,8 @@ namespace BurnOutSharp.Builder
             // Create a new executable to fill
             var executable = new Executable();
 
+            #region MS-DOS Stub
+
             // Parse the MS-DOS stub
             var stub = MSDOS.ParseExecutable(data);
             if (stub?.Header == null || stub.Header.NewExeHeaderAddr == 0)
@@ -87,6 +106,21 @@ namespace BurnOutSharp.Builder
             // Set the MS-DOS stub
             executable.Stub = stub;
 
+            #endregion
+
+            #region Signature
+
+            data.Seek(initialOffset + stub.Header.NewExeHeaderAddr, SeekOrigin.Begin);
+            executable.Signature = new byte[4];
+            for (int i = 0; i < executable.Signature.Length; i++)
+            {
+                executable.Signature[i] = data.ReadByteValue();
+            }
+            if (executable.Signature[0] != 'P' || executable.Signature[1] != 'E' || executable.Signature[2] != '\0' || executable.Signature[3] != '\0')
+                return null;
+
+            #endregion
+
             // TODO: Implement PE parsing
             return null;
         }