fumiama/paper-manager
1
0
Fork 0
mirror of https://github.com/fumiama/paper-manager.git synced 2026-06-12 04:00:23 +08:00
Code Activity

优化 login

Browse Source
This commit is contained in:
源文雨
2023-05-08 14:43:47 +08:00
parent 8d659cc5d7
commit d25fbac8e9
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/login.go
View File

@@ -27,6 +27,7 @@ var (
errWrongPassword = errors.New("invalid username or password") errWrongPassword = errors.New("invalid username or password")
errTooManyFailedLogins = errors.New("too many failed logins") errTooManyFailedLogins = errors.New("too many failed logins")
errAccountIsDisabled = errors.New("account is disabled") errAccountIsDisabled = errors.New("account is disabled")
errUserTokenFull = errors.New("user token full")
) )
const ( const (
@@ -143,6 +144,14 @@ func login(username, challenge string) (*loginResult, error) {
return nil, err return nil, err
} }
token := base64.RawStdEncoding.EncodeToString(buf[:]) token := base64.RawStdEncoding.EncodeToString(buf[:])
cnt := 0
for usertokens.Get(token) != nil && cnt < 4096 {
token = base64.RawStdEncoding.EncodeToString(buf[:])
cnt++
}
if cnt >= 4096 {
return nil, errUserTokenFull
}
usertokens.Set(token, &user) usertokens.Set(token, &user)
loginstatus.Store(username, loginStatusYes) loginstatus.Store(username, loginStatusYes)
return &loginResult{ return &loginResult{