package outbound import ( "context" "net" "github.com/sagernet/sing-box/adapter" C "github.com/sagernet/sing-box/constant" "github.com/sagernet/sing-box/log" "github.com/sagernet/sing-box/option" E "github.com/sagernet/sing/common/exceptions" M "github.com/sagernet/sing/common/metadata" N "github.com/sagernet/sing/common/network" ) var ( _ adapter.Outbound = (*Selector)(nil) _ adapter.OutboundGroup = (*Selector)(nil) ) type Selector struct { myOutboundAdapter tags []string defaultTag string outbounds map[string]adapter.Outbound selected adapter.Outbound } func NewSelector(router adapter.Router, logger log.ContextLogger, tag string, options option.SelectorOutboundOptions) (*Selector, error) { outbound := &Selector{ myOutboundAdapter: myOutboundAdapter{ protocol: C.TypeSelector, router: router, logger: logger, tag: tag, }, tags: options.Outbounds, defaultTag: options.Default, outbounds: make(map[string]adapter.Outbound), } if len(outbound.tags) == 0 { return nil, E.New("missing tags") } return outbound, nil } func (s *Selector) Network() []string { if s.selected == nil { return []string{N.NetworkTCP, N.NetworkUDP} } return s.selected.Network() } func (s *Selector) Start() error { for i, tag := range s.tags { detour, loaded := s.router.Outbound(tag) if !loaded { return E.New("outbound ", i, " not found: ", tag) } s.outbounds[tag] = detour } if s.defaultTag != "" { detour, loaded := s.outbounds[s.defaultTag] if !loaded { return E.New("default outbound not found: ", s.defaultTag) } s.selected = detour } else { s.selected = s.outbounds[s.tags[0]] } return nil } func (s *Selector) Now() string { return s.selected.Tag() } func (s *Selector) All() []string { return s.tags } func (s *Selector) SelectOutbound(tag string) bool { detour, loaded := s.outbounds[tag] if !loaded { return false } s.selected = detour return true } func (s *Selector) DialContext(ctx context.Context, network string, destination M.Socksaddr) (net.Conn, error) { return s.selected.DialContext(ctx, network, destination) } func (s *Selector) ListenPacket(ctx context.Context, destination M.Socksaddr) (net.PacketConn, error) { return s.selected.ListenPacket(ctx, destination) } func (s *Selector) NewConnection(ctx context.Context, conn net.Conn, metadata adapter.InboundContext) error { return s.selected.NewConnection(ctx, conn, metadata) } func (s *Selector) NewPacketConnection(ctx context.Context, conn N.PacketConn, metadata adapter.InboundContext) error { return s.selected.NewPacketConnection(ctx, conn, metadata) } func RealTag(detour adapter.Outbound) string { if group, isGroup := detour.(adapter.OutboundGroup); isGroup { return group.Now() } return detour.Tag() }