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

WAF验证码验证后返回时判断是否已通过验证

Browse Source
This commit is contained in:
GoEdgeLab
2023-11-28 20:39:42 +08:00
parent b7e88b4f4a
commit 18d9fbf409
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
internal/waf/captcha_validator.go
View File

@@ -99,12 +99,18 @@ func (this *CaptchaValidator) Run(req requests.Request, writer http.ResponseWrit
} else { } else {
// 增加计数 // 增加计数
CaptchaIncreaseFails(req, captchaActionConfig, policyId, groupId, setId, CaptchaPageCodeShow) CaptchaIncreaseFails(req, captchaActionConfig, policyId, groupId, setId, CaptchaPageCodeShow)
this.show(captchaActionConfig, req, writer, captchaType) this.show(captchaActionConfig, setId, originURL, req, writer, captchaType)
} }
} }
} }
func (this *CaptchaValidator) show(actionConfig *CaptchaAction, req requests.Request, writer http.ResponseWriter, captchaType firewallconfigs.ServerCaptchaType) { func (this *CaptchaValidator) show(actionConfig *CaptchaAction, setId int64, originURL string, req requests.Request, writer http.ResponseWriter, captchaType firewallconfigs.ServerCaptchaType) {
// validated yet?
if SharedIPWhiteList.Contains(wafutils.ComposeIPType(setId, req), actionConfig.Scope, req.WAFServerId(), req.WAFRemoteIP()) {
http.Redirect(writer, req.WAFRaw(), originURL, http.StatusSeeOther)
return
}
switch captchaType { switch captchaType {
case firewallconfigs.CaptchaTypeOneClick: case firewallconfigs.CaptchaTypeOneClick:
this.showOneClickForm(actionConfig, req, writer) this.showOneClickForm(actionConfig, req, writer)