|
|
@@ -8,13 +8,12 @@ import (
|
|
|
"github.com/v2ray/v2ray-core/common/alloc"
|
|
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
|
|
"github.com/v2ray/v2ray-core/proxy"
|
|
|
- v2testing "github.com/v2ray/v2ray-core/testing"
|
|
|
"github.com/v2ray/v2ray-core/testing/assert"
|
|
|
"github.com/v2ray/v2ray-core/transport"
|
|
|
)
|
|
|
|
|
|
func TestHasAuthenticationMethod(t *testing.T) {
|
|
|
- v2testing.Current(t)
|
|
|
+ assert := assert.On(t)
|
|
|
|
|
|
request := Socks5AuthenticationRequest{
|
|
|
version: socksVersion,
|
|
|
@@ -29,7 +28,7 @@ func TestHasAuthenticationMethod(t *testing.T) {
|
|
|
}
|
|
|
|
|
|
func TestAuthenticationRequestRead(t *testing.T) {
|
|
|
- v2testing.Current(t)
|
|
|
+ assert := assert.On(t)
|
|
|
|
|
|
buffer := alloc.NewBuffer().Clear().AppendBytes(
|
|
|
0x05, // version
|
|
|
@@ -38,13 +37,13 @@ func TestAuthenticationRequestRead(t *testing.T) {
|
|
|
)
|
|
|
request, _, err := ReadAuthentication(buffer)
|
|
|
assert.Error(err).IsNil()
|
|
|
- assert.Byte(request.version).Named("Version").Equals(0x05)
|
|
|
- assert.Byte(request.nMethods).Named("#Methods").Equals(0x01)
|
|
|
- assert.Byte(request.authMethods[0]).Named("Auth Method").Equals(0x02)
|
|
|
+ assert.Byte(request.version).Equals(0x05)
|
|
|
+ assert.Byte(request.nMethods).Equals(0x01)
|
|
|
+ assert.Byte(request.authMethods[0]).Equals(0x02)
|
|
|
}
|
|
|
|
|
|
func TestAuthenticationResponseWrite(t *testing.T) {
|
|
|
- v2testing.Current(t)
|
|
|
+ assert := assert.On(t)
|
|
|
|
|
|
response := NewAuthenticationResponse(byte(0x05))
|
|
|
|
|
|
@@ -54,7 +53,7 @@ func TestAuthenticationResponseWrite(t *testing.T) {
|
|
|
}
|
|
|
|
|
|
func TestRequestRead(t *testing.T) {
|
|
|
- v2testing.Current(t)
|
|
|
+ assert := assert.On(t)
|
|
|
|
|
|
rawRequest := []byte{
|
|
|
0x05, // version
|
|
|
@@ -66,15 +65,15 @@ func TestRequestRead(t *testing.T) {
|
|
|
}
|
|
|
request, err := ReadRequest(bytes.NewReader(rawRequest))
|
|
|
assert.Error(err).IsNil()
|
|
|
- assert.Byte(request.Version).Named("Version").Equals(0x05)
|
|
|
- assert.Byte(request.Command).Named("Command").Equals(0x01)
|
|
|
- assert.Byte(request.AddrType).Named("Address Type").Equals(0x01)
|
|
|
- assert.Bytes(request.IPv4[:]).Named("IPv4").Equals([]byte{0x72, 0x72, 0x72, 0x72})
|
|
|
- assert.Port(request.Port).Named("Port").Equals(v2net.Port(53))
|
|
|
+ assert.Byte(request.Version).Equals(0x05)
|
|
|
+ assert.Byte(request.Command).Equals(0x01)
|
|
|
+ assert.Byte(request.AddrType).Equals(0x01)
|
|
|
+ assert.Bytes(request.IPv4[:]).Equals([]byte{0x72, 0x72, 0x72, 0x72})
|
|
|
+ assert.Port(request.Port).Equals(v2net.Port(53))
|
|
|
}
|
|
|
|
|
|
func TestResponseWrite(t *testing.T) {
|
|
|
- v2testing.Current(t)
|
|
|
+ assert := assert.On(t)
|
|
|
|
|
|
response := Socks5Response{
|
|
|
socksVersion,
|
|
|
@@ -97,11 +96,11 @@ func TestResponseWrite(t *testing.T) {
|
|
|
0x72, 0x72, 0x72, 0x72,
|
|
|
byte(0x00), byte(0x035),
|
|
|
}
|
|
|
- assert.Bytes(buffer.Value).Named("raw response").Equals(expectedBytes)
|
|
|
+ assert.Bytes(buffer.Value).Equals(expectedBytes)
|
|
|
}
|
|
|
|
|
|
func TestSetIPv6(t *testing.T) {
|
|
|
- v2testing.Current(t)
|
|
|
+ assert := assert.On(t)
|
|
|
|
|
|
response := NewSocks5Response()
|
|
|
response.SetIPv6([]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15})
|
|
|
@@ -114,7 +113,7 @@ func TestSetIPv6(t *testing.T) {
|
|
|
}
|
|
|
|
|
|
func TestSetDomain(t *testing.T) {
|
|
|
- v2testing.Current(t)
|
|
|
+ assert := assert.On(t)
|
|
|
|
|
|
response := NewSocks5Response()
|
|
|
response.SetDomain("v2ray.com")
|
|
|
@@ -127,28 +126,28 @@ func TestSetDomain(t *testing.T) {
|
|
|
}
|
|
|
|
|
|
func TestEmptyAuthRequest(t *testing.T) {
|
|
|
- v2testing.Current(t)
|
|
|
+ assert := assert.On(t)
|
|
|
|
|
|
_, _, err := ReadAuthentication(alloc.NewBuffer().Clear())
|
|
|
assert.Error(err).Equals(io.EOF)
|
|
|
}
|
|
|
|
|
|
func TestSingleByteAuthRequest(t *testing.T) {
|
|
|
- v2testing.Current(t)
|
|
|
+ assert := assert.On(t)
|
|
|
|
|
|
_, _, err := ReadAuthentication(bytes.NewReader(make([]byte, 1)))
|
|
|
assert.Error(err).Equals(transport.ErrorCorruptedPacket)
|
|
|
}
|
|
|
|
|
|
func TestZeroAuthenticationMethod(t *testing.T) {
|
|
|
- v2testing.Current(t)
|
|
|
+ assert := assert.On(t)
|
|
|
|
|
|
buffer := alloc.NewBuffer().Clear().AppendBytes(5, 0)
|
|
|
_, _, err := ReadAuthentication(buffer)
|
|
|
assert.Error(err).Equals(proxy.ErrorInvalidAuthentication)
|
|
|
}
|
|
|
func TestWrongProtocolVersion(t *testing.T) {
|
|
|
- v2testing.Current(t)
|
|
|
+ assert := assert.On(t)
|
|
|
|
|
|
buffer := alloc.NewBuffer().Clear().AppendBytes(6, 1, 0)
|
|
|
_, _, err := ReadAuthentication(buffer)
|
|
|
@@ -156,14 +155,14 @@ func TestWrongProtocolVersion(t *testing.T) {
|
|
|
}
|
|
|
|
|
|
func TestEmptyRequest(t *testing.T) {
|
|
|
- v2testing.Current(t)
|
|
|
+ assert := assert.On(t)
|
|
|
|
|
|
_, err := ReadRequest(alloc.NewBuffer().Clear())
|
|
|
assert.Error(err).Equals(io.EOF)
|
|
|
}
|
|
|
|
|
|
func TestIPv6Request(t *testing.T) {
|
|
|
- v2testing.Current(t)
|
|
|
+ assert := assert.On(t)
|
|
|
|
|
|
request, err := ReadRequest(alloc.NewBuffer().Clear().AppendBytes(5, 1, 0, 4, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 0, 8))
|
|
|
assert.Error(err).IsNil()
|