gitea源码

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. // Copyright 2021 The Gitea Authors. All rights reserved.
  2. // SPDX-License-Identifier: MIT
  3. package pam
  4. import (
  5. "context"
  6. "fmt"
  7. "strings"
  8. "code.gitea.io/gitea/models/auth"
  9. user_model "code.gitea.io/gitea/models/user"
  10. "code.gitea.io/gitea/modules/auth/pam"
  11. "code.gitea.io/gitea/modules/optional"
  12. "code.gitea.io/gitea/modules/setting"
  13. "github.com/google/uuid"
  14. )
  15. // Authenticate queries if login/password is valid against the PAM,
  16. // and create a local user if success when enabled.
  17. func (source *Source) Authenticate(ctx context.Context, user *user_model.User, userName, password string) (*user_model.User, error) {
  18. pamLogin, err := pam.Auth(source.ServiceName, userName, password)
  19. if err != nil {
  20. if strings.Contains(err.Error(), "Authentication failure") {
  21. return nil, user_model.ErrUserNotExist{Name: userName}
  22. }
  23. return nil, err
  24. }
  25. if user != nil {
  26. return user, nil
  27. }
  28. // Allow PAM sources with `@` in their name, like from Active Directory
  29. username := pamLogin
  30. email := pamLogin
  31. idx := strings.Index(pamLogin, "@")
  32. if idx > -1 {
  33. username = pamLogin[:idx]
  34. }
  35. if user_model.ValidateEmail(email) != nil {
  36. if source.EmailDomain != "" {
  37. email = fmt.Sprintf("%s@%s", username, source.EmailDomain)
  38. } else {
  39. email = fmt.Sprintf("%s@%s", username, setting.Service.NoReplyAddress)
  40. }
  41. if user_model.ValidateEmail(email) != nil {
  42. email = uuid.New().String() + "@localhost"
  43. }
  44. }
  45. user = &user_model.User{
  46. LowerName: strings.ToLower(username),
  47. Name: username,
  48. Email: email,
  49. Passwd: password,
  50. LoginType: auth.PAM,
  51. LoginSource: source.AuthSource.ID,
  52. LoginName: userName, // This is what the user typed in
  53. }
  54. overwriteDefault := &user_model.CreateUserOverwriteOptions{
  55. IsActive: optional.Some(true),
  56. }
  57. if err := user_model.CreateUser(ctx, user, &user_model.Meta{}, overwriteDefault); err != nil {
  58. return user, err
  59. }
  60. return user, nil
  61. }