feat: support tuicV5

listener/tuic/server.go

@@ -6,8 +6,6 @@ import (
 	"strings"
 	"time"
 
-	"github.com/metacubex/quic-go"
-
 	"github.com/Dreamacro/clash/adapter/inbound"
 	CN "github.com/Dreamacro/clash/common/net"
 	"github.com/Dreamacro/clash/common/sockopt"
@@ -16,6 +14,10 @@ import (
 	"github.com/Dreamacro/clash/log"
 	"github.com/Dreamacro/clash/transport/socks5"
 	"github.com/Dreamacro/clash/transport/tuic"
+
+	"github.com/gofrs/uuid/v5"
+	"github.com/metacubex/quic-go"
+	"golang.org/x/exp/slices"
 )
 
 const ServerMaxIncomingStreams = (1 << 32) - 1
@@ -24,7 +26,7 @@ type Listener struct {
 	closed       bool
 	config       LC.TuicServer
 	udpListeners []net.PacketConn
-	servers      []*tuic.Server
+	servers      []tuic.Server
 }
 
 func New(config LC.TuicServer, tcpIn chan<- C.ConnContext, udpIn chan<- C.PacketAdapter, additions ...inbound.Addition) (*Listener, error) {
@@ -59,39 +61,77 @@ func New(config LC.TuicServer, tcpIn chan<- C.ConnContext, udpIn chan<- C.Packet
 	quicConfig.InitialConnectionReceiveWindow = tuic.DefaultConnectionReceiveWindow / 10
 	quicConfig.MaxConnectionReceiveWindow = tuic.DefaultConnectionReceiveWindow
 
+	packetOverHead := tuic.PacketOverHeadV4
+	if len(config.Token) == 0 {
+		packetOverHead = tuic.PacketOverHeadV5
+	}
+
 	if config.MaxUdpRelayPacketSize == 0 {
 		config.MaxUdpRelayPacketSize = 1500
 	}
-	maxDatagramFrameSize := config.MaxUdpRelayPacketSize + tuic.PacketOverHead
+	maxDatagramFrameSize := config.MaxUdpRelayPacketSize + packetOverHead
 	if maxDatagramFrameSize > 1400 {
 		maxDatagramFrameSize = 1400
 	}
-	config.MaxUdpRelayPacketSize = maxDatagramFrameSize - tuic.PacketOverHead
+	config.MaxUdpRelayPacketSize = maxDatagramFrameSize - packetOverHead
 	quicConfig.MaxDatagramFrameSize = int64(maxDatagramFrameSize)
 
-	tokens := make([][32]byte, len(config.Token))
-	for i, token := range config.Token {
-		tokens[i] = tuic.GenTKN(token)
+	handleTcpFn := func(conn net.Conn, addr socks5.Addr, _additions ...inbound.Addition) error {
+		newAdditions := additions
+		if len(_additions) > 0 {
+			newAdditions = slices.Clone(additions)
+			newAdditions = append(newAdditions, _additions...)
+		}
+		tcpIn <- inbound.NewSocket(addr, conn, C.TUIC, newAdditions...)
+		return nil
+	}
+	handleUdpFn := func(addr socks5.Addr, packet C.UDPPacket, _additions ...inbound.Addition) error {
+		newAdditions := additions
+		if len(_additions) > 0 {
+			newAdditions = slices.Clone(additions)
+			newAdditions = append(newAdditions, _additions...)
+		}
+		select {
+		case udpIn <- inbound.NewPacket(addr, packet, C.TUIC, newAdditions...):
+		default:
+		}
+		return nil
 	}
 
-	option := &tuic.ServerOption{
-		HandleTcpFn: func(conn net.Conn, addr socks5.Addr) error {
-			tcpIn <- inbound.NewSocket(addr, conn, C.TUIC, additions...)
-			return nil
-		},
-		HandleUdpFn: func(addr socks5.Addr, packet C.UDPPacket) error {
-			select {
-			case udpIn <- inbound.NewPacket(addr, packet, C.TUIC, additions...):
-			default:
-			}
-			return nil
-		},
-		TlsConfig:             tlsConfig,
-		QuicConfig:            quicConfig,
-		Tokens:                tokens,
-		CongestionController:  config.CongestionController,
-		AuthenticationTimeout: time.Duration(config.AuthenticationTimeout) * time.Millisecond,
-		MaxUdpRelayPacketSize: config.MaxUdpRelayPacketSize,
+	var optionV4 *tuic.ServerOptionV4
+	var optionV5 *tuic.ServerOptionV5
+	if len(config.Token) > 0 {
+		tokens := make([][32]byte, len(config.Token))
+		for i, token := range config.Token {
+			tokens[i] = tuic.GenTKN(token)
+		}
+
+		optionV4 = &tuic.ServerOptionV4{
+			HandleTcpFn:           handleTcpFn,
+			HandleUdpFn:           handleUdpFn,
+			TlsConfig:             tlsConfig,
+			QuicConfig:            quicConfig,
+			Tokens:                tokens,
+			CongestionController:  config.CongestionController,
+			AuthenticationTimeout: time.Duration(config.AuthenticationTimeout) * time.Millisecond,
+			MaxUdpRelayPacketSize: config.MaxUdpRelayPacketSize,
+		}
+	} else {
+		users := make(map[[16]byte]string)
+		for _uuid, password := range config.Users {
+			users[uuid.FromStringOrNil(_uuid)] = password
+		}
+
+		optionV5 = &tuic.ServerOptionV5{
+			HandleTcpFn:           handleTcpFn,
+			HandleUdpFn:           handleUdpFn,
+			TlsConfig:             tlsConfig,
+			QuicConfig:            quicConfig,
+			Users:                 users,
+			CongestionController:  config.CongestionController,
+			AuthenticationTimeout: time.Duration(config.AuthenticationTimeout) * time.Millisecond,
+			MaxUdpRelayPacketSize: config.MaxUdpRelayPacketSize,
+		}
 	}
 
 	sl := &Listener{false, config, nil, nil}
@@ -111,7 +151,12 @@ func New(config LC.TuicServer, tcpIn chan<- C.ConnContext, udpIn chan<- C.Packet
 
 		sl.udpListeners = append(sl.udpListeners, ul)
 
-		server, err := tuic.NewServer(option, ul)
+		var server tuic.Server
+		if optionV4 != nil {
+			server, err = tuic.NewServerV4(optionV4, ul)
+		} else {
+			server, err = tuic.NewServerV5(optionV5, ul)
+		}
 		if err != nil {
 			return nil, err
 		}