| // Copyright 2015 The Vanadium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package discovery |
| |
| import ( |
| "bytes" |
| "strconv" |
| "strings" |
| "testing" |
| |
| "v.io/v23/discovery" |
| ) |
| |
| func TestValidateAd(t *testing.T) { |
| tests := []struct { |
| ad discovery.Advertisement |
| valid bool |
| }{ |
| { |
| discovery.Advertisement{ |
| Id: discovery.AdId{1, 2, 3}, |
| InterfaceName: "v.io/v23/a", |
| Addresses: []string{"/h:123/x"}, |
| Attributes: discovery.Attributes{ |
| "key": "v", |
| "k_e.y": "v\u0ac0", |
| "k!": "v\n", |
| }, |
| Attachments: discovery.Attachments{ |
| "key": []byte{1, 2, 3}, |
| "k!": []byte{4, 5, 6}, |
| }, |
| }, |
| true, |
| }, |
| { |
| discovery.Advertisement{ |
| Id: discovery.AdId{1, 2, 3}, |
| InterfaceName: "v.io/v23/a", |
| Addresses: []string{"/h:123/x"}, |
| Attachments: discovery.Attachments{ |
| "k": nil, |
| }, |
| }, |
| true, |
| }, |
| { |
| discovery.Advertisement{ |
| Id: discovery.AdId{}, // Invalid id. |
| InterfaceName: "v.io/v23/a", |
| Addresses: []string{"/h:123/x"}, |
| }, |
| false, |
| }, |
| { |
| discovery.Advertisement{ // No interface name. |
| Id: discovery.AdId{1, 2, 3}, |
| Addresses: []string{"/h:123/z"}, |
| }, |
| false, |
| }, |
| { |
| discovery.Advertisement{ // No addresses. |
| Id: discovery.AdId{1, 2, 3}, |
| InterfaceName: "v.io/v23/a", |
| }, |
| false, |
| }, |
| { |
| discovery.Advertisement{ |
| Id: discovery.AdId{1, 2, 3}, |
| InterfaceName: "v.io/v23/a", |
| Addresses: []string{"/h:123/x"}, |
| Attributes: discovery.Attributes{ |
| "_key": "v", // Invalid key. |
| }, |
| }, |
| false, |
| }, |
| { |
| discovery.Advertisement{ |
| Id: discovery.AdId{1, 2, 3}, |
| InterfaceName: "v.io/v23/a", |
| Addresses: []string{"/h:123/x"}, |
| Attributes: discovery.Attributes{ |
| "k=ey": "v", // Invalid key. |
| }, |
| }, |
| false, |
| }, |
| { |
| discovery.Advertisement{ |
| Id: discovery.AdId{1, 2, 3}, |
| InterfaceName: "v.io/v23/a", |
| Addresses: []string{"/h:123/x"}, |
| Attachments: discovery.Attachments{ |
| "key\n": nil, // Invalid key. |
| }, |
| }, |
| false, |
| }, |
| { |
| discovery.Advertisement{ |
| Id: discovery.AdId{1, 2, 3}, |
| InterfaceName: "v.io/v23/a", |
| Addresses: []string{"/h:123/x"}, |
| Attributes: discovery.Attributes{ |
| "k": "\xd8\x00", // Invalid UTF-8. |
| }, |
| }, |
| false, |
| }, |
| { |
| discovery.Advertisement{ |
| Id: discovery.AdId{1, 2, 3}, |
| InterfaceName: "v.io/v23/a", |
| Addresses: []string{"/h:123/x"}, |
| Attributes: discovery.Attributes{ |
| "k": "\x12\x34\xab\xcd", // Invalid UTF-8. |
| }, |
| }, |
| false, |
| }, |
| { |
| discovery.Advertisement{ // Too large. |
| Id: discovery.AdId{1, 2, 3}, |
| InterfaceName: strings.Repeat("i", 100), |
| Addresses: []string{strings.Repeat("a", 100), strings.Repeat("b", 100)}, |
| Attributes: discovery.Attributes{ |
| "k12345": strings.Repeat("v", 100), |
| "k67890a": strings.Repeat("v", 100), |
| // TODO(jhahn): Remove this after rolling back this temporary increase of size. |
| "tmp": strings.Repeat("v", 1000), |
| }, |
| }, |
| false, |
| }, |
| { |
| discovery.Advertisement{ |
| Id: discovery.AdId{1, 2, 3}, |
| InterfaceName: "v.io/v23/a", |
| Addresses: []string{"/h:123/x"}, |
| Attributes: genAttributes(33), // Too many. |
| }, |
| false, |
| }, |
| { |
| discovery.Advertisement{ |
| Id: discovery.AdId{1, 2, 3}, |
| InterfaceName: "v.io/v23/a", |
| Addresses: []string{"/h:123/x"}, |
| Attachments: discovery.Attachments{ |
| "k1": bytes.Repeat([]byte{1}, 100), |
| "k2": bytes.Repeat([]byte{1}, 4097), // Too large. |
| }, |
| }, |
| false, |
| }, |
| { |
| discovery.Advertisement{ |
| Id: discovery.AdId{1, 2, 3}, |
| InterfaceName: "v.io/v23/a", |
| Addresses: []string{"/h:123/x"}, |
| Attachments: genAttachments(33), // Too many. |
| }, |
| false, |
| }, |
| // Real advertisement seen in the wild, with blessings replaced by equal-length strings |
| { |
| discovery.Advertisement{ |
| Id: discovery.AdId{0xd2, 0xa9, 0x48, 0x63, 0x5a, 0xbd, 0xd6, 0x85, 0xde, 0x3b, 0x2c, 0x1a, 0x4a, 0x23, 0xe7, 0x3e}, |
| InterfaceName: "v.io/x/ref/services/syncbase/server/interfaces/Sync", |
| Addresses: []string{ |
| "/@6@wsh@100.110.64.64:47011@@2c8d255c1b25e90cba07f5c857086e3b@s@idprovider:o:there_was_a_very_long_application_identifier_at_this_point_right_here:username@someplace.com@@", |
| "/@6@wsh@8.34.219.227:8100@@2c8d255c1b25e90cba07f5c857086e3b@s@idprovider:o:there_was_a_very_long_application_identifier_at_this_point_right_here:username@someplace.com@@", |
| "/@6@wsh@[2620:0:1000:fd86:66bc:cff:fe51:6cb4]:47011@@2c8d255c1b25e90cba07f5c857086e3b@s@idprovider:o:there_was_a_very_long_application_identifier_at_this_point_right_here:username@someplace.com@@", |
| "/@6@wsh@[2620:0:1000:fd86:d94b:86d7:caff:b48f]:47011@@2c8d255c1b25e90cba07f5c857086e3b@s@idprovider:o:there_was_a_very_long_application_identifier_at_this_point_right_here:username@someplace.com@@", |
| "/@6@wsh@[fe80::2409:8aff:fe2e:f60e]:47011@@2c8d255c1b25e90cba07f5c857086e3b@s@idprovider:o:there_was_a_very_long_application_identifier_at_this_point_right_here:username@someplace.com@@", |
| "/@6@wsh@[fe80::66bc:cff:fe51:6cb4]:47011@@2c8d255c1b25e90cba07f5c857086e3b@s@idprovider:o:there_was_a_very_long_application_identifier_at_this_point_right_here:username@someplace.com@@", |
| "/@6@wsh@[fe80::f33f:4a65:4fe7:fc38]:47011@@2c8d255c1b25e90cba07f5c857086e3b@s@idprovider:o:there_was_a_very_long_application_identifier_at_this_point_right_here:username@someplace.com@@", |
| }, |
| Attributes: discovery.Attributes{ |
| "sg": "/ns.dev.v.io:8101/tmp/todos/users/cloud/%%sync/list_lists_JMxfYoYJzPaQGV9Utj1V_KB6epir3Zz8J", |
| }, |
| }, |
| true, |
| }, |
| } |
| |
| for i, test := range tests { |
| err := validateAd(&test.ad) |
| if test.valid { |
| if err != nil { |
| t.Errorf("[%d]: unexpected error: %v", i, err) |
| } |
| } else { |
| if err == nil { |
| t.Errorf("[%d]: expected an error; but got none", i) |
| } |
| } |
| } |
| } |
| |
| func genAttributes(n int) discovery.Attributes { |
| attributes := make(discovery.Attributes, n) |
| for i := 0; i < n; i++ { |
| attributes[strconv.Itoa(i)] = "" |
| } |
| return attributes |
| } |
| |
| func genAttachments(n int) discovery.Attachments { |
| attachments := make(discovery.Attachments, n) |
| for i := 0; i < n; i++ { |
| attachments[strconv.Itoa(i)] = nil |
| } |
| return attachments |
| } |
