李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
59 lines
1.9 KiB
Go
59 lines
1.9 KiB
Go
package proxyutil
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"net"
|
||
"net/http"
|
||
"net/url"
|
||
"testing"
|
||
"time"
|
||
|
||
"github.com/stretchr/testify/require"
|
||
)
|
||
|
||
var errStub = errors.New("stub dial")
|
||
|
||
// 回归：SOCKS5 分支覆盖了调用方在 Transport 上设置的 DialContext，
|
||
// 底层 forward dialer 必须自带建连超时。proxy.Direct 是零值 net.Dialer，
|
||
// 代理不可达时会一直卡到内核 TCP 重传耗尽（Linux 约 130 秒）。
|
||
func TestSOCKS5ForwardDialerHasBoundedTimeout(t *testing.T) {
|
||
require.Greater(t, socks5ForwardDialer.Timeout, time.Duration(0))
|
||
require.Equal(t, socks5DialTimeout, socks5ForwardDialer.Timeout)
|
||
require.Equal(t, socks5DialKeepAlive, socks5ForwardDialer.KeepAlive)
|
||
}
|
||
|
||
func TestConfigureTransportProxySOCKS5SetsDialContext(t *testing.T) {
|
||
for _, scheme := range []string{"socks5", "socks5h"} {
|
||
t.Run(scheme, func(t *testing.T) {
|
||
proxyURL, err := url.Parse(scheme + "://127.0.0.1:1080")
|
||
require.NoError(t, err)
|
||
|
||
transport := &http.Transport{}
|
||
require.NoError(t, ConfigureTransportProxy(transport, proxyURL))
|
||
require.NotNil(t, transport.DialContext)
|
||
require.Nil(t, transport.Proxy, "SOCKS5 不应设置 Transport.Proxy")
|
||
})
|
||
}
|
||
}
|
||
|
||
// HTTP 代理走 Transport.Proxy，不得覆盖调用方设置的 DialContext。
|
||
func TestConfigureTransportProxyHTTPPreservesDialContext(t *testing.T) {
|
||
proxyURL, err := url.Parse("http://127.0.0.1:8080")
|
||
require.NoError(t, err)
|
||
|
||
called := false
|
||
transport := &http.Transport{}
|
||
transport.DialContext = func(_ context.Context, _, _ string) (net.Conn, error) {
|
||
called = true
|
||
return nil, errStub
|
||
}
|
||
|
||
require.NoError(t, ConfigureTransportProxy(transport, proxyURL))
|
||
require.NotNil(t, transport.Proxy)
|
||
require.NotNil(t, transport.DialContext)
|
||
|
||
_, _ = transport.DialContext(context.Background(), "tcp", "127.0.0.1:1")
|
||
require.True(t, called, "HTTP 代理分支不应替换调用方的 DialContext")
|
||
}
|