2014-07-26 12:24:27 +08:00
// Copyright 2014 The Gogs Authors. All rights reserved.
2018-07-20 01:58:33 +08:00
// Copyright 2018 The Gitea Authors. All rights reserved.
2014-07-26 12:24:27 +08:00
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package user
import (
2017-01-16 10:14:29 +08:00
"errors"
2016-03-12 00:56:52 +08:00
"fmt"
2017-02-25 22:57:06 +08:00
"net/http"
"strings"
2014-08-01 05:25:34 +08:00
2016-11-11 00:24:48 +08:00
"code.gitea.io/gitea/models"
"code.gitea.io/gitea/modules/auth"
2017-02-25 22:57:06 +08:00
"code.gitea.io/gitea/modules/auth/oauth2"
2016-11-11 00:24:48 +08:00
"code.gitea.io/gitea/modules/base"
"code.gitea.io/gitea/modules/context"
"code.gitea.io/gitea/modules/log"
2018-07-05 12:13:05 +08:00
"code.gitea.io/gitea/modules/recaptcha"
2016-11-11 00:24:48 +08:00
"code.gitea.io/gitea/modules/setting"
2019-08-15 22:46:21 +08:00
"code.gitea.io/gitea/modules/timeutil"
2018-06-26 22:45:18 +08:00
"code.gitea.io/gitea/modules/util"
2017-02-25 22:57:06 +08:00
2019-08-24 00:40:30 +08:00
"gitea.com/macaron/captcha"
2017-02-22 15:14:37 +08:00
"github.com/markbates/goth"
2018-05-19 22:12:37 +08:00
"github.com/tstranex/u2f"
2014-07-26 12:24:27 +08:00
)
const (
2018-09-13 20:04:25 +08:00
// tplMustChangePassword template for updating a user's password
tplMustChangePassword = "user/auth/change_passwd"
2016-11-18 11:03:03 +08:00
// tplSignIn template for sign in page
tplSignIn base . TplName = "user/auth/signin"
// tplSignUp template path for sign up page
tplSignUp base . TplName = "user/auth/signup"
// TplActivate template path for activate user
TplActivate base . TplName = "user/auth/activate"
tplForgotPassword base . TplName = "user/auth/forgot_passwd"
tplResetPassword base . TplName = "user/auth/reset_passwd"
2017-01-16 10:14:29 +08:00
tplTwofa base . TplName = "user/auth/twofa"
tplTwofaScratch base . TplName = "user/auth/twofa_scratch"
2017-02-22 15:14:37 +08:00
tplLinkAccount base . TplName = "user/auth/link_account"
2018-05-19 22:12:37 +08:00
tplU2F base . TplName = "user/auth/u2f"
2014-07-26 12:24:27 +08:00
)
2016-03-12 00:56:52 +08:00
// AutoSignIn reads cookie and try to auto-login.
func AutoSignIn ( ctx * context . Context ) ( bool , error ) {
if ! models . HasEngine {
return false , nil
}
uname := ctx . GetCookie ( setting . CookieUserName )
if len ( uname ) == 0 {
return false , nil
}
isSucceed := false
defer func ( ) {
if ! isSucceed {
log . Trace ( "auto-login cookie cleared: %s" , uname )
2019-07-12 21:57:31 +08:00
ctx . SetCookie ( setting . CookieUserName , "" , - 1 , setting . AppSubURL , setting . SessionConfig . Domain , setting . SessionConfig . Secure , true )
ctx . SetCookie ( setting . CookieRememberName , "" , - 1 , setting . AppSubURL , setting . SessionConfig . Domain , setting . SessionConfig . Secure , true )
2016-03-12 00:56:52 +08:00
}
} ( )
u , err := models . GetUserByName ( uname )
if err != nil {
if ! models . IsErrUserNotExist ( err ) {
return false , fmt . Errorf ( "GetUserByName: %v" , err )
}
return false , nil
}
2019-07-06 23:47:09 +08:00
if val , ok := ctx . GetSuperSecureCookie (
base . EncodeMD5 ( u . Rands + u . Passwd ) , setting . CookieRememberName ) ; ! ok || val != u . Name {
2016-03-12 00:56:52 +08:00
return false , nil
}
isSucceed = true
2019-06-13 03:41:28 +08:00
err = ctx . Session . Set ( "uid" , u . ID )
if err != nil {
return false , err
}
err = ctx . Session . Set ( "uname" , u . Name )
if err != nil {
return false , err
}
2019-07-12 21:57:31 +08:00
ctx . SetCookie ( setting . CSRFCookieName , "" , - 1 , setting . AppSubURL , setting . SessionConfig . Domain , setting . SessionConfig . Secure , true )
2016-03-12 00:56:52 +08:00
return true , nil
}
2017-01-16 10:14:29 +08:00
func checkAutoLogin ( ctx * context . Context ) bool {
2014-07-26 12:24:27 +08:00
// Check auto-login.
2016-03-12 00:56:52 +08:00
isSucceed , err := AutoSignIn ( ctx )
2014-07-26 12:24:27 +08:00
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "AutoSignIn" , err )
2017-01-16 10:14:29 +08:00
return true
2014-07-26 12:24:27 +08:00
}
2016-08-28 06:07:02 +08:00
redirectTo := ctx . Query ( "redirect_to" )
if len ( redirectTo ) > 0 {
2018-08-15 04:16:37 +08:00
ctx . SetCookie ( "redirect_to" , redirectTo , 0 , setting . AppSubURL , "" , setting . SessionConfig . Secure , true )
2016-08-28 06:07:02 +08:00
} else {
2019-03-21 10:06:16 +08:00
redirectTo = ctx . GetCookie ( "redirect_to" )
2016-08-28 06:07:02 +08:00
}
2015-08-14 02:43:40 +08:00
if isSucceed {
2018-08-15 04:16:37 +08:00
ctx . SetCookie ( "redirect_to" , "" , - 1 , setting . AppSubURL , "" , setting . SessionConfig . Secure , true )
2018-03-16 05:13:34 +08:00
ctx . RedirectToFirst ( redirectTo , setting . AppSubURL + string ( setting . LandingPageURL ) )
2017-01-16 10:14:29 +08:00
return true
}
return false
}
// SignIn render sign in page
func SignIn ( ctx * context . Context ) {
2017-05-01 21:26:53 +08:00
ctx . Data [ "Title" ] = ctx . Tr ( "sign_in" )
2017-01-16 10:14:29 +08:00
// Check auto-login.
if checkAutoLogin ( ctx ) {
2014-07-26 12:24:27 +08:00
return
}
2017-05-01 21:26:53 +08:00
orderedOAuth2Names , oauth2Providers , err := models . GetActiveOAuth2Providers ( )
2017-02-22 15:14:37 +08:00
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , err )
2017-02-22 15:14:37 +08:00
return
}
2017-05-01 21:26:53 +08:00
ctx . Data [ "OrderedOAuth2Names" ] = orderedOAuth2Names
2017-02-22 15:14:37 +08:00
ctx . Data [ "OAuth2Providers" ] = oauth2Providers
2017-03-17 22:16:08 +08:00
ctx . Data [ "Title" ] = ctx . Tr ( "sign_in" )
2017-03-07 18:47:56 +08:00
ctx . Data [ "SignInLink" ] = setting . AppSubURL + "/user/login"
2017-03-17 22:16:08 +08:00
ctx . Data [ "PageIsSignIn" ] = true
ctx . Data [ "PageIsLogin" ] = true
2017-02-22 15:14:37 +08:00
2016-11-18 11:03:03 +08:00
ctx . HTML ( 200 , tplSignIn )
2014-07-26 12:24:27 +08:00
}
2016-11-18 11:03:03 +08:00
// SignInPost response for sign in request
2016-03-12 00:56:52 +08:00
func SignInPost ( ctx * context . Context , form auth . SignInForm ) {
2017-05-01 21:26:53 +08:00
ctx . Data [ "Title" ] = ctx . Tr ( "sign_in" )
orderedOAuth2Names , oauth2Providers , err := models . GetActiveOAuth2Providers ( )
2017-02-22 15:14:37 +08:00
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , err )
2017-02-22 15:14:37 +08:00
return
}
2017-05-01 21:26:53 +08:00
ctx . Data [ "OrderedOAuth2Names" ] = orderedOAuth2Names
2017-02-22 15:14:37 +08:00
ctx . Data [ "OAuth2Providers" ] = oauth2Providers
2017-03-07 18:47:56 +08:00
ctx . Data [ "Title" ] = ctx . Tr ( "sign_in" )
ctx . Data [ "SignInLink" ] = setting . AppSubURL + "/user/login"
ctx . Data [ "PageIsSignIn" ] = true
ctx . Data [ "PageIsLogin" ] = true
2017-02-22 15:14:37 +08:00
2014-07-26 12:24:27 +08:00
if ctx . HasError ( ) {
2016-11-18 11:03:03 +08:00
ctx . HTML ( 200 , tplSignIn )
2014-07-26 12:24:27 +08:00
return
}
u , err := models . UserSignIn ( form . UserName , form . Password )
if err != nil {
2015-08-05 11:14:17 +08:00
if models . IsErrUserNotExist ( err ) {
2016-11-18 11:03:03 +08:00
ctx . RenderWithErr ( ctx . Tr ( "form.username_password_incorrect" ) , tplSignIn , & form )
2017-08-24 13:57:54 +08:00
log . Info ( "Failed authentication attempt for %s from %s" , form . UserName , ctx . RemoteAddr ( ) )
2017-02-25 22:57:06 +08:00
} else if models . IsErrEmailAlreadyUsed ( err ) {
ctx . RenderWithErr ( ctx . Tr ( "form.email_been_used" ) , tplSignIn , & form )
2017-08-24 13:57:54 +08:00
log . Info ( "Failed authentication attempt for %s from %s" , form . UserName , ctx . RemoteAddr ( ) )
2019-02-19 15:19:28 +08:00
} else if models . IsErrUserProhibitLogin ( err ) {
log . Info ( "Failed authentication attempt for %s from %s" , form . UserName , ctx . RemoteAddr ( ) )
ctx . Data [ "Title" ] = ctx . Tr ( "auth.prohibit_login" )
ctx . HTML ( 200 , "user/auth/prohibit_login" )
} else if models . IsErrUserInactive ( err ) {
if setting . Service . RegisterEmailConfirm {
ctx . Data [ "Title" ] = ctx . Tr ( "auth.active_your_account" )
ctx . HTML ( 200 , TplActivate )
} else {
log . Info ( "Failed authentication attempt for %s from %s" , form . UserName , ctx . RemoteAddr ( ) )
ctx . Data [ "Title" ] = ctx . Tr ( "auth.prohibit_login" )
ctx . HTML ( 200 , "user/auth/prohibit_login" )
}
2014-08-10 12:02:00 +08:00
} else {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , err )
2014-07-26 12:24:27 +08:00
}
return
}
2017-01-16 10:14:29 +08:00
// If this user is enrolled in 2FA, we can't sign the user in just yet.
// Instead, redirect them to the 2FA authentication page.
_ , err = models . GetTwoFactorByUID ( u . ID )
if err != nil {
if models . IsErrTwoFactorNotEnrolled ( err ) {
handleSignIn ( ctx , u , form . Remember )
} else {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , err )
2017-01-16 10:14:29 +08:00
}
return
}
// User needs to use 2FA, save data and redirect to 2FA page.
2019-06-13 03:41:28 +08:00
err = ctx . Session . Set ( "twofaUid" , u . ID )
if err != nil {
ctx . ServerError ( "UserSignIn" , err )
return
}
err = ctx . Session . Set ( "twofaRemember" , form . Remember )
if err != nil {
ctx . ServerError ( "UserSignIn" , err )
return
}
2018-05-19 22:12:37 +08:00
regs , err := models . GetU2FRegistrationsByUID ( u . ID )
if err == nil && len ( regs ) > 0 {
ctx . Redirect ( setting . AppSubURL + "/user/u2f" )
return
}
2017-01-16 10:14:29 +08:00
ctx . Redirect ( setting . AppSubURL + "/user/two_factor" )
}
// TwoFactor shows the user a two-factor authentication page.
func TwoFactor ( ctx * context . Context ) {
ctx . Data [ "Title" ] = ctx . Tr ( "twofa" )
// Check auto-login.
if checkAutoLogin ( ctx ) {
return
}
// Ensure user is in a 2FA session.
if ctx . Session . Get ( "twofaUid" ) == nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , errors . New ( "not in 2FA session" ) )
2017-01-16 10:14:29 +08:00
return
}
ctx . HTML ( 200 , tplTwofa )
}
// TwoFactorPost validates a user's two-factor authentication token.
func TwoFactorPost ( ctx * context . Context , form auth . TwoFactorAuthForm ) {
ctx . Data [ "Title" ] = ctx . Tr ( "twofa" )
// Ensure user is in a 2FA session.
idSess := ctx . Session . Get ( "twofaUid" )
if idSess == nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , errors . New ( "not in 2FA session" ) )
2017-01-16 10:14:29 +08:00
return
}
id := idSess . ( int64 )
twofa , err := models . GetTwoFactorByUID ( id )
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , err )
2017-01-16 10:14:29 +08:00
return
}
// Validate the passcode with the stored TOTP secret.
ok , err := twofa . ValidateTOTP ( form . Passcode )
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , err )
2017-01-16 10:14:29 +08:00
return
}
2018-05-02 23:02:02 +08:00
if ok && twofa . LastUsedPasscode != form . Passcode {
2017-01-16 10:14:29 +08:00
remember := ctx . Session . Get ( "twofaRemember" ) . ( bool )
u , err := models . GetUserByID ( id )
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , err )
2017-01-16 10:14:29 +08:00
return
}
2017-02-27 18:10:26 +08:00
if ctx . Session . Get ( "linkAccount" ) != nil {
gothUser := ctx . Session . Get ( "linkAccountGothUser" )
if gothUser == nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , errors . New ( "not in LinkAccount session" ) )
2017-02-27 18:10:26 +08:00
return
}
err = models . LinkAccountToUser ( u , gothUser . ( goth . User ) )
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , err )
2017-02-27 18:10:26 +08:00
return
}
}
2018-05-02 23:02:02 +08:00
twofa . LastUsedPasscode = form . Passcode
if err = models . UpdateTwoFactor ( twofa ) ; err != nil {
ctx . ServerError ( "UserSignIn" , err )
return
}
2017-01-16 10:14:29 +08:00
handleSignIn ( ctx , u , remember )
return
}
ctx . RenderWithErr ( ctx . Tr ( "auth.twofa_passcode_incorrect" ) , tplTwofa , auth . TwoFactorAuthForm { } )
}
// TwoFactorScratch shows the scratch code form for two-factor authentication.
func TwoFactorScratch ( ctx * context . Context ) {
ctx . Data [ "Title" ] = ctx . Tr ( "twofa_scratch" )
// Check auto-login.
if checkAutoLogin ( ctx ) {
return
}
// Ensure user is in a 2FA session.
if ctx . Session . Get ( "twofaUid" ) == nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , errors . New ( "not in 2FA session" ) )
2017-01-16 10:14:29 +08:00
return
}
ctx . HTML ( 200 , tplTwofaScratch )
}
// TwoFactorScratchPost validates and invalidates a user's two-factor scratch token.
func TwoFactorScratchPost ( ctx * context . Context , form auth . TwoFactorScratchAuthForm ) {
ctx . Data [ "Title" ] = ctx . Tr ( "twofa_scratch" )
// Ensure user is in a 2FA session.
idSess := ctx . Session . Get ( "twofaUid" )
if idSess == nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , errors . New ( "not in 2FA session" ) )
2017-01-16 10:14:29 +08:00
return
}
id := idSess . ( int64 )
twofa , err := models . GetTwoFactorByUID ( id )
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , err )
2017-01-16 10:14:29 +08:00
return
}
// Validate the passcode with the stored TOTP secret.
if twofa . VerifyScratchToken ( form . Token ) {
// Invalidate the scratch token.
2018-07-27 20:54:50 +08:00
_ , err = twofa . GenerateScratchToken ( )
if err != nil {
ctx . ServerError ( "UserSignIn" , err )
return
}
2017-01-16 10:14:29 +08:00
if err = models . UpdateTwoFactor ( twofa ) ; err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , err )
2017-01-16 10:14:29 +08:00
return
}
remember := ctx . Session . Get ( "twofaRemember" ) . ( bool )
u , err := models . GetUserByID ( id )
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , err )
2017-01-16 10:14:29 +08:00
return
}
handleSignInFull ( ctx , u , remember , false )
ctx . Flash . Info ( ctx . Tr ( "auth.twofa_scratch_used" ) )
2018-07-20 01:58:33 +08:00
ctx . Redirect ( setting . AppSubURL + "/user/settings/security" )
2017-01-16 10:14:29 +08:00
return
}
ctx . RenderWithErr ( ctx . Tr ( "auth.twofa_scratch_token_incorrect" ) , tplTwofaScratch , auth . TwoFactorScratchAuthForm { } )
}
2018-05-19 22:12:37 +08:00
// U2F shows the U2F login page
func U2F ( ctx * context . Context ) {
ctx . Data [ "Title" ] = ctx . Tr ( "twofa" )
ctx . Data [ "RequireU2F" ] = true
// Check auto-login.
if checkAutoLogin ( ctx ) {
return
}
// Ensure user is in a 2FA session.
if ctx . Session . Get ( "twofaUid" ) == nil {
ctx . ServerError ( "UserSignIn" , errors . New ( "not in U2F session" ) )
return
}
ctx . HTML ( 200 , tplU2F )
}
// U2FChallenge submits a sign challenge to the browser
func U2FChallenge ( ctx * context . Context ) {
// Ensure user is in a U2F session.
idSess := ctx . Session . Get ( "twofaUid" )
if idSess == nil {
ctx . ServerError ( "UserSignIn" , errors . New ( "not in U2F session" ) )
return
}
id := idSess . ( int64 )
regs , err := models . GetU2FRegistrationsByUID ( id )
if err != nil {
ctx . ServerError ( "UserSignIn" , err )
return
}
if len ( regs ) == 0 {
ctx . ServerError ( "UserSignIn" , errors . New ( "no device registered" ) )
return
}
challenge , err := u2f . NewChallenge ( setting . U2F . AppID , setting . U2F . TrustedFacets )
2019-06-13 03:41:28 +08:00
if err != nil {
ctx . ServerError ( "u2f.NewChallenge" , err )
return
}
2018-05-19 22:12:37 +08:00
if err = ctx . Session . Set ( "u2fChallenge" , challenge ) ; err != nil {
ctx . ServerError ( "UserSignIn" , err )
return
}
ctx . JSON ( 200 , challenge . SignRequest ( regs . ToRegistrations ( ) ) )
}
// U2FSign authenticates the user by signResp
func U2FSign ( ctx * context . Context , signResp u2f . SignResponse ) {
challSess := ctx . Session . Get ( "u2fChallenge" )
idSess := ctx . Session . Get ( "twofaUid" )
if challSess == nil || idSess == nil {
ctx . ServerError ( "UserSignIn" , errors . New ( "not in U2F session" ) )
return
}
challenge := challSess . ( * u2f . Challenge )
id := idSess . ( int64 )
regs , err := models . GetU2FRegistrationsByUID ( id )
if err != nil {
ctx . ServerError ( "UserSignIn" , err )
return
}
for _ , reg := range regs {
r , err := reg . Parse ( )
if err != nil {
2019-04-02 15:48:31 +08:00
log . Fatal ( "parsing u2f registration: %v" , err )
2018-05-19 22:12:37 +08:00
continue
}
newCounter , authErr := r . Authenticate ( signResp , * challenge , reg . Counter )
if authErr == nil {
reg . Counter = newCounter
user , err := models . GetUserByID ( id )
if err != nil {
ctx . ServerError ( "UserSignIn" , err )
return
}
remember := ctx . Session . Get ( "twofaRemember" ) . ( bool )
if err := reg . UpdateCounter ( ) ; err != nil {
ctx . ServerError ( "UserSignIn" , err )
return
}
if ctx . Session . Get ( "linkAccount" ) != nil {
gothUser := ctx . Session . Get ( "linkAccountGothUser" )
if gothUser == nil {
ctx . ServerError ( "UserSignIn" , errors . New ( "not in LinkAccount session" ) )
return
}
err = models . LinkAccountToUser ( user , gothUser . ( goth . User ) )
if err != nil {
ctx . ServerError ( "UserSignIn" , err )
return
}
}
redirect := handleSignInFull ( ctx , user , remember , false )
if redirect == "" {
redirect = setting . AppSubURL + "/"
}
ctx . PlainText ( 200 , [ ] byte ( redirect ) )
return
}
}
ctx . Error ( 401 )
}
2017-01-16 10:14:29 +08:00
// This handles the final part of the sign-in process of the user.
func handleSignIn ( ctx * context . Context , u * models . User , remember bool ) {
handleSignInFull ( ctx , u , remember , true )
}
2018-05-19 22:12:37 +08:00
func handleSignInFull ( ctx * context . Context , u * models . User , remember bool , obeyRedirect bool ) string {
2017-01-16 10:14:29 +08:00
if remember {
2014-07-26 12:24:27 +08:00
days := 86400 * setting . LogInRememberDays
2019-07-12 21:57:31 +08:00
ctx . SetCookie ( setting . CookieUserName , u . Name , days , setting . AppSubURL , setting . SessionConfig . Domain , setting . SessionConfig . Secure , true )
2017-02-25 22:57:06 +08:00
ctx . SetSuperSecureCookie ( base . EncodeMD5 ( u . Rands + u . Passwd ) ,
2019-07-12 21:57:31 +08:00
setting . CookieRememberName , u . Name , days , setting . AppSubURL , setting . SessionConfig . Domain , setting . SessionConfig . Secure , true )
2014-07-26 12:24:27 +08:00
}
2019-06-13 03:41:28 +08:00
_ = ctx . Session . Delete ( "openid_verified_uri" )
_ = ctx . Session . Delete ( "openid_signin_remember" )
_ = ctx . Session . Delete ( "openid_determined_email" )
_ = ctx . Session . Delete ( "openid_determined_username" )
_ = ctx . Session . Delete ( "twofaUid" )
_ = ctx . Session . Delete ( "twofaRemember" )
_ = ctx . Session . Delete ( "u2fChallenge" )
_ = ctx . Session . Delete ( "linkAccount" )
err := ctx . Session . Set ( "uid" , u . ID )
if err != nil {
log . Error ( fmt . Sprintf ( "Error setting session: %v" , err ) )
}
err = ctx . Session . Set ( "uname" , u . Name )
if err != nil {
log . Error ( fmt . Sprintf ( "Error setting session: %v" , err ) )
}
2016-03-13 09:56:03 +08:00
2018-05-05 08:28:30 +08:00
// Language setting of the user overwrites the one previously set
// If the user does not have a locale set, we save the current one.
if len ( u . Language ) == 0 {
u . Language = ctx . Locale . Language ( )
if err := models . UpdateUserCols ( u , "language" ) ; err != nil {
2019-04-02 15:48:31 +08:00
log . Error ( fmt . Sprintf ( "Error updating user language [user: %d, locale: %s]" , u . ID , u . Language ) )
2018-05-19 22:12:37 +08:00
return setting . AppSubURL + "/"
2018-05-05 08:28:30 +08:00
}
}
2019-07-12 21:57:31 +08:00
ctx . SetCookie ( "lang" , u . Language , nil , setting . AppSubURL , setting . SessionConfig . Domain , setting . SessionConfig . Secure , true )
2018-05-05 08:28:30 +08:00
2016-03-13 09:56:03 +08:00
// Clear whatever CSRF has right now, force to generate a new one
2019-07-12 21:57:31 +08:00
ctx . SetCookie ( setting . CSRFCookieName , "" , - 1 , setting . AppSubURL , setting . SessionConfig . Domain , setting . SessionConfig . Secure , true )
2016-03-13 09:56:03 +08:00
2016-11-09 18:53:45 +08:00
// Register last login
u . SetLastLogin ( )
2017-08-12 22:18:44 +08:00
if err := models . UpdateUserCols ( u , "last_login_unix" ) ; err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UpdateUserCols" , err )
2018-05-19 22:12:37 +08:00
return setting . AppSubURL + "/"
2016-11-09 18:53:45 +08:00
}
2019-03-21 10:06:16 +08:00
if redirectTo := ctx . GetCookie ( "redirect_to" ) ; len ( redirectTo ) > 0 && ! util . IsExternalURL ( redirectTo ) {
2018-08-15 04:16:37 +08:00
ctx . SetCookie ( "redirect_to" , "" , - 1 , setting . AppSubURL , "" , setting . SessionConfig . Secure , true )
2017-01-16 10:14:29 +08:00
if obeyRedirect {
2018-03-16 05:13:34 +08:00
ctx . RedirectToFirst ( redirectTo )
2017-01-16 10:14:29 +08:00
}
2018-05-19 22:12:37 +08:00
return redirectTo
2014-07-26 12:24:27 +08:00
}
2017-01-16 10:14:29 +08:00
if obeyRedirect {
ctx . Redirect ( setting . AppSubURL + "/" )
}
2018-05-19 22:12:37 +08:00
return setting . AppSubURL + "/"
2014-07-26 12:24:27 +08:00
}
2017-02-22 15:14:37 +08:00
// SignInOAuth handles the OAuth2 login buttons
func SignInOAuth ( ctx * context . Context ) {
provider := ctx . Params ( ":provider" )
loginSource , err := models . GetActiveOAuth2LoginSourceByName ( provider )
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "SignIn" , err )
2017-02-22 15:14:37 +08:00
return
}
// try to do a direct callback flow, so we don't authenticate the user again but use the valid accesstoken to get the user
user , gothUser , err := oAuth2UserLoginCallback ( loginSource , ctx . Req . Request , ctx . Resp )
if err == nil && user != nil {
// we got the user without going through the whole OAuth2 authentication flow again
handleOAuth2SignIn ( user , gothUser , ctx , err )
return
}
err = oauth2 . Auth ( loginSource . Name , ctx . Req . Request , ctx . Resp )
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "SignIn" , err )
2017-02-22 15:14:37 +08:00
}
// redirect is done in oauth2.Auth
}
// SignInOAuthCallback handles the callback from the given provider
func SignInOAuthCallback ( ctx * context . Context ) {
provider := ctx . Params ( ":provider" )
// first look if the provider is still active
loginSource , err := models . GetActiveOAuth2LoginSourceByName ( provider )
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "SignIn" , err )
2017-02-22 15:14:37 +08:00
return
}
if loginSource == nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "SignIn" , errors . New ( "No valid provider found, check configured callback url in provider" ) )
2017-02-22 15:14:37 +08:00
return
}
u , gothUser , err := oAuth2UserLoginCallback ( loginSource , ctx . Req . Request , ctx . Resp )
handleOAuth2SignIn ( u , gothUser , ctx , err )
}
func handleOAuth2SignIn ( u * models . User , gothUser goth . User , ctx * context . Context , err error ) {
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , err )
2017-02-22 15:14:37 +08:00
return
}
if u == nil {
// no existing user is found, request attach or new account
2019-06-13 03:41:28 +08:00
err = ctx . Session . Set ( "linkAccountGothUser" , gothUser )
if err != nil {
log . Error ( fmt . Sprintf ( "Error setting session: %v" , err ) )
}
2017-02-22 15:14:37 +08:00
ctx . Redirect ( setting . AppSubURL + "/user/link_account" )
return
}
// If this user is enrolled in 2FA, we can't sign the user in just yet.
// Instead, redirect them to the 2FA authentication page.
_ , err = models . GetTwoFactorByUID ( u . ID )
if err != nil {
if models . IsErrTwoFactorNotEnrolled ( err ) {
2019-06-13 03:41:28 +08:00
err = ctx . Session . Set ( "uid" , u . ID )
if err != nil {
log . Error ( fmt . Sprintf ( "Error setting session: %v" , err ) )
}
err = ctx . Session . Set ( "uname" , u . Name )
if err != nil {
log . Error ( fmt . Sprintf ( "Error setting session: %v" , err ) )
}
2017-02-22 15:14:37 +08:00
// Clear whatever CSRF has right now, force to generate a new one
2019-07-12 21:57:31 +08:00
ctx . SetCookie ( setting . CSRFCookieName , "" , - 1 , setting . AppSubURL , setting . SessionConfig . Domain , setting . SessionConfig . Secure , true )
2017-02-22 15:14:37 +08:00
// Register last login
u . SetLastLogin ( )
2017-08-12 22:18:44 +08:00
if err := models . UpdateUserCols ( u , "last_login_unix" ) ; err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UpdateUserCols" , err )
2017-02-22 15:14:37 +08:00
return
}
2019-03-21 10:06:16 +08:00
if redirectTo := ctx . GetCookie ( "redirect_to" ) ; len ( redirectTo ) > 0 {
2018-08-15 04:16:37 +08:00
ctx . SetCookie ( "redirect_to" , "" , - 1 , setting . AppSubURL , "" , setting . SessionConfig . Secure , true )
2018-03-16 05:13:34 +08:00
ctx . RedirectToFirst ( redirectTo )
2017-02-22 15:14:37 +08:00
return
}
ctx . Redirect ( setting . AppSubURL + "/" )
} else {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , err )
2017-02-22 15:14:37 +08:00
}
return
}
// User needs to use 2FA, save data and redirect to 2FA page.
2019-06-13 03:41:28 +08:00
err = ctx . Session . Set ( "twofaUid" , u . ID )
if err != nil {
log . Error ( fmt . Sprintf ( "Error setting session: %v" , err ) )
}
err = ctx . Session . Set ( "twofaRemember" , false )
if err != nil {
log . Error ( fmt . Sprintf ( "Error setting session: %v" , err ) )
}
2018-05-19 22:12:37 +08:00
// If U2F is enrolled -> Redirect to U2F instead
regs , err := models . GetU2FRegistrationsByUID ( u . ID )
if err == nil && len ( regs ) > 0 {
ctx . Redirect ( setting . AppSubURL + "/user/u2f" )
return
}
2017-02-22 15:14:37 +08:00
ctx . Redirect ( setting . AppSubURL + "/user/two_factor" )
}
// OAuth2UserLoginCallback attempts to handle the callback from the OAuth2 provider and if successful
// login the user
func oAuth2UserLoginCallback ( loginSource * models . LoginSource , request * http . Request , response http . ResponseWriter ) ( * models . User , goth . User , error ) {
gothUser , err := oauth2 . ProviderCallback ( loginSource . Name , request , response )
if err != nil {
return nil , goth . User { } , err
}
user := & models . User {
LoginName : gothUser . UserID ,
LoginType : models . LoginOAuth2 ,
LoginSource : loginSource . ID ,
}
hasUser , err := models . GetUser ( user )
if err != nil {
return nil , goth . User { } , err
}
if hasUser {
return user , goth . User { } , nil
}
// search in external linked users
externalLoginUser := & models . ExternalLoginUser {
ExternalID : gothUser . UserID ,
LoginSourceID : loginSource . ID ,
}
hasUser , err = models . GetExternalLogin ( externalLoginUser )
if err != nil {
return nil , goth . User { } , err
}
if hasUser {
user , err = models . GetUserByID ( externalLoginUser . UserID )
return user , goth . User { } , err
}
// no user found to login
return nil , gothUser , nil
}
// LinkAccount shows the page where the user can decide to login or create a new account
func LinkAccount ( ctx * context . Context ) {
2019-07-07 03:48:02 +08:00
ctx . Data [ "DisablePassword" ] = ! setting . Service . RequireExternalRegistrationCaptcha || setting . Service . AllowOnlyExternalRegistration
2017-02-22 15:14:37 +08:00
ctx . Data [ "Title" ] = ctx . Tr ( "link_account" )
ctx . Data [ "LinkAccountMode" ] = true
2019-07-07 03:48:02 +08:00
ctx . Data [ "EnableCaptcha" ] = setting . Service . EnableCaptcha && setting . Service . RequireExternalRegistrationCaptcha
2018-07-05 12:13:05 +08:00
ctx . Data [ "CaptchaType" ] = setting . Service . CaptchaType
2019-05-02 21:09:39 +08:00
ctx . Data [ "RecaptchaURL" ] = setting . Service . RecaptchaURL
2018-07-05 12:13:05 +08:00
ctx . Data [ "RecaptchaSitekey" ] = setting . Service . RecaptchaSitekey
2017-02-22 15:14:37 +08:00
ctx . Data [ "DisableRegistration" ] = setting . Service . DisableRegistration
ctx . Data [ "ShowRegistrationButton" ] = false
// use this to set the right link into the signIn and signUp templates in the link_account template
ctx . Data [ "SignInLink" ] = setting . AppSubURL + "/user/link_account_signin"
ctx . Data [ "SignUpLink" ] = setting . AppSubURL + "/user/link_account_signup"
gothUser := ctx . Session . Get ( "linkAccountGothUser" )
if gothUser == nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , errors . New ( "not in LinkAccount session" ) )
2017-02-22 15:14:37 +08:00
return
}
2018-10-29 06:46:16 +08:00
uname := gothUser . ( goth . User ) . NickName
email := gothUser . ( goth . User ) . Email
ctx . Data [ "user_name" ] = uname
ctx . Data [ "email" ] = email
if len ( email ) != 0 {
u , err := models . GetUserByEmail ( email )
if err != nil && ! models . IsErrUserNotExist ( err ) {
ctx . ServerError ( "UserSignIn" , err )
return
}
if u != nil {
ctx . Data [ "user_exists" ] = true
}
} else if len ( uname ) != 0 {
u , err := models . GetUserByName ( uname )
if err != nil && ! models . IsErrUserNotExist ( err ) {
ctx . ServerError ( "UserSignIn" , err )
return
}
if u != nil {
ctx . Data [ "user_exists" ] = true
}
}
2017-02-22 15:14:37 +08:00
ctx . HTML ( 200 , tplLinkAccount )
}
// LinkAccountPostSignIn handle the coupling of external account with another account using signIn
func LinkAccountPostSignIn ( ctx * context . Context , signInForm auth . SignInForm ) {
2019-07-07 03:48:02 +08:00
ctx . Data [ "DisablePassword" ] = setting . Service . AllowOnlyExternalRegistration
2017-02-22 15:14:37 +08:00
ctx . Data [ "Title" ] = ctx . Tr ( "link_account" )
ctx . Data [ "LinkAccountMode" ] = true
ctx . Data [ "LinkAccountModeSignIn" ] = true
2019-07-07 03:48:02 +08:00
ctx . Data [ "EnableCaptcha" ] = setting . Service . EnableCaptcha && setting . Service . RequireExternalRegistrationCaptcha
2019-05-02 21:09:39 +08:00
ctx . Data [ "RecaptchaURL" ] = setting . Service . RecaptchaURL
2018-07-05 12:13:05 +08:00
ctx . Data [ "CaptchaType" ] = setting . Service . CaptchaType
ctx . Data [ "RecaptchaSitekey" ] = setting . Service . RecaptchaSitekey
2017-02-22 15:14:37 +08:00
ctx . Data [ "DisableRegistration" ] = setting . Service . DisableRegistration
ctx . Data [ "ShowRegistrationButton" ] = false
// use this to set the right link into the signIn and signUp templates in the link_account template
ctx . Data [ "SignInLink" ] = setting . AppSubURL + "/user/link_account_signin"
ctx . Data [ "SignUpLink" ] = setting . AppSubURL + "/user/link_account_signup"
gothUser := ctx . Session . Get ( "linkAccountGothUser" )
if gothUser == nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignIn" , errors . New ( "not in LinkAccount session" ) )
2017-02-22 15:14:37 +08:00
return
}
if ctx . HasError ( ) {
ctx . HTML ( 200 , tplLinkAccount )
return
}
u , err := models . UserSignIn ( signInForm . UserName , signInForm . Password )
if err != nil {
if models . IsErrUserNotExist ( err ) {
ctx . RenderWithErr ( ctx . Tr ( "form.username_password_incorrect" ) , tplLinkAccount , & signInForm )
} else {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserLinkAccount" , err )
2017-02-22 15:14:37 +08:00
}
return
}
// If this user is enrolled in 2FA, we can't sign the user in just yet.
// Instead, redirect them to the 2FA authentication page.
_ , err = models . GetTwoFactorByUID ( u . ID )
if err != nil {
if models . IsErrTwoFactorNotEnrolled ( err ) {
2017-02-27 18:10:26 +08:00
err = models . LinkAccountToUser ( u , gothUser . ( goth . User ) )
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserLinkAccount" , err )
2017-02-27 18:10:26 +08:00
} else {
handleSignIn ( ctx , u , signInForm . Remember )
}
2017-02-22 15:14:37 +08:00
} else {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserLinkAccount" , err )
2017-02-22 15:14:37 +08:00
}
return
}
// User needs to use 2FA, save data and redirect to 2FA page.
2019-06-13 03:41:28 +08:00
err = ctx . Session . Set ( "twofaUid" , u . ID )
if err != nil {
log . Error ( fmt . Sprintf ( "Error setting session: %v" , err ) )
}
err = ctx . Session . Set ( "twofaRemember" , signInForm . Remember )
if err != nil {
log . Error ( fmt . Sprintf ( "Error setting session: %v" , err ) )
}
err = ctx . Session . Set ( "linkAccount" , true )
if err != nil {
log . Error ( fmt . Sprintf ( "Error setting session: %v" , err ) )
}
2017-02-22 15:14:37 +08:00
2018-05-19 22:12:37 +08:00
// If U2F is enrolled -> Redirect to U2F instead
regs , err := models . GetU2FRegistrationsByUID ( u . ID )
if err == nil && len ( regs ) > 0 {
ctx . Redirect ( setting . AppSubURL + "/user/u2f" )
return
}
2017-02-22 15:14:37 +08:00
ctx . Redirect ( setting . AppSubURL + "/user/two_factor" )
}
// LinkAccountPostRegister handle the creation of a new account for an external account using signUp
func LinkAccountPostRegister ( ctx * context . Context , cpt * captcha . Captcha , form auth . RegisterForm ) {
2019-07-07 03:48:02 +08:00
// TODO Make insecure passwords optional for local accounts also,
// once email-based Second-Factor Auth is available
ctx . Data [ "DisablePassword" ] = ! setting . Service . RequireExternalRegistrationCaptcha || setting . Service . AllowOnlyExternalRegistration
2017-02-22 15:14:37 +08:00
ctx . Data [ "Title" ] = ctx . Tr ( "link_account" )
ctx . Data [ "LinkAccountMode" ] = true
ctx . Data [ "LinkAccountModeRegister" ] = true
2019-07-07 03:48:02 +08:00
ctx . Data [ "EnableCaptcha" ] = setting . Service . EnableCaptcha && setting . Service . RequireExternalRegistrationCaptcha
2019-05-02 21:09:39 +08:00
ctx . Data [ "RecaptchaURL" ] = setting . Service . RecaptchaURL
2018-07-05 12:13:05 +08:00
ctx . Data [ "CaptchaType" ] = setting . Service . CaptchaType
ctx . Data [ "RecaptchaSitekey" ] = setting . Service . RecaptchaSitekey
2017-02-22 15:14:37 +08:00
ctx . Data [ "DisableRegistration" ] = setting . Service . DisableRegistration
ctx . Data [ "ShowRegistrationButton" ] = false
// use this to set the right link into the signIn and signUp templates in the link_account template
ctx . Data [ "SignInLink" ] = setting . AppSubURL + "/user/link_account_signin"
ctx . Data [ "SignUpLink" ] = setting . AppSubURL + "/user/link_account_signup"
gothUser := ctx . Session . Get ( "linkAccountGothUser" )
if gothUser == nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UserSignUp" , errors . New ( "not in LinkAccount session" ) )
2017-02-22 15:14:37 +08:00
return
}
if ctx . HasError ( ) {
ctx . HTML ( 200 , tplLinkAccount )
return
}
if setting . Service . DisableRegistration {
ctx . Error ( 403 )
return
}
2019-07-07 03:48:02 +08:00
if setting . Service . EnableCaptcha && setting . Service . RequireExternalRegistrationCaptcha {
var valid bool
switch setting . Service . CaptchaType {
case setting . ImageCaptcha :
valid = cpt . VerifyReq ( ctx . Req )
case setting . ReCaptcha :
valid , _ = recaptcha . Verify ( form . GRecaptchaResponse )
default :
ctx . ServerError ( "Unknown Captcha Type" , fmt . Errorf ( "Unknown Captcha Type: %s" , setting . Service . CaptchaType ) )
return
}
2017-02-22 15:14:37 +08:00
2018-07-05 12:13:05 +08:00
if ! valid {
ctx . Data [ "Err_Captcha" ] = true
ctx . RenderWithErr ( ctx . Tr ( "form.captcha_incorrect" ) , tplLinkAccount , & form )
return
}
}
2019-07-07 03:48:02 +08:00
if setting . Service . AllowOnlyExternalRegistration || ! setting . Service . RequireExternalRegistrationPassword {
// In models.User an empty password is classed as not set, so we set form.Password to empty.
// Eventually the database should be changed to indicate "Second Factor"-enabled accounts
// (accounts that do not introduce the security vulnerabilities of a password).
// If a user decides to circumvent second-factor security, and purposefully create a password,
// they can still do so using the "Recover Account" option.
form . Password = ""
} else {
if ( len ( strings . TrimSpace ( form . Password ) ) > 0 || len ( strings . TrimSpace ( form . Retype ) ) > 0 ) && form . Password != form . Retype {
ctx . Data [ "Err_Password" ] = true
ctx . RenderWithErr ( ctx . Tr ( "form.password_not_match" ) , tplLinkAccount , & form )
return
}
if len ( strings . TrimSpace ( form . Password ) ) > 0 && len ( form . Password ) < setting . MinPasswordLength {
ctx . Data [ "Err_Password" ] = true
ctx . RenderWithErr ( ctx . Tr ( "auth.password_too_short" , setting . MinPasswordLength ) , tplLinkAccount , & form )
return
}
2017-02-22 15:14:37 +08:00
}
loginSource , err := models . GetActiveOAuth2LoginSourceByName ( gothUser . ( goth . User ) . Provider )
if err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "CreateUser" , err )
2017-02-22 15:14:37 +08:00
}
u := & models . User {
Name : form . UserName ,
Email : form . Email ,
Passwd : form . Password ,
IsActive : ! setting . Service . RegisterEmailConfirm ,
LoginType : models . LoginOAuth2 ,
LoginSource : loginSource . ID ,
LoginName : gothUser . ( goth . User ) . UserID ,
}
if err := models . CreateUser ( u ) ; err != nil {
switch {
case models . IsErrUserAlreadyExist ( err ) :
ctx . Data [ "Err_UserName" ] = true
ctx . RenderWithErr ( ctx . Tr ( "form.username_been_taken" ) , tplLinkAccount , & form )
case models . IsErrEmailAlreadyUsed ( err ) :
ctx . Data [ "Err_Email" ] = true
ctx . RenderWithErr ( ctx . Tr ( "form.email_been_used" ) , tplLinkAccount , & form )
case models . IsErrNameReserved ( err ) :
ctx . Data [ "Err_UserName" ] = true
ctx . RenderWithErr ( ctx . Tr ( "user.form.name_reserved" , err . ( models . ErrNameReserved ) . Name ) , tplLinkAccount , & form )
case models . IsErrNamePatternNotAllowed ( err ) :
ctx . Data [ "Err_UserName" ] = true
ctx . RenderWithErr ( ctx . Tr ( "user.form.name_pattern_not_allowed" , err . ( models . ErrNamePatternNotAllowed ) . Pattern ) , tplLinkAccount , & form )
default :
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "CreateUser" , err )
2017-02-22 15:14:37 +08:00
}
return
}
log . Trace ( "Account created: %s" , u . Name )
// Auto-set admin for the only user.
if models . CountUsers ( ) == 1 {
u . IsAdmin = true
u . IsActive = true
2017-08-12 22:18:44 +08:00
u . SetLastLogin ( )
if err := models . UpdateUserCols ( u , "is_admin" , "is_active" , "last_login_unix" ) ; err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UpdateUser" , err )
2017-02-22 15:14:37 +08:00
return
}
}
// Send confirmation email
if setting . Service . RegisterEmailConfirm && u . ID > 1 {
models . SendActivateAccountMail ( ctx . Context , u )
ctx . Data [ "IsSendRegisterMail" ] = true
ctx . Data [ "Email" ] = u . Email
2019-08-15 22:46:21 +08:00
ctx . Data [ "ActiveCodeLives" ] = timeutil . MinutesToFriendly ( setting . Service . ActiveCodeLives , ctx . Locale . Language ( ) )
2017-02-22 15:14:37 +08:00
ctx . HTML ( 200 , TplActivate )
if err := ctx . Cache . Put ( "MailResendLimit_" + u . LowerName , u . LowerName , 180 ) ; err != nil {
2019-04-02 15:48:31 +08:00
log . Error ( "Set cache(MailResendLimit) fail: %v" , err )
2017-02-22 15:14:37 +08:00
}
return
}
ctx . Redirect ( setting . AppSubURL + "/user/login" )
}
2019-04-18 15:23:59 +08:00
func handleSignOut ( ctx * context . Context ) {
2019-06-13 03:41:28 +08:00
_ = ctx . Session . Delete ( "uid" )
_ = ctx . Session . Delete ( "uname" )
_ = ctx . Session . Delete ( "socialId" )
_ = ctx . Session . Delete ( "socialName" )
_ = ctx . Session . Delete ( "socialEmail" )
2019-07-12 21:57:31 +08:00
ctx . SetCookie ( setting . CookieUserName , "" , - 1 , setting . AppSubURL , setting . SessionConfig . Domain , setting . SessionConfig . Secure , true )
ctx . SetCookie ( setting . CookieRememberName , "" , - 1 , setting . AppSubURL , setting . SessionConfig . Domain , setting . SessionConfig . Secure , true )
ctx . SetCookie ( setting . CSRFCookieName , "" , - 1 , setting . AppSubURL , setting . SessionConfig . Domain , setting . SessionConfig . Secure , true )
ctx . SetCookie ( "lang" , "" , - 1 , setting . AppSubURL , setting . SessionConfig . Domain , setting . SessionConfig . Secure , true ) // Setting the lang cookie will trigger the middleware to reset the language ot previous state.
2019-04-18 15:23:59 +08:00
}
// SignOut sign out from login status
func SignOut ( ctx * context . Context ) {
handleSignOut ( ctx )
2016-11-27 18:14:25 +08:00
ctx . Redirect ( setting . AppSubURL + "/" )
2014-07-26 12:24:27 +08:00
}
2016-11-18 11:03:03 +08:00
// SignUp render the register page
2016-03-12 00:56:52 +08:00
func SignUp ( ctx * context . Context ) {
2014-07-26 12:24:27 +08:00
ctx . Data [ "Title" ] = ctx . Tr ( "sign_up" )
2017-03-07 18:47:56 +08:00
ctx . Data [ "SignUpLink" ] = setting . AppSubURL + "/user/sign_up"
2015-09-13 23:07:21 +08:00
ctx . Data [ "EnableCaptcha" ] = setting . Service . EnableCaptcha
2019-05-02 21:09:39 +08:00
ctx . Data [ "RecaptchaURL" ] = setting . Service . RecaptchaURL
2018-07-05 12:13:05 +08:00
ctx . Data [ "CaptchaType" ] = setting . Service . CaptchaType
ctx . Data [ "RecaptchaSitekey" ] = setting . Service . RecaptchaSitekey
2017-02-22 15:14:37 +08:00
ctx . Data [ "DisableRegistration" ] = setting . Service . DisableRegistration
2014-07-26 12:24:27 +08:00
2016-11-18 11:03:03 +08:00
ctx . HTML ( 200 , tplSignUp )
2014-07-26 12:24:27 +08:00
}
2016-11-18 11:03:03 +08:00
// SignUpPost response for sign up information submission
2016-03-12 00:56:52 +08:00
func SignUpPost ( ctx * context . Context , cpt * captcha . Captcha , form auth . RegisterForm ) {
2014-07-26 12:24:27 +08:00
ctx . Data [ "Title" ] = ctx . Tr ( "sign_up" )
2017-03-07 18:47:56 +08:00
ctx . Data [ "SignUpLink" ] = setting . AppSubURL + "/user/sign_up"
2015-09-13 23:07:21 +08:00
ctx . Data [ "EnableCaptcha" ] = setting . Service . EnableCaptcha
2019-05-02 21:09:39 +08:00
ctx . Data [ "RecaptchaURL" ] = setting . Service . RecaptchaURL
2018-07-05 12:13:05 +08:00
ctx . Data [ "CaptchaType" ] = setting . Service . CaptchaType
ctx . Data [ "RecaptchaSitekey" ] = setting . Service . RecaptchaSitekey
2018-05-13 15:51:16 +08:00
//Permission denied if DisableRegistration or AllowOnlyExternalRegistration options are true
if ! setting . Service . ShowRegistrationButton {
2014-07-26 12:24:27 +08:00
ctx . Error ( 403 )
return
}
if ctx . HasError ( ) {
2016-11-18 11:03:03 +08:00
ctx . HTML ( 200 , tplSignUp )
2014-07-26 12:24:27 +08:00
return
}
2019-07-07 03:48:02 +08:00
if setting . Service . EnableCaptcha {
var valid bool
switch setting . Service . CaptchaType {
case setting . ImageCaptcha :
valid = cpt . VerifyReq ( ctx . Req )
case setting . ReCaptcha :
valid , _ = recaptcha . Verify ( form . GRecaptchaResponse )
default :
ctx . ServerError ( "Unknown Captcha Type" , fmt . Errorf ( "Unknown Captcha Type: %s" , setting . Service . CaptchaType ) )
return
}
2015-09-13 21:51:51 +08:00
2018-07-05 12:13:05 +08:00
if ! valid {
ctx . Data [ "Err_Captcha" ] = true
ctx . RenderWithErr ( ctx . Tr ( "form.captcha_incorrect" ) , tplSignUp , & form )
return
}
}
2018-11-15 09:00:04 +08:00
if ! form . IsEmailDomainWhitelisted ( ) {
ctx . RenderWithErr ( ctx . Tr ( "auth.email_domain_blacklisted" ) , tplSignUp , & form )
return
}
2015-09-13 21:51:51 +08:00
if form . Password != form . Retype {
2014-07-26 12:24:27 +08:00
ctx . Data [ "Err_Password" ] = true
2016-11-18 11:03:03 +08:00
ctx . RenderWithErr ( ctx . Tr ( "form.password_not_match" ) , tplSignUp , & form )
2014-07-26 12:24:27 +08:00
return
}
2016-12-24 21:40:44 +08:00
if len ( form . Password ) < setting . MinPasswordLength {
ctx . Data [ "Err_Password" ] = true
ctx . RenderWithErr ( ctx . Tr ( "auth.password_too_short" , setting . MinPasswordLength ) , tplSignUp , & form )
return
}
2014-07-26 12:24:27 +08:00
u := & models . User {
Name : form . UserName ,
Email : form . Email ,
Passwd : form . Password ,
2015-09-18 04:11:44 +08:00
IsActive : ! setting . Service . RegisterEmailConfirm ,
2014-07-26 12:24:27 +08:00
}
if err := models . CreateUser ( u ) ; err != nil {
2015-03-27 05:11:47 +08:00
switch {
case models . IsErrUserAlreadyExist ( err ) :
2014-07-26 12:24:27 +08:00
ctx . Data [ "Err_UserName" ] = true
2016-11-18 11:03:03 +08:00
ctx . RenderWithErr ( ctx . Tr ( "form.username_been_taken" ) , tplSignUp , & form )
2015-03-27 05:11:47 +08:00
case models . IsErrEmailAlreadyUsed ( err ) :
2014-07-26 12:24:27 +08:00
ctx . Data [ "Err_Email" ] = true
2016-11-18 11:03:03 +08:00
ctx . RenderWithErr ( ctx . Tr ( "form.email_been_used" ) , tplSignUp , & form )
2015-03-27 05:11:47 +08:00
case models . IsErrNameReserved ( err ) :
2014-07-26 12:24:27 +08:00
ctx . Data [ "Err_UserName" ] = true
2016-11-18 11:03:03 +08:00
ctx . RenderWithErr ( ctx . Tr ( "user.form.name_reserved" , err . ( models . ErrNameReserved ) . Name ) , tplSignUp , & form )
2015-03-27 05:11:47 +08:00
case models . IsErrNamePatternNotAllowed ( err ) :
ctx . Data [ "Err_UserName" ] = true
2016-11-18 11:03:03 +08:00
ctx . RenderWithErr ( ctx . Tr ( "user.form.name_pattern_not_allowed" , err . ( models . ErrNamePatternNotAllowed ) . Pattern ) , tplSignUp , & form )
2014-07-26 12:24:27 +08:00
default :
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "CreateUser" , err )
2014-07-26 12:24:27 +08:00
}
return
}
log . Trace ( "Account created: %s" , u . Name )
2015-08-19 04:58:45 +08:00
// Auto-set admin for the only user.
if models . CountUsers ( ) == 1 {
u . IsAdmin = true
u . IsActive = true
2017-08-12 22:18:44 +08:00
u . SetLastLogin ( )
if err := models . UpdateUserCols ( u , "is_admin" , "is_active" , "last_login_unix" ) ; err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UpdateUser" , err )
2015-08-19 04:58:45 +08:00
return
}
}
2016-07-16 00:36:39 +08:00
// Send confirmation email, no need for social account.
2016-07-24 01:08:22 +08:00
if setting . Service . RegisterEmailConfirm && u . ID > 1 {
2016-07-16 00:36:39 +08:00
models . SendActivateAccountMail ( ctx . Context , u )
2014-08-10 08:25:02 +08:00
ctx . Data [ "IsSendRegisterMail" ] = true
ctx . Data [ "Email" ] = u . Email
2019-08-15 22:46:21 +08:00
ctx . Data [ "ActiveCodeLives" ] = timeutil . MinutesToFriendly ( setting . Service . ActiveCodeLives , ctx . Locale . Language ( ) )
2016-11-18 11:03:03 +08:00
ctx . HTML ( 200 , TplActivate )
2014-08-10 08:25:02 +08:00
if err := ctx . Cache . Put ( "MailResendLimit_" + u . LowerName , u . LowerName , 180 ) ; err != nil {
2019-04-02 15:48:31 +08:00
log . Error ( "Set cache(MailResendLimit) fail: %v" , err )
2014-08-10 08:25:02 +08:00
}
return
}
2014-07-26 12:24:27 +08:00
2018-08-12 03:33:19 +08:00
ctx . Flash . Success ( ctx . Tr ( "auth.sign_up_successful" ) )
handleSignInFull ( ctx , u , false , true )
2014-07-26 12:24:27 +08:00
}
2016-11-18 11:03:03 +08:00
// Activate render activate user page
2016-03-12 00:56:52 +08:00
func Activate ( ctx * context . Context ) {
2014-08-10 12:02:00 +08:00
code := ctx . Query ( "code" )
if len ( code ) == 0 {
ctx . Data [ "IsActivatePage" ] = true
if ctx . User . IsActive {
ctx . Error ( 404 )
return
}
2016-07-16 00:36:39 +08:00
// Resend confirmation email.
2014-08-10 12:02:00 +08:00
if setting . Service . RegisterEmailConfirm {
if ctx . Cache . IsExist ( "MailResendLimit_" + ctx . User . LowerName ) {
ctx . Data [ "ResendLimited" ] = true
} else {
2019-08-15 22:46:21 +08:00
ctx . Data [ "ActiveCodeLives" ] = timeutil . MinutesToFriendly ( setting . Service . ActiveCodeLives , ctx . Locale . Language ( ) )
2016-07-16 00:36:39 +08:00
models . SendActivateAccountMail ( ctx . Context , ctx . User )
2014-08-10 12:02:00 +08:00
if err := ctx . Cache . Put ( "MailResendLimit_" + ctx . User . LowerName , ctx . User . LowerName , 180 ) ; err != nil {
2019-04-02 15:48:31 +08:00
log . Error ( "Set cache(MailResendLimit) fail: %v" , err )
2014-08-10 12:02:00 +08:00
}
}
} else {
ctx . Data [ "ServiceNotEnabled" ] = true
}
2016-11-18 11:03:03 +08:00
ctx . HTML ( 200 , TplActivate )
2014-08-10 12:02:00 +08:00
return
}
// Verify code.
if user := models . VerifyUserActiveCode ( code ) ; user != nil {
user . IsActive = true
2016-12-20 20:32:02 +08:00
var err error
if user . Rands , err = models . GetUserSalt ( ) ; err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UpdateUser" , err )
2016-12-20 20:32:02 +08:00
return
}
2017-08-12 22:18:44 +08:00
if err := models . UpdateUserCols ( user , "is_active" , "rands" ) ; err != nil {
2015-08-05 11:14:17 +08:00
if models . IsErrUserNotExist ( err ) {
2014-08-10 12:02:00 +08:00
ctx . Error ( 404 )
} else {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "UpdateUser" , err )
2014-08-10 12:02:00 +08:00
}
return
}
log . Trace ( "User activated: %s" , user . Name )
2019-06-13 03:41:28 +08:00
err = ctx . Session . Set ( "uid" , user . ID )
if err != nil {
log . Error ( fmt . Sprintf ( "Error setting session: %v" , err ) )
}
err = ctx . Session . Set ( "uname" , user . Name )
if err != nil {
log . Error ( fmt . Sprintf ( "Error setting session: %v" , err ) )
}
2018-07-26 01:54:56 +08:00
ctx . Flash . Success ( ctx . Tr ( "auth.account_activated" ) )
2016-11-27 18:14:25 +08:00
ctx . Redirect ( setting . AppSubURL + "/" )
2014-08-10 12:02:00 +08:00
return
}
ctx . Data [ "IsActivateFailed" ] = true
2016-11-18 11:03:03 +08:00
ctx . HTML ( 200 , TplActivate )
2014-07-26 12:24:27 +08:00
}
2016-11-18 11:03:03 +08:00
// ActivateEmail render the activate email page
2016-03-12 00:56:52 +08:00
func ActivateEmail ( ctx * context . Context ) {
2014-12-17 23:41:49 +08:00
code := ctx . Query ( "code" )
2016-11-18 11:03:03 +08:00
emailStr := ctx . Query ( "email" )
2014-12-17 23:41:49 +08:00
// Verify code.
2016-11-18 11:03:03 +08:00
if email := models . VerifyActiveEmailCode ( code , emailStr ) ; email != nil {
2015-02-22 11:13:47 +08:00
if err := email . Activate ( ) ; err != nil {
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "ActivateEmail" , err )
2014-12-17 23:41:49 +08:00
}
log . Trace ( "Email activated: %s" , email . Email )
2015-11-20 00:52:39 +08:00
ctx . Flash . Success ( ctx . Tr ( "settings.add_email_success" ) )
2014-12-17 23:41:49 +08:00
}
2016-11-27 18:14:25 +08:00
ctx . Redirect ( setting . AppSubURL + "/user/settings/email" )
2014-12-17 23:41:49 +08:00
}
2016-11-18 11:03:03 +08:00
// ForgotPasswd render the forget pasword page
2016-03-12 00:56:52 +08:00
func ForgotPasswd ( ctx * context . Context ) {
2017-02-21 02:27:20 +08:00
ctx . Data [ "Title" ] = ctx . Tr ( "auth.forgot_password_title" )
2014-07-26 12:24:27 +08:00
if setting . MailService == nil {
ctx . Data [ "IsResetDisable" ] = true
2016-11-18 11:03:03 +08:00
ctx . HTML ( 200 , tplForgotPassword )
2014-07-26 12:24:27 +08:00
return
}
2017-03-11 17:11:54 +08:00
email := ctx . Query ( "email" )
ctx . Data [ "Email" ] = email
2014-07-26 12:24:27 +08:00
ctx . Data [ "IsResetRequest" ] = true
2016-11-18 11:03:03 +08:00
ctx . HTML ( 200 , tplForgotPassword )
2014-07-26 12:24:27 +08:00
}
2016-11-18 11:03:03 +08:00
// ForgotPasswdPost response for forget password request
2016-03-12 00:56:52 +08:00
func ForgotPasswdPost ( ctx * context . Context ) {
2017-02-21 02:27:20 +08:00
ctx . Data [ "Title" ] = ctx . Tr ( "auth.forgot_password_title" )
2014-08-10 12:02:00 +08:00
if setting . MailService == nil {
2018-01-11 05:34:17 +08:00
ctx . NotFound ( "ForgotPasswdPost" , nil )
2014-08-10 12:02:00 +08:00
return
}
ctx . Data [ "IsResetRequest" ] = true
email := ctx . Query ( "email" )
2015-09-18 02:57:24 +08:00
ctx . Data [ "Email" ] = email
2014-08-10 12:02:00 +08:00
u , err := models . GetUserByEmail ( email )
if err != nil {
2015-08-05 11:14:17 +08:00
if models . IsErrUserNotExist ( err ) {
2019-08-15 22:46:21 +08:00
ctx . Data [ "ResetPwdCodeLives" ] = timeutil . MinutesToFriendly ( setting . Service . ResetPwdCodeLives , ctx . Locale . Language ( ) )
2016-10-17 10:08:40 +08:00
ctx . Data [ "IsResetSent" ] = true
2016-11-18 11:03:03 +08:00
ctx . HTML ( 200 , tplForgotPassword )
2016-10-17 10:08:40 +08:00
return
2014-08-10 12:02:00 +08:00
}
2016-11-18 11:03:03 +08:00
2018-01-11 05:34:17 +08:00
ctx . ServerError ( "user.ResetPasswd(check existence)" , err )
2014-08-10 12:02:00 +08:00
return
}
2017-02-22 15:14:37 +08:00
if ! u . IsLocal ( ) && ! u . IsOAuth2 ( ) {
2016-03-14 22:40:16 +08:00
ctx . Data [ "Err_Email" ] = true
2016-11-18 11:03:03 +08:00
ctx . RenderWithErr ( ctx . Tr ( "auth.non_local_account" ) , tplForgotPassword , nil )
2016-03-14 22:40:16 +08:00
return
}
2014-08-10 12:02:00 +08:00
if ctx . Cache . IsExist ( "MailResendLimit_" + u . LowerName ) {
ctx . Data [ "ResendLimited" ] = true
2016-11-18 11:03:03 +08:00
ctx . HTML ( 200 , tplForgotPassword )
2014-08-10 12:02:00 +08:00
return
}
2016-07-16 00:36:39 +08:00
models . SendResetPasswordMail ( ctx . Context , u )
2014-08-10 12:02:00 +08:00
if err = ctx . Cache . Put ( "MailResendLimit_" + u . LowerName , u . LowerName , 180 ) ; err != nil {
2019-04-02 15:48:31 +08:00
log . Error ( "Set cache(MailResendLimit) fail: %v" , err )
2014-08-10 12:02:00 +08:00
}
2019-08-15 22:46:21 +08:00
ctx . Data [ "ResetPwdCodeLives" ] = timeutil . MinutesToFriendly ( setting . Service . ResetPwdCodeLives , ctx . Locale . Language ( ) )
2014-08-10 12:02:00 +08:00
ctx . Data [ "IsResetSent" ] = true
2016-11-18 11:03:03 +08:00
ctx . HTML ( 200 , tplForgotPassword )
2014-07-26 12:24:27 +08:00
}
2019-04-18 15:23:59 +08:00
func commonResetPassword ( ctx * context . Context ) * models . User {
code := ctx . Query ( "code" )
2014-08-10 12:02:00 +08:00
ctx . Data [ "Title" ] = ctx . Tr ( "auth.reset_password" )
2019-04-18 15:23:59 +08:00
ctx . Data [ "Code" ] = code
if nil != ctx . User {
ctx . Data [ "user_signed_in" ] = true
}
2014-07-26 12:24:27 +08:00
if len ( code ) == 0 {
2019-04-18 15:23:59 +08:00
ctx . Flash . Error ( ctx . Tr ( "auth.invalid_code" ) )
return nil
2014-07-26 12:24:27 +08:00
}
2019-02-20 07:09:47 +08:00
2019-04-18 15:23:59 +08:00
// Fail early, don't frustrate the user
u := models . VerifyUserActiveCode ( code )
if u == nil {
ctx . Flash . Error ( ctx . Tr ( "auth.invalid_code" ) )
return nil
2019-02-20 07:09:47 +08:00
}
2019-04-18 15:23:59 +08:00
// Show the user that they are affecting the account that they intended to
ctx . Data [ "user_email" ] = u . Email
if nil != ctx . User && u . ID != ctx . User . ID {
ctx . Flash . Error ( ctx . Tr ( "auth.reset_password_wrong_user" , ctx . User . Email , u . Email ) )
return nil
}
return u
}
// ResetPasswd render the account recovery page
func ResetPasswd ( ctx * context . Context ) {
ctx . Data [ "IsResetForm" ] = true
commonResetPassword ( ctx )
2016-11-18 11:03:03 +08:00
ctx . HTML ( 200 , tplResetPassword )
2014-07-26 12:24:27 +08:00
}
2019-04-18 15:23:59 +08:00
// ResetPasswdPost response from account recovery request
2016-03-12 00:56:52 +08:00
func ResetPasswdPost ( ctx * context . Context ) {
2019-04-18 15:23:59 +08:00
u := commonResetPassword ( ctx )
2014-08-10 12:02:00 +08:00
2019-04-18 15:23:59 +08:00
if u == nil {
// Flash error has been set
ctx . HTML ( 200 , tplResetPassword )
2014-08-10 12:02:00 +08:00
return
}
2019-04-18 15:23:59 +08:00
// Validate password length.
passwd := ctx . Query ( "password" )
if len ( passwd ) < setting . MinPasswordLength {
ctx . Data [ "IsResetForm" ] = true
ctx . Data [ "Err_Password" ] = true
ctx . RenderWithErr ( ctx . Tr ( "auth.password_too_short" , setting . MinPasswordLength ) , tplResetPassword , nil )
return
}
2014-08-10 12:02:00 +08:00
2019-04-18 15:23:59 +08:00
var err error
if u . Rands , err = models . GetUserSalt ( ) ; err != nil {
ctx . ServerError ( "UpdateUser" , err )
return
}
if u . Salt , err = models . GetUserSalt ( ) ; err != nil {
ctx . ServerError ( "UpdateUser" , err )
return
}
2014-08-10 12:02:00 +08:00
2019-04-18 15:23:59 +08:00
u . HashPassword ( passwd )
u . MustChangePassword = false
if err := models . UpdateUserCols ( u , "must_change_password" , "passwd" , "rands" , "salt" ) ; err != nil {
ctx . ServerError ( "UpdateUser" , err )
2014-08-10 12:02:00 +08:00
return
}
2019-04-18 15:23:59 +08:00
log . Trace ( "User password reset: %s" , u . Name )
2014-08-10 12:02:00 +08:00
ctx . Data [ "IsResetFailed" ] = true
2019-04-18 15:23:59 +08:00
remember := len ( ctx . Query ( "remember" ) ) != 0
handleSignInFull ( ctx , u , remember , true )
2014-07-26 12:24:27 +08:00
}
2018-09-13 20:04:25 +08:00
// MustChangePassword renders the page to change a user's password
func MustChangePassword ( ctx * context . Context ) {
ctx . Data [ "Title" ] = ctx . Tr ( "auth.must_change_password" )
ctx . Data [ "ChangePasscodeLink" ] = setting . AppSubURL + "/user/settings/change_password"
ctx . HTML ( 200 , tplMustChangePassword )
}
// MustChangePasswordPost response for updating a user's password after his/her
// account was created by an admin
func MustChangePasswordPost ( ctx * context . Context , cpt * captcha . Captcha , form auth . MustChangePasswordForm ) {
ctx . Data [ "Title" ] = ctx . Tr ( "auth.must_change_password" )
ctx . Data [ "ChangePasscodeLink" ] = setting . AppSubURL + "/user/settings/change_password"
if ctx . HasError ( ) {
ctx . HTML ( 200 , tplMustChangePassword )
return
}
u := ctx . User
// Make sure only requests for users who are eligible to change their password via
// this method passes through
if ! u . MustChangePassword {
ctx . ServerError ( "MustUpdatePassword" , errors . New ( "cannot update password.. Please visit the settings page" ) )
return
}
if form . Password != form . Retype {
ctx . Data [ "Err_Password" ] = true
ctx . RenderWithErr ( ctx . Tr ( "form.password_not_match" ) , tplMustChangePassword , & form )
return
}
if len ( form . Password ) < setting . MinPasswordLength {
ctx . Data [ "Err_Password" ] = true
ctx . RenderWithErr ( ctx . Tr ( "auth.password_too_short" , setting . MinPasswordLength ) , tplMustChangePassword , & form )
return
}
var err error
if u . Salt , err = models . GetUserSalt ( ) ; err != nil {
ctx . ServerError ( "UpdateUser" , err )
return
}
u . HashPassword ( form . Password )
u . MustChangePassword = false
if err := models . UpdateUserCols ( u , "must_change_password" , "passwd" , "salt" ) ; err != nil {
ctx . ServerError ( "UpdateUser" , err )
return
}
ctx . Flash . Success ( ctx . Tr ( "settings.change_password_success" ) )
log . Trace ( "User updated password: %s" , u . Name )
2019-03-21 10:06:16 +08:00
if redirectTo := ctx . GetCookie ( "redirect_to" ) ; len ( redirectTo ) > 0 && ! util . IsExternalURL ( redirectTo ) {
2018-09-13 20:04:25 +08:00
ctx . SetCookie ( "redirect_to" , "" , - 1 , setting . AppSubURL )
ctx . RedirectToFirst ( redirectTo )
return
}
ctx . Redirect ( setting . AppSubURL + "/" )
}