2021-09-06 15:07:34 +00:00
|
|
|
package dialer
|
|
|
|
|
2022-08-28 05:41:19 +00:00
|
|
|
import (
|
2023-03-07 01:30:51 +00:00
|
|
|
"context"
|
|
|
|
"net"
|
|
|
|
|
2022-12-07 12:01:44 +00:00
|
|
|
"github.com/Dreamacro/clash/component/resolver"
|
|
|
|
|
2022-08-28 05:41:19 +00:00
|
|
|
"go.uber.org/atomic"
|
|
|
|
)
|
2021-09-06 15:07:34 +00:00
|
|
|
|
2021-11-07 08:48:51 +00:00
|
|
|
var (
|
2022-02-17 06:23:47 +00:00
|
|
|
DefaultOptions []Option
|
|
|
|
DefaultInterface = atomic.NewString("")
|
|
|
|
DefaultRoutingMark = atomic.NewInt32(0)
|
2021-11-07 08:48:51 +00:00
|
|
|
)
|
|
|
|
|
2023-03-07 01:30:51 +00:00
|
|
|
type NetDialer interface {
|
|
|
|
DialContext(ctx context.Context, network, address string) (net.Conn, error)
|
|
|
|
}
|
|
|
|
|
2021-11-07 08:48:51 +00:00
|
|
|
type option struct {
|
2021-09-06 15:07:34 +00:00
|
|
|
interfaceName string
|
|
|
|
addrReuse bool
|
2021-11-08 08:59:48 +00:00
|
|
|
routingMark int
|
2022-08-28 05:41:19 +00:00
|
|
|
network int
|
|
|
|
prefer int
|
2023-02-24 05:53:44 +00:00
|
|
|
tfo bool
|
2022-12-07 12:01:44 +00:00
|
|
|
resolver resolver.Resolver
|
2023-03-07 01:30:51 +00:00
|
|
|
netDialer NetDialer
|
2021-09-06 15:07:34 +00:00
|
|
|
}
|
|
|
|
|
2021-11-07 08:48:51 +00:00
|
|
|
type Option func(opt *option)
|
2021-09-06 15:07:34 +00:00
|
|
|
|
|
|
|
func WithInterface(name string) Option {
|
2021-11-07 08:48:51 +00:00
|
|
|
return func(opt *option) {
|
2021-09-06 15:07:34 +00:00
|
|
|
opt.interfaceName = name
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func WithAddrReuse(reuse bool) Option {
|
2021-11-07 08:48:51 +00:00
|
|
|
return func(opt *option) {
|
2021-09-06 15:07:34 +00:00
|
|
|
opt.addrReuse = reuse
|
|
|
|
}
|
|
|
|
}
|
2021-11-08 08:59:48 +00:00
|
|
|
|
|
|
|
func WithRoutingMark(mark int) Option {
|
|
|
|
return func(opt *option) {
|
|
|
|
opt.routingMark = mark
|
|
|
|
}
|
|
|
|
}
|
2022-03-27 16:44:13 +00:00
|
|
|
|
2022-12-07 12:01:44 +00:00
|
|
|
func WithResolver(r resolver.Resolver) Option {
|
2022-03-27 16:44:13 +00:00
|
|
|
return func(opt *option) {
|
2022-12-07 12:01:44 +00:00
|
|
|
opt.resolver = r
|
2022-03-27 16:44:13 +00:00
|
|
|
}
|
|
|
|
}
|
2022-08-28 05:41:19 +00:00
|
|
|
|
|
|
|
func WithPreferIPv4() Option {
|
|
|
|
return func(opt *option) {
|
|
|
|
opt.prefer = 4
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func WithPreferIPv6() Option {
|
|
|
|
return func(opt *option) {
|
|
|
|
opt.prefer = 6
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func WithOnlySingleStack(isIPv4 bool) Option {
|
|
|
|
return func(opt *option) {
|
|
|
|
if isIPv4 {
|
|
|
|
opt.network = 4
|
|
|
|
} else {
|
|
|
|
opt.network = 6
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2022-12-19 16:11:02 +00:00
|
|
|
|
2023-02-24 05:53:44 +00:00
|
|
|
func WithTFO(tfo bool) Option {
|
|
|
|
return func(opt *option) {
|
|
|
|
opt.tfo = tfo
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-03-07 01:30:51 +00:00
|
|
|
func WithNetDialer(netDialer NetDialer) Option {
|
|
|
|
return func(opt *option) {
|
|
|
|
opt.netDialer = netDialer
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-12-22 01:53:11 +00:00
|
|
|
func WithOption(o option) Option {
|
2022-12-19 16:11:02 +00:00
|
|
|
return func(opt *option) {
|
|
|
|
*opt = o
|
|
|
|
}
|
|
|
|
}
|