-
Notifications
You must be signed in to change notification settings - Fork 354
/
fuzz_test.go
95 lines (79 loc) · 2.02 KB
/
fuzz_test.go
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
//go:build go1.18
// +build go1.18
package ldap
import (
"testing"
"github.com/stretchr/testify/assert"
)
func FuzzParseDN(f *testing.F) {
f.Add("*")
f.Add("cn=Jim\\0Test")
f.Add("cn=Jim\\0")
f.Add("DC=example,=net")
f.Add("o=a+o=B")
f.Fuzz(func(t *testing.T, input_data string) {
_, _ = ParseDN(input_data)
})
}
func FuzzDecodeEscapedSymbols(f *testing.F) {
f.Add([]byte("a\u0100\x80"))
f.Add([]byte(`start\d`))
f.Add([]byte(`\`))
f.Add([]byte(`start\--end`))
f.Add([]byte(`start\d0\hh`))
f.Fuzz(func(t *testing.T, input_data []byte) {
_, _ = decodeEscapedSymbols(input_data)
})
}
func FuzzEscapeDN(f *testing.F) {
f.Add("test,user")
f.Add("#test#user#")
f.Add("\\test\\user\\")
f.Add(" test user ")
f.Add("\u0000te\x00st\x00user" + string(rune(0)))
f.Add("test\"+,;<>\\-_user")
f.Add("test\u0391user ")
f.Fuzz(func(t *testing.T, input_data string) {
_ = EscapeDN(input_data)
})
}
func FuzzRoundTripRDNSequence(f *testing.F) {
f.Add("CN=foo-long.com,OU=FooLong,OU=Barq,OU=Baz,OU=Dept.,O=Corp.,C=US")
f.Add("CN=foo-lon❤️\\,g.com,OU=Foo===Long,OU=Ba # rq,OU=Baz,O=C\\; orp.,C=US")
f.Add("CN=fo\x00o-long.com,OU=\x04FooLong")
f.Add("İ=")
f.Fuzz(func(t *testing.T, subjectString string) {
t.Parallel()
rdnSeq, err := ParseDN(subjectString)
if err != nil {
t.Skip()
}
newRDNSeq, err := ParseDN(rdnSeq.String())
if err != nil {
t.Fatal(err)
}
assert.True(t, rdnSeq.Equal(newRDNSeq))
assert.True(t, rdnSeq.EqualFold(newRDNSeq))
})
}
func FuzzRoundTripEncodeDecode(f *testing.F) {
f.Add("dffad=-fasdfsd")
f.Add("❤️\\,")
f.Add("aaa\x00o-long.c\x04FooLong")
f.Add("İ")
f.Fuzz(func(t *testing.T, rawString string) {
t.Parallel()
keyEncoded := encodeString(rawString, true)
keyDecoded, err := decodeString(keyEncoded)
if err != nil {
t.Fatal(err)
}
assert.Equal(t, rawString, keyDecoded)
valueEncoded := encodeString(rawString, false)
valueDecoded, err := decodeString(valueEncoded)
if err != nil {
t.Fatal(err)
}
assert.Equal(t, rawString, valueDecoded)
})
}