feat:KCP协议

cmd/internal/transport/udp_test.go

@@ -141,10 +141,11 @@ func TestUDPReceiveLoopDeliversMessages(t *testing.T) {
 	go func() {
 		loopErr <- receiver.ReceiveLoop(func(msg protocol.Message, _ *net.UDPAddr) error {
 			mu.Lock()
-			defer mu.Unlock()
 			got = append(got, msg)
-			if len(got) >= len(want) {
-				return nil
+			done := len(got) >= len(want)
+			mu.Unlock()
+			if done {
+				return receiver.Close()
 			}
 			return nil
 		})
@@ -156,17 +157,12 @@ func TestUDPReceiveLoopDeliversMessages(t *testing.T) {
 		}
 	}
 
-	// 关闭发送端，ReceiveLoop 会因读取错误退出
-	if err := sender.Close(); err != nil {
-		t.Fatalf("sender.Close() error = %v", err)
-	}
-
 	err := <-loopErr
 	if err == nil {
-		t.Fatal("ReceiveLoop() error = nil, want non-nil after peer close")
+		t.Fatal("ReceiveLoop() error = nil, want non-nil after receiver close")
 	}
-	if !strings.Contains(err.Error(), "udp receive loop read") {
-		t.Fatalf("ReceiveLoop() error = %v, want read context", err)
+	if !strings.Contains(err.Error(), "closed") && !strings.Contains(err.Error(), "use of closed network connection") {
+		t.Fatalf("ReceiveLoop() error = %v, want close-related error", err)
 	}
 
 	mu.Lock()
@@ -268,6 +264,7 @@ func newUDPConnPair(t *testing.T, senderOpts []UDPOption, receiverOpts []UDPOpti
 		_ = conn1.Close()
 		t.Fatalf("ListenUDP(2) error = %v", err)
 	}
+	receiverLocalAddr := conn2.LocalAddr().(*net.UDPAddr)
 
 	// 用 Dial 模式连接对端
 	senderRaw, err := net.DialUDP("udp", nil, conn2.LocalAddr().(*net.UDPAddr))
@@ -277,14 +274,13 @@ func newUDPConnPair(t *testing.T, senderOpts []UDPOption, receiverOpts []UDPOpti
 		t.Fatalf("DialUDP(sender) error = %v", err)
 	}
 	_ = conn1.Close() // 不再需要 conn1
+	_ = conn2.Close() // 释放 receiver 计划使用的本地地址
 
-	receiverRaw, err := net.DialUDP("udp", conn2.LocalAddr().(*net.UDPAddr), senderRaw.LocalAddr().(*net.UDPAddr))
+	receiverRaw, err := net.DialUDP("udp", receiverLocalAddr, senderRaw.LocalAddr().(*net.UDPAddr))
 	if err != nil {
 		_ = senderRaw.Close()
-		_ = conn2.Close()
 		t.Fatalf("DialUDP(receiver) error = %v", err)
 	}
-	_ = conn2.Close() // 不再需要 conn2
 
 	sender, err := NewUDPConn(senderRaw, nil, senderOpts...)
 	if err != nil {