Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 16 additions & 11 deletions models/user.go
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@ import (
"errors"
"fmt"
_ "image/jpeg" // Needed for jpeg support
"net/mail"
"os"
"path/filepath"
"regexp"
Expand Down Expand Up @@ -809,9 +808,8 @@ func CreateUser(u *User) (err error) {
return ErrEmailAlreadyUsed{u.Email}
}

_, err = mail.ParseAddress(u.Email)
if err != nil {
return ErrEmailInvalid{u.Email}
if err = ValidateEmail(u.Email); err != nil {
return err
}

isExist, err = isEmailUsed(sess, u.Email)
Expand Down Expand Up @@ -956,11 +954,10 @@ func checkDupEmail(e Engine, u *User) error {
return nil
}

func updateUser(e Engine, u *User) error {
func updateUser(e Engine, u *User) (err error) {
u.Email = strings.ToLower(u.Email)
_, err := mail.ParseAddress(u.Email)
if err != nil {
return ErrEmailInvalid{u.Email}
if err = ValidateEmail(u.Email); err != nil {
return err
}
_, err = e.ID(u.ID).AllCols().Update(u)
return err
Expand All @@ -982,13 +979,21 @@ func updateUserCols(e Engine, u *User, cols ...string) error {
}

// UpdateUserSetting updates user's settings.
func UpdateUserSetting(u *User) error {
func UpdateUserSetting(u *User) (err error) {
sess := x.NewSession()
defer sess.Close()
if err = sess.Begin(); err != nil {
return err
}
if !u.IsOrganization() {
if err := checkDupEmail(x, u); err != nil {
if err = checkDupEmail(sess, u); err != nil {
return err
}
}
return updateUser(x, u)
if err = updateUser(sess, u); err != nil {
return err
}
return sess.Commit()
}

// deleteBeans deletes all given beans, beans should contain delete conditions.
Expand Down
25 changes: 19 additions & 6 deletions models/user_mail.go
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,21 @@ type EmailAddress struct {
IsPrimary bool `xorm:"-"`
}

// ValidateEmail check if email is a allowed address
func ValidateEmail(email string) error {
if len(email) == 0 {
return nil
}

if _, err := mail.ParseAddress(email); err != nil {
return ErrEmailInvalid{email}
}

// TODO: add an email allow/block list

return nil
}

// GetEmailAddresses returns all email addresses belongs to given user.
func GetEmailAddresses(uid int64) ([]*EmailAddress, error) {
emails := make([]*EmailAddress, 0, 5)
Expand Down Expand Up @@ -144,9 +159,8 @@ func addEmailAddress(e Engine, email *EmailAddress) error {
return ErrEmailAlreadyUsed{email.Email}
}

_, err = mail.ParseAddress(email.Email)
if err != nil {
return ErrEmailInvalid{email.Email}
if err = ValidateEmail(email.Email); err != nil {
return err
}

_, err = e.Insert(email)
Expand All @@ -173,9 +187,8 @@ func AddEmailAddresses(emails []*EmailAddress) error {
} else if used {
return ErrEmailAlreadyUsed{emails[i].Email}
}
_, err = mail.ParseAddress(emails[i].Email)
if err != nil {
return ErrEmailInvalid{emails[i].Email}
if err = ValidateEmail(emails[i].Email); err != nil {
return err
}
}

Expand Down