mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-04 15:51:54 +08:00
264 lines
8.1 KiB
Go
264 lines
8.1 KiB
Go
// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
|
|
|
|
package re_test
|
|
|
|
import (
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
|
|
"github.com/TeaOSLab/EdgeNode/internal/re"
|
|
"github.com/iwind/TeaGo/assert"
|
|
"regexp"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestRegexp(t *testing.T) {
|
|
for _, s := range []string{"(?i)(abc|efg)", "abc|efg", "abc(.+)"} {
|
|
var reg = regexp.MustCompile(s)
|
|
t.Log("===" + s + "===")
|
|
t.Log(reg.LiteralPrefix())
|
|
t.Log(reg.NumSubexp())
|
|
t.Log(reg.SubexpNames())
|
|
}
|
|
}
|
|
|
|
func TestRegexp_MatchString(t *testing.T) {
|
|
var a = assert.NewAssertion(t)
|
|
|
|
{
|
|
var r = re.MustCompile("abc")
|
|
a.IsTrue(r.MatchString("abc"))
|
|
a.IsFalse(r.MatchString("ab"))
|
|
a.IsFalse(r.MatchString("ABC"))
|
|
}
|
|
|
|
{
|
|
var r = re.MustCompile("(?i)abc|def|ghi")
|
|
a.IsTrue(r.MatchString("DEF"))
|
|
a.IsFalse(r.MatchString("ab"))
|
|
a.IsTrue(r.MatchString("ABC"))
|
|
}
|
|
}
|
|
|
|
func TestRegexp_Sub(t *testing.T) {
|
|
{
|
|
reg := regexp.MustCompile(`(a|b|c)(e|f|g)`)
|
|
for _, subName := range reg.SubexpNames() {
|
|
t.Log(subName)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestRegexp_ParseKeywords(t *testing.T) {
|
|
var r = re.MustCompile("")
|
|
|
|
{
|
|
var keywords = r.ParseKeywords(`\n\t\n\f\r\v\x123`)
|
|
t.Log(keywords)
|
|
}
|
|
}
|
|
|
|
func TestRegexp_Special(t *testing.T) {
|
|
for _, s := range []string{
|
|
`\\s`,
|
|
`\s\W`,
|
|
`aaaa/\W`,
|
|
`aaaa\/\W`,
|
|
`aaaa\=\W`,
|
|
`aaaa\\=\W`,
|
|
`aaaa\\\=\W`,
|
|
`aaaa\\\\=\W`,
|
|
} {
|
|
var es = testUnescape(t, s)
|
|
t.Log(s, "=>", es)
|
|
_, err := re.Compile(es)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestRegexp_Special2(t *testing.T) {
|
|
r, err := re.Compile(testUnescape(t, `/api/ios/a
|
|
/api/ios/b
|
|
/api/ios/c
|
|
/report`))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Log(r.Keywords())
|
|
}
|
|
|
|
func TestRegexp_ParseKeywords2(t *testing.T) {
|
|
var a = assert.NewAssertion(t)
|
|
|
|
var r = re.MustCompile("")
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("(abc)def"), []string{"abcdef"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("(abc)|(?:def)"), []string{"abc", "def"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("(abc)"), []string{"abc"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("(abc|def|ghi)"), []string{"abc", "def", "ghi"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("(?i:abc)"), []string{}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(`\babc`), []string{"abc"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(` \babc`), []string{" "}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(`\babc\b`), []string{"abc"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(`\b(abc)`), []string{"abc"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("abc"), []string{"abc"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("abc|efg|hij"), []string{"abc", "efg", "hij"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(`abc\|efg|hij`), []string{"abc|efg", "hij"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(`abc\|efg*|hij`), []string{"abc|ef", "hij"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(`abc\|efg?|hij`), []string{"abc|ef", "hij"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(`abc\|efg+|hij`), []string{"abc|ef", "hij"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(`abc\|efg{2,10}|hij`), []string{"abc|ef", "hij"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(`abc\|efg{0,10}|hij`), []string{"abc|ef", "hij"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(`abc\|efg.+|hij`), []string{"abc|efg", "hij"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("A(abc|bcd)"), []string{"Aabc", "Abcd"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("^abc"), []string{"abc"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("abc$"), []string{"abc"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(`abc$`), []string{"abc"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("abc\\d"), []string{"abc"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("abc{0,4}"), []string{"ab"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("{0,4}"), []string{}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("{1,4}"), []string{}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords("中文|北京|上海|golang"), []string{"中文", "北京", "上海", "golang"}))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(`(onmouseover|onmousemove|onmousedown|onmouseup|onerror|onload|onclick|ondblclick)\s*=`), strings.Split("onmouseover|onmousemove|onmousedown|onmouseup|onerror|onload|onclick|ondblclick", "|")))
|
|
a.IsTrue(testCompareStrings(r.ParseKeywords(`/\*(!|\x00)`), []string{"/*"}))
|
|
}
|
|
|
|
func TestRegexp_ParseKeywords3(t *testing.T) {
|
|
var r = re.MustCompile("")
|
|
|
|
var policy = firewallconfigs.HTTPFirewallTemplate()
|
|
for _, group := range policy.Inbound.Groups {
|
|
for _, set := range group.Sets {
|
|
for _, rule := range set.Rules {
|
|
if rule.Operator == firewallconfigs.HTTPFirewallRuleOperatorMatch || rule.Operator == firewallconfigs.HTTPFirewallRuleOperatorNotMatch {
|
|
t.Log(set.Name+":", rule.Value, "=>", r.ParseKeywords(rule.Value))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkRegexp_MatchString(b *testing.B) {
|
|
var r = re.MustCompile("(?i)(onmouseover|onmousemove|onmousedown|onmouseup|onerror|onload|onclick|ondblclick|onkeydown|onkeyup|onkeypress)(\\s|%09|%0A|(\\+|%20))*(=|%3D)")
|
|
b.ResetTimer()
|
|
|
|
//b.Log("keywords:", r.Keywords())
|
|
for i := 0; i < b.N; i++ {
|
|
r.MatchString("Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36")
|
|
}
|
|
}
|
|
|
|
func BenchmarkRegexp_MatchString2(b *testing.B) {
|
|
var r = regexp.MustCompile(`(?i)(onmouseover|onmousemove|onmousedown|onmouseup|onerror|onload|onclick|ondblclick|onkeydown|onkeyup|onkeypress)(\s|%09|%0A|(\+|%20))*(=|%3D)`)
|
|
b.ResetTimer()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
r.MatchString("Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36")
|
|
}
|
|
}
|
|
|
|
func BenchmarkRegexp_MatchString_CaseSensitive(b *testing.B) {
|
|
var r = re.MustCompile("(abc|def|ghi)")
|
|
b.Log("keywords:", r.Keywords())
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
r.MatchString("Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36")
|
|
}
|
|
}
|
|
|
|
func BenchmarkRegexp_MatchString_CaseSensitive2(b *testing.B) {
|
|
var r = regexp.MustCompile("(abc|def|ghi)")
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
r.MatchString("Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36")
|
|
}
|
|
}
|
|
|
|
func BenchmarkRegexp_MatchString_VS_FindSubString1(b *testing.B) {
|
|
var r = re.MustCompile("(?i)(chrome)")
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_ = r.Raw().MatchString("Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36")
|
|
}
|
|
}
|
|
|
|
func BenchmarkRegexp_MatchString_VS_FindSubString2(b *testing.B) {
|
|
var r = re.MustCompile("(?i)(chrome)")
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_ = r.Raw().FindStringSubmatch("Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36")
|
|
}
|
|
}
|
|
|
|
func TestSplitAndJoin(t *testing.T) {
|
|
var pieces = strings.Split(`/api/ios/a
|
|
/api/ios/b
|
|
/api/ios/c
|
|
/report`, "/")
|
|
t.Log(strings.Join(pieces, `(/|%2F)`))
|
|
}
|
|
|
|
func testCompareStrings(s1 []string, s2 []string) bool {
|
|
if len(s1) != len(s2) {
|
|
return false
|
|
}
|
|
for index, s := range s1 {
|
|
if s != s2[index] {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func testUnescape(t *testing.T, v string) string {
|
|
// replace urlencoded characters
|
|
var unescapeChars = [][2]string{
|
|
{`\s`, `(\s|%09|%0A|\+)`},
|
|
{`\(`, `(\(|%28)`},
|
|
{`=`, `(=|%3D)`},
|
|
{`<`, `(<|%3C)`},
|
|
{`\*`, `(\*|%2A)`},
|
|
{`\\`, `(\\|%2F)`},
|
|
{`!`, `(!|%21)`},
|
|
{`/`, `(/|%2F)`},
|
|
{`;`, `(;|%3B)`},
|
|
{`\+`, `(\+|%20)`},
|
|
}
|
|
|
|
for _, c := range unescapeChars {
|
|
if !strings.Contains(v, c[0]) {
|
|
continue
|
|
}
|
|
var pieces = strings.Split(v, c[0])
|
|
|
|
// 修复piece中错误的\
|
|
for pieceIndex, piece := range pieces {
|
|
var l = len(piece)
|
|
if l == 0 {
|
|
continue
|
|
}
|
|
if piece[l-1] != '\\' {
|
|
continue
|
|
}
|
|
|
|
// 计算\的数量
|
|
var countBackSlashes = 0
|
|
for i := l - 1; i >= 0; i-- {
|
|
if piece[i] == '\\' {
|
|
countBackSlashes++
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
if countBackSlashes%2 == 1 {
|
|
// 去掉最后一个
|
|
pieces[pieceIndex] = piece[:len(piece)-1]
|
|
}
|
|
}
|
|
|
|
v = strings.Join(pieces, c[1])
|
|
}
|
|
|
|
return v
|
|
}
|