li
66ad405dd6
http.Transport 未设置 DialContext 时使用零值 net.Dialer（Timeout=0）， DNS 解析与 TCP 握手没有任何上限，只能等内核 TCP 重传耗尽（Linux 约 130 秒）。 ResponseHeaderTimeout 只约束连接建立之后等待响应头的阶段，覆盖不到建连。 上游域名被解析到 443 不可达的 IP 时（DNS 污染 / 路由异常），单个账号就要卡满 内核超时；而多账号故障转移是串行的，一次请求会阻塞数分钟且不写中间错误， 客户端表现为"长时间无首字后直接结束"。 - buildUpstreamTransport 显式设置 DialContext（10s）与 TLSHandshakeTimeout（10s） - proxyutil SOCKS5 分支的 forward dialer 由 proxy.Direct（零值 dialer，同样无超时） 换成带 10s 超时的 net.Dialer；该分支会覆盖 Transport.DialContext， 调用方设置的建连超时对它无效，必须在此补上 Fixes #5151
87 lines
2.8 KiB
Go
87 lines
2.8 KiB
Go
// Package proxyutil 提供统一的代理配置功能
|
||
//
|
||
// 支持的代理协议：
|
||
// - HTTP/HTTPS: 通过 Transport.Proxy 设置
|
||
// - SOCKS5: 通过 Transport.DialContext 设置（客户端本地解析 DNS）
|
||
// - SOCKS5H: 通过 Transport.DialContext 设置（代理端远程解析 DNS，推荐）
|
||
//
|
||
// 注意：proxyurl.Parse() 会自动将 socks5:// 升级为 socks5h://，
|
||
// 确保 DNS 也由代理端解析，防止 DNS 泄漏。
|
||
package proxyutil
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"net"
|
||
"net/http"
|
||
"net/url"
|
||
"strings"
|
||
"time"
|
||
|
||
"golang.org/x/net/proxy"
|
||
)
|
||
|
||
const (
|
||
// socks5DialTimeout 限制到 SOCKS5 代理自身的 TCP 建连耗时。
|
||
socks5DialTimeout = 10 * time.Second
|
||
// socks5DialKeepAlive 与 Go 默认 keepalive 探测间隔保持一致。
|
||
socks5DialKeepAlive = 30 * time.Second
|
||
)
|
||
|
||
// socks5ForwardDialer 是 SOCKS5 dialer 的底层拨号器。
|
||
//
|
||
// proxy.FromURL 的默认 forward dialer 是 proxy.Direct（零值 net.Dialer，无超时），
|
||
// 代理地址不可达时会一直卡到内核 TCP 重传耗尽（Linux 约 130 秒）。SOCKS5 分支会
|
||
// 覆盖 Transport.DialContext，因此调用方在 Transport 上设置的建连超时对这条路径
|
||
// 无效，必须在这里补上。
|
||
var socks5ForwardDialer = &net.Dialer{
|
||
Timeout: socks5DialTimeout,
|
||
KeepAlive: socks5DialKeepAlive,
|
||
}
|
||
|
||
// ConfigureTransportProxy 根据代理 URL 配置 Transport
|
||
//
|
||
// 支持的协议：
|
||
// - http/https: 设置 transport.Proxy
|
||
// - socks5: 设置 transport.DialContext（客户端本地解析 DNS）
|
||
// - socks5h: 设置 transport.DialContext（代理端远程解析 DNS，推荐）
|
||
//
|
||
// 参数：
|
||
// - transport: 需要配置的 http.Transport
|
||
// - proxyURL: 代理地址，nil 表示直连
|
||
//
|
||
// 返回：
|
||
// - error: 代理配置错误（协议不支持或 dialer 创建失败）
|
||
func ConfigureTransportProxy(transport *http.Transport, proxyURL *url.URL) error {
|
||
if proxyURL == nil {
|
||
return nil
|
||
}
|
||
|
||
scheme := strings.ToLower(proxyURL.Scheme)
|
||
switch scheme {
|
||
case "http", "https":
|
||
transport.Proxy = http.ProxyURL(proxyURL)
|
||
return nil
|
||
|
||
case "socks5", "socks5h":
|
||
dialer, err := proxy.FromURL(proxyURL, socks5ForwardDialer)
|
||
if err != nil {
|
||
return fmt.Errorf("create socks5 dialer: %w", err)
|
||
}
|
||
// 优先使用支持 context 的 DialContext，以支持请求取消和超时
|
||
if contextDialer, ok := dialer.(proxy.ContextDialer); ok {
|
||
transport.DialContext = contextDialer.DialContext
|
||
} else {
|
||
// 回退路径：如果 dialer 不支持 ContextDialer，则包装为简单的 DialContext
|
||
// 注意：此回退不支持请求取消和超时控制
|
||
transport.DialContext = func(_ context.Context, network, addr string) (net.Conn, error) {
|
||
return dialer.Dial(network, addr)
|
||
}
|
||
}
|
||
return nil
|
||
|
||
default:
|
||
return fmt.Errorf("unsupported proxy scheme: %s", scheme)
|
||
}
|
||
}
|