From 948da435a95cb5573f132d05944b62a332c13d07 Mon Sep 17 00:00:00 2001
From: bestest <jojloj3@gmail.com>
Date: Fri, 25 Oct 2024 00:01:04 -0400
Subject: [PATCH] Add preliminary copy-X protection checking

---
 BinaryObjectScanner/Protection/CopyX.cs | 78 +++++++++++++++++++++++++
 1 file changed, 78 insertions(+)
 create mode 100644 BinaryObjectScanner/Protection/CopyX.cs

diff --git a/BinaryObjectScanner/Protection/CopyX.cs b/BinaryObjectScanner/Protection/CopyX.cs
new file mode 100644
index 00000000..d8451279
--- /dev/null
+++ b/BinaryObjectScanner/Protection/CopyX.cs
@@ -0,0 +1,78 @@
+using System;
+#if NET40_OR_GREATER || NETCOREAPP
+using System.Collections.Concurrent;
+#endif
+using System.Collections.Generic;
+using System.IO;
+using System.Text;
+using System.Linq;
+using BinaryObjectScanner.Interfaces;
+using SabreTools.IO;
+using SabreTools.IO.Extensions;
+using SabreTools.Serialization.Wrappers;
+
+namespace BinaryObjectScanner.Protection
+{
+    // TODO: Technically not necessary, but just check for light/pro first and only if it isn't found look for the other. It should be an Or situation and not an And situation.
+   // TODO: figure out if that's actually what light and professional mean
+   // TODO: add documentation/comments to methods
+    public class CopyX : IPathCheck, IPortableExecutableCheck
+    {
+        /// <inheritdoc/>
+        public string? CheckPortableExecutable(string file, PortableExecutable pex, bool includeDebug) //checks for Professional
+        {
+            //I need to change this to check all executables, not just portable ones, but including portable ones. I think. I'll figure out how to do that later. Do i need to? does rdata work for this?
+            //i dont need this to check optgraph.dll, that's redundant
+            var sections = pex.Model.SectionTable; 
+            if (sections == null)
+                return null;
+
+            // Get the .rdata section strings, if they exist
+            var name = pex.FileDescription;
+
+            
+            if (pex.OverlayStrings != null)
+            {
+                if (pex.OverlayStrings.Any(s => s.Contains("optgraph.dll")))
+                    return "copy-X";
+            }
+            return null;
+        }
+
+        /// <inheritdoc/>
+#if NET20 || NET35
+        public Queue<string> CheckDirectoryPath(string path, IEnumerable<string>? files)
+#else
+        public ConcurrentQueue<string> CheckDirectoryPath(string path, IEnumerable<string>? files) // checks for Light
+#endif
+        {
+#if NET20 || NET35
+            var protections = new Queue<string>();
+#else
+            var protections = new ConcurrentQueue<string>();
+#endif
+            if (files == null)
+                return protections;
+            var zdatFiles = files.Where(f => f.Remove(0, path.Length + 1).StartsWith("ZDAT"));
+            var fileList = zdatFiles.ToList();
+            fileList.Sort();
+            FileStream stream = new FileStream(fileList[0], FileMode.Open, FileAccess.Read);         
+            byte[] block = new byte[1024];
+            stream.Read(block, 0, 1024);
+            if (block.All(thisByte => thisByte.Equals(0x00)))
+            {
+                protections.Enqueue("x-Copy");
+            }
+            return protections;
+        }
+        /// <inheritdoc/>
+        public string? CheckFilePath(string path)
+        {
+            if (Path.GetFileName(path).Equals("optgraph.dll", StringComparison.OrdinalIgnoreCase))
+            {
+                return "copy-X";
+            }
+            return null;
+        }
+    }
+}
\ No newline at end of file