Fix copy early conn

2022-08-08 20:56:53 +08:00
parent 6d78cf6b58
commit 4067e0f25c
2 changed files with 18 additions and 1 deletions
--- a/experimental/clashapi/trafficontrol/tracker.go
+++ b/experimental/clashapi/trafficontrol/tracker.go
@@ -79,6 +79,10 @@ func (tt *tcpTracker) Leave() {
 	tt.manager.Leave(tt)
 }

+func (tt *tcpTracker) Upstream() any {
+	return tt.Conn
+}
+
 func NewTCPTracker(conn net.Conn, manager *Manager, metadata Metadata, router adapter.Router, rule adapter.Rule) *tcpTracker {
 	uuid, _ := uuid.NewV4()

@@ -166,6 +170,10 @@ func (ut *udpTracker) Leave() {
 	ut.manager.Leave(ut)
 }

+func (ut *udpTracker) Upstream() any {
+	return ut.PacketConn
+}
+
 func NewUDPTracker(conn N.PacketConn, manager *Manager, metadata Metadata, router adapter.Router, rule adapter.Rule) *udpTracker {
 	uuid, _ := uuid.NewV4()