Full implementation of Socks5 protocol, proxying is on the way

io/socks/socks.go

@@ -68,12 +68,15 @@ func (r *Socks5AuthenticationResponse) ToBytes() []byte {
 
 func WriteAuthentication(writer io.Writer, response Socks5AuthenticationResponse) error {
 	_, err := writer.Write(response.ToBytes())
-	if err != nil {
-		return err
-	}
-	return nil
+	return err
 }
 
+const (
+	AddrTypeIPv4   = byte(0x01)
+	AddrTypeIPv6   = byte(0x04)
+	AddrTypeDomain = byte(0x03)
+)
+
 type Socks5Request struct {
 	version  byte
 	command  byte
@@ -149,3 +152,51 @@ func ReadRequest(reader io.Reader) (request *Socks5Request, err error) {
 	request.port = binary.BigEndian.Uint16(buffer)
 	return
 }
+
+const (
+	ErrorSuccess                 = byte(0x00)
+	ErrorGeneralFailure          = byte(0x01)
+	ErrorConnectionNotAllowed    = byte(0x02)
+	ErrorNetworkUnreachable      = byte(0x03)
+	ErrorHostUnUnreachable       = byte(0x04)
+	ErrorConnectionRefused       = byte(0x05)
+	ErrorTTLExpired              = byte(0x06)
+	ErrorCommandNotSupported     = byte(0x07)
+	ErrorAddressTypeNotSupported = byte(0x08)
+)
+
+type Socks5Response struct {
+	Version  byte
+	Error    byte
+	AddrType byte
+	IPv4     [4]byte
+	Domain   string
+	IPv6     [16]byte
+	Port     uint16
+}
+
+func (r Socks5Response) toBytes() []byte {
+	buffer := make([]byte, 0, 300)
+	buffer = append(buffer, r.Version)
+	buffer = append(buffer, r.Error)
+	buffer = append(buffer, 0x00) // reserved
+	buffer = append(buffer, r.AddrType)
+	switch r.AddrType {
+	case 0x01:
+		buffer = append(buffer, r.IPv4[:]...)
+	case 0x03:
+		buffer = append(buffer, byte(len(r.Domain)))
+		buffer = append(buffer, []byte(r.Domain)...)
+	case 0x04:
+		buffer = append(buffer, r.IPv6[:]...)
+	}
+	portBuffer := make([]byte, 2)
+	binary.BigEndian.PutUint16(portBuffer, r.Port)
+	buffer = append(buffer, portBuffer...)
+	return buffer
+}
+
+func WriteResponse(writer io.Writer, response Socks5Response) error {
+	_, err := writer.Write(response.toBytes())
+	return err
+}

io/socks/socks_test.go

@@ -68,3 +68,27 @@ func TestRequestRead(t *testing.T) {
 		t.Errorf("Expected port 53, but got %d", request.port)
 	}
 }
+
+func TestResponseToBytes(t *testing.T) {
+	response := Socks5Response{
+		socksVersion,
+		ErrorSuccess,
+		AddrTypeIPv4,
+		[4]byte{0x72, 0x72, 0x72, 0x72},
+		"",
+		[16]byte{},
+		uint16(53),
+	}
+	rawResponse := response.toBytes()
+	expectedBytes := []byte{
+		socksVersion,
+		ErrorSuccess,
+		byte(0x00),
+		AddrTypeIPv4,
+		0x72, 0x72, 0x72, 0x72,
+		byte(0x00), byte(0x035),
+	}
+	if !bytes.Equal(rawResponse, expectedBytes) {
+		t.Errorf("Expected response %v, but got %v", expectedBytes, rawResponse)
+	}
+}

net/sockshandler.go → net/socks/socks.go

@@ -1,4 +1,4 @@
-package net
+package socks
 
 import (
 	"net"

vconfig.go

@@ -1,17 +1,17 @@
 package core
 
-// User account that is used for connection to a VPoint
+// VUser is the user account that is used for connection to a VPoint
 type VUser struct {
-	id VID // The ID of this VUser.
+	Id VID // The ID of this VUser.
 }
 
-// The next VPoint server in the connection chain.
+// VNext is the next VPoint server in the connection chain.
 type VNext struct {
 	ServerAddress string  // Address of VNext server, in the form of "IP:Port"
 	User          []VUser // User accounts for accessing VNext.
 }
 
-// The config for VPoint server.
+// VConfig is the config for VPoint server.
 type VConfig struct {
 	Port           uint16 // Port of this VPoint server.
 	AllowedClients []VUser

vid.go

@@ -9,6 +9,7 @@ import (
 // The ID of en entity, in the form of an UUID.
 type VID [16]byte
 
+// Hash generates a MD5 hash based on current VID and a suffix string.
 func (v VID) Hash(suffix []byte) []byte {
 	md5 := md5.New()
 	md5.Write(v[:])

vpoint.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 )
 
+// VPoint is an single server in V2Ray system.
 type VPoint struct {
 	config      VConfig
 	connHandler ConnectionHandler

vuserset.go

@@ -1,5 +1,39 @@
 package core
 
+import (
+	"encoding/base64"
+)
+
 type VUserSet struct {
-	validUserIds [][]byte
+	validUserIds   []VID
+	userIdsAskHash map[string]int
+}
+
+func NewVUserSet() *VUserSet {
+	vuSet := new(VUserSet)
+	vuSet.validUserIds = make([]VID, 0, 16)
+	return vuSet
+}
+
+func hashBytesToString(hash []byte) string {
+	return base64.StdEncoding.EncodeToString(hash)
+}
+
+func (us *VUserSet) AddUser(user VUser) error {
+	id := user.Id
+	us.validUserIds = append(us.validUserIds, id)
+
+	idBase64 := hashBytesToString(id.Hash([]byte("ASK")))
+	us.userIdsAskHash[idBase64] = len(us.validUserIds) - 1
+
+	return nil
+}
+
+func (us VUserSet) IsValidUserId(askHash []byte) (*VID, bool) {
+	askBase64 := hashBytesToString(askHash)
+	idIndex, found := us.userIdsAskHash[askBase64]
+	if found {
+		return &us.validUserIds[idIndex], true
+	}
+	return nil, false
 }