fix: tuic pool client should only cache the system's UDPConn

xishang0128 · Apr 10, 2023 · 4d12ed4 · 4d12ed4
1 parent 9afcb70
commit 4d12ed4
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/transport/tuic/pool_client.go b/transport/tuic/pool_client.go
@@ -67,11 +67,14 @@ func (t *PoolClient) dial(ctx context.Context, dialer C.Dialer, dialFn DialFunc)
 		return nil, nil, err
 	}
 
-	dr.pc, dr.addr, dr.err = pc, addr, err
+	if _, ok := pc.(*net.UDPConn); ok { // only cache the system's UDPConn
+		dr.pc, dr.addr, dr.err = pc, addr, err
+
+		t.dialResultMutex.Lock()
+		t.dialResultMap[dialer] = dr
+		t.dialResultMutex.Unlock()
+	}
 
-	t.dialResultMutex.Lock()
-	t.dialResultMap[dialer] = dr
-	t.dialResultMutex.Unlock()
 	return pc, addr, err
 }