mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-22 10:08:15 -05:00
43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
package protocol
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/v2ray/v2ray-core/common/alloc"
|
|
"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))
|
|
assert.Error(err).Equals(Socks4Downgrade)
|
|
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)
|
|
|
|
response := &Socks4AuthenticationResponse{
|
|
result: byte(0x10),
|
|
port: 443,
|
|
ip: []byte{1, 2, 3, 4},
|
|
}
|
|
|
|
buffer := alloc.NewSmallBuffer().Clear()
|
|
defer buffer.Release()
|
|
|
|
response.Write(buffer)
|
|
assert.Bytes(buffer.Value).Equals([]byte{0x00, 0x10, 0x01, 0xBB, 0x01, 0x02, 0x03, 0x04})
|
|
}
|