package shadowsocks_2022 import ( "context" "strconv" "strings" "github.com/sagernet/sing-shadowsocks/shadowaead_2022" C "github.com/sagernet/sing/common" A "github.com/sagernet/sing/common/auth" B "github.com/sagernet/sing/common/buf" "github.com/sagernet/sing/common/bufio" E "github.com/sagernet/sing/common/exceptions" M "github.com/sagernet/sing/common/metadata" N "github.com/sagernet/sing/common/network" "github.com/xtls/xray-core/common" "github.com/xtls/xray-core/common/buf" "github.com/xtls/xray-core/common/log" "github.com/xtls/xray-core/common/net" "github.com/xtls/xray-core/common/protocol" "github.com/xtls/xray-core/common/session" "github.com/xtls/xray-core/common/uuid" "github.com/xtls/xray-core/features/routing" "github.com/xtls/xray-core/transport/internet/stat" ) func init() { common.Must(common.RegisterConfig((*RelayServerConfig)(nil), func(ctx context.Context, config interface{}) (interface{}, error) { return NewRelayServer(ctx, config.(*RelayServerConfig)) })) } type RelayInbound struct { networks []net.Network destinations []*RelayDestination service *shadowaead_2022.RelayService[int] } func NewRelayServer(ctx context.Context, config *RelayServerConfig) (*RelayInbound, error) { networks := config.Network if len(networks) == 0 { networks = []net.Network{ net.Network_TCP, net.Network_UDP, } } inbound := &RelayInbound{ networks: networks, destinations: config.Destinations, } if !C.Contains(shadowaead_2022.List, config.Method) || !strings.Contains(config.Method, "aes") { return nil, newError("unsupported method ", config.Method) } service, err := shadowaead_2022.NewRelayServiceWithPassword[int](config.Method, config.Key, 500, inbound) if err != nil { return nil, newError("create service").Base(err) } for i, destination := range config.Destinations { if destination.Email == "" { u := uuid.New() destination.Email = "unnamed-destination-" + strconv.Itoa(i) + "-" + u.String() } } err = service.UpdateUsersWithPasswords( C.MapIndexed(config.Destinations, func(index int, it *RelayDestination) int { return index }), C.Map(config.Destinations, func(it *RelayDestination) string { return it.Key }), C.Map(config.Destinations, func(it *RelayDestination) M.Socksaddr { return toSocksaddr(net.Destination{ Address: it.Address.AsAddress(), Port: net.Port(it.Port), }) }), ) if err != nil { return nil, newError("create service").Base(err) } inbound.service = service return inbound, nil } func (i *RelayInbound) Network() []net.Network { return i.networks } func (i *RelayInbound) Process(ctx context.Context, network net.Network, connection stat.Connection, dispatcher routing.Dispatcher) error { inbound := session.InboundFromContext(ctx) var metadata M.Metadata if inbound.Source.IsValid() { metadata.Source = M.ParseSocksaddr(inbound.Source.NetAddr()) } ctx = session.ContextWithDispatcher(ctx, dispatcher) if network == net.Network_TCP { return returnError(i.service.NewConnection(ctx, connection, metadata)) } else { reader := buf.NewReader(connection) pc := &natPacketConn{connection} for { mb, err := reader.ReadMultiBuffer() if err != nil { buf.ReleaseMulti(mb) return returnError(err) } for _, buffer := range mb { packet := B.As(buffer.Bytes()).ToOwned() err = i.service.NewPacket(ctx, pc, packet, metadata) if err != nil { packet.Release() buf.ReleaseMulti(mb) return err } buffer.Release() } } } } func (i *RelayInbound) NewConnection(ctx context.Context, conn net.Conn, metadata M.Metadata) error { inbound := session.InboundFromContext(ctx) user, _ := A.UserFromContext[User](ctx) inbound.User = &protocol.MemoryUser{ Email: user.Email, Level: uint32(user.Level), } ctx = log.ContextWithAccessMessage(ctx, &log.AccessMessage{ From: metadata.Source, To: metadata.Destination, Status: log.AccessAccepted, Email: user.Email, }) newError("tunnelling request to tcp:", metadata.Destination).WriteToLog(session.ExportIDToError(ctx)) dispatcher := session.DispatcherFromContext(ctx) link, err := dispatcher.Dispatch(ctx, toDestination(metadata.Destination, net.Network_TCP)) if err != nil { return err } outConn := &pipeConnWrapper{ &buf.BufferedReader{Reader: link.Reader}, link.Writer, conn, } return bufio.CopyConn(ctx, conn, outConn) } func (i *RelayInbound) NewPacketConnection(ctx context.Context, conn N.PacketConn, metadata M.Metadata) error { inbound := session.InboundFromContext(ctx) user, _ := A.UserFromContext[User](ctx) inbound.User = &protocol.MemoryUser{ Email: user.Email, Level: uint32(user.Level), } ctx = log.ContextWithAccessMessage(ctx, &log.AccessMessage{ From: metadata.Source, To: metadata.Destination, Status: log.AccessAccepted, Email: user.Email, }) newError("tunnelling request to udp:", metadata.Destination).WriteToLog(session.ExportIDToError(ctx)) dispatcher := session.DispatcherFromContext(ctx) destination := toDestination(metadata.Destination, net.Network_UDP) link, err := dispatcher.Dispatch(ctx, destination) if err != nil { return err } outConn := &packetConnWrapper{ Reader: link.Reader, Writer: link.Writer, Dest: destination, } return bufio.CopyPacketConn(ctx, conn, outConn) } func (i *RelayInbound) NewError(ctx context.Context, err error) { if E.IsClosed(err) { return } newError(err).AtWarning().WriteToLog() }