Andreas/siyuan
1
0
Fork 0
mirror of https://github.com/siyuan-note/siyuan.git synced 2026-01-21 07:46:09 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

🎨 访问授权码输入错误 3 次后加入验证码 https://github.com/siyuan-note/siyuan/issues/5429

Browse source
This commit is contained in:
Liang Ding 2022-07-16 10:48:33 +08:00
parent 527753113b
commit b4208bbf36
No known key found for this signature in database
GPG key ID: 136F30F901A2231D
6 changed files with 85 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

65
kernel/model/session.go
View file

@ -24,6 +24,7 @@ import (
ginSessions "github.com/gin-contrib/sessions"
"github.com/gin-gonic/gin"
"github.com/siyuan-note/siyuan/kernel/util"
"github.com/steambap/captcha"
)
func LogoutAuth(c *gin.Context) {
@ -53,27 +54,83 @@ func LogoutAuth(c *gin.Context) {
func LoginAuth(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var inputCaptcha string
session := util.GetSession(c)
if session.NeedCaptcha() {
captchaArg := arg["captcha"]
if nil == captchaArg {
c.Status(400)
ret.Code = -1
ret.Msg = "need input captcha"
return
}
inputCaptcha = captchaArg.(string)
if session.Captcha != inputCaptcha {
ret.Code = -1
ret.Msg = "invalid captcha"
return
}
}
authCode := arg["authCode"].(string)
if Conf.AccessAuthCode != authCode {
ret.Code = -1
ret.Msg = Conf.Language(83)
session.WrongAuthCount++
session.Captcha = gulu.Rand.String(7)
if session.NeedCaptcha() {
ret.Code = 1 // 需要渲染验证码
}
if err := session.Save(c); nil != err {
util.LogErrorf("save session failed: " + err.Error())
c.Status(500)
return
}
return
}
session := &util.SessionData{ID: gulu.Rand.Int(0, 1024), AccessAuthCode: authCode}
session.WrongAuthCount = 0
session.Captcha = gulu.Rand.String(7)
if err := session.Save(c); nil != err {
util.LogErrorf("saves session failed: " + err.Error())
ret.Code = -1
ret.Msg = "save session failed"
util.LogErrorf("save session failed: " + err.Error())
c.Status(500)
return
}
}
func GetCaptcha(c *gin.Context) {
img, err := captcha.NewMathExpr(150, 30)
if nil != err {
util.LogErrorf("generates captcha failed: " + err.Error())
c.Status(500)
return
}
session := util.GetSession(c)
session.Captcha = img.Text
if err = session.Save(c); nil != err {
util.LogErrorf("save session failed: " + err.Error())
c.Status(500)
return
}
if err = img.WriteImage(c.Writer); nil != err {
util.LogErrorf("writes captcha image failed: " + err.Error())
c.Status(500)
return
}
c.Status(200)
}
func CheckReadonly(c *gin.Context) {
if util.ReadOnly {
result := util.NewResult()