防盗链功能增加禁止的来源域名

internal/waf/checkpoints/request_referer_block.go

@@ -41,19 +41,45 @@ func (this *RequestRefererBlockCheckpoint) RequestValue(req requests.Request, pa
 		return
 	}
 
-	var domains = options.GetSlice("allowDomains")
+	// allow domains
-	var domainStrings = []string{}
+	var allowDomains = options.GetSlice("allowDomains")
-	for _, domain := range domains {
+	var allowDomainStrings = []string{}
-		domainStrings = append(domainStrings, types.String(domain))
+	for _, domain := range allowDomains {
+		allowDomainStrings = append(allowDomainStrings, types.String(domain))
 	}
 
-	if len(domainStrings) == 0 {
+	// deny domains
+	var denyDomains = options.GetSlice("denyDomains")
+	var denyDomainStrings = []string{}
+	for _, domain := range denyDomains {
+		denyDomainStrings = append(denyDomainStrings, types.String(domain))
+	}
+
+	if len(allowDomainStrings) == 0 {
+		if len(denyDomainStrings) > 0 {
+			if configutils.MatchDomains(denyDomainStrings, host) {
+				value = 0
+			} else {
+				value = 1
+			}
+			return
+		}
+
 		value = 0
 		return
 	}
 
-	if configutils.MatchDomains(domainStrings, host) {
+	if configutils.MatchDomains(allowDomainStrings, host) {
+		if len(denyDomainStrings) > 0 {
+			if configutils.MatchDomains(denyDomainStrings, host) {
+				value = 0
+			} else {
+				value = 1
+			}
+			return
+		}
 		value = 1
+		return
 	} else {
 		value = 0
 	}