Add project files.

2025-09-02 15:32:24 +01:00
parent 6a633eed7a
commit 50bb9c9781
53 changed files with 9925 additions and 0 deletions
--- a/Microsoft/Windows.cs
+++ b/Microsoft/Windows.cs
@@ -0,0 +1,137 @@
+using System.Diagnostics;
+using System.Reflection;
+using System.Runtime.InteropServices;
+using System.Security.Principal;
+
+namespace AiQ_GUI
+{
+    internal class Windows
+    {
+        private static readonly string[] targetProcesses = ["IP_Tool", "Rapier", "IPConfig", "BackdoorGUI"];
+
+        // Closes other MAV and Rudstone tools. 
+        public static async Task CloseProcesses()
+        {
+            IEnumerable<Task> tasks = Process.GetProcesses()
+                .Where(p => targetProcesses.Any(tp => p.ProcessName.Contains(tp)))
+                .Select(clsProcess =>
+                {
+                    using (clsProcess)
+                    {
+                        try
+                        {
+                            clsProcess.CloseMainWindow();
+                        }
+                        catch { }
+                    }
+                    return Task.CompletedTask;
+                });
+
+            await Task.WhenAll(tasks).ConfigureAwait(false); // Run all tasks concurrently
+        }
+
+        public static void StartAsAdmin(string ExeLoc)
+        {
+            Logging.LogMessage($"Starting exe from {ExeLoc}");
+
+            ProcessStartInfo processInfo = new ProcessStartInfo(ExeLoc)
+            {
+                UseShellExecute = true,
+                Verb = "runas"
+            };
+
+            try
+            {
+                Process.Start(processInfo);
+                Properties.Settings.Default.FirstRun = false;
+                Properties.Settings.Default.Save();
+                Application.Exit(); // Exit now that we have admin rights version
+            }
+            catch (Exception ex)
+            {
+                Logging.LogErrorMessage("Failed to restart with admin rights. " + ex.Message);
+                MessageBox.Show("Sorry, but I don't seem to be able to start this program with administrator rights!");
+            }
+        }
+
+        public static void UpdateFirewall()
+        {
+            WindowsPrincipal wp = new(WindowsIdentity.GetCurrent());
+            bool runAsAdmin = wp.IsInRole(WindowsBuiltInRole.Administrator);
+            string ExeLoc = Assembly.GetEntryAssembly().Location.Replace("dll", "exe"); // Sometimes trys to open the dll instead of exe
+
+            if (Properties.Settings.Default.FirstRun && !runAsAdmin) // On first run, put into admin mode to allow defender.
+            {
+                StartAsAdmin(ExeLoc);
+            }
+            else if (runAsAdmin)
+            {
+                try
+                {
+                    // Use dynamic for COM interop
+                    Type ruleType = Type.GetTypeFromProgID("HNetCfg.FWRule");
+                    Type policyType = Type.GetTypeFromProgID("HNetCfg.FwPolicy2");
+                    dynamic firewallRule = Activator.CreateInstance(ruleType);
+                    dynamic firewallPolicy = Activator.CreateInstance(policyType);
+
+                    firewallRule.ApplicationName = ExeLoc;
+                    firewallRule.Action = 1; // NET_FW_ACTION_ALLOW
+                    firewallRule.Description = "Programmatically added rule to allow the GUI to work";
+                    firewallRule.Enabled = true;
+                    firewallRule.InterfaceTypes = "All";
+                    firewallRule.Name = "AiQ_GUI";
+                    firewallRule.Protocol = 17; // UDP
+
+                    firewallPolicy.Rules.Add(firewallRule);
+
+                    Properties.Settings.Default.FirstRun = false;
+                    Properties.Settings.Default.Save();
+                }
+                catch (Exception ex)
+                {
+                    Logging.LogErrorMessage("Failed to install firewall. " + ex.Message);
+                    MessageBox.Show("Sorry, but I couldn't install the firewall rule!");
+                }
+            }
+        }
+    }
+
+    [ComImport, Guid("AF230D27-BABA-4E42-ACED-F524F22CFCE2")]
+    public interface INetFwRule
+    {
+        string Name { get; set; }
+        string Description { get; set; }
+        string ApplicationName { get; set; }
+        string ServiceName { get; set; }
+        int Protocol { get; set; }
+        string LocalPorts { get; set; }
+        string RemotePorts { get; set; }
+        string LocalAddresses { get; set; }
+        string RemoteAddresses { get; set; }
+        string IcmpTypesAndCodes { get; set; }
+        int Direction { get; set; }
+        object Interfaces { get; set; }
+        string InterfaceTypes { get; set; }
+        bool Enabled { get; set; }
+        string Grouping { get; set; }
+        int Profiles { get; set; }
+        bool EdgeTraversal { get; set; }
+        int Action { get; set; }
+    }
+
+    [ComImport, Guid("98325047-C671-4174-8D81-DEFCD3F03186")]
+    public interface INetFwPolicy2
+    {
+        int CurrentProfileTypes { get; }
+        void get_FirewallEnabled(int profileType, out bool enabled);
+        void put_FirewallEnabled(int profileType, bool enabled);
+        void get_ExcludedInterfaces(int profileType, out object interfaces);
+        void put_ExcludedInterfaces(int profileType, object interfaces);
+        int BlockAllInboundTraffic { get; set; }
+        int NotificationsDisabled { get; set; }
+        int UnicastResponsesToMulticastBroadcastDisabled { get; set; }
+        object Rules { get; }
+        object ServiceRestriction { get; }
+        // ...other members omitted for brevity
+    }
+}