TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-06 01:50:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复默认WAF策略模板中分组不能默认关闭的问题

Browse Source
This commit is contained in:
刘祥超
2023-08-02 17:15:26 +08:00
parent 41a1a6a2e5
commit c2635b0d04
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
internal/db/models/http_firewall_policy_dao.go
View File

@@ -172,16 +172,18 @@ func (this *HTTPFirewallPolicyDAO) CreateDefaultFirewallPolicy(tx *dbs.Tx, name
// 初始化 // 初始化
var groupCodes = []string{} var groupCodes = []string{}
templatePolicy := firewallconfigs.HTTPFirewallTemplate() var templatePolicy = firewallconfigs.HTTPFirewallTemplate()
for _, group := range templatePolicy.AllRuleGroups() { for _, group := range templatePolicy.AllRuleGroups() {
groupCodes = append(groupCodes, group.Code) if group.IsOn {
groupCodes = append(groupCodes, group.Code)
}
} }
var inboundConfig = &firewallconfigs.HTTPFirewallInboundConfig{IsOn: true} var inboundConfig = &firewallconfigs.HTTPFirewallInboundConfig{IsOn: true}
var outboundConfig = &firewallconfigs.HTTPFirewallOutboundConfig{IsOn: true} var outboundConfig = &firewallconfigs.HTTPFirewallOutboundConfig{IsOn: true}
if templatePolicy.Inbound != nil { if templatePolicy.Inbound != nil {
for _, group := range templatePolicy.Inbound.Groups { for _, group := range templatePolicy.Inbound.Groups {
isOn := lists.ContainsString(groupCodes, group.Code) var isOn = lists.ContainsString(groupCodes, group.Code)
group.IsOn = isOn group.IsOn = isOn
groupId, err := SharedHTTPFirewallRuleGroupDAO.CreateGroupFromConfig(tx, group) groupId, err := SharedHTTPFirewallRuleGroupDAO.CreateGroupFromConfig(tx, group)
@@ -196,7 +198,7 @@ func (this *HTTPFirewallPolicyDAO) CreateDefaultFirewallPolicy(tx *dbs.Tx, name
} }
if templatePolicy.Outbound != nil { if templatePolicy.Outbound != nil {
for _, group := range templatePolicy.Outbound.Groups { for _, group := range templatePolicy.Outbound.Groups {
isOn := lists.ContainsString(groupCodes, group.Code) var isOn = lists.ContainsString(groupCodes, group.Code)
group.IsOn = isOn group.IsOn = isOn
groupId, err := SharedHTTPFirewallRuleGroupDAO.CreateGroupFromConfig(tx, group) groupId, err := SharedHTTPFirewallRuleGroupDAO.CreateGroupFromConfig(tx, group)