feat: 检测并要求以管理员权限运行

incoming: 提供服务
SaltWood-Studio · Mar 9, 2024 · e56f318 · e56f318
1 parent 3d462de
commit e56f318
Show file tree

Hide file tree

Showing 14 changed files with 102 additions and 106 deletions.
diff --git a/CSharp-OpenBMCLAPI/CSharp-OpenBMCLAPI.csproj b/CSharp-OpenBMCLAPI/CSharp-OpenBMCLAPI.csproj
@@ -15,9 +15,9 @@
   <ItemGroup>
     <PackageReference Include="Apache.Avro" Version="1.11.3" />
     <PackageReference Include="Downloader" Version="3.0.6" />
-    <PackageReference Include="log4net" Version="2.0.15" />
+    <PackageReference Include="log4net" Version="2.0.16" />
     <PackageReference Include="Newtonsoft.Json" Version="13.0.3" />
-    <PackageReference Include="SaltWood.TeraIO" Version="1.0.2" />
+    <PackageReference Include="SaltWood.TeraIO" Version="1.0.4" />
     <PackageReference Include="SocketIOClient" Version="3.1.1" />
     <PackageReference Include="ZstdSharp.Port" Version="0.7.5" />
   </ItemGroup>

diff --git a/CSharp-OpenBMCLAPI/Logger.cs b/CSharp-OpenBMCLAPI/Logger.cs
@@ -1,10 +1,4 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
-
-namespace CSharpOpenBMCLAPI
+namespace CSharpOpenBMCLAPI
 {
     public class Logger
     {

diff --git a/CSharp-OpenBMCLAPI/Modules/Challenge.cs b/CSharp-OpenBMCLAPI/Modules/Challenge.cs
@@ -1,10 +1,4 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
-
-namespace CSharpOpenBMCLAPI.Modules
+namespace CSharpOpenBMCLAPI.Modules
 {
     public class Challenge
     {

diff --git a/CSharp-OpenBMCLAPI/Modules/Cluster.cs b/CSharp-OpenBMCLAPI/Modules/Cluster.cs
@@ -1,22 +1,16 @@
 using Avro;
-using Avro.IO;
-using Avro.File;
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
-using TeraIO.Runnable;
-using ZstdSharp;
 using Avro.Generic;
+using Avro.IO;
 using Downloader;
-using System.Security.Cryptography;
 using SocketIOClient;
-using SocketIO.Core;
-using TeraIO.Extension;
-using System.Diagnostics;
+using System.Runtime.InteropServices;
+using System.Security.Cryptography;
+using System.Security.Principal;
+using System.Text;
 using System.Text.Json;
-using System.Net;
+using TeraIO.Network.Http;
+using TeraIO.Runnable;
+using ZstdSharp;
 
 namespace CSharpOpenBMCLAPI.Modules
 {
@@ -30,6 +24,7 @@ public class Cluster : RunnableBase
         public bool IsEnabled { get; private set; }
         private Task? _keepAlive;
         //List<Task> tasks = new List<Task>();
+        private HttpServer server;
 
         public Cluster(ClusterInfo info, TokenManager token) : base()
         {
@@ -54,6 +49,7 @@ public Cluster(ClusterInfo info, TokenManager token) : base()
                 }
             });
 
+            this.server = new();
         }
 
         private void HandleError(SocketIOResponse resp)
@@ -80,13 +76,7 @@ protected async Task<int> AsyncRun()
 
             Connect();
 
-            HttpServer hs = new();
-            hs.PORT = 4000;
-            hs.UriPrefixes = new()
-            {
-                "http://localhost:4000/"
-            };
-            hs.Start();
+            InitializeHttpsServer(this.server);
 
             await RequestCertification();
 
@@ -109,6 +99,34 @@ protected async Task<int> AsyncRun()
             return returns;
         }
 
+        private void InitializeHttpsServer(HttpServerAppBase server)
+        {
+            server.UriPrefixes = new()
+            {
+                $"http://*:4000/"
+            };
+            server.Started += (current, e) => SharedData.Logger.LogInfo($"HTTP 服务实例 \"<{server} {server.GetHashCode()}>\" 已启动");
+            server.Stopped += (current, e) => SharedData.Logger.LogInfo($"HTTP 服务实例 \"<{server} {server.GetHashCode()}>\" 已停止");
+            server.Start();
+        }
+
+        /*
+    private void RequestFirewall()
+    {
+        FtpWebRequest requestDownload;
+        requestDownload = (FtpWebRequest)WebRequest.Create(uri);
+        requestDownload.UsePassive = false;
+        requestDownload.KeepAlive = false;
+        requestDownload.UseBinary = true;
+        requestDownload.Method = WebRequestMethods.Ftp.DownloadFile;
+
+
+        requestDownload.Credentials = new NetworkCredential(ftpInfoDownload[3], ftpInfoDownload[4]);
+
+        responseDownload = (FtpWebResponse)requestDownload.GetResponse();
+        Stream ftpStream = responseDownload.GetResponseStream();
+    }*/
+
         public void Connect()
         {
             this.socket.ConnectAsync().Wait();
@@ -128,6 +146,8 @@ public async Task Enable()
             await socket.EmitAsync("enable",
                 (SocketIOResponse resp) =>
                 {
+                    SharedData.Logger.LogInfo(resp);
+                    // Debugger.Break();
                     SharedData.Logger.LogInfo($"启用成功");
                 },
                 new

diff --git a/CSharp-OpenBMCLAPI/Modules/ClusterInfo.cs b/CSharp-OpenBMCLAPI/Modules/ClusterInfo.cs
@@ -1,11 +1,4 @@
-using System;
-using System.Collections.Generic;
-using System.Configuration;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
-
-namespace CSharpOpenBMCLAPI.Modules
+namespace CSharpOpenBMCLAPI.Modules
 {
     public struct ClusterInfo
     {

diff --git a/CSharp-OpenBMCLAPI/Modules/Config.cs b/CSharp-OpenBMCLAPI/Modules/Config.cs
@@ -1,10 +1,5 @@
 using Newtonsoft.Json;
-using System;
-using System.Collections.Generic;
 using System.ComponentModel;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
 
 namespace CSharpOpenBMCLAPI.Modules
 {
@@ -25,8 +20,9 @@ public string HOST
             {
                 if (string.IsNullOrEmpty(this._host))
                 {
-                    HttpClient client = new HttpClient();
-                    this._host = client.GetAsync("https://4.ipw.cn/").Result.Content.ReadAsStringAsync().Result;
+                    //HttpClient client = new HttpClient();
+                    //this._host = client.GetAsync("https://4.ipw.cn/").Result.Content.ReadAsStringAsync().Result;
+                    this._host = $"";
                 }
                 return this._host;
             }

diff --git a/CSharp-OpenBMCLAPI/Modules/ExtensionMethods.cs b/CSharp-OpenBMCLAPI/Modules/ExtensionMethods.cs
@@ -1,9 +1,4 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Reflection;
-using System.Text;
-using System.Threading.Tasks;
+using System.Reflection;
 
 namespace CSharpOpenBMCLAPI.Modules
 {

diff --git a/CSharp-OpenBMCLAPI/Modules/HttpRequest.cs b/CSharp-OpenBMCLAPI/Modules/HttpRequest.cs
@@ -1,10 +1,4 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
-
-namespace CSharpOpenBMCLAPI.Modules
+namespace CSharpOpenBMCLAPI.Modules
 {
     internal class HttpRequest
     {

diff --git a/CSharp-OpenBMCLAPI/Modules/HttpServer.cs b/CSharp-OpenBMCLAPI/Modules/HttpServer.cs
@@ -1,17 +1,18 @@
-using System;
-using System.Collections.Generic;
-using System.Diagnostics;
-using System.Linq;
-using System.Security.Policy;
-using System.Text;
-using System.Threading.Tasks;
+using System.Security.Cryptography.X509Certificates;
 using TeraIO.Network.Http;
 
 namespace CSharpOpenBMCLAPI.Modules
 {
     public class HttpServer : HttpServerAppBase
     {
-        public HttpServer() : base() => LoadNew();
+        public HttpServer() : base()
+        {
+            LoadNew();
+
+            X509Certificate2 certificate = new($"{SharedData.Config.clusterFileDirectory}certifications/cert.pem",
+                $"{SharedData.Config.clusterFileDirectory}certifications/key.pem");
+
+        }
 
         [HttpHandler("/measure")]
         public void Measure(HttpClientRequest req)

diff --git a/CSharp-OpenBMCLAPI/Modules/SharedData.cs b/CSharp-OpenBMCLAPI/Modules/SharedData.cs
@@ -1,10 +1,4 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
-
-namespace CSharpOpenBMCLAPI.Modules
+namespace CSharpOpenBMCLAPI.Modules
 {
     public static class SharedData
     {

diff --git a/CSharp-OpenBMCLAPI/Modules/Token.cs b/CSharp-OpenBMCLAPI/Modules/Token.cs
@@ -1,10 +1,4 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
-
-namespace CSharpOpenBMCLAPI.Modules
+namespace CSharpOpenBMCLAPI.Modules
 {
     public struct Token
     {

diff --git a/CSharp-OpenBMCLAPI/Modules/TokenManager.cs b/CSharp-OpenBMCLAPI/Modules/TokenManager.cs
@@ -1,11 +1,7 @@
 using Newtonsoft.Json;
-using System;
-using System.Collections.Generic;
-using System.Linq;
 using System.Net.Http.Headers;
 using System.Security.Cryptography;
 using System.Text;
-using System.Threading.Tasks;
 
 namespace CSharpOpenBMCLAPI.Modules
 {

diff --git a/CSharp-OpenBMCLAPI/Modules/Utils.cs b/CSharp-OpenBMCLAPI/Modules/Utils.cs
@@ -1,9 +1,8 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
+using System.Diagnostics;
+using System.Runtime.InteropServices;
 using System.Security.Cryptography;
+using System.Security.Principal;
 using System.Text;
-using System.Threading.Tasks;
 
 namespace CSharpOpenBMCLAPI.Modules
 {
@@ -44,5 +43,35 @@ public static bool CheckSign(string? hash, string? secret, string? s, string? e)
             return sign == s && timestamp < (ToDecimal(e) / 100);
         }
 
+        public static bool IsAdministrator()
+        {
+            if (!RuntimeInformation.IsOSPlatform(OSPlatform.Windows)) return true;
+            WindowsIdentity current = WindowsIdentity.GetCurrent();
+            WindowsPrincipal windowsPrincipal = new WindowsPrincipal(current);
+            //WindowsBuiltInRole可以枚举出很多权限，例如系统用户、User、Guest等等
+            return windowsPrincipal.IsInRole(WindowsBuiltInRole.Administrator);
+        }
+
+        public static void RunAsAdministrator()
+        {
+            //创建启动对象
+            ProcessStartInfo startInfo = new ProcessStartInfo()
+            {
+                FileName = Process.GetCurrentProcess().MainModule?.FileName,
+                WorkingDirectory = Environment.CurrentDirectory,
+                Verb = "runas",
+                UseShellExecute = true,
+                CreateNoWindow = false
+            };
+            try
+            {
+                Process.Start(startInfo);
+            }
+            catch
+            {
+                return;
+            }
+            Environment.Exit(0);
+        }
     }
 }
diff --git a/CSharp-OpenBMCLAPI/Program.cs b/CSharp-OpenBMCLAPI/Program.cs
@@ -1,15 +1,6 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Net.Http.Headers;
-using System.Net.Http.Json;
-using System.Reflection;
-using System.Security.Cryptography;
-using System.Text;
-using System.Threading.Tasks;
-using System.Xml.Linq;
-using CSharpOpenBMCLAPI.Modules;
+using CSharpOpenBMCLAPI.Modules;
 using Newtonsoft.Json;
+using System.Reflection;
 using TeraIO.Runnable;
 
 namespace CSharpOpenBMCLAPI
@@ -82,6 +73,11 @@ protected async Task<int> AsyncRun()
         {
             int returns = 0;
 
+            if (!Utils.IsAdministrator())
+            {
+                Utils.RunAsAdministrator();
+            }
+
             // 从 .env.json 读取密钥然后 FetchToken
             ClusterInfo info = JsonConvert.DeserializeObject<ClusterInfo>(await File.ReadAllTextAsync(".env.json"));
             SharedData.ClusterInfo = info;