v2ray-core/proxy/socks/protocol/socks4_test.go

39 lines
1.0 KiB
Go
Raw Normal View History

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
}