netch/Netch/Controllers/HTTPController.cs

using System;
using System.Diagnostics;
using System.Windows.Forms;
using Microsoft.Win32;
using Netch.Models;
using Netch.Utils;

namespace Netch.Controllers
{
    public class HTTPController
    {
        private bool prevEnabled;
        private string prevBypass, prevHTTP, prevPAC;

        /// <summary>
        ///     实例
        /// </summary>
        public PrivoxyController pPrivoxyController = new PrivoxyController();

        /// <summary>
        ///		启动
        /// </summary>
        /// <param name="server">服务器</param>
        /// <param name="mode">模式</param>
        /// <returns>是否启动成功</returns>
        public bool Start(Server server, Mode mode)
        {
            RecordPrevious();
            try
            {
                if (server.Type == "Socks5")
                {
                    if (!string.IsNullOrWhiteSpace(server.Username) && !string.IsNullOrWhiteSpace(server.Password))
                    {
                        return false;
                    }

                    pPrivoxyController.Start(server, mode);
                }
                else
                {
                    pPrivoxyController.Start(server, mode);
                }

                if (mode.Type != 5)
                {
                    NativeMethods.SetGlobal($"127.0.0.1:{Global.Settings.HTTPLocalPort}", "<local>");
                }
            }
            catch (Exception e)
            {
                if (MessageBox.Show(i18N.Translate("Failed to set the system proxy, it may be caused by the lack of dependent programs. Do you want to jump to Netch's official website to download dependent programs?"), i18N.Translate("Information"), MessageBoxButtons.OKCancel, MessageBoxIcon.Information) == DialogResult.OK)
                {
                    Process.Start("https://netch.org/#/?id=%e4%be%9d%e8%b5%96");
                }
                Logging.Info("设置系统代理失败" + e);
                return false;
            }

            return true;
        }

        private void RecordPrevious()
        {
            var registry = Registry.CurrentUser.OpenSubKey("Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings");
            if (registry == null)
            {
                prevEnabled = false;
                prevPAC = prevHTTP = prevBypass = "";
                return;
            }
            
            prevPAC = registry.GetValue("AutoConfigURL")?.ToString() ?? "";
            if ((registry.GetValue("ProxyEnable")?.Equals(1) ?? false) || prevPAC != "")
            {
                prevEnabled = true;
            }
            prevHTTP = registry.GetValue("ProxyServer")?.ToString() ?? "";
            prevBypass = registry.GetValue("ProxyOverride")?.ToString() ?? "";
        }

        /// <summary>
        ///		停止
        /// </summary>
        public void Stop()
        {
            try
            {
                try
                {
                    pPrivoxyController.Stop();
                }
                catch (Exception e)
                {
                    Logging.Info(e.ToString());
                }

                NativeMethods.SetGlobal(prevHTTP, prevBypass);
                if (prevPAC != "")
                    NativeMethods.SetURL(prevPAC);
                if (!prevEnabled)
                    NativeMethods.SetDIRECT();
                prevEnabled = false;
            }
            catch (Exception e)
            {
                Logging.Info(e.ToString());
            }
        }
    }
}