From 3ce6bb474d70039214f1766960b85266ddb0506b Mon Sep 17 00:00:00 2001 From: GoEdgeLab Date: Sun, 10 Dec 2023 09:27:29 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0WAF=E7=9B=B8=E5=85=B3?= =?UTF-8?q?=E6=B5=8B=E8=AF=95=E7=94=A8=E4=BE=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/waf/template_test.go | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) diff --git a/internal/waf/template_test.go b/internal/waf/template_test.go index 7a32b73..89d7d2b 100644 --- a/internal/waf/template_test.go +++ b/internal/waf/template_test.go @@ -467,3 +467,35 @@ func BenchmarkTemplatePathTraversal(b *testing.B) { } }) } + +func BenchmarkTemplateCC2(b *testing.B) { + runtime.GOMAXPROCS(4) + + template, err := waf.Template() + if err != nil { + b.Fatal(err) + } + var group = template.FindRuleGroupWithCode("cc2") + if group == nil { + b.Fatal("group not found") + return + } + + b.ResetTimer() + + b.RunParallel(func(pb *testing.PB) { + for pb.Next() { + req, err := http.NewRequest(http.MethodPost, "https://example.com/?id=1234"+types.String(rand.Int()%10000)+"&name=lily&time=12345678910", nil) + if err != nil { + b.Fatal(err) + } + req.Header.Set("User-Agent", testUserAgent) + + _, _, result, err := group.MatchRequest(requests.NewTestRequest(req)) + if err != nil { + b.Fatal(err) + } + _ = result + } + }) +}