[waf]改进正则表达式缓存

Author: iwind

teawaf/utils/utils.go

@@ -13,6 +13,11 @@ var grid = teamemory.NewGrid(32, teamemory.NewLimitCountOpt(1000_0000))
 
 // 正则表达式匹配字符串，并缓存结果
 func MatchStringCache(regex *regexp.Regexp, s string) bool {
+	// 如果长度超过4096，大概率是不能重用的
+	if len(s) > 4096 {
+		return regex.MatchString(s)
+	}
+
 	hash := siphash.Hash(0, 0, teautils.UnsafeStringToBytes(s))
 	key := []byte(fmt.Sprintf("%p_", regex) + strconv.FormatUint(hash, 10))
 	item := grid.Read(key)
@@ -30,6 +35,11 @@ func MatchStringCache(regex *regexp.Regexp, s string) bool {
 
 // 正则表达式匹配字节slice，并缓存结果
 func MatchBytesCache(regex *regexp.Regexp, byteSlice []byte) bool {
+	// 如果长度超过4096，大概率是不能重用的
+	if len(byteSlice) > 4096 {
+		return regex.Match(byteSlice)
+	}
+
 	hash := siphash.Hash(0, 0, byteSlice)
 	key := []byte(fmt.Sprintf("%p_", regex) + strconv.FormatUint(hash, 10))
 	item := grid.Read(key)

teawaf/utils/utils_test.go

@@ -21,7 +21,7 @@ func TestMatchBytesCache(t *testing.T) {
 }
 
 func BenchmarkMatchStringCache(b *testing.B) {
-	data := strings.Repeat("HELLO", 1024*4)
+	data := strings.Repeat("HELLO", 512)
 	regex := regexp.MustCompile(`(?iU)\b(eval|system|exec|execute|passthru|shell_exec|phpinfo)\b`)
 
 	for i := 0; i < b.N; i++ {
@@ -30,7 +30,7 @@ func BenchmarkMatchStringCache(b *testing.B) {
 }
 
 func BenchmarkMatchStringCache_WithoutCache(b *testing.B) {
-	data := strings.Repeat("HELLO", 1024*4)
+	data := strings.Repeat("HELLO", 512)
 	regex := regexp.MustCompile(`(?iU)\b(eval|system|exec|execute|passthru|shell_exec|phpinfo)\b`)
 
 	for i := 0; i < b.N; i++ {