|
@@ -0,0 +1,225 @@
|
|
|
+package outbound
|
|
|
+
|
|
|
+import (
|
|
|
+ "context"
|
|
|
+ "net"
|
|
|
+ "time"
|
|
|
+
|
|
|
+ "github.com/sagernet/sing-box/adapter"
|
|
|
+ "github.com/sagernet/sing-box/common/urltest"
|
|
|
+ C "github.com/sagernet/sing-box/constant"
|
|
|
+ "github.com/sagernet/sing-box/log"
|
|
|
+ "github.com/sagernet/sing-box/option"
|
|
|
+ "github.com/sagernet/sing/common"
|
|
|
+ "github.com/sagernet/sing/common/batch"
|
|
|
+ E "github.com/sagernet/sing/common/exceptions"
|
|
|
+ M "github.com/sagernet/sing/common/metadata"
|
|
|
+ N "github.com/sagernet/sing/common/network"
|
|
|
+)
|
|
|
+
|
|
|
+var (
|
|
|
+ _ adapter.Outbound = (*URLTest)(nil)
|
|
|
+ _ adapter.OutboundGroup = (*URLTest)(nil)
|
|
|
+)
|
|
|
+
|
|
|
+type URLTest struct {
|
|
|
+ myOutboundAdapter
|
|
|
+ router adapter.Router
|
|
|
+ tags []string
|
|
|
+ link string
|
|
|
+ interval time.Duration
|
|
|
+ tolerance uint16
|
|
|
+ group *URLTestGroup
|
|
|
+}
|
|
|
+
|
|
|
+func NewURLTest(router adapter.Router, logger log.ContextLogger, tag string, options option.URLTestOutboundOptions) (*URLTest, error) {
|
|
|
+ outbound := &URLTest{
|
|
|
+ myOutboundAdapter: myOutboundAdapter{
|
|
|
+ protocol: C.TypeURLTest,
|
|
|
+ logger: logger,
|
|
|
+ tag: tag,
|
|
|
+ },
|
|
|
+ router: router,
|
|
|
+ tags: options.Outbounds,
|
|
|
+ link: options.URL,
|
|
|
+ interval: time.Duration(options.Interval),
|
|
|
+ tolerance: options.Tolerance,
|
|
|
+ }
|
|
|
+ if len(outbound.tags) == 0 {
|
|
|
+ return nil, E.New("missing tags")
|
|
|
+ }
|
|
|
+ return outbound, nil
|
|
|
+}
|
|
|
+
|
|
|
+func (s *URLTest) Network() []string {
|
|
|
+ if s.group == nil {
|
|
|
+ return []string{C.NetworkTCP, C.NetworkUDP}
|
|
|
+ }
|
|
|
+ return s.group.Select().Network()
|
|
|
+}
|
|
|
+
|
|
|
+func (s *URLTest) Start() error {
|
|
|
+ outbounds := make([]adapter.Outbound, 0, len(s.tags))
|
|
|
+ for i, tag := range s.tags {
|
|
|
+ detour, loaded := s.router.Outbound(tag)
|
|
|
+ if !loaded {
|
|
|
+ return E.New("outbound ", i, " not found: ", tag)
|
|
|
+ }
|
|
|
+ outbounds = append(outbounds, detour)
|
|
|
+ }
|
|
|
+ s.group = NewURLTestGroup(s.router, s.logger, outbounds, s.link, s.interval, s.tolerance)
|
|
|
+ return s.group.Start()
|
|
|
+}
|
|
|
+
|
|
|
+func (s URLTest) Close() error {
|
|
|
+ return common.Close(
|
|
|
+ common.PtrOrNil(s.group),
|
|
|
+ )
|
|
|
+}
|
|
|
+
|
|
|
+func (s *URLTest) Now() string {
|
|
|
+ return s.group.Select().Tag()
|
|
|
+}
|
|
|
+
|
|
|
+func (s *URLTest) All() []string {
|
|
|
+ return s.tags
|
|
|
+}
|
|
|
+
|
|
|
+func (s *URLTest) DialContext(ctx context.Context, network string, destination M.Socksaddr) (net.Conn, error) {
|
|
|
+ return s.group.Select().DialContext(ctx, network, destination)
|
|
|
+}
|
|
|
+
|
|
|
+func (s *URLTest) ListenPacket(ctx context.Context, destination M.Socksaddr) (net.PacketConn, error) {
|
|
|
+ return s.group.Select().ListenPacket(ctx, destination)
|
|
|
+}
|
|
|
+
|
|
|
+func (s *URLTest) NewConnection(ctx context.Context, conn net.Conn, metadata adapter.InboundContext) error {
|
|
|
+ return s.group.Select().NewConnection(ctx, conn, metadata)
|
|
|
+}
|
|
|
+
|
|
|
+func (s *URLTest) NewPacketConnection(ctx context.Context, conn N.PacketConn, metadata adapter.InboundContext) error {
|
|
|
+ return s.group.Select().NewPacketConnection(ctx, conn, metadata)
|
|
|
+}
|
|
|
+
|
|
|
+type URLTestGroup struct {
|
|
|
+ router adapter.Router
|
|
|
+ logger log.Logger
|
|
|
+ outbounds []adapter.Outbound
|
|
|
+ link string
|
|
|
+ interval time.Duration
|
|
|
+ tolerance uint16
|
|
|
+
|
|
|
+ ticker *time.Ticker
|
|
|
+ close chan struct{}
|
|
|
+}
|
|
|
+
|
|
|
+func NewURLTestGroup(router adapter.Router, logger log.Logger, outbounds []adapter.Outbound, link string, interval time.Duration, tolerance uint16) *URLTestGroup {
|
|
|
+ if link == "" {
|
|
|
+ //goland:noinspection HttpUrlsUsage
|
|
|
+ link = "http://www.gstatic.com/generate_204"
|
|
|
+ }
|
|
|
+ if interval == 0 {
|
|
|
+ interval = C.DefaultURLTestInterval
|
|
|
+ }
|
|
|
+ if tolerance == 0 {
|
|
|
+ tolerance = 50
|
|
|
+ }
|
|
|
+ return &URLTestGroup{
|
|
|
+ router: router,
|
|
|
+ logger: logger,
|
|
|
+ outbounds: outbounds,
|
|
|
+ link: link,
|
|
|
+ interval: interval,
|
|
|
+ tolerance: tolerance,
|
|
|
+ close: make(chan struct{}),
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func (g *URLTestGroup) Start() error {
|
|
|
+ g.ticker = time.NewTicker(g.interval)
|
|
|
+ go g.loopCheck()
|
|
|
+ return nil
|
|
|
+}
|
|
|
+
|
|
|
+func (g *URLTestGroup) Close() error {
|
|
|
+ g.ticker.Stop()
|
|
|
+ close(g.close)
|
|
|
+ return nil
|
|
|
+}
|
|
|
+
|
|
|
+func (g *URLTestGroup) Select() adapter.Outbound {
|
|
|
+ var minDelay uint16
|
|
|
+ var minTime time.Time
|
|
|
+ var minOutbound adapter.Outbound
|
|
|
+ for _, detour := range g.outbounds {
|
|
|
+ history := g.router.URLTestHistoryStorage(false).LoadURLTestHistory(RealTag(detour))
|
|
|
+ if history == nil {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ if minDelay == 0 || minDelay > history.Delay+g.tolerance || minDelay > history.Delay-g.tolerance && minTime.Before(history.Time) {
|
|
|
+ minDelay = history.Delay
|
|
|
+ minTime = history.Time
|
|
|
+ minOutbound = detour
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if minOutbound == nil {
|
|
|
+ minOutbound = g.outbounds[0]
|
|
|
+ }
|
|
|
+ return minOutbound
|
|
|
+}
|
|
|
+
|
|
|
+func (g *URLTestGroup) loopCheck() {
|
|
|
+ go g.checkOutbounds()
|
|
|
+ for {
|
|
|
+ select {
|
|
|
+ case <-g.close:
|
|
|
+ return
|
|
|
+ case <-g.ticker.C:
|
|
|
+ g.checkOutbounds()
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func (g *URLTestGroup) checkOutbounds() {
|
|
|
+ b, _ := batch.New(context.Background(), batch.WithConcurrencyNum[any](10))
|
|
|
+ checked := make(map[string]bool)
|
|
|
+ for _, detour := range g.outbounds {
|
|
|
+ realTag := RealTag(detour)
|
|
|
+ if checked[realTag] {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ history := g.router.URLTestHistoryStorage(false).LoadURLTestHistory(realTag)
|
|
|
+ if history != nil && time.Now().Sub(history.Time) < g.interval {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ checked[realTag] = true
|
|
|
+ p, loaded := g.router.Outbound(realTag)
|
|
|
+ if !loaded {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ b.Go(realTag, func() (any, error) {
|
|
|
+ ctx, cancel := context.WithTimeout(context.Background(), C.URLTestTimeout)
|
|
|
+ defer cancel()
|
|
|
+ t, err := urltest.URLTest(ctx, g.link, p)
|
|
|
+ if err != nil {
|
|
|
+ g.logger.Debug("outbound ", detour.Tag(), " unavailable: ", err)
|
|
|
+ g.router.URLTestHistoryStorage(true).DeleteURLTestHistory(realTag)
|
|
|
+ } else {
|
|
|
+ g.logger.Debug("outbound ", detour.Tag(), " available: ", t, "ms")
|
|
|
+ g.router.URLTestHistoryStorage(true).StoreURLTestHistory(realTag, &urltest.History{
|
|
|
+ Time: time.Now(),
|
|
|
+ Delay: t,
|
|
|
+ })
|
|
|
+ }
|
|
|
+ return nil, nil
|
|
|
+ })
|
|
|
+ }
|
|
|
+ b.Wait()
|
|
|
+}
|
|
|
+
|
|
|
+func RealTag(detour adapter.Outbound) string {
|
|
|
+ if group, isGroup := detour.(adapter.OutboundGroup); isGroup {
|
|
|
+ return group.Now()
|
|
|
+ }
|
|
|
+ return detour.Tag()
|
|
|
+}
|