增加CC防护相关API定义、配置

pkg/serverconfigs/shared/url_pattern.go

@@ -34,7 +34,11 @@ func (this *URLPattern) Init() error {
 		for index, piece := range pieces {
 			pieces[index] = regexp.QuoteMeta(piece)
 		}
-		reg, err := regexp.Compile("(?i)" /** 大小写不敏感 **/ + "^" + strings.Join(pieces, "(.*)") + "$")
+		var pattern = strings.Join(pieces, "(.*)")
+		if len(pattern) > 0 && pattern[0] == '/' {
+			pattern = "(http|https)://[\\w.-]+" + pattern
+		}
+		reg, err := regexp.Compile("(?i)" /** 大小写不敏感 **/ + "^" + pattern + "$")
 		if err != nil {
 			return err
 		}

pkg/serverconfigs/shared/url_pattern_test.go

@@ -4,13 +4,10 @@ package shared_test
 
 import (
 	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
-	"github.com/iwind/TeaGo/assert"
 	"testing"
 )
 
 func TestURLPattern_Match(t *testing.T) {
-	var a = assert.NewAssertion(t)
-
 	type unitTest struct {
 		patternType string
 		pattern     string
@@ -55,6 +52,24 @@ func TestURLPattern_Match(t *testing.T) {
 			url:         "https://example.com",
 			result:      false,
 		},
+		{
+			patternType: "wildcard",
+			pattern:     "https://example.com",
+			url:         "https://example.com",
+			result:      true,
+		},
+		{
+			patternType: "wildcard",
+			pattern:     "/hello/world",
+			url:         "https://example-test.com/hello/world",
+			result:      true,
+		},
+		{
+			patternType: "wildcard",
+			pattern:     "/hello/world",
+			url:         "https://example-test.com/123/hello/world",
+			result:      false,
+		},
 		{
 			patternType: "regexp",
 			pattern:     ".*",
@@ -94,6 +109,9 @@ func TestURLPattern_Match(t *testing.T) {
 		if err != nil {
 			t.Fatal(err)
 		}
-		a.IsTrue(p.Match(ut.url) == ut.result)
+		var b = p.Match(ut.url) == ut.result
+		if !b {
+			t.Fatal("not matched pattern:", ut.pattern, "url:", ut.url)
+		}
 	}
 }