mirror of
https://github.com/SagerNet/sing-box.git
synced 2026-11-10 13:14:06 +08:00
404 lines
11 KiB
Go
404 lines
11 KiB
Go
package http
|
|
|
|
import (
|
|
std_bufio "bufio"
|
|
"context"
|
|
"encoding/base64"
|
|
"errors"
|
|
"maps"
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"sync"
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
"github.com/sagernet/sing-box/common/badhttp"
|
|
"github.com/sagernet/sing-box/common/httpclient"
|
|
C "github.com/sagernet/sing-box/constant"
|
|
"github.com/sagernet/sing-box/option"
|
|
"github.com/sagernet/sing/common/buf"
|
|
"github.com/sagernet/sing/common/bufio"
|
|
"github.com/sagernet/sing/common/bufio/deadline"
|
|
E "github.com/sagernet/sing/common/exceptions"
|
|
M "github.com/sagernet/sing/common/metadata"
|
|
N "github.com/sagernet/sing/common/network"
|
|
aTLS "github.com/sagernet/sing/common/tls"
|
|
|
|
"golang.org/x/net/http2"
|
|
)
|
|
|
|
type tlsDialer interface {
|
|
DialTLSContext(ctx context.Context, destination M.Socksaddr) (aTLS.Conn, error)
|
|
}
|
|
|
|
type ClientOptions struct {
|
|
Dialer N.Dialer
|
|
HTTP1Dialer N.Dialer
|
|
RawDialer N.Dialer
|
|
TLSConfig aTLS.Config
|
|
Server M.Socksaddr
|
|
Username string
|
|
Password string
|
|
Path string
|
|
Headers http.Header
|
|
Version int
|
|
DisableVersionFallback bool
|
|
HTTP2Options option.HTTP2Options
|
|
HTTP3Options option.QUICOptions
|
|
}
|
|
|
|
type http3Client interface {
|
|
DialContext(ctx context.Context, destination M.Socksaddr) (net.Conn, error)
|
|
ListenPacket(ctx context.Context, destination M.Socksaddr) (N.PacketConn, error)
|
|
ResetConnection()
|
|
Close() error
|
|
}
|
|
|
|
var NewHTTP3Client func(options ClientOptions, authorization string) (http3Client, error)
|
|
|
|
type Client struct {
|
|
dialer N.Dialer
|
|
http1Dialer N.Dialer
|
|
tlsDialer tlsDialer
|
|
server M.Socksaddr
|
|
authorization string
|
|
host string
|
|
path string
|
|
headers http.Header
|
|
version int
|
|
disableVersionFallback bool
|
|
http2Transport *http2.Transport
|
|
http2Access sync.Mutex
|
|
http2Conns []*http2ClientConn
|
|
http2Unsupported atomic.Bool
|
|
http2ConnectUDPUnsupported atomic.Bool
|
|
http3 http3Client
|
|
http3Broken atomic.Int64
|
|
http3Backoff atomic.Int64
|
|
}
|
|
|
|
func NewClient(options ClientOptions) (*Client, error) {
|
|
client := &Client{
|
|
dialer: options.Dialer,
|
|
http1Dialer: options.HTTP1Dialer,
|
|
server: options.Server,
|
|
path: options.Path,
|
|
headers: options.Headers.Clone(),
|
|
version: options.Version,
|
|
disableVersionFallback: options.DisableVersionFallback,
|
|
}
|
|
if client.headers != nil {
|
|
client.host = client.headers.Get("Host")
|
|
client.headers.Del("Host")
|
|
}
|
|
if client.host != "" && client.path != "" {
|
|
return nil, E.New("Host header and path are not allowed at the same time")
|
|
}
|
|
client.version = ResolveVersion(client.version, client.path, client.host)
|
|
if client.version >= 2 && (client.path != "" || client.host != "") {
|
|
return nil, E.New("path and Host header are only supported by HTTP/1")
|
|
}
|
|
if client.dialer == nil {
|
|
client.dialer = N.SystemDialer
|
|
}
|
|
if client.http1Dialer == nil {
|
|
client.http1Dialer = client.dialer
|
|
}
|
|
if dialer, isTLSDialer := options.Dialer.(tlsDialer); isTLSDialer && client.version >= 2 {
|
|
client.tlsDialer = dialer
|
|
http2Transport, err := httpclient.ConfigureHTTP2Transport(options.HTTP2Options)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
http2Transport.DisableCompression = true
|
|
client.http2Transport = http2Transport
|
|
} else if client.version == 2 && options.DisableVersionFallback {
|
|
return nil, E.New("HTTP/2 requires TLS")
|
|
}
|
|
if client.version == 3 && NewHTTP3Client == nil {
|
|
return nil, C.ErrQUICNotIncluded
|
|
}
|
|
if options.Username != "" {
|
|
client.authorization = "Basic " + base64.StdEncoding.EncodeToString([]byte(options.Username+":"+options.Password))
|
|
}
|
|
if client.version == 3 {
|
|
http3, err := NewHTTP3Client(options, client.authorization)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
client.http3 = http3
|
|
}
|
|
return client, nil
|
|
}
|
|
|
|
func ResolveVersion(version int, path string, host string) int {
|
|
if version != 0 {
|
|
return version
|
|
}
|
|
if path != "" || host != "" {
|
|
return 1
|
|
}
|
|
return 2
|
|
}
|
|
|
|
func (c *Client) MultiplexEnabled() bool {
|
|
return c.http3Available() || (c.tlsDialer != nil && !c.http2Unsupported.Load())
|
|
}
|
|
|
|
func (c *Client) http3Available() bool {
|
|
if c.http3 == nil {
|
|
return false
|
|
}
|
|
brokenUntil := c.http3Broken.Load()
|
|
return brokenUntil == 0 || time.Now().UnixNano() >= brokenUntil
|
|
}
|
|
|
|
func (c *Client) markHTTP3Broken() {
|
|
backoff := time.Duration(c.http3Backoff.Load())
|
|
if backoff == 0 {
|
|
backoff = http3BrokenBackoffInitial
|
|
} else {
|
|
backoff = min(backoff*2, http3BrokenBackoffMax)
|
|
}
|
|
c.http3Backoff.Store(int64(backoff))
|
|
c.http3Broken.Store(time.Now().Add(backoff).UnixNano())
|
|
}
|
|
|
|
func (c *Client) clearHTTP3Broken() {
|
|
c.http3Broken.Store(0)
|
|
c.http3Backoff.Store(0)
|
|
}
|
|
|
|
func (c *Client) DialContext(ctx context.Context, network string, destination M.Socksaddr) (net.Conn, error) {
|
|
switch N.NetworkName(network) {
|
|
case N.NetworkTCP:
|
|
case N.NetworkUDP:
|
|
return nil, os.ErrInvalid
|
|
default:
|
|
return nil, E.Extend(N.ErrUnknownNetwork, network)
|
|
}
|
|
if c.http3Available() {
|
|
conn, err := c.http3.DialContext(ctx, destination)
|
|
if err == nil {
|
|
c.clearHTTP3Broken()
|
|
return conn, nil
|
|
}
|
|
if c.disableVersionFallback || !errors.Is(err, ErrHTTP3Unavailable) {
|
|
return nil, err
|
|
}
|
|
c.markHTTP3Broken()
|
|
}
|
|
if c.tlsDialer != nil && !c.http2Unsupported.Load() {
|
|
clientConn, conn, err := c.acquireHTTP2(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if clientConn != nil {
|
|
return c.connectHTTP2(ctx, clientConn, destination)
|
|
}
|
|
if c.disableVersionFallback {
|
|
conn.Close()
|
|
return nil, ErrHTTP2Unsupported
|
|
}
|
|
return c.connectAndClose(ctx, conn, destination)
|
|
}
|
|
conn, err := c.http1Dialer.DialContext(ctx, N.NetworkTCP, c.server)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return c.connectAndClose(ctx, conn, destination)
|
|
}
|
|
|
|
func (c *Client) closeHTTP2Locked() {
|
|
for _, clientConn := range c.http2Conns {
|
|
clientConn.Close()
|
|
}
|
|
c.http2Conns = nil
|
|
}
|
|
|
|
func (c *Client) ResetConnections() {
|
|
c.http2Access.Lock()
|
|
c.closeHTTP2Locked()
|
|
c.http2Access.Unlock()
|
|
c.http2Unsupported.Store(false)
|
|
c.http2ConnectUDPUnsupported.Store(false)
|
|
if c.http3 != nil {
|
|
c.http3.ResetConnection()
|
|
c.clearHTTP3Broken()
|
|
}
|
|
}
|
|
|
|
func (c *Client) connect(ctx context.Context, conn net.Conn, destination M.Socksaddr) (net.Conn, error) {
|
|
stop := context.AfterFunc(ctx, func() {
|
|
conn.Close()
|
|
})
|
|
defer stop()
|
|
request := &http.Request{
|
|
Method: http.MethodConnect,
|
|
Header: http.Header{
|
|
"Proxy-Connection": []string{"Keep-Alive"},
|
|
},
|
|
}
|
|
if c.host != "" && c.host != destination.Fqdn {
|
|
request.Host = c.host
|
|
request.URL = &url.URL{Opaque: destination.String()}
|
|
} else {
|
|
request.URL = &url.URL{Host: destination.String()}
|
|
}
|
|
if c.path != "" {
|
|
err := badhttp.URLSetPath(request.URL, c.path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
maps.Copy(request.Header, c.headers)
|
|
if _, loaded := request.Header["User-Agent"]; !loaded {
|
|
request.Header["User-Agent"] = nil
|
|
}
|
|
if c.authorization != "" {
|
|
request.Header.Set("Proxy-Authorization", c.authorization)
|
|
}
|
|
err := request.Write(conn)
|
|
if err != nil {
|
|
return nil, E.Cause(err, "write request")
|
|
}
|
|
reader := std_bufio.NewReader(conn)
|
|
response, err := http.ReadResponse(reader, request)
|
|
if err != nil {
|
|
return nil, E.Cause(err, "read response")
|
|
}
|
|
if response.StatusCode != http.StatusOK {
|
|
return nil, statusError(response)
|
|
}
|
|
if !stop() {
|
|
return nil, ctx.Err()
|
|
}
|
|
if reader.Buffered() > 0 {
|
|
buffer := buf.NewSize(reader.Buffered())
|
|
_, err = buffer.ReadFullFrom(reader, buffer.FreeLen())
|
|
if err != nil {
|
|
buffer.Release()
|
|
return nil, err
|
|
}
|
|
return bufio.NewCachedConn(conn, buffer), nil
|
|
}
|
|
return conn, nil
|
|
}
|
|
|
|
func (c *Client) Close() error {
|
|
c.http2Access.Lock()
|
|
defer c.http2Access.Unlock()
|
|
c.closeHTTP2Locked()
|
|
if c.http3 != nil {
|
|
c.http3.Close()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
const (
|
|
http3BrokenBackoffInitial = 5 * time.Second
|
|
http3BrokenBackoffMax = 5 * time.Minute
|
|
)
|
|
|
|
var (
|
|
ErrHTTP2Unsupported = E.New("server does not support HTTP/2")
|
|
ErrHTTP3Unavailable = E.New("HTTP/3 unavailable")
|
|
errExtendedConnectUnsupported = E.New("server does not support HTTP/2 extended CONNECT")
|
|
)
|
|
|
|
func statusError(response *http.Response) error {
|
|
switch response.StatusCode {
|
|
case http.StatusProxyAuthRequired:
|
|
return E.New("authentication required")
|
|
case http.StatusMethodNotAllowed:
|
|
return E.New("method not allowed")
|
|
default:
|
|
return E.New("unexpected status: ", response.Status)
|
|
}
|
|
}
|
|
|
|
func (c *Client) ListenPacket(ctx context.Context, destination M.Socksaddr) (net.PacketConn, error) {
|
|
packetConn, err := c.listenPacket(ctx, destination)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return deadline.NewPacketConn(bufio.NewNetPacketConn(&boundPacketConn{PacketConn: packetConn, destination: destination.Unwrap()})), nil
|
|
}
|
|
|
|
func (c *Client) listenPacket(ctx context.Context, destination M.Socksaddr) (N.PacketConn, error) {
|
|
if c.http3Available() {
|
|
packetConn, err := c.http3.ListenPacket(ctx, destination)
|
|
if err == nil {
|
|
c.clearHTTP3Broken()
|
|
return packetConn, nil
|
|
}
|
|
if c.disableVersionFallback || !errors.Is(err, ErrHTTP3Unavailable) {
|
|
return nil, err
|
|
}
|
|
c.markHTTP3Broken()
|
|
}
|
|
if c.tlsDialer != nil && !c.http2Unsupported.Load() && !c.http2ConnectUDPUnsupported.Load() {
|
|
clientConn, conn, err := c.acquireHTTP2(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if clientConn != nil {
|
|
packetConn, connectErr := c.connectUDPHTTP2(ctx, clientConn, destination)
|
|
if connectErr == nil {
|
|
return packetConn, nil
|
|
}
|
|
if !errors.Is(connectErr, errExtendedConnectUnsupported) || c.disableVersionFallback {
|
|
return nil, connectErr
|
|
}
|
|
c.http2ConnectUDPUnsupported.Store(true)
|
|
} else if c.disableVersionFallback {
|
|
conn.Close()
|
|
return nil, ErrHTTP2Unsupported
|
|
} else {
|
|
return c.connectUDPHTTP1AndClose(ctx, conn, destination)
|
|
}
|
|
}
|
|
conn, err := c.http1Dialer.DialContext(ctx, N.NetworkTCP, c.server)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return c.connectUDPHTTP1AndClose(ctx, conn, destination)
|
|
}
|
|
|
|
func (c *Client) connectUDPHTTP1AndClose(ctx context.Context, conn net.Conn, destination M.Socksaddr) (N.PacketConn, error) {
|
|
packetConn, err := c.connectUDPHTTP1(ctx, conn, destination)
|
|
if err != nil {
|
|
conn.Close()
|
|
if ctx.Err() != nil {
|
|
return nil, ctx.Err()
|
|
}
|
|
return nil, err
|
|
}
|
|
return packetConn, nil
|
|
}
|
|
|
|
var _ N.Dialer = (*Client)(nil)
|
|
|
|
type boundPacketConn struct {
|
|
N.PacketConn
|
|
destination M.Socksaddr
|
|
}
|
|
|
|
func (c *boundPacketConn) WritePacket(buffer *buf.Buffer, destination M.Socksaddr) error {
|
|
if destination.Unwrap() != c.destination {
|
|
buffer.Release()
|
|
return E.New("connect-udp: destination mismatch: ", destination)
|
|
}
|
|
return c.PacketConn.WritePacket(buffer, destination)
|
|
}
|
|
|
|
func (c *boundPacketConn) NeedAdditionalReadDeadline() bool {
|
|
return true
|
|
}
|
|
|
|
func (c *boundPacketConn) Upstream() any {
|
|
return c.PacketConn
|
|
}
|