Update VMess protocol

io/vmess/vmess.go

@@ -13,6 +13,7 @@ import (
 
 	"github.com/v2ray/v2ray-core"
 	v2io "github.com/v2ray/v2ray-core/io"
+	"github.com/v2ray/v2ray-core/log"
 	v2net "github.com/v2ray/v2ray-core/net"
 )
 
@@ -24,12 +25,11 @@ const (
 	Version = byte(0x01)
 
 	blockSize = 16
-
-	CryptoMessage = "c48619fe-8f02-49e0-b9e9-edf763e17e21"
 )
 
 var (
-	ErrorInvalidUser = errors.New("Invalid User")
+	ErrorInvalidUser   = errors.New("Invalid User")
+	ErrorInvalidVerion = errors.New("Invalid Version")
 
 	emptyIV = make([]byte, blockSize)
 )
@@ -62,17 +62,13 @@ func (r *VMessRequestReader) Read(reader io.Reader) (*VMessRequest, error) {
 	request := new(VMessRequest)
 
 	buffer := make([]byte, 256)
-	nBytes, err := reader.Read(buffer[0:1])
-	if err != nil {
-		return nil, err
-	}
-	// TODO: verify version number
-	request.Version = buffer[0]
 
-	nBytes, err = reader.Read(buffer[:core.IDBytesLen])
+	nBytes, err := reader.Read(buffer[:core.IDBytesLen])
 	if err != nil {
 		return nil, err
 	}
+  
+  log.Debug("Read user hash: %v", buffer[:nBytes])
 
 	userId, valid := r.vUserSet.GetUser(buffer[:nBytes])
 	if !valid {
@@ -80,7 +76,7 @@ func (r *VMessRequestReader) Read(reader io.Reader) (*VMessRequest, error) {
 	}
 	request.UserId = *userId
 
-	aesCipher, err := aes.NewCipher(userId.Hash([]byte(CryptoMessage)))
+	aesCipher, err := aes.NewCipher(userId.CmdKey())
 	if err != nil {
 		return nil, err
 	}
@@ -105,6 +101,17 @@ func (r *VMessRequestReader) Read(reader io.Reader) (*VMessRequest, error) {
 		return nil, err
 	}
 
+	nBytes, err = decryptor.Read(buffer[0:1])
+	if err != nil {
+		return nil, err
+	}
+
+	request.Version = buffer[0]
+	if request.Version != Version {
+		log.Error("Unknown VMess version %d", request.Version)
+		return nil, ErrorInvalidVerion
+	}
+
 	// TODO: check number of bytes returned
 	_, err = decryptor.Read(request.RequestIV[:])
 	if err != nil {
@@ -182,8 +189,10 @@ func NewVMessRequestWriter() *VMessRequestWriter {
 
 func (w *VMessRequestWriter) Write(writer io.Writer, request *VMessRequest) error {
 	buffer := make([]byte, 0, 300)
-	buffer = append(buffer, request.Version)
-	buffer = append(buffer, request.UserId.TimeRangeHash(30)...)
+  userHash := request.UserId.TimeRangeHash(30)
+  
+  log.Debug("Writing userhash: %v", userHash)
+	buffer = append(buffer, userHash...)
 
 	encryptionBegin := len(buffer)
 
@@ -196,6 +205,7 @@ func (w *VMessRequestWriter) Write(writer io.Writer, request *VMessRequest) erro
 	buffer = append(buffer, byte(randomLength))
 	buffer = append(buffer, randomContent...)
 
+	buffer = append(buffer, request.Version)
 	buffer = append(buffer, request.RequestIV[:]...)
 	buffer = append(buffer, request.RequestKey[:]...)
 	buffer = append(buffer, request.ResponseHeader[:]...)
@@ -231,7 +241,7 @@ func (w *VMessRequestWriter) Write(writer io.Writer, request *VMessRequest) erro
 	buffer = append(buffer, paddingBuffer...)
 	encryptionEnd := len(buffer)
 
-	aesCipher, err := aes.NewCipher(request.UserId.Hash([]byte(CryptoMessage)))
+	aesCipher, err := aes.NewCipher(request.UserId.CmdKey())
 	if err != nil {
 		return err
 	}

io/vmess/vmess_test.go

@@ -3,6 +3,7 @@ package vmess
 import (
 	"bytes"
 	"crypto/rand"
+	"io/ioutil"
 	"testing"
 
 	"github.com/v2ray/v2ray-core"
@@ -51,7 +52,7 @@ func TestVMessSerialization(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	userSet.UserHashes[string(buffer.Bytes()[1:17])] = 0
+	userSet.UserHashes[string(buffer.Bytes()[:16])] = 0
 
 	requestReader := NewVMessRequestReader(&userSet)
 	actualRequest, err := requestReader.Read(buffer)
@@ -67,3 +68,25 @@ func TestVMessSerialization(t *testing.T) {
 	assert.Byte(actualRequest.Command).Named("Command").Equals(request.Command)
 	assert.String(actualRequest.Address.String()).Named("Address").Equals(request.Address.String())
 }
+
+func BenchmarkVMessRequestWriting(b *testing.B) {
+	userId, _ := core.NewID("2b2966ac-16aa-4fbf-8d81-c5f172a3da51")
+	userSet := mocks.MockUserSet{[]core.ID{}, make(map[string]int)}
+	userSet.AddUser(core.User{userId})
+
+	request := new(VMessRequest)
+	request.Version = byte(0x01)
+	request.UserId = userId
+
+	rand.Read(request.RequestIV[:])
+	rand.Read(request.RequestKey[:])
+	rand.Read(request.ResponseHeader[:])
+
+	request.Command = byte(0x01)
+	request.Address = v2net.DomainAddress("v2ray.com", 80)
+
+	requestWriter := NewVMessRequestWriter()
+	for i := 0; i < b.N; i++ {
+		requestWriter.Write(ioutil.Discard, request)
+	}
+}