WMI for managment, nssm for editing paramns of service

MicrosocksGUI/MicrosocksGUI.csproj

@@ -1,7 +1,7 @@
 <Project Sdk="Microsoft.NET.Sdk">
     <PropertyGroup>
         <OutputType>WinExe</OutputType>
-        <TargetFramework>net9.0</TargetFramework>
+        <TargetFramework>net9.0-windows</TargetFramework>
         <Nullable>enable</Nullable>
         <ApplicationManifest>app.manifest</ApplicationManifest>
         <AvaloniaUseCompiledBindingsByDefault>true</AvaloniaUseCompiledBindingsByDefault>
@@ -27,5 +27,7 @@
           <PrivateAssets>all</PrivateAssets>
           <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
         </PackageReference>
+        <PackageReference Include="System.Management" Version="10.0.0" />
+        <PackageReference Include="System.ServiceProcess.ServiceController" Version="10.0.0" />
     </ItemGroup>
 </Project>

MicrosocksGUI/ViewModels/MainWindowViewModel.cs

@@ -1,10 +1,13 @@
 using System;
 using System.Diagnostics;
 using System.IO;
+using System.Management;
+using System.Runtime.Versioning;
 using System.Threading;
 using System.Threading.Tasks;
 using ReactiveUI;
 using ReactiveUI.SourceGenerators;
+using System.ServiceProcess;
 
 namespace MicrosocksGUI.ViewModels;
 
@@ -16,105 +19,60 @@ public partial class MainWindowViewModel : ViewModelBase
     [Reactive] private string? _port;
     
     private Timer _timer;
-    
+    private readonly ManagementObject _serviceManager = new("Win32_Service.Name='Microsocks'");
-    private ProcessStartInfo _getStatusProcessInfo = new()
-    {
-        FileName = ".\\nssm.exe",
-        Arguments = "status Microsocks",
-        UseShellExecute = false,
-        RedirectStandardOutput = true,
-        RedirectStandardError = true,
-        CreateNoWindow = true, 
-        WorkingDirectory = Path.GetDirectoryName(".")
-    };
-        
-    private ProcessStartInfo _getStartTypeProcessInfo = new()
-    {
-        FileName = ".\\nssm.exe",
-        Arguments = "get Microsocks Start",
-        UseShellExecute = false,
-        RedirectStandardOutput = true,
-        RedirectStandardError = true,
-        CreateNoWindow = true, 
-        WorkingDirectory = Path.GetDirectoryName(".")
-    };
-    
-    
     
     public MainWindowViewModel()
     {
         // TODO Костыль, таймер запускается через секунду, а должен после инициализации ReactiveUI
-        _timer = new Timer(GetWorkingStatusFromService, null, TimeSpan.FromSeconds(1), TimeSpan.FromMilliseconds(100));
+        _timer = new Timer(GetServiceStatus, null, TimeSpan.FromSeconds(1), TimeSpan.FromMilliseconds(100));
+
+        GetMicroSocksArguments();
     }
     
     [ReactiveCommand]
-    private async Task SetWorkingStatusToService()
+    private void SetServiceStatus()
     {
-        
+        var serviceProps = _serviceManager.GetMethodParameters("Change");
-        var args = _isWorking ? "start Microsocks" : "stop Microsocks";
+        serviceProps["StartMode"] = IsAutoloading ? "Automatic" : "Manual";
-        
+        _serviceManager.InvokeMethod("Change", serviceProps, null);
-        
+        _serviceManager.InvokeMethod(IsWorking ? "StartService" : "StopService", null, null);
-        ProcessStartInfo processInfo = new()
+    }
+    
+    private void GetServiceStatus(object? state)
+    {
+        _serviceManager.Get();
+        IsWorking = _serviceManager["State"]?.ToString() == "Running";
+        IsAutoloading = _serviceManager["StartMode"]?.ToString() == "Auto";
+    }
+
+    private void GetMicroSocksArguments()
+    {
+        using var process = new Process();
+        process.StartInfo = new ProcessStartInfo
         {
             FileName = ".\\nssm.exe",
-            Arguments = args,
+            Arguments = "get MicroSocks AppParameters",
             UseShellExecute = false,
             RedirectStandardOutput = true,
             RedirectStandardError = true,
             CreateNoWindow = true, 
             WorkingDirectory = Path.GetDirectoryName(".")
         };
-
+            
-        using var process = new Process();
-        process.StartInfo = processInfo;
         process.Start();
-    }
+            
-    
+        var output = process.StandardOutput.ReadToEnd();
-    private async void GetWorkingStatusFromService(object? state)
+        var error = process.StandardError.ReadToEnd();
-    {
+            
-        using (Process process = new Process())
+        process.WaitForExit();
+            
+        if (!string.IsNullOrEmpty(error))
         {
-            process.StartInfo = _getStatusProcessInfo;
+            Console.WriteLine(error);
-            process.Start();
-            
-            
-            var output = await process.StandardOutput.ReadToEndAsync();
-            var error = await process.StandardError.ReadToEndAsync();
-            
-            await process.WaitForExitAsync();
-            
-            if (!string.IsNullOrEmpty(error))
-            {
-                //Console.WriteLine(error);
-            }
-
-            //Console.WriteLine(output.Replace("\0", "").Trim());
-
-            if (output.Replace("\0", "").Trim() == "SERVICE_RUNNING")
-            {
-                //Console.WriteLine("yes");
-                //IsWorking = true;
-            }
-            
-            IsWorking = output.Replace("\0", "").Trim() == "SERVICE_RUNNING";
         }
-        
+
-        using (Process process = new Process())
+        Console.WriteLine(output.Replace("\0", "").Trim());
-        {
-            process.StartInfo = _getStartTypeProcessInfo;
-            process.Start();
             
-            var output = await process.StandardOutput.ReadToEndAsync();
+        //IsAutoloading = output.Replace("\0", "").Trim() == "SERVICE_AUTO_START";
-            var error = await process.StandardError.ReadToEndAsync();
-            
-            await process.WaitForExitAsync();
-            
-            if (!string.IsNullOrEmpty(error))
-            {
-                Console.WriteLine(error);
-            }
-            
-            IsAutoloading = output.Replace("\0", "").Trim() == "SERVICE_AUTO_START";
-        }
     }
 }

MicrosocksGUI/Views/MainWindow.axaml

@@ -29,7 +29,7 @@
                 HorizontalAlignment="Stretch" 
                 HorizontalContentAlignment="Center" 
                 IsChecked="{Binding IsWorking}"
-                Command="{Binding SetWorkingStatusToServiceCommand}">
+                Command="{Binding SetServiceStatusCommand}">
                 Работа
             </ToggleButton>
             
@@ -37,7 +37,8 @@
                 Grid.Column="1" 
                 HorizontalAlignment="Stretch" 
                 HorizontalContentAlignment="Center" 
-                IsChecked="{Binding IsAutoloading}">
+                IsChecked="{Binding IsAutoloading}"
+                Command="{Binding SetServiceStatusCommand}">
                 Автозапуск
             </ToggleButton>
         </Grid>

MicrosocksGUI/app.manifest

@@ -15,4 +15,15 @@
       <supportedOS Id="{8e0f7a12-bfb3-4fe8-b9a5-48fd50a15a9a}" />
     </application>
   </compatibility>
+
+  <!--<trustInfo xmlns="urn:schemas-microsoft-com:asm.v2">
+    <security>
+      <requestedPrivileges xmlns="urn:schemas-microsoft-com:asm.v3">
+        &lt;!&ndash; UAC Запрос прав администратора &ndash;&gt;
+        <requestedExecutionLevel level="requireAdministrator" uiAccess="false" />
+      </requestedPrivileges>
+    </security>
+  </trustInfo>-->
+  
+ 
 </assembly>