|
|
@@ -1,16 +1,10 @@
|
|
|
package kcp_test
|
|
|
|
|
|
import (
|
|
|
- "crypto/rand"
|
|
|
- "io"
|
|
|
- "net"
|
|
|
"testing"
|
|
|
"time"
|
|
|
|
|
|
- v2net "v2ray.com/core/common/net"
|
|
|
"v2ray.com/core/testing/assert"
|
|
|
- "v2ray.com/core/transport/internet"
|
|
|
- "v2ray.com/core/transport/internet/authenticators/srtp"
|
|
|
. "v2ray.com/core/transport/internet/kcp"
|
|
|
)
|
|
|
|
|
|
@@ -37,43 +31,3 @@ func TestConnectionReadTimeout(t *testing.T) {
|
|
|
|
|
|
conn.Terminate()
|
|
|
}
|
|
|
-
|
|
|
-func TestConnectionReadWrite(t *testing.T) {
|
|
|
- assert := assert.On(t)
|
|
|
-
|
|
|
- upReader, upWriter := io.Pipe()
|
|
|
- downReader, downWriter := io.Pipe()
|
|
|
-
|
|
|
- auth := internet.NewAuthenticatorChain(srtp.SRTPFactory{}.Create(nil), NewSimpleAuthenticator())
|
|
|
-
|
|
|
- connClient := NewConnection(1, upWriter, &net.UDPAddr{IP: v2net.LocalHostIP.IP(), Port: 1}, &net.UDPAddr{IP: v2net.LocalHostIP.IP(), Port: 2}, auth, &Config{})
|
|
|
- connClient.FetchInputFrom(downReader)
|
|
|
-
|
|
|
- connServer := NewConnection(1, downWriter, &net.UDPAddr{IP: v2net.LocalHostIP.IP(), Port: 2}, &net.UDPAddr{IP: v2net.LocalHostIP.IP(), Port: 1}, auth, &Config{})
|
|
|
- connServer.FetchInputFrom(upReader)
|
|
|
-
|
|
|
- totalWritten := 1024 * 1024
|
|
|
- clientSend := make([]byte, totalWritten)
|
|
|
- rand.Read(clientSend)
|
|
|
- go func() {
|
|
|
- nBytes, err := connClient.Write(clientSend)
|
|
|
- assert.Int(nBytes).Equals(totalWritten)
|
|
|
- assert.Error(err).IsNil()
|
|
|
- }()
|
|
|
-
|
|
|
- serverReceived := make([]byte, totalWritten)
|
|
|
- totalRead := 0
|
|
|
- for totalRead < totalWritten {
|
|
|
- nBytes, err := connServer.Read(serverReceived[totalRead:])
|
|
|
- assert.Error(err).IsNil()
|
|
|
- totalRead += nBytes
|
|
|
- }
|
|
|
- assert.Bytes(serverReceived).Equals(clientSend)
|
|
|
-
|
|
|
- connClient.Close()
|
|
|
- connServer.Close()
|
|
|
-
|
|
|
- for connClient.State() != StateTerminated || connServer.State() != StateTerminated {
|
|
|
- time.Sleep(time.Second)
|
|
|
- }
|
|
|
-}
|