Merge branch 'bug/166'

Author: Voltstro

src/Packages/UnityWebBrowser/Runtime/Core/Engines/EngineProcess.cs

@@ -0,0 +1,80 @@
+// UnityWebBrowser (UWB)
+// Copyright (c) 2021-2024 Voltstro-Studios
+// 
+// This project is under the MIT license. See the LICENSE.md file for more details.
+
+using System;
+using System.Diagnostics;
+using VoltstroStudios.UnityWebBrowser.Core.Engines.Process;
+using VoltstroStudios.UnityWebBrowser.Helper;
+using VoltstroStudios.UnityWebBrowser.Logging;
+
+namespace VoltstroStudios.UnityWebBrowser.Core.Engines
+{
+    /// <summary>
+    ///     Handler for the engine process
+    /// </summary>
+    internal sealed class EngineProcess : IDisposable
+    {
+        private readonly IProcess processHandle;
+        private readonly Engine engine;
+        private readonly IWebBrowserLogger logger;
+        
+        /// <summary>
+        ///     Creates a new <see cref="EngineProcess"/> instance
+        /// </summary>
+        /// <param name="engine"></param>
+        /// <param name="logger"></param>
+        public EngineProcess(Engine engine, IWebBrowserLogger logger)
+        {
+#if UNITY_STANDALONE_WIN
+            processHandle = new WindowProcess();
+#elif UNITY_STANDALONE_LINUX
+            processHandle = new LinuxProcess(logger);
+#endif
+            
+            this.engine = engine;
+            this.logger = logger;
+        }
+
+        /// <summary>
+        ///     Has the process exited?
+        /// </summary>
+        public bool HasExited => processHandle.HasExited;
+
+        /// <summary>
+        ///     What was the exit code of the process
+        /// </summary>
+        public int ExitCode => processHandle.ExitCode;
+
+        /// <summary>
+        ///     Starts the engine process
+        /// </summary>
+        /// <param name="arguments"></param>
+        /// <param name="onLogEvent"></param>
+        /// <param name="onErrorLogEvent"></param>
+        public void StartProcess(string arguments, DataReceivedEventHandler onLogEvent, DataReceivedEventHandler onErrorLogEvent)
+        {
+            string engineFullProcessPath = WebBrowserUtils.GetBrowserEngineProcessPath(engine);
+            string engineDirectory = WebBrowserUtils.GetBrowserEnginePath(engine);
+            
+            logger.Debug($"Process Path: '{engineFullProcessPath}'\nWorking: '{engineDirectory}'");
+            logger.Debug($"Arguments: '{arguments}'");
+            
+            processHandle.StartProcess(engineFullProcessPath, engineDirectory, arguments, onLogEvent, onErrorLogEvent);
+        }
+
+        /// <summary>
+        ///     Kills the engine process
+        /// </summary>
+        public void KillProcess()
+        {
+            processHandle.KillProcess();
+        }
+        
+        public void Dispose()
+        {
+            processHandle.Dispose();
+        }
+    }
+}

src/Packages/UnityWebBrowser/Runtime/Core/Engines/EngineProcess.cs.meta

@@ -0,0 +1,21 @@
+// UnityWebBrowser (UWB)
+// Copyright (c) 2021-2024 Voltstro-Studios
+// 
+// This project is under the MIT license. See the LICENSE.md file for more details.
+
+using System;
+using System.Diagnostics;
+
+namespace VoltstroStudios.UnityWebBrowser.Core.Engines.Process
+{
+    internal interface IProcess : IDisposable
+    {
+        public void StartProcess(string executable, string workingDir, string arguments, DataReceivedEventHandler onLogEvent, DataReceivedEventHandler onErrorLogEvent);
+
+        public void KillProcess();
+        
+        public bool HasExited { get; }
+        
+        public int ExitCode { get; }
+    }
+}

src/Packages/UnityWebBrowser/Runtime/Core/Engines/Process.meta

@@ -0,0 +1,126 @@
+// UnityWebBrowser (UWB)
+// Copyright (c) 2021-2024 Voltstro-Studios
+// 
+// This project is under the MIT license. See the LICENSE.md file for more details.
+
+#if UNITY_STANDALONE_LINUX
+
+using System;
+using System.Collections.Generic;
+using System.Diagnostics;
+using System.IO;
+using VoltstroStudios.UnityWebBrowser.Logging;
+using VoltstroStudios.UnityWebBrowser.UnixSupport;
+
+namespace VoltstroStudios.UnityWebBrowser.Core.Engines.Process
+{
+    internal sealed class LinuxProcess : IProcess
+    {
+        private readonly System.Diagnostics.Process process;
+        private readonly IWebBrowserLogger logger;
+
+        public LinuxProcess(IWebBrowserLogger logger)
+        {
+            process = new System.Diagnostics.Process();
+            this.logger = logger;
+        }
+        
+        public bool HasExited { get; }
+        public int ExitCode { get; }
+
+        public void StartProcess(string executable, string workingDir, string arguments, DataReceivedEventHandler onLogEvent,
+            DataReceivedEventHandler onErrorLogEvent)
+        {
+            if (PermissionsManager.CheckAndSetIfNeededFileExecutablePermission(executable))
+                logger.Warn(
+                    "UWB engine process did not have +rwx permissions! Engine process permission's were updated for the user.");
+            
+            ProcessStartInfo startInfo = new(executable, arguments)
+            {
+                CreateNoWindow = true,
+                UseShellExecute = false,
+                RedirectStandardOutput = true,
+                RedirectStandardError = true,
+                WorkingDirectory = workingDir
+            };
+
+            process.StartInfo = startInfo;
+            process.OutputDataReceived += onLogEvent;
+            process.ErrorDataReceived += onErrorLogEvent;
+            process.Start();
+            process.BeginOutputReadLine();
+            process.BeginErrorReadLine();
+        }
+
+        public void KillProcess()
+        {
+            TimeSpan timeout = TimeSpan.FromSeconds(30);
+            HashSet<int> children = new();
+            GetAllChildIdsUnix(process.Id, children, timeout);
+            foreach (int childId in children) KillProcessUnix(childId, timeout);
+            KillProcessUnix(process.Id, timeout);
+        }
+        
+        public void Dispose()
+        {
+            process.Dispose();
+        }
+        
+        private static void GetAllChildIdsUnix(int parentId, ISet<int> children, TimeSpan timeout)
+        {
+            int exitCode = RunProcessAndWaitForExit(
+                "pgrep",
+                $"-P {parentId}",
+                timeout,
+                out string stdout);
+
+            if (exitCode != 0 || string.IsNullOrEmpty(stdout)) return;
+            using StringReader reader = new(stdout);
+            while (true)
+            {
+                string text = reader.ReadLine();
+                if (text == null) return;
+
+                if (!int.TryParse(text, out int id)) continue;
+
+                children.Add(id);
+                // Recursively get the children
+                GetAllChildIdsUnix(id, children, timeout);
+            }
+        }
+
+        private static void KillProcessUnix(int processId, TimeSpan timeout)
+        {
+            RunProcessAndWaitForExit(
+                "kill",
+                $"-TERM {processId}",
+                timeout,
+                out string _);
+        }
+
+        private static int RunProcessAndWaitForExit(string fileName, string arguments, TimeSpan timeout,
+            out string stdout)
+        {
+            ProcessStartInfo startInfo = new()
+            {
+                FileName = fileName,
+                Arguments = arguments,
+                RedirectStandardOutput = true,
+                UseShellExecute = false,
+                CreateNoWindow = true
+            };
+
+            System.Diagnostics.Process process = System.Diagnostics.Process.Start(startInfo);
+
+            stdout = null;
+            if (process.WaitForExit((int)timeout.TotalMilliseconds))
+                stdout = process.StandardOutput.ReadToEnd();
+            else
+                process.Kill();
+
+            return process.ExitCode;
+        }
+    }
+}
+
+#endif