2015-09-25 18:48:55 +00:00
|
|
|
package protocol
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"testing"
|
|
|
|
|
2015-10-08 21:06:12 +00:00
|
|
|
"github.com/v2ray/v2ray-core/common/alloc"
|
2015-09-25 18:48:55 +00:00
|
|
|
"github.com/v2ray/v2ray-core/testing/unit"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestSocks4AuthenticationRequestRead(t *testing.T) {
|
|
|
|
assert := unit.Assert(t)
|
|
|
|
|
|
|
|
rawRequest := []byte{
|
|
|
|
0x04, // version
|
|
|
|
0x01, // command
|
|
|
|
0x00, 0x35,
|
|
|
|
0x72, 0x72, 0x72, 0x72,
|
|
|
|
}
|
|
|
|
_, request4, err := ReadAuthentication(bytes.NewReader(rawRequest))
|
2015-10-13 11:55:06 +00:00
|
|
|
assert.Error(err).Equals(Socks4Downgrade)
|
2015-09-25 18:48:55 +00:00
|
|
|
assert.Byte(request4.Version).Named("Version").Equals(0x04)
|
|
|
|
assert.Byte(request4.Command).Named("Command").Equals(0x01)
|
|
|
|
assert.Uint16(request4.Port).Named("Port").Equals(53)
|
|
|
|
assert.Bytes(request4.IP[:]).Named("IP").Equals([]byte{0x72, 0x72, 0x72, 0x72})
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestSocks4AuthenticationResponseToBytes(t *testing.T) {
|
|
|
|
assert := unit.Assert(t)
|
|
|
|
|
2015-11-02 22:55:10 +00:00
|
|
|
response := NewSocks4AuthenticationResponse(byte(0x10), 443, []byte{1, 2, 3, 4})
|
2015-10-08 21:06:12 +00:00
|
|
|
|
|
|
|
buffer := alloc.NewSmallBuffer().Clear()
|
|
|
|
defer buffer.Release()
|
|
|
|
|
|
|
|
response.Write(buffer)
|
|
|
|
assert.Bytes(buffer.Value).Equals([]byte{0x00, 0x10, 0x01, 0xBB, 0x01, 0x02, 0x03, 0x04})
|
2015-09-25 18:48:55 +00:00
|
|
|
}
|