156 lines
3.3 KiB
Go
156 lines
3.3 KiB
Go
package outboundgroup
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/Dreamacro/clash/adapter/outbound"
|
|
"github.com/Dreamacro/clash/common/callback"
|
|
N "github.com/Dreamacro/clash/common/net"
|
|
"github.com/Dreamacro/clash/component/dialer"
|
|
C "github.com/Dreamacro/clash/constant"
|
|
"github.com/Dreamacro/clash/constant/provider"
|
|
)
|
|
|
|
type Fallback struct {
|
|
*GroupBase
|
|
disableUDP bool
|
|
testUrl string
|
|
selected string
|
|
}
|
|
|
|
func (f *Fallback) Now() string {
|
|
proxy := f.findAliveProxy(false)
|
|
return proxy.Name()
|
|
}
|
|
|
|
// DialContext implements C.ProxyAdapter
|
|
func (f *Fallback) DialContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (C.Conn, error) {
|
|
proxy := f.findAliveProxy(true)
|
|
c, err := proxy.DialContext(ctx, metadata, f.Base.DialOptions(opts...)...)
|
|
if err == nil {
|
|
c.AppendToChains(f)
|
|
} else {
|
|
f.onDialFailed(proxy.Type(), err)
|
|
}
|
|
|
|
if N.NeedHandshake(c) {
|
|
c = &callback.FirstWriteCallBackConn{
|
|
Conn: c,
|
|
Callback: func(err error) {
|
|
if err == nil {
|
|
f.onDialSuccess()
|
|
} else {
|
|
f.onDialFailed(proxy.Type(), err)
|
|
}
|
|
},
|
|
}
|
|
}
|
|
|
|
return c, err
|
|
}
|
|
|
|
// ListenPacketContext implements C.ProxyAdapter
|
|
func (f *Fallback) ListenPacketContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (C.PacketConn, error) {
|
|
proxy := f.findAliveProxy(true)
|
|
pc, err := proxy.ListenPacketContext(ctx, metadata, f.Base.DialOptions(opts...)...)
|
|
if err == nil {
|
|
pc.AppendToChains(f)
|
|
}
|
|
|
|
return pc, err
|
|
}
|
|
|
|
// SupportUDP implements C.ProxyAdapter
|
|
func (f *Fallback) SupportUDP() bool {
|
|
if f.disableUDP {
|
|
return false
|
|
}
|
|
|
|
proxy := f.findAliveProxy(false)
|
|
return proxy.SupportUDP()
|
|
}
|
|
|
|
// MarshalJSON implements C.ProxyAdapter
|
|
func (f *Fallback) MarshalJSON() ([]byte, error) {
|
|
all := []string{}
|
|
for _, proxy := range f.GetProxies(false) {
|
|
all = append(all, proxy.Name())
|
|
}
|
|
return json.Marshal(map[string]any{
|
|
"type": f.Type().String(),
|
|
"now": f.Now(),
|
|
"all": all,
|
|
})
|
|
}
|
|
|
|
// Unwrap implements C.ProxyAdapter
|
|
func (f *Fallback) Unwrap(metadata *C.Metadata, touch bool) C.Proxy {
|
|
proxy := f.findAliveProxy(touch)
|
|
return proxy
|
|
}
|
|
|
|
func (f *Fallback) findAliveProxy(touch bool) C.Proxy {
|
|
proxies := f.GetProxies(touch)
|
|
for _, proxy := range proxies {
|
|
if len(f.selected) == 0 {
|
|
if proxy.Alive() {
|
|
return proxy
|
|
}
|
|
} else {
|
|
if proxy.Name() == f.selected {
|
|
if proxy.Alive() {
|
|
return proxy
|
|
} else {
|
|
f.selected = ""
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return proxies[0]
|
|
}
|
|
|
|
func (f *Fallback) Set(name string) error {
|
|
var p C.Proxy
|
|
for _, proxy := range f.GetProxies(false) {
|
|
if proxy.Name() == name {
|
|
p = proxy
|
|
break
|
|
}
|
|
}
|
|
|
|
if p == nil {
|
|
return errors.New("proxy not exist")
|
|
}
|
|
|
|
f.selected = name
|
|
if !p.Alive() {
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond*time.Duration(5000))
|
|
defer cancel()
|
|
_, _ = p.URLTest(ctx, f.testUrl)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func NewFallback(option *GroupCommonOption, providers []provider.ProxyProvider) *Fallback {
|
|
return &Fallback{
|
|
GroupBase: NewGroupBase(GroupBaseOption{
|
|
outbound.BaseOption{
|
|
Name: option.Name,
|
|
Type: C.Fallback,
|
|
Interface: option.Interface,
|
|
RoutingMark: option.RoutingMark,
|
|
},
|
|
option.Filter,
|
|
option.ExcludeFilter,
|
|
option.ExcludeType,
|
|
providers,
|
|
}),
|
|
disableUDP: option.DisableUDP,
|
|
testUrl: option.URL,
|
|
}
|
|
}
|