Skip to content

Commit

Permalink
Merge pull request #11735 from wswcfan/optimize-auth-lock-performance
Browse files Browse the repository at this point in the history
auth: dramatically improve checkPassword performance
  • Loading branch information
jingyih committed Apr 1, 2020
2 parents dd816f0 + 9cf3162 commit 054b0e5
Showing 1 changed file with 20 additions and 10 deletions.
30 changes: 20 additions & 10 deletions auth/store.go
Original file line number Diff line number Diff line change
Expand Up @@ -319,24 +319,34 @@ func (as *authStore) CheckPassword(username, password string) (uint64, error) {
return 0, ErrAuthNotEnabled
}

tx := as.be.BatchTx()
tx.Lock()
defer tx.Unlock()
var user *authpb.User
// CompareHashAndPassword is very expensive, so we use closures
// to avoid putting it in the critical section of the tx lock.
revision, err := func() (uint64, error) {
tx := as.be.BatchTx()
tx.Lock()
defer tx.Unlock()

user = getUser(as.lg, tx, username)
if user == nil {
return 0, ErrAuthFailed
}

user := getUser(as.lg, tx, username)
if user == nil {
return 0, ErrAuthFailed
}
if user.Options != nil && user.Options.NoPassword {
return 0, ErrAuthFailed
}

if user.Options != nil && user.Options.NoPassword {
return 0, ErrAuthFailed
return getRevision(tx), nil
}()
if err != nil {
return 0, err
}

if bcrypt.CompareHashAndPassword(user.Password, []byte(password)) != nil {
as.lg.Info("invalid password", zap.String("user-name", username))
return 0, ErrAuthFailed
}
return getRevision(tx), nil
return revision, nil
}

func (as *authStore) Recover(be backend.Backend) {
Expand Down

0 comments on commit 054b0e5

Please sign in to comment.