2019-12-08 04:17:24 +00:00
|
|
|
package outbound
|
2019-10-09 10:46:23 +00:00
|
|
|
|
|
|
|
import (
|
2019-10-12 15:55:39 +00:00
|
|
|
"context"
|
2019-10-09 10:46:23 +00:00
|
|
|
"fmt"
|
|
|
|
"net"
|
|
|
|
"strconv"
|
|
|
|
|
|
|
|
"github.com/Dreamacro/clash/common/structure"
|
2020-02-15 13:42:46 +00:00
|
|
|
"github.com/Dreamacro/clash/component/dialer"
|
2019-10-09 10:46:23 +00:00
|
|
|
C "github.com/Dreamacro/clash/constant"
|
2021-05-13 14:18:49 +00:00
|
|
|
obfs "github.com/Dreamacro/clash/transport/simple-obfs"
|
|
|
|
"github.com/Dreamacro/clash/transport/snell"
|
2019-10-09 10:46:23 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type Snell struct {
|
|
|
|
*Base
|
|
|
|
psk []byte
|
2020-09-20 16:33:13 +00:00
|
|
|
pool *snell.Pool
|
2019-10-09 10:46:23 +00:00
|
|
|
obfsOption *simpleObfsOption
|
2020-09-20 16:33:13 +00:00
|
|
|
version int
|
2019-10-09 10:46:23 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
type SnellOption struct {
|
2021-11-07 08:48:51 +00:00
|
|
|
BasicOption
|
2022-03-16 04:10:13 +00:00
|
|
|
Name string `proxy:"name"`
|
|
|
|
Server string `proxy:"server"`
|
|
|
|
Port int `proxy:"port"`
|
|
|
|
Psk string `proxy:"psk"`
|
|
|
|
UDP bool `proxy:"udp,omitempty"`
|
|
|
|
Version int `proxy:"version,omitempty"`
|
|
|
|
ObfsOpts map[string]any `proxy:"obfs-opts,omitempty"`
|
2019-10-09 10:46:23 +00:00
|
|
|
}
|
|
|
|
|
2020-09-20 16:33:13 +00:00
|
|
|
type streamOption struct {
|
|
|
|
psk []byte
|
|
|
|
version int
|
|
|
|
addr string
|
|
|
|
obfsOption *simpleObfsOption
|
|
|
|
}
|
|
|
|
|
|
|
|
func streamConn(c net.Conn, option streamOption) *snell.Snell {
|
|
|
|
switch option.obfsOption.Mode {
|
2019-10-09 10:46:23 +00:00
|
|
|
case "tls":
|
2020-09-20 16:33:13 +00:00
|
|
|
c = obfs.NewTLSObfs(c, option.obfsOption.Host)
|
2019-10-09 10:46:23 +00:00
|
|
|
case "http":
|
2020-09-20 16:33:13 +00:00
|
|
|
_, port, _ := net.SplitHostPort(option.addr)
|
|
|
|
c = obfs.NewHTTPObfs(c, option.obfsOption.Host, port)
|
2019-10-09 10:46:23 +00:00
|
|
|
}
|
2020-09-20 16:33:13 +00:00
|
|
|
return snell.StreamConn(c, option.psk, option.version)
|
|
|
|
}
|
|
|
|
|
2021-04-29 03:23:14 +00:00
|
|
|
// StreamConn implements C.ProxyAdapter
|
2020-09-20 16:33:13 +00:00
|
|
|
func (s *Snell) StreamConn(c net.Conn, metadata *C.Metadata) (net.Conn, error) {
|
|
|
|
c = streamConn(c, streamOption{s.psk, s.version, s.addr, s.obfsOption})
|
2022-01-01 17:09:29 +00:00
|
|
|
port, _ := strconv.ParseUint(metadata.DstPort, 10, 16)
|
2020-09-20 16:33:13 +00:00
|
|
|
err := snell.WriteHeader(c, metadata.String(), uint(port), s.version)
|
2020-03-21 15:46:49 +00:00
|
|
|
return c, err
|
|
|
|
}
|
|
|
|
|
2021-04-29 03:23:14 +00:00
|
|
|
// DialContext implements C.ProxyAdapter
|
2021-11-07 08:48:51 +00:00
|
|
|
func (s *Snell) DialContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (_ C.Conn, err error) {
|
|
|
|
if s.version == snell.Version2 && len(opts) == 0 {
|
2020-09-20 16:33:13 +00:00
|
|
|
c, err := s.pool.Get()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2021-11-07 16:31:08 +00:00
|
|
|
port, _ := strconv.ParseUint(metadata.DstPort, 10, 16)
|
2021-03-22 15:26:20 +00:00
|
|
|
if err = snell.WriteHeader(c, metadata.String(), uint(port), s.version); err != nil {
|
|
|
|
c.Close()
|
|
|
|
return nil, err
|
|
|
|
}
|
2020-09-20 16:33:13 +00:00
|
|
|
return NewConn(c, s), err
|
|
|
|
}
|
|
|
|
|
2021-11-07 08:48:51 +00:00
|
|
|
c, err := dialer.DialContext(ctx, "tcp", s.addr, s.Base.DialOptions(opts...)...)
|
2020-03-21 15:46:49 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, fmt.Errorf("%s connect error: %w", s.addr, err)
|
|
|
|
}
|
|
|
|
tcpKeepAlive(c)
|
|
|
|
|
2021-03-22 15:26:20 +00:00
|
|
|
defer safeConnClose(c, err)
|
|
|
|
|
2020-03-21 15:46:49 +00:00
|
|
|
c, err = s.StreamConn(c, metadata)
|
|
|
|
return NewConn(c, s), err
|
2019-10-09 10:46:23 +00:00
|
|
|
}
|
|
|
|
|
2022-01-10 12:24:20 +00:00
|
|
|
// ListenPacketContext implements C.ProxyAdapter
|
|
|
|
func (s *Snell) ListenPacketContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (C.PacketConn, error) {
|
|
|
|
c, err := dialer.DialContext(ctx, "tcp", s.addr, s.Base.DialOptions(opts...)...)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
tcpKeepAlive(c)
|
|
|
|
c = streamConn(c, streamOption{s.psk, s.version, s.addr, s.obfsOption})
|
|
|
|
|
|
|
|
err = snell.WriteUDPHeader(c, s.version)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
pc := snell.PacketConn(c)
|
|
|
|
return newPacketConn(pc, s), nil
|
|
|
|
}
|
|
|
|
|
2019-10-09 10:46:23 +00:00
|
|
|
func NewSnell(option SnellOption) (*Snell, error) {
|
2020-03-21 15:46:49 +00:00
|
|
|
addr := net.JoinHostPort(option.Server, strconv.Itoa(option.Port))
|
2019-10-09 10:46:23 +00:00
|
|
|
psk := []byte(option.Psk)
|
|
|
|
|
|
|
|
decoder := structure.NewDecoder(structure.Option{TagName: "obfs", WeaklyTypedInput: true})
|
|
|
|
obfsOption := &simpleObfsOption{Host: "bing.com"}
|
|
|
|
if err := decoder.Decode(option.ObfsOpts, obfsOption); err != nil {
|
2020-03-21 15:46:49 +00:00
|
|
|
return nil, fmt.Errorf("snell %s initialize obfs error: %w", addr, err)
|
2019-10-09 10:46:23 +00:00
|
|
|
}
|
|
|
|
|
2020-12-24 05:47:56 +00:00
|
|
|
switch obfsOption.Mode {
|
|
|
|
case "tls", "http", "":
|
|
|
|
break
|
|
|
|
default:
|
2020-03-21 15:46:49 +00:00
|
|
|
return nil, fmt.Errorf("snell %s obfs mode error: %s", addr, obfsOption.Mode)
|
2019-10-09 10:46:23 +00:00
|
|
|
}
|
|
|
|
|
2020-09-20 16:33:13 +00:00
|
|
|
// backward compatible
|
|
|
|
if option.Version == 0 {
|
|
|
|
option.Version = snell.DefaultSnellVersion
|
|
|
|
}
|
2022-01-10 12:24:20 +00:00
|
|
|
switch option.Version {
|
|
|
|
case snell.Version1, snell.Version2:
|
|
|
|
if option.UDP {
|
|
|
|
return nil, fmt.Errorf("snell version %d not support UDP", option.Version)
|
|
|
|
}
|
|
|
|
case snell.Version3:
|
|
|
|
default:
|
2020-09-20 16:33:13 +00:00
|
|
|
return nil, fmt.Errorf("snell version error: %d", option.Version)
|
|
|
|
}
|
|
|
|
|
|
|
|
s := &Snell{
|
2019-10-09 10:46:23 +00:00
|
|
|
Base: &Base{
|
2021-11-07 08:48:51 +00:00
|
|
|
name: option.Name,
|
|
|
|
addr: addr,
|
|
|
|
tp: C.Snell,
|
2022-01-10 12:24:20 +00:00
|
|
|
udp: option.UDP,
|
2021-11-07 08:48:51 +00:00
|
|
|
iface: option.Interface,
|
2019-10-09 10:46:23 +00:00
|
|
|
},
|
|
|
|
psk: psk,
|
|
|
|
obfsOption: obfsOption,
|
2020-09-20 16:33:13 +00:00
|
|
|
version: option.Version,
|
|
|
|
}
|
|
|
|
|
|
|
|
if option.Version == snell.Version2 {
|
|
|
|
s.pool = snell.NewPool(func(ctx context.Context) (*snell.Snell, error) {
|
2021-11-07 08:48:51 +00:00
|
|
|
c, err := dialer.DialContext(ctx, "tcp", addr, s.Base.DialOptions()...)
|
2020-09-20 16:33:13 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
tcpKeepAlive(c)
|
|
|
|
return streamConn(c, streamOption{psk, option.Version, addr, obfsOption}), nil
|
|
|
|
})
|
|
|
|
}
|
|
|
|
return s, nil
|
2019-10-09 10:46:23 +00:00
|
|
|
}
|