TRKJ/EdgeCommon
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeCommon.git synced 2025-11-03 12:20:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

WAF-区域封禁增加提示内容设置

Browse Source
This commit is contained in:
GoEdgeLab
2023-07-14 11:01:01 +08:00
parent fb2cb8192f
commit 0cad34cea2
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/serverconfigs/firewallconfigs/http_firewall_region_config.go
View File

@@ -3,6 +3,7 @@ package firewallconfigs
import ( import (
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/regionconfigs" "github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/regionconfigs"
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared" "github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
"strings"
) )
type HTTPFirewallRegionConfig struct { type HTTPFirewallRegionConfig struct {
@@ -15,9 +16,11 @@ type HTTPFirewallRegionConfig struct {
CountryOnlyURLPatterns []*shared.URLPattern `yaml:"countryOnlyURLPatterns" json:"countryOnlyURLPatterns"` // 仅限的URL CountryOnlyURLPatterns []*shared.URLPattern `yaml:"countryOnlyURLPatterns" json:"countryOnlyURLPatterns"` // 仅限的URL
CountryExceptURLPatterns []*shared.URLPattern `yaml:"countryExceptURLPatterns" json:"countryExceptURLPatterns"` // 排除的URL CountryExceptURLPatterns []*shared.URLPattern `yaml:"countryExceptURLPatterns" json:"countryExceptURLPatterns"` // 排除的URL
CountryHTML string `yaml:"countryHTML" json:"countryHTML"` // 提示HTML
ProvinceOnlyURLPatterns []*shared.URLPattern `yaml:"provinceOnlyURLPatterns" json:"provinceOnlyURLPatterns"` // 仅限的URL ProvinceOnlyURLPatterns []*shared.URLPattern `yaml:"provinceOnlyURLPatterns" json:"provinceOnlyURLPatterns"` // 仅限的URL
ProvinceExceptURLPatterns []*shared.URLPattern `yaml:"provinceExceptURLPatterns" json:"provinceExceptURLPatterns"` // 排除的URL ProvinceExceptURLPatterns []*shared.URLPattern `yaml:"provinceExceptURLPatterns" json:"provinceExceptURLPatterns"` // 排除的URL
ProvinceHTML string `yaml:"provinceHTML" json:"provinceHTML"` // 提示HTML
isNotEmpty bool isNotEmpty bool
@@ -40,6 +43,8 @@ func (this *HTTPFirewallRegionConfig) Init() error {
this.denyCountryIdMap[countryId] = true this.denyCountryIdMap[countryId] = true
} }
this.CountryHTML = strings.TrimSpace(this.CountryHTML)
this.allowProvinceIdMap = map[int64]bool{} this.allowProvinceIdMap = map[int64]bool{}
for _, provinceId := range this.AllowProvinceIds { for _, provinceId := range this.AllowProvinceIds {
this.allowProvinceIdMap[provinceId] = true this.allowProvinceIdMap[provinceId] = true
@@ -50,6 +55,8 @@ func (this *HTTPFirewallRegionConfig) Init() error {
this.denyProvinceIdMap[provinceId] = true this.denyProvinceIdMap[provinceId] = true
} }
this.ProvinceHTML = strings.TrimSpace(this.ProvinceHTML)
// url patterns // url patterns
for _, pattern := range this.CountryExceptURLPatterns { for _, pattern := range this.CountryExceptURLPatterns {
err := pattern.Init() err := pattern.Init()