1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
package nametransform
import (
"bytes"
"strings"
"testing"
)
func TestPad16(t *testing.T) {
s := [][]byte{
[]byte("foo"),
[]byte("12345678901234567"),
[]byte("12345678901234567abcdefg"),
}
for i := range s {
orig := s[i]
padded := pad16(orig)
if len(padded) <= len(orig) {
t.Errorf("Padded length not bigger than orig: %d", len(padded))
}
if len(padded)%16 != 0 {
t.Errorf("Length is not aligend: %d", len(padded))
}
unpadded, err := unPad16(padded)
if err != nil {
t.Error("unPad16 returned error:", err)
}
if len(unpadded) != len(orig) {
t.Errorf("Size mismatch: orig=%d unpadded=%d", len(s[i]), len(unpadded))
}
if !bytes.Equal(orig, unpadded) {
t.Error("Content mismatch orig vs unpadded")
}
}
}
// TestUnpad16Garbage - unPad16 should never crash on corrupt or malicious inputs
func TestUnpad16Garbage(t *testing.T) {
testCases := [][]byte{
make([]byte, 0),
make([]byte, 16),
make([]byte, 1),
make([]byte, 17),
bytes.Repeat([]byte{16}, 16),
bytes.Repeat([]byte{17}, 16),
}
for _, v := range testCases {
_, err := unPad16([]byte(v))
if err == nil {
t.Fail()
}
}
}
func TestIsValidName(t *testing.T) {
cases := []struct {
in string
want bool
}{
{"", false},
{".", false},
{"..", false},
{"...", true},
{"asdasd/asdasd", false},
{"asdasd\000asdasd", false},
{"hello", true},
{strings.Repeat("x", 255), true},
{strings.Repeat("x", 256), false},
}
for _, c := range cases {
have := IsValidName(c.in)
if (have == nil) != c.want {
t.Errorf("IsValidName(%q): want %v have %v", c.in, c.want, have)
}
}
}
|
