func TestVMessUDPReadWrite(t *testing.T) { assert := unit.Assert(t) userId, err := user.NewID("2b2966ac-16aa-4fbf-8d81-c5f172a3da51") assert.Error(err).IsNil() userSet := mocks.MockUserSet{[]user.ID{}, make(map[string]int), make(map[string]int64)} userSet.AddUser(user.User{userId}) message := &VMessUDP{ user: userId, version: byte(0x01), token: 1234, address: v2net.DomainAddress("v2ray.com", 8372), data: []byte("An UDP message."), } mockTime := int64(1823730) buffer := message.ToBytes(user.NewTimeHash(user.HMACHash{}), func(base int64, delta int) int64 { return mockTime }, nil) userSet.UserHashes[string(buffer[:16])] = 0 userSet.Timestamps[string(buffer[:16])] = mockTime messageRestored, err := ReadVMessUDP(buffer, &userSet) assert.Error(err).IsNil() assert.String(messageRestored.user.String).Equals(message.user.String) assert.Byte(messageRestored.version).Equals(message.version) assert.Uint16(messageRestored.token).Equals(message.token) assert.String(messageRestored.address.String()).Equals(message.address.String()) assert.Bytes(messageRestored.data).Equals(message.data) }
func BenchmarkVMessRequestWriting(b *testing.B) { userId, _ := config.NewID("2b2966ac-16aa-4fbf-8d81-c5f172a3da51") userSet := mocks.MockUserSet{[]config.User{}, make(map[string]int), make(map[string]int64)} testUser := &TestUser{ id: userId, } userSet.AddUser(testUser) request := new(VMessRequest) request.Version = byte(0x01) request.User = testUser randBytes := make([]byte, 36) rand.Read(randBytes) request.RequestIV = randBytes[:16] request.RequestKey = randBytes[16:32] request.ResponseHeader = randBytes[32:] request.Command = byte(0x01) request.Address = v2net.DomainAddress("v2ray.com", 80) for i := 0; i < b.N; i++ { request.ToBytes(user.NewTimeHash(user.HMACHash{}), user.GenerateRandomInt64InRange, nil) } }
func TestVMessSerialization(t *testing.T) { assert := unit.Assert(t) userId, err := core.NewID("2b2966ac-16aa-4fbf-8d81-c5f172a3da51") if err != nil { t.Fatal(err) } userSet := mocks.MockUserSet{[]core.ID{}, make(map[string]int), make(map[string]int64)} userSet.AddUser(core.User{userId}) request := new(VMessRequest) request.Version = byte(0x01) request.UserId = userId _, err = rand.Read(request.RequestIV[:]) if err != nil { t.Fatal(err) } _, err = rand.Read(request.RequestKey[:]) if err != nil { t.Fatal(err) } _, err = rand.Read(request.ResponseHeader[:]) if err != nil { t.Fatal(err) } request.Command = byte(0x01) request.Address = v2net.DomainAddress("v2ray.com", 80) buffer := bytes.NewBuffer(make([]byte, 0, 300)) mockTime := int64(1823730) requestWriter := NewVMessRequestWriter(v2hash.NewTimeHash(v2hash.HMACHash{}), func(base int64, delta int) int64 { return mockTime }) err = requestWriter.Write(buffer, request) if err != nil { t.Fatal(err) } userSet.UserHashes[string(buffer.Bytes()[:16])] = 0 userSet.Timestamps[string(buffer.Bytes()[:16])] = mockTime requestReader := NewVMessRequestReader(&userSet) actualRequest, err := requestReader.Read(buffer) if err != nil { t.Fatal(err) } assert.Byte(actualRequest.Version).Named("Version").Equals(byte(0x01)) assert.String(actualRequest.UserId.String).Named("UserId").Equals(request.UserId.String) assert.Bytes(actualRequest.RequestIV[:]).Named("RequestIV").Equals(request.RequestIV[:]) assert.Bytes(actualRequest.RequestKey[:]).Named("RequestKey").Equals(request.RequestKey[:]) assert.Bytes(actualRequest.ResponseHeader[:]).Named("ResponseHeader").Equals(request.ResponseHeader[:]) assert.Byte(actualRequest.Command).Named("Command").Equals(request.Command) assert.String(actualRequest.Address.String()).Named("Address").Equals(request.Address.String()) }
func TestVMessSerialization(t *testing.T) { assert := unit.Assert(t) userId, err := config.NewID("2b2966ac-16aa-4fbf-8d81-c5f172a3da51") if err != nil { t.Fatal(err) } testUser := &TestUser{ id: userId, } userSet := mocks.MockUserSet{[]config.User{}, make(map[string]int), make(map[string]int64)} userSet.AddUser(testUser) request := new(VMessRequest) request.Version = byte(0x01) request.User = testUser randBytes := make([]byte, 36) _, err = rand.Read(randBytes) assert.Error(err).IsNil() request.RequestIV = randBytes[:16] request.RequestKey = randBytes[16:32] request.ResponseHeader = randBytes[32:] request.Command = byte(0x01) request.Address = v2net.DomainAddress("v2ray.com", 80) mockTime := int64(1823730) buffer, err := request.ToBytes(user.NewTimeHash(user.HMACHash{}), func(base int64, delta int) int64 { return mockTime }, nil) if err != nil { t.Fatal(err) } userSet.UserHashes[string(buffer.Value[:16])] = 0 userSet.Timestamps[string(buffer.Value[:16])] = mockTime requestReader := NewVMessRequestReader(&userSet) actualRequest, err := requestReader.Read(bytes.NewReader(buffer.Value)) if err != nil { t.Fatal(err) } assert.Byte(actualRequest.Version).Named("Version").Equals(byte(0x01)) assert.String(actualRequest.User.ID().String).Named("UserId").Equals(request.User.ID().String) assert.Bytes(actualRequest.RequestIV).Named("RequestIV").Equals(request.RequestIV[:]) assert.Bytes(actualRequest.RequestKey).Named("RequestKey").Equals(request.RequestKey[:]) assert.Bytes(actualRequest.ResponseHeader).Named("ResponseHeader").Equals(request.ResponseHeader[:]) 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, _ := user.NewID("2b2966ac-16aa-4fbf-8d81-c5f172a3da51") userSet := mocks.MockUserSet{[]user.ID{}, make(map[string]int), make(map[string]int64)} userSet.AddUser(user.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) for i := 0; i < b.N; i++ { request.ToBytes(user.NewTimeHash(user.HMACHash{}), user.GenerateRandomInt64InRange, nil) } }
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) } }