/
fuzz_test.go
129 lines (103 loc) · 2.97 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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
//go:build go1.18
// +build go1.18
package main_test
import (
"regexp"
"testing"
"github.com/CAFxX/regexp2go/examples/dna"
"github.com/CAFxX/regexp2go/examples/ipv6"
"github.com/CAFxX/regexp2go/examples/log_parse"
"github.com/CAFxX/regexp2go/examples/mail_crawler"
"github.com/CAFxX/regexp2go/examples/unicode"
"github.com/CAFxX/regexp2go/internal"
)
func check(t *testing.T, str string, re *regexp.Regexp, matches []string, index int, found bool) {
res := re.FindStringSubmatchIndex(str)
// refound := re.MatchString(str)
refound := res != nil
if found != refound {
t.Errorf("found: %v/%v", found, refound)
}
if !found {
return
}
// reindex := re.FindStringIndex(str)
reindex := res[0:2]
if index != reindex[0] {
t.Errorf("index start: %v/%v", index, reindex[0])
}
if index+len(matches[0]) != reindex[1] {
t.Errorf("index end: %v/%v", index+len(matches[0]), reindex[1])
}
// rematches := re.FindStringSubmatch(str)
for i := range matches {
var rematch string
if res[i*2] >= 0 {
rematch = str[res[i*2]:res[i*2+1]]
}
if matches[i] != rematch {
t.Errorf("match %d: %v/%v", i, matches[i], rematch)
}
}
}
func FuzzDna(f *testing.F) {
re := regexp.MustCompile(dna.MatchRegexp)
f.Add("aaaagggtaaagagattttt")
f.Add("agggcaaa")
f.Add("agggaaa")
f.Fuzz(func(t *testing.T, str string) {
matches, index, found := dna.Match{}.FindString(str)
check(t, str, re, matches[:], index, found)
})
}
func FuzzIpv6(f *testing.F) {
re := regexp.MustCompile(ipv6.MatchRegexp)
f.Add("::1")
f.Add("0000::")
f.Add("1234:5678:90ab:cdef:0000:0000:0000:0000")
f.Fuzz(func(t *testing.T, str string) {
matches, index, found := ipv6.Match{}.FindString(str)
check(t, str, re, matches[:], index, found)
})
}
func FuzzUnicode(f *testing.F) {
re := regexp.MustCompile(unicode.MatchRegexp)
f.Add("私はばかです")
f.Add("私はCAFxXです")
f.Add("私は私は私はですですです")
f.Add("私は🤣です")
f.Fuzz(func(t *testing.T, str string) {
matches, index, found := unicode.Match{}.FindString(str)
check(t, str, re, matches[:], index, found)
})
}
func FuzzMailCrawler(f *testing.F) {
re := regexp.MustCompile(mail_crawler.MatchRegexp)
f.Add("hello@example.com")
f.Add("my address is hello+world@example.com.")
f.Fuzz(func(t *testing.T, str string) {
matches, index, found := mail_crawler.Match{}.FindString(str)
check(t, str, re, matches[:], index, found)
})
}
func FuzzLogParse(f *testing.F) {
re := regexp.MustCompile(log_parse.MatchRegexp)
f.Add("no match\nINFO res=7 msg=hello world\nINFO res=5 msg=oh noes\n")
f.Add("INFO res=0000 msg=xxx")
f.Fuzz(func(t *testing.T, str string) {
matches, index, found := log_parse.Match{}.FindString(str)
check(t, str, re, matches[:], index, found)
})
}
func FuzzRegexp2Go(f *testing.F) {
f.Fuzz(func(t *testing.T, regex string) {
_, err := regexp.Compile(regex)
if err != nil {
return
}
_, err = internal.Generate(regex, "fuzz", "Fuzz", 212, true)
if err != nil {
t.Fatal(err)
}
})
}