From bab8dc0b820199ccc47f1bbcceb0037ee5a1fd28 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= <i@sekai.icu>
Date: Fri, 31 Jan 2025 12:57:35 +0800
Subject: [PATCH] Fix missing handshake for early conn

---
 route/conn.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/route/conn.go b/route/conn.go
index e010c2cd..cbed3491 100644
--- a/route/conn.go
+++ b/route/conn.go
@@ -225,6 +225,17 @@ func (m *ConnectionManager) connectionCopy(ctx context.Context, source io.Reader
 		}
 		break
 	}
+	if earlyConn, isEarlyConn := common.Cast[N.EarlyConn](destination); isEarlyConn && earlyConn.NeedHandshake() {
+		_, err := destination.Write(nil)
+		if err != nil {
+			if !direction {
+				m.logger.ErrorContext(ctx, "connection upload handshake: ", err)
+			} else {
+				m.logger.ErrorContext(ctx, "connection download handshake: ", err)
+			}
+			return
+		}
+	}
 	_, err := bufio.CopyWithCounters(destination, source, originSource, readCounters, writeCounters)
 	if err != nil {
 		common.Close(originDestination)