Make unit test runnable in user mode under linux

src/Fluxzy.Core/Misc/ProcessUtilX.cs

@@ -2,13 +2,18 @@
 
 using System;
 using System.Collections.Generic;
+using System.Collections.ObjectModel;
 using System.Diagnostics;
+using System.Linq;
 using System.Threading.Tasks;
 
 namespace Fluxzy.Misc
 {
     internal static class ProcessUtilX
     {
+        private static readonly string[] AskPassBinaries = new[] { "ssh-askpass", "ksshaskpass", "lxqt-sudo" };
+
+
         private static readonly HashSet<string> RequiredCapabilities = new HashSet<string>(
             new[] { "cap_net_raw", "cap_net_admin" }, 
             StringComparer.OrdinalIgnoreCase);
@@ -35,5 +40,48 @@ public static async Task<bool> HasCaptureCapabilities()
 
             return availableCapabilities.IsSupersetOf(RequiredCapabilities);
         }
+
+        public static async Task<string?> GetExecutablePath(string executableName)
+        {
+            var processRunResult = await ProcessUtils.QuickRunAsync("which", executableName, null);
+
+            if (processRunResult.ExitCode != 0)
+                return null;
+
+            return processRunResult.StandardOutputMessage?.Trim('\r', '\n');
+        }
+
+
+        public static async Task<Process> RunElevatedSudoALinux(
+            string commandName, string[] args, bool redirectStdOut,
+            string askPasswordPrompt, bool redirectStandardError = false)
+        {
+            var tasks = AskPassBinaries
+                .Select(binary => GetExecutablePath(binary));
+
+            var results = await Task.WhenAll(tasks);
+            var result = results.FirstOrDefault(x => x != null) ?? 
+                         throw new Exception(
+                             $"No askpass binary found. Must install one of the following:" +
+                             $" {string.Join(", ", AskPassBinaries)}");
+
+            var execCommandName = "sudo";
+            var preArgs = new List<string>() { "-A", "-E", "-p", $"{askPasswordPrompt}", commandName };
+
+            var startInfo = new ProcessStartInfo() {
+                FileName = execCommandName,
+                RedirectStandardOutput = redirectStdOut,
+                RedirectStandardError = redirectStandardError,
+                UseShellExecute = false,
+            }; 
+
+            startInfo.EnvironmentVariables["SUDO_ASKPASS"] = result;
+            foreach (var arg in preArgs.Concat(args))
+                startInfo.ArgumentList.Add(arg);
+
+            var process = Process.Start(startInfo);
+
+            return process!;
+        }
     }
 }

src/Fluxzy.Core/Misc/ProcessUtils.cs

@@ -244,6 +244,9 @@ public static bool IsCommandAvailable(string commandName)
             return process;
         }
 
+
+
+
         public static Process? RunElevated(
             string commandName, string[] args, bool redirectStdOut,
             string askPasswordPrompt)

src/Fluxzy/System/OutOfProcAuthorityManager.cs

@@ -31,7 +31,6 @@ await ProcessUtils.QuickRunAsync($"{_currentBinaryFullPath.RemoveFileExtension()
                     $" cert uninstall {thumbPrint}");
 
                 // We are using pkexec for linux 
-
                 var result = await ProcessUtils.QuickRunAsync("pkexec",
                     $"{_currentBinaryFullPath.RemoveFileExtension()} cert uninstall {thumbPrint}");
 

test/Fluxzy.Tests/Startup.cs

@@ -5,6 +5,7 @@
 using System.IO.Compression;
 using System.Linq;
 using System.Net;
+using System.Reflection;
 using System.Runtime.InteropServices;
 using System.Threading.Tasks;
 using Fluxzy.Certificates;
@@ -50,8 +51,8 @@ public Startup(IMessageSink messageSink)
             CertificateContext.InstallDefaultCertificate();
 
             if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux)) {
-                _ = Task.Run(async () =>
-                            await Utility.AcquireCapabilities(new FileInfo("fluxzynetcap").FullName))
+                Task.Run(async () => { await Utility.AcquireCapabilitiesLinux(new FileInfo("fluxzynetcap").FullName);
+                        })
                         .GetAwaiter().GetResult();
             }
         }

test/Fluxzy.Tests/UnitTests/Rules/AverageThrottlerTests.cs

@@ -62,6 +62,8 @@ public async Task SimpleThrottle(int delaySeconds)
             var stream = await response.Content.ReadAsStreamAsync();
             await stream.CopyToAsync(Stream.Null);
 
+            await Task.Delay(500);
+
             watch.Stop();
 
             Assert.True(watch.ElapsedMilliseconds > delaySeconds * 1000, 

test/Fluxzy.Tests/Utility.cs

@@ -1,5 +1,7 @@
 // Copyright 2021 - Haga Rakotoharivelo - https://github.com/haga-rak
 
+using System;
+using System.IO;
 using System.Threading.Tasks;
 using Fluxzy.Misc;
 
@@ -11,25 +13,24 @@ public static class Utility
         ///  Make an executable have the required capabilities
         /// </summary>
         /// <returns></returns>
-        public static async Task<bool> AcquireCapabilities(string executablePath)
+        public static async Task<bool> AcquireCapabilitiesLinux(string executablePath)
         {
+            if (!File.Exists(executablePath)) {
+                executablePath = (await ProcessUtilX.GetExecutablePath(executablePath))
+                    ?? throw new InvalidOperationException("Executable not found");
+            }
+
             if (await ProcessUtilX.CanElevated())
                 return true; // Already root  - no need to set capabilities
 
             if (!ProcessUtils.IsCommandAvailable("setcap"))
                 return false; 
 
-            if (!ProcessUtils.IsCommandAvailable("pkexec"))
-                return false; 
-
-            var fullCommand = $"setcap cap_net_raw,cap_net_admin=eip \"{executablePath}\""; 
-
-            var process = await ProcessUtils.RunElevatedAsync("pkexec", 
-                new []{ "setcap", "cap_net_raw,cap_net_admin=eip", executablePath},
+            var process = await ProcessUtilX.RunElevatedSudoALinux("setcap", 
+                new []{ "cap_net_raw,cap_net_admin=eip", executablePath},
                 false, "Please enter your password to set the required capabilities");
 
             await process!.WaitForExitAsync();
-
             return process.ExitCode == 0; 
         }
     }