Austin/mihomo
1
0
Fork 0
mirror of https://github.com/MetaCubeX/mihomo.git synced 2024-11-16 03:32:33 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: dual stack fallback

Browse Source
This commit is contained in:
Skyxim 2023-02-26 10:42:22 +08:00
parent cdd91f5132
commit 8df44a7c3d
1 changed files with 44 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

71
component/dialer/dialer.go
View File

@ -8,6 +8,7 @@ import (
"net/netip"
"strings"
"sync"
"time"
"github.com/Dreamacro/clash/component/resolver"
)
@ -19,6 +20,7 @@ var (
tcpConcurrent = false
ErrorInvalidedNetworkStack = errors.New("invalided network stack")
ErrorConnTimeout = errors.New("connect timeout")
fallbackTimeout = 300 * time.Millisecond
)
func applyOptions(options ...Option) *option {
@ -131,9 +133,10 @@ func serialDualStackDialContext(ctx context.Context, network, address string, op
if err != nil {
return nil, err
}
return dualStackDial(
func() (net.Conn, error) { return serialDialContext(ctx, network, ips, port, opt) },
func() (net.Conn, error) { return serialDialContext(ctx, network, ips, port, opt) },
return dualStackDialContext(
ctx,
func(ctx context.Context) (net.Conn, error) { return serialDialContext(ctx, network, ips, port, opt) },
func(ctx context.Context) (net.Conn, error) { return serialDialContext(ctx, network, ips, port, opt) },
opt.prefer == 4)
}
@ -159,9 +162,14 @@ func concurrentDualStackDialContext(ctx context.Context, network, address string
return parallelDialContext(ctx, network, ips, port, opt)
}
ipv4s, ipv6s := sortationAddr(ips)
return dualStackDial(
func() (net.Conn, error) { return parallelDialContext(ctx, network, ipv4s, port, opt) },
func() (net.Conn, error) { return parallelDialContext(ctx, network, ipv6s, port, opt) },
return dualStackDialContext(
ctx,
func(ctx context.Context) (net.Conn, error) {
return parallelDialContext(ctx, network, ipv4s, port, opt)
},
func(ctx context.Context) (net.Conn, error) {
return parallelDialContext(ctx, network, ipv6s, port, opt)
},
opt.prefer == 4)
}
@ -182,14 +190,19 @@ func NewDialer(options ...Option) Dialer {
return Dialer{Opt: *opt}
}
func dualStackDial(
ipv4DialFn func() (net.Conn, error),
ipv6DialFn func() (net.Conn, error),
func dualStackDialContext(
ctx context.Context,
ipv4DialFn func(ctx context.Context) (net.Conn, error),
ipv6DialFn func(ctx context.Context) (net.Conn, error),
preferIPv4 bool) (net.Conn, error) {
fallbackTimer := time.NewTimer(fallbackTimeout)
defer fallbackTimer.Stop()
fallbackCtx, fallbackCancel := context.WithCancel(ctx)
defer fallbackCancel()
results := make(chan dialResult)
returned := make(chan struct{})
defer close(returned)
racer := func(dial func() (net.Conn, error), isPrimary bool) {
racer := func(dial func(ctx context.Context) (net.Conn, error), isPrimary bool) {
result := dialResult{isPrimary: isPrimary}
defer func() {
select {
@ -200,29 +213,33 @@ func dualStackDial(
}
}
}()
result.Conn, result.error = dial()
result.Conn, result.error = dial(fallbackCtx)
}
go racer(ipv4DialFn, preferIPv4)
go racer(ipv6DialFn, !preferIPv4)
var fallbackErr dialResult
var primaryErr dialResult
for res := range results {
if res.error == nil {
if res.isPrimary {
return res.Conn, nil
var fallback dialResult
var err error
for {
select {
case <-ctx.Done():
if fallback.error == nil && fallback.Conn != nil {
return fallback.Conn, nil
}
fallbackErr = res
}
if res.isPrimary {
primaryErr = res
} else {
fallbackErr = res
return nil, fmt.Errorf("dual stack connect failed: %w", err)
case <-fallbackTimer.C:
if fallback.error == nil && fallback.Conn != nil {
return fallback.Conn, nil
}
case res := <-results:
if res.error == nil {
if res.isPrimary {
return res.Conn, nil
}
fallback = res
}
err = res.error
}
}
if primaryErr.error != nil {
return nil, primaryErr.error
}
return nil, fallbackErr.error
}
func parallelDialContext(ctx context.Context, network string, ips []netip.Addr, port string, opt *option) (net.Conn, error) {