EdgePatcher.cs

// =============================================================================
// GPAL - Generally Positive Automation Library
// Copyright © 2026 Software Decisions, Inc. All rights reserved.
//
// This file is part of GPAL.
// Licensed under the Business Source License 1.1
//
// Primary development, architecture, and vision by Michael B. Vederman,
// CEO of Software Decisions, Inc., Texas.
//
// Internal development maintained privately.
// Public releases appear on GitHub: https://github.com/SoftwareDecisionsInc/GPAL.
//
// See LICENSE for full terms, including Additional Use Grant.
// =============================================================================
 
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using GenerallyPositive.Browser;
 
namespace GenerallyPositive.Browser
{
    internal static class EdgePatcher
    {
        static readonly string[] edgedriverSourceMatches = new[]
        {
            "WebDriver"
            , "W3C"
            , "Execute-Script"
            , "Chromium"
            //, "shadow-6066-11e4-a52e-4f735466cecf"    // part of w3c protocol - other scripts could use this, change breaks selenium - use tostring override
            //, "element-6066-11e4-a52e-4f735466cecf"    // part of w3c protocol - other scripts could use this, change breaks selenium - use tostring override
            , "STALE_ELEMENT_REFERENCE"     // enum will still work with munged name
            , "crbug.com"
            , "shadow root is detached from the current frame"
            , "stale element not found in the current frame"
        };

        public static void PatchEdgeDriver(IBrowser browser)
        {
            if (null == ((Browser)browser).BrowserSettings.DriverLocation)
                // driver is located with executable
                ((Browser)browser).BrowserSettings.DriverLocation = System.IO.Path.GetDirectoryName(System.Reflection.Assembly.GetExecutingAssembly().GetName().CodeBase).Substring(6);    // skip file:\\
 
            string filePath = FileHelper.EnsureDirectoryEndsWithBackslash(((Browser)browser).BrowserSettings.DriverLocation) + $@"{GPAL.GPALSettings.EdgeDriverFilename}";
 
            long fileSize = new FileInfo(filePath).Length;
            int totalSteps = 6 + edgedriverSourceMatches.Length;
            GPAL.PublishSimpleEvent(Enums.GPALEventType.INFO, $"Stealth patching [{filePath}] ({fileSize / 1024 / 1024} MB, {totalSteps} steps)", browser, Enums.GPALObjectType.Browser);
 
            if (true == CheckIfPatched(filePath))
            {
                GPAL.PublishSimpleEvent(Enums.GPALEventType.INFO, $"[{filePath}] already patched.", browser, Enums.GPALObjectType.Browser);
                return;
            }
 
            // Create a backup of the original file before patching
            string backupPath = filePath + ".org";
            if (File.Exists(backupPath))
            {
                GPAL.PublishSimpleEvent(Enums.GPALEventType.INFO, $"[{backupPath}] backup already exists. Deleting.", browser, Enums.GPALObjectType.Browser);
                File.Delete(backupPath);
            }
 
            File.Copy(filePath, backupPath);
            GPAL.PublishSimpleEvent(Enums.GPALEventType.INFO, $"[{backupPath}] backup created.", browser, Enums.GPALObjectType.Browser);
 
            Action<int, int, string> onProgress = (step, total, description) =>
                GPAL.PublishSimpleEvent(Enums.GPALEventType.INFO, $"Patch step [{step}/{total}]: {description}", browser, Enums.GPALObjectType.Browser);
 
            // Patch the executable file directly (no renaming the patched file)
            if (PatchExe(filePath, onProgress))
            {
                if (true == CheckIfPatched(filePath))
                    GPAL.PublishSimpleEvent(Enums.GPALEventType.INFO, $"[{filePath}] patched successfully.", browser, Enums.GPALObjectType.Browser);
                else
                    GPAL.PublishSimpleEvent(Enums.GPALEventType.ERROR, $"[{filePath}] patching failed.", browser, Enums.GPALObjectType.Browser);
            }
            else
                GPAL.PublishSimpleEvent(Enums.GPALEventType.ERROR, $"[{filePath}] patching failed.", browser, Enums.GPALObjectType.Browser);
        }

        static bool PatchExe(string executablePath, Action<int, int, string> onProgress)
        {
            // Read the binary content of the file
            byte[] content = File.ReadAllBytes(executablePath);
 
            // Apply the regex replacements
            content = ApplyRegexPatch(content, onProgress);
 
            // Write the patched content back to the original file
            File.WriteAllBytes(executablePath, content);
            return true;
        }

        static byte[] ApplyRegexPatch(byte[] content, Action<int, int, string> onProgress)
        {
            int total = 6 + edgedriverSourceMatches.Length;
            int step = 0;
 
            onProgress(++step, total, "cdc variable assignments");
            content = ApplyPattern(content,
                @"window\.cdc_[a-zA-Z0-9]{22,}_(Array|Promise|Symbol|Object|Proxy|JSON) = window\.(Array|Promise|Symbol|Object|Proxy|JSON);",
                match => new string('\n', match.Length));
 
            onProgress(++step, total, "cdc variable guards");
            content = ApplyPattern(content,
                @"window\.cdc_[a-zA-Z0-9]{22,}_(Array|Promise|Symbol|Object|Proxy|JSON) \|\|",
                match => new string('\n', match.Length));
 
            onProgress(++step, total, "cdc Window references");
            content = ApplyPattern(content,
                @"window\.cdc_[a-zA-Z0-9]{22,}_Window",
                match => "window.Window");
 
            onProgress(++step, total, "WindowProxy binding");
            content = ApplyPattern(content,
                @"let WindowProxy = window\.cdc_[a-zA-Z0-9]{22,}_Window(?:\s*\|\|\s*window\.Window;)?",
                match => "let WindowProxy = window.Window");
 
            onProgress(++step, total, "item.cdc Window references");
            content = ApplyPattern(content,
                @"item\.cdc_[a-zA-Z0-9]{22,}_Window",
                match => "item.Window");
 
            onProgress(++step, total, "console.log stamp");
            byte[] logMessageBytes = Encoding.ASCII.GetBytes($"{{console.log(\"GPAL {GPAL.Version} msedgedriver patch!\")}}");
            content = ApplyPattern(content,
                @"console\.log\‍(([^)]+)\‍)",
                match =>
                {
                    byte[] newTargetBytes = new byte[match.Length];
                    Array.Copy(logMessageBytes, 0, newTargetBytes, 0, logMessageBytes.Length);
                    for (int i = logMessageBytes.Length; i < match.Length; i++)
                    {
                        newTargetBytes[i] = (byte)' '; // Fill the rest with space characters
                    }
                    return Encoding.ASCII.GetString(newTargetBytes);
                });
 
            foreach (var originalString in edgedriverSourceMatches)
            {
                onProgress(++step, total, $"string [{originalString}]");
                string alteredString = AlterString(originalString);
                content = ApplyPattern(content, Regex.Escape(originalString), _ => alteredString);
            }
 
            return content;
        }
        static string AlterString(string input)
        {
            if (string.IsNullOrEmpty(input)) return input;
 
            // Shift the first character by one
            char firstChar = input[0];
            char shiftedChar = (char)(firstChar + 1);
 
            // Handle wrap-around for alphabets
            if (char.IsLetter(firstChar))
            {
                if (char.IsLower(firstChar) && shiftedChar > 'z') shiftedChar = 'a';
                if (char.IsUpper(firstChar) && shiftedChar > 'Z') shiftedChar = 'A';
            }
            // Handle wrap-around for digits
            else if (char.IsDigit(firstChar))
            {
                if (shiftedChar > '9') shiftedChar = '0';
            }
 
            // Reconstruct the string
            return shiftedChar + input.Substring(1);
        }

        static byte[] ApplyPattern(byte[] content, string pattern, Func<Match, string> replacementFunc)
        {
            // Compile regex
            Regex regex = new Regex(pattern, RegexOptions.Compiled);
            string contentStr = Encoding.UTF8.GetString(content);
            MatchCollection matches = regex.Matches(contentStr);
 
            foreach (Match match in matches)
            {
                byte[] targetBytes = Encoding.UTF8.GetBytes(match.Value);
                string replacement = replacementFunc(match);
                byte[] replacementBytes = Encoding.UTF8.GetBytes(replacement);
 
                // Pad the replacementBytes to match the targetBytes length
                byte[] paddedReplacementBytes = PadByteArray(replacementBytes, targetBytes.Length);
 
                content = ReplaceBytes(content, targetBytes, paddedReplacementBytes);
            }
 
            return content;
        }

        static byte[] PadByteArray(byte[] byteArray, int desiredLength)
        {
            if (byteArray.Length >= desiredLength)
                return byteArray;
 
            byte[] paddedArray = new byte[desiredLength];
            byteArray.CopyTo(paddedArray, 0);
            for (int i = byteArray.Length; i < desiredLength; i++)
            {
                paddedArray[i] = (byte)' '; // Pad with space character
            }
            return paddedArray;
        }

        static byte[] ReplaceBytes(byte[] content, byte[] oldBytes, byte[] newBytes)
        {
            using (MemoryStream stream = new MemoryStream())
            {
                int index = 0;
 
                while (index <= content.Length - oldBytes.Length)
                {
                    bool match = true;
 
                    for (int j = 0; j < oldBytes.Length; j++)
                    {
                        if (content[index + j] != oldBytes[j])
                        {
                            match = false;
                            break;
                        }
                    }
 
                    if (match)
                    {
                        stream.Write(newBytes, 0, newBytes.Length);
                        index += oldBytes.Length;
                    }
                    else
                    {
                        stream.WriteByte(content[index]);
                        index++;
                    }
                }
 
                stream.Write(content, index, content.Length - index);
                return stream.ToArray();
            }
        }
        static bool CheckIfPatched(string filePath)
        {
            byte[] content = File.ReadAllBytes(filePath);
            string contentStr = Encoding.UTF8.GetString(content);
 
            // Check for the GPAL version message in the console.log
            if (Regex.IsMatch(contentStr, @"console\.log\‍(\"".*?msedgedriver patch!\""\‍)"))
            {
//                GPAL.PublishSimpleEvent(Enums.GPALEventType.DEBUG, "EdgeDriver is patched.", filePath, Enums.GPALObjectType.Other);
                return true;
            }
            return false;
        }
    }
}