优化鉴权

pkg/serverconfigs/http_auth_method_base.go

@@ -0,0 +1,99 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package serverconfigs
+
+import (
+	"github.com/TeaOSLab/EdgeCommon/pkg/configutils"
+	"github.com/iwind/TeaGo/lists"
+	"net/http"
+	"path/filepath"
+	"regexp"
+	"strings"
+)
+
+var httpAuthTimestampRegexp = regexp.MustCompile(`^\d{10}$`)
+
+type HTTPAuthBaseMethod struct {
+	Exts    []string `json:"exts"`
+	Domains []string `json:"domains"`
+}
+
+func (this *HTTPAuthBaseMethod) SetExts(exts []string) {
+	this.Exts = exts
+}
+
+func (this *HTTPAuthBaseMethod) SetDomains(domains []string) {
+	this.Domains = domains
+}
+
+func (this *HTTPAuthBaseMethod) removeQueryArgs(query string, args []string) string {
+	var pieces = strings.Split(query, "&")
+	var result = []string{}
+Loop:
+	for _, piece := range pieces {
+		for _, arg := range args {
+			if strings.HasPrefix(piece, arg+"=") {
+				continue Loop
+			}
+		}
+		result = append(result, piece)
+	}
+	return strings.Join(result, "&")
+}
+
+func (this *HTTPAuthBaseMethod) matchTimestamp(timestamp string) bool {
+	return httpAuthTimestampRegexp.MatchString(timestamp)
+}
+
+func (this *HTTPAuthBaseMethod) MatchRequest(req *http.Request) bool {
+	if len(this.Exts) > 0 {
+		var ext = filepath.Ext(req.URL.Path)
+		if len(ext) == 0 {
+			return false
+		}
+
+		// ext中包含点符号
+		ext = strings.ToLower(ext)
+		if !lists.ContainsString(this.Exts, ext) {
+			return false
+		}
+	}
+
+	if len(this.Domains) > 0 {
+		var domain = req.Host
+		if len(domain) == 0 {
+			return false
+		}
+		if !configutils.MatchDomains(this.Domains, domain) {
+			return false
+		}
+	}
+
+	return true
+}
+
+// cleanPath 清理Path中的多余的字符
+func (this *HTTPAuthBaseMethod) cleanPath(path string) string {
+	var l = len(path)
+	if l == 0 {
+		return "/"
+	}
+	var result = []byte{'/'}
+	var isSlash = true
+	for i := 0; i < l; i++ {
+		if path[i] == '?' {
+			result = append(result, path[i:]...)
+			break
+		}
+		if path[i] == '\\' || path[i] == '/' {
+			if !isSlash {
+				isSlash = true
+				result = append(result, '/')
+			}
+		} else {
+			isSlash = false
+			result = append(result, path[i])
+		}
+	}
+	return string(result)
+}