From 99dfb7349ccce58b962f538e468bbc9016cd9efb Mon Sep 17 00:00:00 2001 From: samwaf Date: Sun, 2 Apr 2023 14:55:19 +0800 Subject: [PATCH] =?UTF-8?q?=E5=88=A0=E9=99=A4=E4=B8=BB=E6=9C=BA=E6=97=B6?= =?UTF-8?q?=E5=80=99=EF=BC=8C=E9=A1=BA=E4=BE=BF=E5=88=A0=E6=8E=89=E6=89=80?= =?UTF-8?q?=E6=9C=89=E7=9A=84=E5=85=B3=E8=81=94=E4=BF=9D=E6=8A=A4=E4=BF=A1?= =?UTF-8?q?=E6=81=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.go | 2 +- service/waf_service/waf_host.go | 14 ++++++++++++++ 2 files changed, 15 insertions(+), 1 deletion(-) diff --git a/main.go b/main.go index 2b5840b..11d9f9f 100644 --- a/main.go +++ b/main.go @@ -125,7 +125,7 @@ func main() { for { select { case msg := <-global.GWAF_CHAN_MSG: - if wafEngine.HostCode[msg.HostCode] != "" && wafEngine.HostTarget[wafEngine.HostCode[msg.HostCode]] != nil { + if wafEngine.HostTarget[wafEngine.HostCode[msg.HostCode]] != nil && wafEngine.HostCode[msg.HostCode] != "" { switch msg.Type { case enums.ChanTypeWhiteIP: wafEngine.HostTarget[wafEngine.HostCode[msg.HostCode]].IPWhiteLists = msg.Content.([]model.IPWhiteList) diff --git a/service/waf_service/waf_host.go b/service/waf_service/waf_host.go index 190b493..a650465 100644 --- a/service/waf_service/waf_host.go +++ b/service/waf_service/waf_host.go @@ -88,6 +88,20 @@ func (receiver *WafHostService) DelHostApi(req request.WafHostDelReq) error { return err } err = global.GWAF_LOCAL_DB.Where("CODE = ?", req.CODE).Delete(model.Hosts{}).Error + //删除规则 + err = global.GWAF_LOCAL_DB.Where("HostCode = ?", req.CODE).Delete(model.Rules{}).Error + //删除Anticc + err = global.GWAF_LOCAL_DB.Where("Host_Code = ?", req.CODE).Delete(model.AntiCC{}).Error + //删除禁用ip + err = global.GWAF_LOCAL_DB.Where("Host_Code = ?", req.CODE).Delete(model.IPBlockList{}).Error + //删除禁用url + err = global.GWAF_LOCAL_DB.Where("Host_Code = ?", req.CODE).Delete(model.URLBlockList{}).Error + //删除隐私保护url + err = global.GWAF_LOCAL_DB.Where("Host_Code = ?", req.CODE).Delete(model.LDPUrl{}).Error + //删除白名单ip + err = global.GWAF_LOCAL_DB.Where("Host_Code = ?", req.CODE).Delete(model.IPWhiteList{}).Error + //删除白名单URL + err = global.GWAF_LOCAL_DB.Where("Host_Code = ?", req.CODE).Delete(model.URLWhiteList{}).Error return err } func (receiver *WafHostService) ModifyGuardStatusApi(req request.WafHostGuardStatusReq) error {