|  | @@ -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)
 | 
	
		
			
				|  |  | -	}
 | 
	
		
			
				|  |  | -}
 |