XdCxRhDW/Service/CpuCgServer/Program.cs

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;
                XdCxRhDW.Framework.LogHelper.Error("出现未处理的异常,程序即将退出!", args);
            };
            Application.ThreadException += (sender, e) =>
            {
                XdCxRhDW.UI.Lib.LogUI.Error("出现未处理的线程异常!", e.Exception).Wait(5000);
            };
            AppDomain.CurrentDomain.AssemblyResolve += (sender, args) =>
            {
                //if (args.Name.EndsWith(".dll"))
                //    return Assembly.Load(args.Name);
                // 1. 提取程序集简单名称
                string assemblyName = new AssemblyName(args.Name).Name;

                // 2. 定义程序集可能的查找路径
                string assemblyPath1 = Path.Combine(Application.StartupPath, "AddIns", $"{assemblyName}.dll");
                string assemblyPath2 = Path.Combine(Application.StartupPath, $"{assemblyName}.dll");

                if (File.Exists(assemblyPath1))
                    return Assembly.LoadFrom(assemblyPath1);
                if (File.Exists(assemblyPath2))
                    return Assembly.LoadFrom(assemblyPath2);
                return null;
            };
        }
        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();
                if (args.Length > 0)
                {
                    mainForm.platAddr = args[0].Trim();
                }
                System.Windows.Forms.Application.Run(mainForm);
            }
            else
            {
                RestartAsAdmin();
            }
        }
    }
}