|  | @@ -5,7 +5,7 @@ import (
 | 
	
		
			
				|  |  |  	"io"
 | 
	
		
			
				|  |  |  	"testing"
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -	"v2ray.com/core/common/alloc"
 | 
	
		
			
				|  |  | +	"v2ray.com/core/common/buf"
 | 
	
		
			
				|  |  |  	v2net "v2ray.com/core/common/net"
 | 
	
		
			
				|  |  |  	"v2ray.com/core/proxy"
 | 
	
		
			
				|  |  |  	"v2ray.com/core/testing/assert"
 | 
	
	
		
			
				|  | @@ -30,7 +30,7 @@ func TestHasAuthenticationMethod(t *testing.T) {
 | 
	
		
			
				|  |  |  func TestAuthenticationRequestRead(t *testing.T) {
 | 
	
		
			
				|  |  |  	assert := assert.On(t)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -	buffer := alloc.NewBuffer()
 | 
	
		
			
				|  |  | +	buffer := buf.NewBuffer()
 | 
	
		
			
				|  |  |  	buffer.AppendBytes(
 | 
	
		
			
				|  |  |  		0x05, // version
 | 
	
		
			
				|  |  |  		0x01, // nMethods
 | 
	
	
		
			
				|  | @@ -85,7 +85,7 @@ func TestResponseWrite(t *testing.T) {
 | 
	
		
			
				|  |  |  		[16]byte{},
 | 
	
		
			
				|  |  |  		v2net.Port(53),
 | 
	
		
			
				|  |  |  	}
 | 
	
		
			
				|  |  | -	buffer := alloc.NewLocalBuffer(2048)
 | 
	
		
			
				|  |  | +	buffer := buf.NewLocalBuffer(2048)
 | 
	
		
			
				|  |  |  	defer buffer.Release()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  	response.Write(buffer)
 | 
	
	
		
			
				|  | @@ -106,7 +106,7 @@ func TestSetIPv6(t *testing.T) {
 | 
	
		
			
				|  |  |  	response := NewSocks5Response()
 | 
	
		
			
				|  |  |  	response.SetIPv6([]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15})
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -	buffer := alloc.NewLocalBuffer(2048)
 | 
	
		
			
				|  |  | +	buffer := buf.NewLocalBuffer(2048)
 | 
	
		
			
				|  |  |  	defer buffer.Release()
 | 
	
		
			
				|  |  |  	response.Write(buffer)
 | 
	
		
			
				|  |  |  	assert.Bytes(buffer.Bytes()).Equals([]byte{
 | 
	
	
		
			
				|  | @@ -119,7 +119,7 @@ func TestSetDomain(t *testing.T) {
 | 
	
		
			
				|  |  |  	response := NewSocks5Response()
 | 
	
		
			
				|  |  |  	response.SetDomain("v2ray.com")
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -	buffer := alloc.NewLocalBuffer(2048)
 | 
	
		
			
				|  |  | +	buffer := buf.NewLocalBuffer(2048)
 | 
	
		
			
				|  |  |  	defer buffer.Release()
 | 
	
		
			
				|  |  |  	response.Write(buffer)
 | 
	
		
			
				|  |  |  	assert.Bytes(buffer.Bytes()).Equals([]byte{
 | 
	
	
		
			
				|  | @@ -129,7 +129,7 @@ func TestSetDomain(t *testing.T) {
 | 
	
		
			
				|  |  |  func TestEmptyAuthRequest(t *testing.T) {
 | 
	
		
			
				|  |  |  	assert := assert.On(t)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -	_, _, err := ReadAuthentication(alloc.NewBuffer())
 | 
	
		
			
				|  |  | +	_, _, err := ReadAuthentication(buf.NewBuffer())
 | 
	
		
			
				|  |  |  	assert.Error(err).Equals(io.EOF)
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
	
		
			
				|  | @@ -143,7 +143,7 @@ func TestSingleByteAuthRequest(t *testing.T) {
 | 
	
		
			
				|  |  |  func TestZeroAuthenticationMethod(t *testing.T) {
 | 
	
		
			
				|  |  |  	assert := assert.On(t)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -	buffer := alloc.NewBuffer()
 | 
	
		
			
				|  |  | +	buffer := buf.NewBuffer()
 | 
	
		
			
				|  |  |  	buffer.AppendBytes(5, 0)
 | 
	
		
			
				|  |  |  	_, _, err := ReadAuthentication(buffer)
 | 
	
		
			
				|  |  |  	assert.Error(err).Equals(crypto.ErrAuthenticationFailed)
 | 
	
	
		
			
				|  | @@ -151,7 +151,7 @@ func TestZeroAuthenticationMethod(t *testing.T) {
 | 
	
		
			
				|  |  |  func TestWrongProtocolVersion(t *testing.T) {
 | 
	
		
			
				|  |  |  	assert := assert.On(t)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -	buffer := alloc.NewBuffer()
 | 
	
		
			
				|  |  | +	buffer := buf.NewBuffer()
 | 
	
		
			
				|  |  |  	buffer.AppendBytes(6, 1, 0)
 | 
	
		
			
				|  |  |  	_, _, err := ReadAuthentication(buffer)
 | 
	
		
			
				|  |  |  	assert.Error(err).Equals(proxy.ErrInvalidProtocolVersion)
 | 
	
	
		
			
				|  | @@ -160,14 +160,14 @@ func TestWrongProtocolVersion(t *testing.T) {
 | 
	
		
			
				|  |  |  func TestEmptyRequest(t *testing.T) {
 | 
	
		
			
				|  |  |  	assert := assert.On(t)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -	_, err := ReadRequest(alloc.NewBuffer())
 | 
	
		
			
				|  |  | +	_, err := ReadRequest(buf.NewBuffer())
 | 
	
		
			
				|  |  |  	assert.Error(err).Equals(io.EOF)
 | 
	
		
			
				|  |  |  }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  func TestIPv6Request(t *testing.T) {
 | 
	
		
			
				|  |  |  	assert := assert.On(t)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -	b := alloc.NewBuffer()
 | 
	
		
			
				|  |  | +	b := buf.NewBuffer()
 | 
	
		
			
				|  |  |  	b.AppendBytes(5, 1, 0, 4, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 0, 8)
 | 
	
		
			
				|  |  |  	request, err := ReadRequest(b)
 | 
	
		
			
				|  |  |  	assert.Error(err).IsNil()
 |