mirror of https://github.com/v2ray/v2ray-core
496 lines
13 KiB
Go
496 lines
13 KiB
Go
package socks
|
|
|
|
import (
|
|
"io"
|
|
|
|
"v2ray.com/core/common/buf"
|
|
"v2ray.com/core/common/net"
|
|
"v2ray.com/core/common/protocol"
|
|
"v2ray.com/core/common/serial"
|
|
)
|
|
|
|
const (
|
|
socks5Version = 0x05
|
|
socks4Version = 0x04
|
|
|
|
cmdTCPConnect = 0x01
|
|
cmdTCPBind = 0x02
|
|
cmdUDPPort = 0x03
|
|
|
|
socks4RequestGranted = 90
|
|
socks4RequestRejected = 91
|
|
|
|
authNotRequired = 0x00
|
|
//authGssAPI = 0x01
|
|
authPassword = 0x02
|
|
authNoMatchingMethod = 0xFF
|
|
|
|
addrTypeIPv4 = 0x01
|
|
addrTypeIPv6 = 0x04
|
|
addrTypeDomain = 0x03
|
|
|
|
statusSuccess = 0x00
|
|
statusCmdNotSupport = 0x07
|
|
)
|
|
|
|
type ServerSession struct {
|
|
config *ServerConfig
|
|
port net.Port
|
|
}
|
|
|
|
func (s *ServerSession) Handshake(reader io.Reader, writer io.Writer) (*protocol.RequestHeader, error) {
|
|
buffer := buf.NewLocal(512)
|
|
request := new(protocol.RequestHeader)
|
|
|
|
if err := buffer.AppendSupplier(buf.ReadFullFrom(reader, 2)); err != nil {
|
|
return nil, newError("insufficient header").Base(err)
|
|
}
|
|
|
|
version := buffer.Byte(0)
|
|
if version == socks4Version {
|
|
if err := buffer.AppendSupplier(buf.ReadFullFrom(reader, 6)); err != nil {
|
|
return nil, newError("insufficient header").Base(err)
|
|
}
|
|
port := net.PortFromBytes(buffer.BytesRange(2, 4))
|
|
address := net.IPAddress(buffer.BytesRange(4, 8))
|
|
_, err := readUntilNull(reader) // user id
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if address.IP()[0] == 0x00 {
|
|
domain, err := readUntilNull(reader)
|
|
if err != nil {
|
|
return nil, newError("failed to read domain for socks 4a").Base(err)
|
|
}
|
|
address = net.DomainAddress(domain)
|
|
}
|
|
|
|
switch buffer.Byte(1) {
|
|
case cmdTCPConnect:
|
|
request.Command = protocol.RequestCommandTCP
|
|
request.Address = address
|
|
request.Port = port
|
|
request.Version = socks4Version
|
|
if err := writeSocks4Response(writer, socks4RequestGranted, net.AnyIP, net.Port(0)); err != nil {
|
|
return nil, err
|
|
}
|
|
return request, nil
|
|
default:
|
|
writeSocks4Response(writer, socks4RequestRejected, net.AnyIP, net.Port(0))
|
|
return nil, newError("unsupported command: ", buffer.Byte(1))
|
|
}
|
|
}
|
|
|
|
if version == socks5Version {
|
|
nMethod := int(buffer.Byte(1))
|
|
if err := buffer.AppendSupplier(buf.ReadFullFrom(reader, nMethod)); err != nil {
|
|
return nil, newError("failed to read auth methods").Base(err)
|
|
}
|
|
|
|
var expectedAuth byte = authNotRequired
|
|
if s.config.AuthType == AuthType_PASSWORD {
|
|
expectedAuth = authPassword
|
|
}
|
|
|
|
if !hasAuthMethod(expectedAuth, buffer.BytesRange(2, 2+nMethod)) {
|
|
writeSocks5AuthenticationResponse(writer, socks5Version, authNoMatchingMethod)
|
|
return nil, newError("no matching auth method")
|
|
}
|
|
|
|
if err := writeSocks5AuthenticationResponse(writer, socks5Version, expectedAuth); err != nil {
|
|
return nil, newError("failed to write auth response").Base(err)
|
|
}
|
|
|
|
if expectedAuth == authPassword {
|
|
username, password, err := readUsernamePassword(reader)
|
|
if err != nil {
|
|
return nil, newError("failed to read username and password for authentication").Base(err)
|
|
}
|
|
|
|
if !s.config.HasAccount(username, password) {
|
|
writeSocks5AuthenticationResponse(writer, 0x01, 0xFF)
|
|
return nil, newError("invalid username or password")
|
|
}
|
|
|
|
if err := writeSocks5AuthenticationResponse(writer, 0x01, 0x00); err != nil {
|
|
return nil, newError("failed to write auth response").Base(err)
|
|
}
|
|
}
|
|
if err := buffer.Reset(buf.ReadFullFrom(reader, 4)); err != nil {
|
|
return nil, newError("failed to read request").Base(err)
|
|
}
|
|
|
|
cmd := buffer.Byte(1)
|
|
if cmd == cmdTCPBind || (cmd == cmdUDPPort && !s.config.UdpEnabled) {
|
|
writeSocks5Response(writer, statusCmdNotSupport, net.AnyIP, net.Port(0))
|
|
return nil, newError("unsupported command: ", cmd)
|
|
}
|
|
|
|
switch cmd {
|
|
case cmdTCPConnect:
|
|
request.Command = protocol.RequestCommandTCP
|
|
case cmdUDPPort:
|
|
request.Command = protocol.RequestCommandUDP
|
|
}
|
|
|
|
addrType := buffer.Byte(3)
|
|
|
|
buffer.Clear()
|
|
|
|
request.Version = socks5Version
|
|
switch addrType {
|
|
case addrTypeIPv4:
|
|
if err := buffer.AppendSupplier(buf.ReadFullFrom(reader, 4)); err != nil {
|
|
return nil, err
|
|
}
|
|
request.Address = net.IPAddress(buffer.Bytes())
|
|
case addrTypeIPv6:
|
|
if err := buffer.AppendSupplier(buf.ReadFullFrom(reader, 16)); err != nil {
|
|
return nil, err
|
|
}
|
|
request.Address = net.IPAddress(buffer.Bytes())
|
|
case addrTypeDomain:
|
|
if err := buffer.AppendSupplier(buf.ReadFullFrom(reader, 1)); err != nil {
|
|
return nil, err
|
|
}
|
|
domainLength := int(buffer.Byte(0))
|
|
if err := buffer.AppendSupplier(buf.ReadFullFrom(reader, domainLength)); err != nil {
|
|
return nil, err
|
|
}
|
|
request.Address = net.ParseAddress(string(buffer.BytesFrom(-domainLength)))
|
|
default:
|
|
return nil, newError("Unknown address type: ", addrType)
|
|
}
|
|
|
|
if err := buffer.AppendSupplier(buf.ReadFullFrom(reader, 2)); err != nil {
|
|
return nil, err
|
|
}
|
|
request.Port = net.PortFromBytes(buffer.BytesFrom(-2))
|
|
|
|
responseAddress := net.AnyIP
|
|
responsePort := net.Port(1717)
|
|
if request.Command == protocol.RequestCommandUDP {
|
|
addr := s.config.Address.AsAddress()
|
|
if addr == nil {
|
|
addr = net.LocalHostIP
|
|
}
|
|
responseAddress = addr
|
|
responsePort = s.port
|
|
}
|
|
if err := writeSocks5Response(writer, statusSuccess, responseAddress, responsePort); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return request, nil
|
|
}
|
|
|
|
return nil, newError("unknown Socks version: ", version)
|
|
}
|
|
|
|
func readUsernamePassword(reader io.Reader) (string, string, error) {
|
|
buffer := buf.NewLocal(512)
|
|
defer buffer.Release()
|
|
|
|
if err := buffer.Reset(buf.ReadFullFrom(reader, 2)); err != nil {
|
|
return "", "", err
|
|
}
|
|
nUsername := int(buffer.Byte(1))
|
|
|
|
if err := buffer.Reset(buf.ReadFullFrom(reader, nUsername)); err != nil {
|
|
return "", "", err
|
|
}
|
|
username := buffer.String()
|
|
|
|
if err := buffer.Reset(buf.ReadFullFrom(reader, 1)); err != nil {
|
|
return "", "", err
|
|
}
|
|
nPassword := int(buffer.Byte(0))
|
|
if err := buffer.Reset(buf.ReadFullFrom(reader, nPassword)); err != nil {
|
|
return "", "", err
|
|
}
|
|
password := buffer.String()
|
|
return username, password, nil
|
|
}
|
|
|
|
func readUntilNull(reader io.Reader) (string, error) {
|
|
var b [256]byte
|
|
size := 0
|
|
for {
|
|
_, err := reader.Read(b[size : size+1])
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if b[size] == 0x00 {
|
|
return string(b[:size]), nil
|
|
}
|
|
size++
|
|
if size == 256 {
|
|
return "", newError("buffer overrun")
|
|
}
|
|
}
|
|
}
|
|
|
|
func hasAuthMethod(expectedAuth byte, authCandidates []byte) bool {
|
|
for _, a := range authCandidates {
|
|
if a == expectedAuth {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func writeSocks5AuthenticationResponse(writer io.Writer, version byte, auth byte) error {
|
|
_, err := writer.Write([]byte{version, auth})
|
|
return err
|
|
}
|
|
|
|
func appendAddress(buffer *buf.Buffer, address net.Address, port net.Port) {
|
|
switch address.Family() {
|
|
case net.AddressFamilyIPv4:
|
|
buffer.AppendBytes(0x01)
|
|
buffer.Append(address.IP())
|
|
case net.AddressFamilyIPv6:
|
|
buffer.AppendBytes(0x04)
|
|
buffer.Append(address.IP())
|
|
case net.AddressFamilyDomain:
|
|
buffer.AppendBytes(0x03, byte(len(address.Domain())))
|
|
buffer.AppendSupplier(serial.WriteString(address.Domain()))
|
|
}
|
|
buffer.AppendSupplier(serial.WriteUint16(port.Value()))
|
|
}
|
|
|
|
func writeSocks5Response(writer io.Writer, errCode byte, address net.Address, port net.Port) error {
|
|
buffer := buf.NewLocal(64)
|
|
buffer.AppendBytes(socks5Version, errCode, 0x00 /* reserved */)
|
|
appendAddress(buffer, address, port)
|
|
|
|
_, err := writer.Write(buffer.Bytes())
|
|
return err
|
|
}
|
|
|
|
func writeSocks4Response(writer io.Writer, errCode byte, address net.Address, port net.Port) error {
|
|
buffer := buf.NewLocal(32)
|
|
buffer.AppendBytes(0x00, errCode)
|
|
buffer.AppendSupplier(serial.WriteUint16(port.Value()))
|
|
buffer.Append(address.IP())
|
|
_, err := writer.Write(buffer.Bytes())
|
|
return err
|
|
}
|
|
|
|
func DecodeUDPPacket(packet []byte) (*protocol.RequestHeader, []byte, error) {
|
|
if len(packet) < 5 {
|
|
return nil, nil, newError("insufficient length of packet.")
|
|
}
|
|
request := &protocol.RequestHeader{
|
|
Version: socks5Version,
|
|
Command: protocol.RequestCommandUDP,
|
|
}
|
|
|
|
// packet[0] and packet[1] are reserved
|
|
if packet[2] != 0 /* fragments */ {
|
|
return nil, nil, newError("discarding fragmented payload.")
|
|
}
|
|
|
|
addrType := packet[3]
|
|
var dataBegin int
|
|
|
|
switch addrType {
|
|
case addrTypeIPv4:
|
|
if len(packet) < 10 {
|
|
return nil, nil, newError("insufficient length of packet")
|
|
}
|
|
ip := packet[4:8]
|
|
request.Port = net.PortFromBytes(packet[8:10])
|
|
request.Address = net.IPAddress(ip)
|
|
dataBegin = 10
|
|
case addrTypeIPv6:
|
|
if len(packet) < 22 {
|
|
return nil, nil, newError("insufficient length of packet")
|
|
}
|
|
ip := packet[4:20]
|
|
request.Port = net.PortFromBytes(packet[20:22])
|
|
request.Address = net.IPAddress(ip)
|
|
dataBegin = 22
|
|
case addrTypeDomain:
|
|
domainLength := int(packet[4])
|
|
if len(packet) < 5+domainLength+2 {
|
|
return nil, nil, newError("insufficient length of packet")
|
|
}
|
|
domain := string(packet[5 : 5+domainLength])
|
|
request.Port = net.PortFromBytes(packet[5+domainLength : 5+domainLength+2])
|
|
request.Address = net.ParseAddress(domain)
|
|
dataBegin = 5 + domainLength + 2
|
|
default:
|
|
return nil, nil, newError("unknown address type ", addrType)
|
|
}
|
|
|
|
return request, packet[dataBegin:], nil
|
|
}
|
|
|
|
func EncodeUDPPacket(request *protocol.RequestHeader, data []byte) *buf.Buffer {
|
|
b := buf.New()
|
|
b.AppendBytes(0, 0, 0 /* Fragment */)
|
|
appendAddress(b, request.Address, request.Port)
|
|
b.Append(data)
|
|
return b
|
|
}
|
|
|
|
type UDPReader struct {
|
|
reader io.Reader
|
|
}
|
|
|
|
func NewUDPReader(reader io.Reader) *UDPReader {
|
|
return &UDPReader{reader: reader}
|
|
}
|
|
|
|
func (r *UDPReader) Read() (buf.MultiBuffer, error) {
|
|
b := buf.New()
|
|
if err := b.AppendSupplier(buf.ReadFrom(r.reader)); err != nil {
|
|
return nil, err
|
|
}
|
|
_, data, err := DecodeUDPPacket(b.Bytes())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
b.Clear()
|
|
b.Append(data)
|
|
return buf.NewMultiBufferValue(b), nil
|
|
}
|
|
|
|
type UDPWriter struct {
|
|
request *protocol.RequestHeader
|
|
writer io.Writer
|
|
}
|
|
|
|
func NewUDPWriter(request *protocol.RequestHeader, writer io.Writer) *UDPWriter {
|
|
return &UDPWriter{
|
|
request: request,
|
|
writer: writer,
|
|
}
|
|
}
|
|
|
|
// Write implements io.Writer.
|
|
func (w *UDPWriter) Write(b []byte) (int, error) {
|
|
eb := EncodeUDPPacket(w.request, b)
|
|
defer eb.Release()
|
|
if _, err := w.writer.Write(eb.Bytes()); err != nil {
|
|
return 0, err
|
|
}
|
|
return len(b), nil
|
|
}
|
|
|
|
func ClientHandshake(request *protocol.RequestHeader, reader io.Reader, writer io.Writer) (*protocol.RequestHeader, error) {
|
|
authByte := byte(authNotRequired)
|
|
if request.User != nil {
|
|
authByte = byte(authPassword)
|
|
}
|
|
authRequest := []byte{socks5Version, 0x01, authByte}
|
|
if _, err := writer.Write(authRequest); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
b := buf.NewLocal(512)
|
|
if err := b.AppendSupplier(buf.ReadFullFrom(reader, 2)); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if b.Byte(0) != socks5Version {
|
|
return nil, newError("unexpected server version: ", b.Byte(0)).AtWarning()
|
|
}
|
|
if b.Byte(1) != authByte {
|
|
return nil, newError("auth method not supported.").AtWarning()
|
|
}
|
|
|
|
if authByte == authPassword {
|
|
rawAccount, err := request.User.GetTypedAccount()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
account := rawAccount.(*Account)
|
|
|
|
b.Clear()
|
|
b.AppendBytes(socks5Version, byte(len(account.Username)))
|
|
b.Append([]byte(account.Username))
|
|
b.AppendBytes(byte(len(account.Password)))
|
|
b.Append([]byte(account.Password))
|
|
if _, err := writer.Write(b.Bytes()); err != nil {
|
|
return nil, err
|
|
}
|
|
b.Clear()
|
|
if err := b.AppendSupplier(buf.ReadFullFrom(reader, 2)); err != nil {
|
|
return nil, err
|
|
}
|
|
if b.Byte(1) != 0x00 {
|
|
return nil, newError("server rejects account: ", b.Byte(1))
|
|
}
|
|
}
|
|
|
|
b.Clear()
|
|
|
|
command := byte(cmdTCPConnect)
|
|
if request.Command == protocol.RequestCommandUDP {
|
|
command = byte(cmdUDPPort)
|
|
}
|
|
b.AppendBytes(socks5Version, command, 0x00 /* reserved */)
|
|
appendAddress(b, request.Address, request.Port)
|
|
if _, err := writer.Write(b.Bytes()); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
b.Clear()
|
|
if err := b.AppendSupplier(buf.ReadFullFrom(reader, 4)); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp := b.Byte(1)
|
|
if resp != 0x00 {
|
|
return nil, newError("server rejects request: ", resp)
|
|
}
|
|
|
|
addrType := b.Byte(3)
|
|
|
|
b.Clear()
|
|
|
|
var address net.Address
|
|
switch addrType {
|
|
case addrTypeIPv4:
|
|
if err := b.AppendSupplier(buf.ReadFullFrom(reader, 4)); err != nil {
|
|
return nil, err
|
|
}
|
|
address = net.IPAddress(b.Bytes())
|
|
case addrTypeIPv6:
|
|
if err := b.AppendSupplier(buf.ReadFullFrom(reader, 16)); err != nil {
|
|
return nil, err
|
|
}
|
|
address = net.IPAddress(b.Bytes())
|
|
case addrTypeDomain:
|
|
if err := b.AppendSupplier(buf.ReadFullFrom(reader, 1)); err != nil {
|
|
return nil, err
|
|
}
|
|
domainLength := int(b.Byte(0))
|
|
if err := b.AppendSupplier(buf.ReadFullFrom(reader, domainLength)); err != nil {
|
|
return nil, err
|
|
}
|
|
address = net.DomainAddress(string(b.BytesFrom(-domainLength)))
|
|
default:
|
|
return nil, newError("unknown address type: ", addrType)
|
|
}
|
|
|
|
if err := b.AppendSupplier(buf.ReadFullFrom(reader, 2)); err != nil {
|
|
return nil, err
|
|
}
|
|
port := net.PortFromBytes(b.BytesFrom(-2))
|
|
|
|
if request.Command == protocol.RequestCommandUDP {
|
|
udpRequest := &protocol.RequestHeader{
|
|
Version: socks5Version,
|
|
Command: protocol.RequestCommandUDP,
|
|
Address: address,
|
|
Port: port,
|
|
}
|
|
return udpRequest, nil
|
|
}
|
|
|
|
return nil, nil
|
|
}
|