mirror of
https://github.com/TeaOSLab/EdgeAdmin.git
synced 2025-12-25 03:26:34 +08:00
登录页尝试使用csrf校验
This commit is contained in:
39
internal/web/actions/default/csrf/token.go
Normal file
39
internal/web/actions/default/csrf/token.go
Normal file
@@ -0,0 +1,39 @@
|
||||
package csrf
|
||||
|
||||
import (
|
||||
"github.com/TeaOSLab/EdgeAdmin/internal/csrf"
|
||||
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
|
||||
"github.com/TeaOSLab/EdgeAdmin/internal/web/helpers"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
var lastTimestamp = int64(0)
|
||||
var locker sync.Mutex
|
||||
|
||||
type TokenAction struct {
|
||||
actionutils.ParentAction
|
||||
}
|
||||
|
||||
func (this *TokenAction) Init() {
|
||||
this.Nav("", "", "")
|
||||
}
|
||||
|
||||
func (this *TokenAction) RunGet(params struct {
|
||||
Auth *helpers.UserShouldAuth
|
||||
}) {
|
||||
locker.Lock()
|
||||
defer locker.Unlock()
|
||||
|
||||
defer func() {
|
||||
lastTimestamp = time.Now().Unix()
|
||||
}()
|
||||
|
||||
// 没有登录,则限制请求速度
|
||||
if params.Auth.AdminId() <= 0 && lastTimestamp > 0 && time.Now().Unix()-lastTimestamp <= 1 {
|
||||
this.Fail("请求速度过快,请稍后刷新后重试")
|
||||
}
|
||||
|
||||
this.Data["token"] = csrf.Generate()
|
||||
this.Success()
|
||||
}
|
||||
Reference in New Issue
Block a user