XdCxRhDW/Service/X2D1TaskServer/Program.cs

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Reflection;
using System.Security.Principal;
using System.Windows.Forms;

namespace X2D1TaskServer
{
    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;
                XdCxRhDW.Framework.LogHelper.Error("出现未处理的异常,程序即将退出!", args);
            };
            Application.ThreadException += (sender, e) =>
            {
                XdCxRhDW.UI.Lib.LogUI.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)
        {
            if (IsRunningAsAdmin())
            {
                MainForm mainForm = new MainForm();
                System.Windows.Forms.Application.Run(mainForm);
            }
            else
            {
                RestartAsAdmin();
            }
        }
    }
}