package kcp_test import ( "context" "crypto/rand" "io" "sync" "testing" "time" "v2ray.com/core/common/net" "v2ray.com/core/transport/internet" . "v2ray.com/core/transport/internet/kcp" . "v2ray.com/ext/assert" ) func TestDialAndListen(t *testing.T) { assert := With(t) lctx := internet.ContextWithStreamSettings(context.Background(), &internet.MemoryStreamConfig{ ProtocolName: "mkcp", ProtocolSettings: &Config{}, }) listerner, err := NewListener(lctx, net.LocalHostIP, net.Port(0), func(conn internet.Connection) { go func(c internet.Connection) { payload := make([]byte, 4096) for { nBytes, err := c.Read(payload) if err != nil { break } for idx, b := range payload[:nBytes] { payload[idx] = b ^ 'c' } c.Write(payload[:nBytes]) } c.Close() }(conn) }) assert(err, IsNil) port := net.Port(listerner.Addr().(*net.UDPAddr).Port) ctx := internet.ContextWithStreamSettings(context.Background(), &internet.MemoryStreamConfig{ ProtocolName: "mkcp", ProtocolSettings: &Config{}, }) wg := new(sync.WaitGroup) for i := 0; i < 10; i++ { clientConn, err := DialKCP(ctx, net.UDPDestination(net.LocalHostIP, port)) assert(err, IsNil) wg.Add(1) go func() { clientSend := make([]byte, 1024*1024) rand.Read(clientSend) go clientConn.Write(clientSend) clientReceived := make([]byte, 1024*1024) nBytes, _ := io.ReadFull(clientConn, clientReceived) assert(nBytes, Equals, len(clientReceived)) clientConn.Close() clientExpected := make([]byte, 1024*1024) for idx, b := range clientSend { clientExpected[idx] = b ^ 'c' } assert(clientReceived, Equals, clientExpected) wg.Done() }() } wg.Wait() for i := 0; i < 60 && listerner.ActiveConnections() > 0; i++ { time.Sleep(500 * time.Millisecond) } assert(listerner.ActiveConnections(), Equals, 0) listerner.Close() }