XdCxRhDW/Service/CpuCgServer/Program.cs

using Serilog;
using System;
using System.Collections.Generic;
using System.Configuration;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Security.Principal;
using System.Threading.Tasks;
using System.Windows.Forms;

namespace CpuCgServer
{
    internal static class Program
    {
        static Program()
        {

            //设置私有路径
            Environment.CurrentDirectory = AppDomain.CurrentDomain.BaseDirectory;
            AppDomain.CurrentDomain.SetData("PRIVATE_BINPATH", "AddIns;");
            var m = typeof(AppDomainSetup).GetMethod("UpdateContextProperty", BindingFlags.NonPublic | BindingFlags.Static);
            var funsion = typeof(AppDomain).GetMethod("GetFusionContext", BindingFlags.NonPublic | BindingFlags.Instance);
            m.Invoke(null, new object[] { funsion.Invoke(AppDomain.CurrentDomain, null), "PRIVATE_BINPATH", "AddIns;" });

            //c++dll加入环境变量
            string paths = Environment.GetEnvironmentVariable("PATH");
            var dirs = Directory.EnumerateDirectories("AddIns", "*", SearchOption.AllDirectories);
            List<string> list = new List<string>
            {
                Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "AddIns")
            };
            foreach (var item in dirs)
            {
                list.Add(Path.Combine(AppDomain.CurrentDomain.BaseDirectory, item));
            }
            Environment.SetEnvironmentVariable("PATH", $"{paths};{string.Join(";", list)}");

            AppDomain.CurrentDomain.UnhandledException += (sender, e) =>
            {
                var args = e.ExceptionObject as Exception;
                LogHelper.Error("出现未处理的异常,程序即将退出!", args).Wait(5000);
            };
            Application.ThreadException += (sender, e) =>
            {
                LogHelper.Error("出现未处理的线程异常!", e.Exception).Wait(5000);
            };
        }
        static bool IsRunningAsAdmin()
        {
            bool result;
            try
            {
                WindowsIdentity identity = WindowsIdentity.GetCurrent();
                WindowsPrincipal principal = new WindowsPrincipal(identity);
                result = principal.IsInRole(WindowsBuiltInRole.Administrator);
            }
            catch
            {
                result = false;
            }
            return result;
        }
        static void RestartAsAdmin()
        {
            var startInfo = new ProcessStartInfo();
            startInfo.FileName = Application.ExecutablePath;
            startInfo.Verb = "runas"; // 以管理员身份运行
            try
            {
                Process.Start(startInfo);
            }
            catch (System.ComponentModel.Win32Exception)
            {
                // 用户取消了管理员权限提示，或者其他错误
                // 可以在此处处理错误情况
            }

            Application.Exit();
        }
        /// <summary>
        /// 应用程序的主入口点。
        /// </summary>
        [STAThread]
        static void Main(string[] args)
        {
            string outputTemplate = "{Timestamp:yyyy-MM-dd HH:mm:ss.fff zzz} [{Level:u3}] {Message:lj}{NewLine}  {Exception}";
            Serilog.Log.Logger = new Serilog.LoggerConfiguration()
                .WriteTo.Console(outputTemplate: outputTemplate)
                .WriteTo.Logger(p => p.Filter.ByIncludingOnly(e => e.Level == Serilog.Events.LogEventLevel.Information)
                    .WriteTo.File("Logs\\Info\\.log", rollingInterval: Serilog.RollingInterval.Day))
                .WriteTo.Logger(p => p.Filter.ByIncludingOnly(e => e.Level == Serilog.Events.LogEventLevel.Warning)
                    .WriteTo.File("Logs\\Warning\\.log", rollingInterval: Serilog.RollingInterval.Day))
                .WriteTo.Logger(p => p.Filter.ByIncludingOnly(e => e.Level == Serilog.Events.LogEventLevel.Error)
                    .WriteTo.File("Logs\\Error\\.log", rollingInterval: Serilog.RollingInterval.Day, outputTemplate: outputTemplate))
                .CreateLogger();
         
            if (IsRunningAsAdmin())
            {
                MainForm mainForm = new MainForm();
                System.Windows.Forms.Application.Run(mainForm);
            }
            else
            {
                RestartAsAdmin();
            }
        }
    }
}