From ae76daf39392d3e0b37f3e0692bb8be1f8cac332 Mon Sep 17 00:00:00 2001
From: wwqgtxx <wwqgtxx@gmail.com>
Date: Sat, 26 Nov 2022 21:14:56 +0800
Subject: [PATCH] chore: tuic add fast-open support

---
 adapter/outbound/tuic.go |  4 +++-
 transport/tuic/client.go | 42 +++++++++++++++++++++++++++++++++++-----
 2 files changed, 40 insertions(+), 6 deletions(-)

diff --git a/adapter/outbound/tuic.go b/adapter/outbound/tuic.go
index a69b44ca..c2f4f917 100644
--- a/adapter/outbound/tuic.go
+++ b/adapter/outbound/tuic.go
@@ -47,6 +47,7 @@ type TuicOption struct {
 	DisableSni            bool     `proxy:"disable_sni,omitempty"`
 	MaxUdpRelayPacketSize int      `proxy:"max_udp_relay_packet_size,omitempty"`
 
+	FastOpen            bool   `proxy:"fast-open,omitempty"`
 	SkipCertVerify      bool   `proxy:"skip-cert-verify,omitempty"`
 	Fingerprint         string `proxy:"fingerprint,omitempty"`
 	CustomCA            string `proxy:"ca,omitempty"`
@@ -263,6 +264,7 @@ func NewTuic(option TuicOption) (*Tuic, error) {
 			ReduceRtt:             option.ReduceRtt,
 			RequestTimeout:        option.RequestTimeout,
 			MaxUdpRelayPacketSize: option.MaxUdpRelayPacketSize,
+			FastOpen:              option.FastOpen,
 			Inference:             t,
 			Key:                   o,
 			LastVisited:           time.Now(),
@@ -303,7 +305,7 @@ func NewTuic(option TuicOption) (*Tuic, error) {
 						}
 					}
 				}
-				if time.Now().Sub(client.LastVisited) > 30*time.Minute {
+				if client.OpenStreams.Load() == 0 && time.Now().Sub(client.LastVisited) > 30*time.Minute {
 					next := it.Next()
 					clients.Remove(it)
 					it = next
diff --git a/transport/tuic/client.go b/transport/tuic/client.go
index 0ac03b5c..95aeeac3 100644
--- a/transport/tuic/client.go
+++ b/transport/tuic/client.go
@@ -38,6 +38,7 @@ type Client struct {
 	ReduceRtt             bool
 	RequestTimeout        int
 	MaxUdpRelayPacketSize int
+	FastOpen              bool
 
 	Inference   any
 	Key         any
@@ -282,18 +283,49 @@ func (t *Client) DialContext(ctx context.Context, metadata *C.Metadata, dialFn f
 	if t.RequestTimeout > 0 {
 		_ = stream.SetReadDeadline(time.Now().Add(time.Duration(t.RequestTimeout) * time.Millisecond))
 	}
-	conn := N.NewBufferedConn(stream)
+	conn := &earlyConn{BufferedConn: N.NewBufferedConn(stream)}
+	if !t.FastOpen {
+		err = conn.Response()
+		if err != nil {
+			return nil, err
+		}
+	}
+	return conn, nil
+}
+
+type earlyConn struct {
+	*N.BufferedConn
+	resOnce sync.Once
+	resErr  error
+}
+
+func (conn *earlyConn) response() error {
 	response, err := ReadResponse(conn)
 	if err != nil {
 		_ = conn.Close()
-		return nil, err
+		return err
 	}
 	if response.IsFailed() {
 		_ = conn.Close()
-		return nil, errors.New("connect failed")
+		return errors.New("connect failed")
 	}
-	_ = stream.SetReadDeadline(time.Time{})
-	return conn, nil
+	_ = conn.SetReadDeadline(time.Time{})
+	return nil
+}
+
+func (conn *earlyConn) Response() error {
+	conn.resOnce.Do(func() {
+		conn.resErr = conn.response()
+	})
+	return conn.resErr
+}
+
+func (conn *earlyConn) Read(b []byte) (n int, err error) {
+	err = conn.Response()
+	if err != nil {
+		return 0, err
+	}
+	return conn.BufferedConn.Read(b)
 }
 
 type quicStreamConn struct {