From 9b923ba301621f1daacda80f3b87ffbb735f2ddf Mon Sep 17 00:00:00 2001 From: zyj Date: Fri, 6 Mar 2026 18:00:58 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E6=9C=AC=E6=9C=BA=E4=BB=A3?= =?UTF-8?q?=E7=90=86=E8=8E=B7=E5=8F=96=E6=96=B9=E6=B3=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/proxy.go | 173 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 173 insertions(+) create mode 100644 internal/proxy.go diff --git a/internal/proxy.go b/internal/proxy.go new file mode 100644 index 0000000..5d9f6cf --- /dev/null +++ b/internal/proxy.go @@ -0,0 +1,173 @@ +package internal + +import ( + "fmt" + "os" + "strings" + + "golang.org/x/sys/windows/registry" +) + +// ProxyInfo 保存当前系统代理信息 +type ProxyInfo struct { + // 是否启用了系统代理 + Enabled bool + // 系统代理服务器地址 (例如 127.0.0.1:7890) + Server string + // 代理协议 (http, https, socks5) + Protocol string + // HTTP_PROXY 环境变量 + HTTPProxy string + // HTTPS_PROXY 环境变量 + HTTPSProxy string + // 代理绕过列表 + Bypass string +} + +// GetCurrentProxy 获取本机当前的代理设置 +// 优先读取 Windows 系统代理（注册表），同时也读取环境变量中的代理配置 +// 支持检测 Clash、V2Ray 等常见代理工具的配置 +func GetCurrentProxy() (*ProxyInfo, error) { + info := &ProxyInfo{} + + // 1. 从环境变量读取代理 + info.HTTPProxy = getEnvProxy("HTTP_PROXY", "http_proxy") + info.HTTPSProxy = getEnvProxy("HTTPS_PROXY", "https_proxy") + + // 2. 从 Windows 注册表读取系统代理设置 + if err := readWindowsSystemProxy(info); err != nil { + // 注册表读取失败不算致命错误，环境变量中可能仍有代理信息 + fmt.Printf("读取系统代理设置时出错: %v\n", err) + } + + // 3. 如果系统代理未启用，尝试从环境变量推断 + if !info.Enabled && (info.HTTPProxy != "" || info.HTTPSProxy != "") { + info.Enabled = true + proxy := info.HTTPProxy + if proxy == "" { + proxy = info.HTTPSProxy + } + info.Server = extractServerFromURL(proxy) + info.Protocol = extractProtocolFromURL(proxy) + } + + return info, nil +} + +// readWindowsSystemProxy 从 Windows 注册表读取系统代理设置 +func readWindowsSystemProxy(info *ProxyInfo) error { + key, err := registry.OpenKey( + registry.CURRENT_USER, + `Software\Microsoft\Windows\CurrentVersion\Internet Settings`, + registry.QUERY_VALUE, + ) + if err != nil { + return fmt.Errorf("无法打开注册表键: %w", err) + } + defer key.Close() + + // 读取代理是否启用 (ProxyEnable: 0=关闭, 1=开启) + proxyEnable, _, err := key.GetIntegerValue("ProxyEnable") + if err != nil { + return fmt.Errorf("无法读取 ProxyEnable: %w", err) + } + info.Enabled = proxyEnable == 1 + + // 读取代理服务器地址 + proxyServer, _, err := key.GetStringValue("ProxyServer") + if err == nil && proxyServer != "" { + info.Server = proxyServer + // 根据端口推断代理协议 + info.Protocol = guessProtocol(proxyServer) + } + + // 读取代理绕过列表 + bypass, _, err := key.GetStringValue("ProxyOverride") + if err == nil { + info.Bypass = bypass + } + + return nil +} + +// getEnvProxy 依次检查多个环境变量名，返回第一个非空值 +func getEnvProxy(names ...string) string { + for _, name := range names { + if v := os.Getenv(name); v != "" { + return v + } + } + return "" +} + +// extractServerFromURL 从代理 URL 中提取 host:port 部分 +// 例如 "http://127.0.0.1:7890" -> "127.0.0.1:7890" +func extractServerFromURL(proxyURL string) string { + s := proxyURL + // 去掉协议前缀 + if idx := strings.Index(s, "://"); idx != -1 { + s = s[idx+3:] + } + // 去掉尾部斜杠 + s = strings.TrimRight(s, "/") + return s +} + +// extractProtocolFromURL 从代理 URL 中提取协议 +// 例如 "socks5://127.0.0.1:7891" -> "socks5" +func extractProtocolFromURL(proxyURL string) string { + if idx := strings.Index(proxyURL, "://"); idx != -1 { + return proxyURL[:idx] + } + return "http" +} + +// guessProtocol 根据代理服务器地址猜测代理协议 +// Clash 默认: 7890(http), 7891(socks5) +// V2Ray 默认: 10808(socks5), 10809(http) +func guessProtocol(server string) string { + knownPorts := map[string]string{ + "7890": "http", // Clash HTTP + "7891": "socks5", // Clash SOCKS5 + "7892": "http", // Clash mixed + "10808": "socks5", // V2Ray SOCKS5 + "10809": "http", // V2Ray HTTP + "1080": "socks5", // 通用 SOCKS5 + "1081": "http", // 通用 HTTP + "8080": "http", // 通用 HTTP + } + + parts := strings.Split(server, ":") + if len(parts) >= 2 { + port := parts[len(parts)-1] + if protocol, ok := knownPorts[port]; ok { + return protocol + } + } + return "http" +} + +// String 返回代理信息的可读字符串 +func (p *ProxyInfo) String() string { + if !p.Enabled { + return "当前未检测到系统代理" + } + + var sb strings.Builder + sb.WriteString("当前系统代理信息:\n") + sb.WriteString(" 状态: 已启用\n") + fmt.Fprintf(&sb, " 服务器: %s\n", p.Server) + fmt.Fprintf(&sb, " 协议: %s\n", p.Protocol) + + if p.Bypass != "" { + fmt.Fprintf(&sb, " 绕过: %s\n", p.Bypass) + } + if p.HTTPProxy != "" { + fmt.Fprintf(&sb, " HTTP_PROXY: %s\n", p.HTTPProxy) + } + if p.HTTPSProxy != "" { + fmt.Fprintf(&sb, " HTTPS_PROXY: %s\n", p.HTTPSProxy) + } + + return sb.String() +}