You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

auth.go 29KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011
  1. // Copyright 2014 The Gogs Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package user
  5. import (
  6. "errors"
  7. "fmt"
  8. "net/http"
  9. "net/url"
  10. "strings"
  11. "code.gitea.io/gitea/models"
  12. "code.gitea.io/gitea/modules/auth"
  13. "code.gitea.io/gitea/modules/auth/oauth2"
  14. "code.gitea.io/gitea/modules/base"
  15. "code.gitea.io/gitea/modules/context"
  16. "code.gitea.io/gitea/modules/log"
  17. "code.gitea.io/gitea/modules/setting"
  18. "github.com/go-macaron/captcha"
  19. "github.com/markbates/goth"
  20. )
  21. const (
  22. // tplSignIn template for sign in page
  23. tplSignIn base.TplName = "user/auth/signin"
  24. // tplSignUp template path for sign up page
  25. tplSignUp base.TplName = "user/auth/signup"
  26. // TplActivate template path for activate user
  27. TplActivate base.TplName = "user/auth/activate"
  28. tplForgotPassword base.TplName = "user/auth/forgot_passwd"
  29. tplResetPassword base.TplName = "user/auth/reset_passwd"
  30. tplTwofa base.TplName = "user/auth/twofa"
  31. tplTwofaScratch base.TplName = "user/auth/twofa_scratch"
  32. tplLinkAccount base.TplName = "user/auth/link_account"
  33. )
  34. // AutoSignIn reads cookie and try to auto-login.
  35. func AutoSignIn(ctx *context.Context) (bool, error) {
  36. if !models.HasEngine {
  37. return false, nil
  38. }
  39. uname := ctx.GetCookie(setting.CookieUserName)
  40. if len(uname) == 0 {
  41. return false, nil
  42. }
  43. isSucceed := false
  44. defer func() {
  45. if !isSucceed {
  46. log.Trace("auto-login cookie cleared: %s", uname)
  47. ctx.SetCookie(setting.CookieUserName, "", -1, setting.AppSubURL)
  48. ctx.SetCookie(setting.CookieRememberName, "", -1, setting.AppSubURL)
  49. }
  50. }()
  51. u, err := models.GetUserByName(uname)
  52. if err != nil {
  53. if !models.IsErrUserNotExist(err) {
  54. return false, fmt.Errorf("GetUserByName: %v", err)
  55. }
  56. return false, nil
  57. }
  58. if val, _ := ctx.GetSuperSecureCookie(
  59. base.EncodeMD5(u.Rands+u.Passwd), setting.CookieRememberName); val != u.Name {
  60. return false, nil
  61. }
  62. isSucceed = true
  63. ctx.Session.Set("uid", u.ID)
  64. ctx.Session.Set("uname", u.Name)
  65. ctx.SetCookie(setting.CSRFCookieName, "", -1, setting.AppSubURL)
  66. return true, nil
  67. }
  68. func checkAutoLogin(ctx *context.Context) bool {
  69. // Check auto-login.
  70. isSucceed, err := AutoSignIn(ctx)
  71. if err != nil {
  72. ctx.ServerError("AutoSignIn", err)
  73. return true
  74. }
  75. redirectTo := ctx.Query("redirect_to")
  76. if len(redirectTo) > 0 {
  77. ctx.SetCookie("redirect_to", redirectTo, 0, setting.AppSubURL)
  78. } else {
  79. redirectTo, _ = url.QueryUnescape(ctx.GetCookie("redirect_to"))
  80. }
  81. if isSucceed {
  82. ctx.SetCookie("redirect_to", "", -1, setting.AppSubURL)
  83. ctx.RedirectToFirst(redirectTo, setting.AppSubURL+string(setting.LandingPageURL))
  84. return true
  85. }
  86. return false
  87. }
  88. // SignIn render sign in page
  89. func SignIn(ctx *context.Context) {
  90. ctx.Data["Title"] = ctx.Tr("sign_in")
  91. // Check auto-login.
  92. if checkAutoLogin(ctx) {
  93. return
  94. }
  95. orderedOAuth2Names, oauth2Providers, err := models.GetActiveOAuth2Providers()
  96. if err != nil {
  97. ctx.ServerError("UserSignIn", err)
  98. return
  99. }
  100. ctx.Data["OrderedOAuth2Names"] = orderedOAuth2Names
  101. ctx.Data["OAuth2Providers"] = oauth2Providers
  102. ctx.Data["Title"] = ctx.Tr("sign_in")
  103. ctx.Data["SignInLink"] = setting.AppSubURL + "/user/login"
  104. ctx.Data["PageIsSignIn"] = true
  105. ctx.Data["PageIsLogin"] = true
  106. ctx.HTML(200, tplSignIn)
  107. }
  108. // SignInPost response for sign in request
  109. func SignInPost(ctx *context.Context, form auth.SignInForm) {
  110. ctx.Data["Title"] = ctx.Tr("sign_in")
  111. orderedOAuth2Names, oauth2Providers, err := models.GetActiveOAuth2Providers()
  112. if err != nil {
  113. ctx.ServerError("UserSignIn", err)
  114. return
  115. }
  116. ctx.Data["OrderedOAuth2Names"] = orderedOAuth2Names
  117. ctx.Data["OAuth2Providers"] = oauth2Providers
  118. ctx.Data["Title"] = ctx.Tr("sign_in")
  119. ctx.Data["SignInLink"] = setting.AppSubURL + "/user/login"
  120. ctx.Data["PageIsSignIn"] = true
  121. ctx.Data["PageIsLogin"] = true
  122. if ctx.HasError() {
  123. ctx.HTML(200, tplSignIn)
  124. return
  125. }
  126. u, err := models.UserSignIn(form.UserName, form.Password)
  127. if err != nil {
  128. if models.IsErrUserNotExist(err) {
  129. ctx.RenderWithErr(ctx.Tr("form.username_password_incorrect"), tplSignIn, &form)
  130. log.Info("Failed authentication attempt for %s from %s", form.UserName, ctx.RemoteAddr())
  131. } else if models.IsErrEmailAlreadyUsed(err) {
  132. ctx.RenderWithErr(ctx.Tr("form.email_been_used"), tplSignIn, &form)
  133. log.Info("Failed authentication attempt for %s from %s", form.UserName, ctx.RemoteAddr())
  134. } else {
  135. ctx.ServerError("UserSignIn", err)
  136. }
  137. return
  138. }
  139. // If this user is enrolled in 2FA, we can't sign the user in just yet.
  140. // Instead, redirect them to the 2FA authentication page.
  141. _, err = models.GetTwoFactorByUID(u.ID)
  142. if err != nil {
  143. if models.IsErrTwoFactorNotEnrolled(err) {
  144. handleSignIn(ctx, u, form.Remember)
  145. } else {
  146. ctx.ServerError("UserSignIn", err)
  147. }
  148. return
  149. }
  150. // User needs to use 2FA, save data and redirect to 2FA page.
  151. ctx.Session.Set("twofaUid", u.ID)
  152. ctx.Session.Set("twofaRemember", form.Remember)
  153. ctx.Redirect(setting.AppSubURL + "/user/two_factor")
  154. }
  155. // TwoFactor shows the user a two-factor authentication page.
  156. func TwoFactor(ctx *context.Context) {
  157. ctx.Data["Title"] = ctx.Tr("twofa")
  158. // Check auto-login.
  159. if checkAutoLogin(ctx) {
  160. return
  161. }
  162. // Ensure user is in a 2FA session.
  163. if ctx.Session.Get("twofaUid") == nil {
  164. ctx.ServerError("UserSignIn", errors.New("not in 2FA session"))
  165. return
  166. }
  167. ctx.HTML(200, tplTwofa)
  168. }
  169. // TwoFactorPost validates a user's two-factor authentication token.
  170. func TwoFactorPost(ctx *context.Context, form auth.TwoFactorAuthForm) {
  171. ctx.Data["Title"] = ctx.Tr("twofa")
  172. // Ensure user is in a 2FA session.
  173. idSess := ctx.Session.Get("twofaUid")
  174. if idSess == nil {
  175. ctx.ServerError("UserSignIn", errors.New("not in 2FA session"))
  176. return
  177. }
  178. id := idSess.(int64)
  179. twofa, err := models.GetTwoFactorByUID(id)
  180. if err != nil {
  181. ctx.ServerError("UserSignIn", err)
  182. return
  183. }
  184. // Validate the passcode with the stored TOTP secret.
  185. ok, err := twofa.ValidateTOTP(form.Passcode)
  186. if err != nil {
  187. ctx.ServerError("UserSignIn", err)
  188. return
  189. }
  190. if ok && twofa.LastUsedPasscode != form.Passcode {
  191. remember := ctx.Session.Get("twofaRemember").(bool)
  192. u, err := models.GetUserByID(id)
  193. if err != nil {
  194. ctx.ServerError("UserSignIn", err)
  195. return
  196. }
  197. if ctx.Session.Get("linkAccount") != nil {
  198. gothUser := ctx.Session.Get("linkAccountGothUser")
  199. if gothUser == nil {
  200. ctx.ServerError("UserSignIn", errors.New("not in LinkAccount session"))
  201. return
  202. }
  203. err = models.LinkAccountToUser(u, gothUser.(goth.User))
  204. if err != nil {
  205. ctx.ServerError("UserSignIn", err)
  206. return
  207. }
  208. }
  209. twofa.LastUsedPasscode = form.Passcode
  210. if err = models.UpdateTwoFactor(twofa); err != nil {
  211. ctx.ServerError("UserSignIn", err)
  212. return
  213. }
  214. handleSignIn(ctx, u, remember)
  215. return
  216. }
  217. ctx.RenderWithErr(ctx.Tr("auth.twofa_passcode_incorrect"), tplTwofa, auth.TwoFactorAuthForm{})
  218. }
  219. // TwoFactorScratch shows the scratch code form for two-factor authentication.
  220. func TwoFactorScratch(ctx *context.Context) {
  221. ctx.Data["Title"] = ctx.Tr("twofa_scratch")
  222. // Check auto-login.
  223. if checkAutoLogin(ctx) {
  224. return
  225. }
  226. // Ensure user is in a 2FA session.
  227. if ctx.Session.Get("twofaUid") == nil {
  228. ctx.ServerError("UserSignIn", errors.New("not in 2FA session"))
  229. return
  230. }
  231. ctx.HTML(200, tplTwofaScratch)
  232. }
  233. // TwoFactorScratchPost validates and invalidates a user's two-factor scratch token.
  234. func TwoFactorScratchPost(ctx *context.Context, form auth.TwoFactorScratchAuthForm) {
  235. ctx.Data["Title"] = ctx.Tr("twofa_scratch")
  236. // Ensure user is in a 2FA session.
  237. idSess := ctx.Session.Get("twofaUid")
  238. if idSess == nil {
  239. ctx.ServerError("UserSignIn", errors.New("not in 2FA session"))
  240. return
  241. }
  242. id := idSess.(int64)
  243. twofa, err := models.GetTwoFactorByUID(id)
  244. if err != nil {
  245. ctx.ServerError("UserSignIn", err)
  246. return
  247. }
  248. // Validate the passcode with the stored TOTP secret.
  249. if twofa.VerifyScratchToken(form.Token) {
  250. // Invalidate the scratch token.
  251. twofa.ScratchToken = ""
  252. if err = models.UpdateTwoFactor(twofa); err != nil {
  253. ctx.ServerError("UserSignIn", err)
  254. return
  255. }
  256. remember := ctx.Session.Get("twofaRemember").(bool)
  257. u, err := models.GetUserByID(id)
  258. if err != nil {
  259. ctx.ServerError("UserSignIn", err)
  260. return
  261. }
  262. handleSignInFull(ctx, u, remember, false)
  263. ctx.Flash.Info(ctx.Tr("auth.twofa_scratch_used"))
  264. ctx.Redirect(setting.AppSubURL + "/user/settings/two_factor")
  265. return
  266. }
  267. ctx.RenderWithErr(ctx.Tr("auth.twofa_scratch_token_incorrect"), tplTwofaScratch, auth.TwoFactorScratchAuthForm{})
  268. }
  269. // This handles the final part of the sign-in process of the user.
  270. func handleSignIn(ctx *context.Context, u *models.User, remember bool) {
  271. handleSignInFull(ctx, u, remember, true)
  272. }
  273. func handleSignInFull(ctx *context.Context, u *models.User, remember bool, obeyRedirect bool) {
  274. if remember {
  275. days := 86400 * setting.LogInRememberDays
  276. ctx.SetCookie(setting.CookieUserName, u.Name, days, setting.AppSubURL)
  277. ctx.SetSuperSecureCookie(base.EncodeMD5(u.Rands+u.Passwd),
  278. setting.CookieRememberName, u.Name, days, setting.AppSubURL)
  279. }
  280. ctx.Session.Delete("openid_verified_uri")
  281. ctx.Session.Delete("openid_signin_remember")
  282. ctx.Session.Delete("openid_determined_email")
  283. ctx.Session.Delete("openid_determined_username")
  284. ctx.Session.Delete("twofaUid")
  285. ctx.Session.Delete("twofaRemember")
  286. ctx.Session.Set("uid", u.ID)
  287. ctx.Session.Set("uname", u.Name)
  288. // Clear whatever CSRF has right now, force to generate a new one
  289. ctx.SetCookie(setting.CSRFCookieName, "", -1, setting.AppSubURL)
  290. // Register last login
  291. u.SetLastLogin()
  292. if err := models.UpdateUserCols(u, "last_login_unix"); err != nil {
  293. ctx.ServerError("UpdateUserCols", err)
  294. return
  295. }
  296. if redirectTo, _ := url.QueryUnescape(ctx.GetCookie("redirect_to")); len(redirectTo) > 0 {
  297. ctx.SetCookie("redirect_to", "", -1, setting.AppSubURL)
  298. if obeyRedirect {
  299. ctx.RedirectToFirst(redirectTo)
  300. }
  301. return
  302. }
  303. if obeyRedirect {
  304. ctx.Redirect(setting.AppSubURL + "/")
  305. }
  306. }
  307. // SignInOAuth handles the OAuth2 login buttons
  308. func SignInOAuth(ctx *context.Context) {
  309. provider := ctx.Params(":provider")
  310. loginSource, err := models.GetActiveOAuth2LoginSourceByName(provider)
  311. if err != nil {
  312. ctx.ServerError("SignIn", err)
  313. return
  314. }
  315. // try to do a direct callback flow, so we don't authenticate the user again but use the valid accesstoken to get the user
  316. user, gothUser, err := oAuth2UserLoginCallback(loginSource, ctx.Req.Request, ctx.Resp)
  317. if err == nil && user != nil {
  318. // we got the user without going through the whole OAuth2 authentication flow again
  319. handleOAuth2SignIn(user, gothUser, ctx, err)
  320. return
  321. }
  322. err = oauth2.Auth(loginSource.Name, ctx.Req.Request, ctx.Resp)
  323. if err != nil {
  324. ctx.ServerError("SignIn", err)
  325. }
  326. // redirect is done in oauth2.Auth
  327. }
  328. // SignInOAuthCallback handles the callback from the given provider
  329. func SignInOAuthCallback(ctx *context.Context) {
  330. provider := ctx.Params(":provider")
  331. // first look if the provider is still active
  332. loginSource, err := models.GetActiveOAuth2LoginSourceByName(provider)
  333. if err != nil {
  334. ctx.ServerError("SignIn", err)
  335. return
  336. }
  337. if loginSource == nil {
  338. ctx.ServerError("SignIn", errors.New("No valid provider found, check configured callback url in provider"))
  339. return
  340. }
  341. u, gothUser, err := oAuth2UserLoginCallback(loginSource, ctx.Req.Request, ctx.Resp)
  342. handleOAuth2SignIn(u, gothUser, ctx, err)
  343. }
  344. func handleOAuth2SignIn(u *models.User, gothUser goth.User, ctx *context.Context, err error) {
  345. if err != nil {
  346. ctx.ServerError("UserSignIn", err)
  347. return
  348. }
  349. if u == nil {
  350. // no existing user is found, request attach or new account
  351. ctx.Session.Set("linkAccountGothUser", gothUser)
  352. ctx.Redirect(setting.AppSubURL + "/user/link_account")
  353. return
  354. }
  355. // If this user is enrolled in 2FA, we can't sign the user in just yet.
  356. // Instead, redirect them to the 2FA authentication page.
  357. _, err = models.GetTwoFactorByUID(u.ID)
  358. if err != nil {
  359. if models.IsErrTwoFactorNotEnrolled(err) {
  360. ctx.Session.Set("uid", u.ID)
  361. ctx.Session.Set("uname", u.Name)
  362. // Clear whatever CSRF has right now, force to generate a new one
  363. ctx.SetCookie(setting.CSRFCookieName, "", -1, setting.AppSubURL)
  364. // Register last login
  365. u.SetLastLogin()
  366. if err := models.UpdateUserCols(u, "last_login_unix"); err != nil {
  367. ctx.ServerError("UpdateUserCols", err)
  368. return
  369. }
  370. if redirectTo, _ := url.QueryUnescape(ctx.GetCookie("redirect_to")); len(redirectTo) > 0 {
  371. ctx.SetCookie("redirect_to", "", -1, setting.AppSubURL)
  372. ctx.RedirectToFirst(redirectTo)
  373. return
  374. }
  375. ctx.Redirect(setting.AppSubURL + "/")
  376. } else {
  377. ctx.ServerError("UserSignIn", err)
  378. }
  379. return
  380. }
  381. // User needs to use 2FA, save data and redirect to 2FA page.
  382. ctx.Session.Set("twofaUid", u.ID)
  383. ctx.Session.Set("twofaRemember", false)
  384. ctx.Redirect(setting.AppSubURL + "/user/two_factor")
  385. }
  386. // OAuth2UserLoginCallback attempts to handle the callback from the OAuth2 provider and if successful
  387. // login the user
  388. func oAuth2UserLoginCallback(loginSource *models.LoginSource, request *http.Request, response http.ResponseWriter) (*models.User, goth.User, error) {
  389. gothUser, err := oauth2.ProviderCallback(loginSource.Name, request, response)
  390. if err != nil {
  391. return nil, goth.User{}, err
  392. }
  393. user := &models.User{
  394. LoginName: gothUser.UserID,
  395. LoginType: models.LoginOAuth2,
  396. LoginSource: loginSource.ID,
  397. }
  398. hasUser, err := models.GetUser(user)
  399. if err != nil {
  400. return nil, goth.User{}, err
  401. }
  402. if hasUser {
  403. return user, goth.User{}, nil
  404. }
  405. // search in external linked users
  406. externalLoginUser := &models.ExternalLoginUser{
  407. ExternalID: gothUser.UserID,
  408. LoginSourceID: loginSource.ID,
  409. }
  410. hasUser, err = models.GetExternalLogin(externalLoginUser)
  411. if err != nil {
  412. return nil, goth.User{}, err
  413. }
  414. if hasUser {
  415. user, err = models.GetUserByID(externalLoginUser.UserID)
  416. return user, goth.User{}, err
  417. }
  418. // no user found to login
  419. return nil, gothUser, nil
  420. }
  421. // LinkAccount shows the page where the user can decide to login or create a new account
  422. func LinkAccount(ctx *context.Context) {
  423. ctx.Data["Title"] = ctx.Tr("link_account")
  424. ctx.Data["LinkAccountMode"] = true
  425. ctx.Data["EnableCaptcha"] = setting.Service.EnableCaptcha
  426. ctx.Data["DisableRegistration"] = setting.Service.DisableRegistration
  427. ctx.Data["ShowRegistrationButton"] = false
  428. // use this to set the right link into the signIn and signUp templates in the link_account template
  429. ctx.Data["SignInLink"] = setting.AppSubURL + "/user/link_account_signin"
  430. ctx.Data["SignUpLink"] = setting.AppSubURL + "/user/link_account_signup"
  431. gothUser := ctx.Session.Get("linkAccountGothUser")
  432. if gothUser == nil {
  433. ctx.ServerError("UserSignIn", errors.New("not in LinkAccount session"))
  434. return
  435. }
  436. ctx.Data["user_name"] = gothUser.(goth.User).NickName
  437. ctx.Data["email"] = gothUser.(goth.User).Email
  438. ctx.HTML(200, tplLinkAccount)
  439. }
  440. // LinkAccountPostSignIn handle the coupling of external account with another account using signIn
  441. func LinkAccountPostSignIn(ctx *context.Context, signInForm auth.SignInForm) {
  442. ctx.Data["Title"] = ctx.Tr("link_account")
  443. ctx.Data["LinkAccountMode"] = true
  444. ctx.Data["LinkAccountModeSignIn"] = true
  445. ctx.Data["EnableCaptcha"] = setting.Service.EnableCaptcha
  446. ctx.Data["DisableRegistration"] = setting.Service.DisableRegistration
  447. ctx.Data["ShowRegistrationButton"] = false
  448. // use this to set the right link into the signIn and signUp templates in the link_account template
  449. ctx.Data["SignInLink"] = setting.AppSubURL + "/user/link_account_signin"
  450. ctx.Data["SignUpLink"] = setting.AppSubURL + "/user/link_account_signup"
  451. gothUser := ctx.Session.Get("linkAccountGothUser")
  452. if gothUser == nil {
  453. ctx.ServerError("UserSignIn", errors.New("not in LinkAccount session"))
  454. return
  455. }
  456. if ctx.HasError() {
  457. ctx.HTML(200, tplLinkAccount)
  458. return
  459. }
  460. u, err := models.UserSignIn(signInForm.UserName, signInForm.Password)
  461. if err != nil {
  462. if models.IsErrUserNotExist(err) {
  463. ctx.RenderWithErr(ctx.Tr("form.username_password_incorrect"), tplLinkAccount, &signInForm)
  464. } else {
  465. ctx.ServerError("UserLinkAccount", err)
  466. }
  467. return
  468. }
  469. // If this user is enrolled in 2FA, we can't sign the user in just yet.
  470. // Instead, redirect them to the 2FA authentication page.
  471. _, err = models.GetTwoFactorByUID(u.ID)
  472. if err != nil {
  473. if models.IsErrTwoFactorNotEnrolled(err) {
  474. err = models.LinkAccountToUser(u, gothUser.(goth.User))
  475. if err != nil {
  476. ctx.ServerError("UserLinkAccount", err)
  477. } else {
  478. handleSignIn(ctx, u, signInForm.Remember)
  479. }
  480. } else {
  481. ctx.ServerError("UserLinkAccount", err)
  482. }
  483. return
  484. }
  485. // User needs to use 2FA, save data and redirect to 2FA page.
  486. ctx.Session.Set("twofaUid", u.ID)
  487. ctx.Session.Set("twofaRemember", signInForm.Remember)
  488. ctx.Session.Set("linkAccount", true)
  489. ctx.Redirect(setting.AppSubURL + "/user/two_factor")
  490. }
  491. // LinkAccountPostRegister handle the creation of a new account for an external account using signUp
  492. func LinkAccountPostRegister(ctx *context.Context, cpt *captcha.Captcha, form auth.RegisterForm) {
  493. ctx.Data["Title"] = ctx.Tr("link_account")
  494. ctx.Data["LinkAccountMode"] = true
  495. ctx.Data["LinkAccountModeRegister"] = true
  496. ctx.Data["EnableCaptcha"] = setting.Service.EnableCaptcha
  497. ctx.Data["DisableRegistration"] = setting.Service.DisableRegistration
  498. ctx.Data["ShowRegistrationButton"] = false
  499. // use this to set the right link into the signIn and signUp templates in the link_account template
  500. ctx.Data["SignInLink"] = setting.AppSubURL + "/user/link_account_signin"
  501. ctx.Data["SignUpLink"] = setting.AppSubURL + "/user/link_account_signup"
  502. gothUser := ctx.Session.Get("linkAccountGothUser")
  503. if gothUser == nil {
  504. ctx.ServerError("UserSignUp", errors.New("not in LinkAccount session"))
  505. return
  506. }
  507. if ctx.HasError() {
  508. ctx.HTML(200, tplLinkAccount)
  509. return
  510. }
  511. if setting.Service.DisableRegistration {
  512. ctx.Error(403)
  513. return
  514. }
  515. if setting.Service.EnableCaptcha && !cpt.VerifyReq(ctx.Req) {
  516. ctx.Data["Err_Captcha"] = true
  517. ctx.RenderWithErr(ctx.Tr("form.captcha_incorrect"), tplLinkAccount, &form)
  518. return
  519. }
  520. if (len(strings.TrimSpace(form.Password)) > 0 || len(strings.TrimSpace(form.Retype)) > 0) && form.Password != form.Retype {
  521. ctx.Data["Err_Password"] = true
  522. ctx.RenderWithErr(ctx.Tr("form.password_not_match"), tplLinkAccount, &form)
  523. return
  524. }
  525. if len(strings.TrimSpace(form.Password)) > 0 && len(form.Password) < setting.MinPasswordLength {
  526. ctx.Data["Err_Password"] = true
  527. ctx.RenderWithErr(ctx.Tr("auth.password_too_short", setting.MinPasswordLength), tplLinkAccount, &form)
  528. return
  529. }
  530. loginSource, err := models.GetActiveOAuth2LoginSourceByName(gothUser.(goth.User).Provider)
  531. if err != nil {
  532. ctx.ServerError("CreateUser", err)
  533. }
  534. u := &models.User{
  535. Name: form.UserName,
  536. Email: form.Email,
  537. Passwd: form.Password,
  538. IsActive: !setting.Service.RegisterEmailConfirm,
  539. LoginType: models.LoginOAuth2,
  540. LoginSource: loginSource.ID,
  541. LoginName: gothUser.(goth.User).UserID,
  542. }
  543. if err := models.CreateUser(u); err != nil {
  544. switch {
  545. case models.IsErrUserAlreadyExist(err):
  546. ctx.Data["Err_UserName"] = true
  547. ctx.RenderWithErr(ctx.Tr("form.username_been_taken"), tplLinkAccount, &form)
  548. case models.IsErrEmailAlreadyUsed(err):
  549. ctx.Data["Err_Email"] = true
  550. ctx.RenderWithErr(ctx.Tr("form.email_been_used"), tplLinkAccount, &form)
  551. case models.IsErrNameReserved(err):
  552. ctx.Data["Err_UserName"] = true
  553. ctx.RenderWithErr(ctx.Tr("user.form.name_reserved", err.(models.ErrNameReserved).Name), tplLinkAccount, &form)
  554. case models.IsErrNamePatternNotAllowed(err):
  555. ctx.Data["Err_UserName"] = true
  556. ctx.RenderWithErr(ctx.Tr("user.form.name_pattern_not_allowed", err.(models.ErrNamePatternNotAllowed).Pattern), tplLinkAccount, &form)
  557. default:
  558. ctx.ServerError("CreateUser", err)
  559. }
  560. return
  561. }
  562. log.Trace("Account created: %s", u.Name)
  563. // Auto-set admin for the only user.
  564. if models.CountUsers() == 1 {
  565. u.IsAdmin = true
  566. u.IsActive = true
  567. u.SetLastLogin()
  568. if err := models.UpdateUserCols(u, "is_admin", "is_active", "last_login_unix"); err != nil {
  569. ctx.ServerError("UpdateUser", err)
  570. return
  571. }
  572. }
  573. // Send confirmation email
  574. if setting.Service.RegisterEmailConfirm && u.ID > 1 {
  575. models.SendActivateAccountMail(ctx.Context, u)
  576. ctx.Data["IsSendRegisterMail"] = true
  577. ctx.Data["Email"] = u.Email
  578. ctx.Data["ActiveCodeLives"] = base.MinutesToFriendly(setting.Service.ActiveCodeLives, ctx.Locale.Language())
  579. ctx.HTML(200, TplActivate)
  580. if err := ctx.Cache.Put("MailResendLimit_"+u.LowerName, u.LowerName, 180); err != nil {
  581. log.Error(4, "Set cache(MailResendLimit) fail: %v", err)
  582. }
  583. return
  584. }
  585. ctx.Redirect(setting.AppSubURL + "/user/login")
  586. }
  587. // SignOut sign out from login status
  588. func SignOut(ctx *context.Context) {
  589. ctx.Session.Delete("uid")
  590. ctx.Session.Delete("uname")
  591. ctx.Session.Delete("socialId")
  592. ctx.Session.Delete("socialName")
  593. ctx.Session.Delete("socialEmail")
  594. ctx.SetCookie(setting.CookieUserName, "", -1, setting.AppSubURL)
  595. ctx.SetCookie(setting.CookieRememberName, "", -1, setting.AppSubURL)
  596. ctx.SetCookie(setting.CSRFCookieName, "", -1, setting.AppSubURL)
  597. ctx.Redirect(setting.AppSubURL + "/")
  598. }
  599. // SignUp render the register page
  600. func SignUp(ctx *context.Context) {
  601. ctx.Data["Title"] = ctx.Tr("sign_up")
  602. ctx.Data["SignUpLink"] = setting.AppSubURL + "/user/sign_up"
  603. ctx.Data["EnableCaptcha"] = setting.Service.EnableCaptcha
  604. ctx.Data["DisableRegistration"] = setting.Service.DisableRegistration
  605. ctx.HTML(200, tplSignUp)
  606. }
  607. // SignUpPost response for sign up information submission
  608. func SignUpPost(ctx *context.Context, cpt *captcha.Captcha, form auth.RegisterForm) {
  609. ctx.Data["Title"] = ctx.Tr("sign_up")
  610. ctx.Data["SignUpLink"] = setting.AppSubURL + "/user/sign_up"
  611. ctx.Data["EnableCaptcha"] = setting.Service.EnableCaptcha
  612. if setting.Service.DisableRegistration {
  613. ctx.Error(403)
  614. return
  615. }
  616. if ctx.HasError() {
  617. ctx.HTML(200, tplSignUp)
  618. return
  619. }
  620. if setting.Service.EnableCaptcha && !cpt.VerifyReq(ctx.Req) {
  621. ctx.Data["Err_Captcha"] = true
  622. ctx.RenderWithErr(ctx.Tr("form.captcha_incorrect"), tplSignUp, &form)
  623. return
  624. }
  625. if form.Password != form.Retype {
  626. ctx.Data["Err_Password"] = true
  627. ctx.RenderWithErr(ctx.Tr("form.password_not_match"), tplSignUp, &form)
  628. return
  629. }
  630. if len(form.Password) < setting.MinPasswordLength {
  631. ctx.Data["Err_Password"] = true
  632. ctx.RenderWithErr(ctx.Tr("auth.password_too_short", setting.MinPasswordLength), tplSignUp, &form)
  633. return
  634. }
  635. u := &models.User{
  636. Name: form.UserName,
  637. Email: form.Email,
  638. Passwd: form.Password,
  639. IsActive: !setting.Service.RegisterEmailConfirm,
  640. }
  641. if err := models.CreateUser(u); err != nil {
  642. switch {
  643. case models.IsErrUserAlreadyExist(err):
  644. ctx.Data["Err_UserName"] = true
  645. ctx.RenderWithErr(ctx.Tr("form.username_been_taken"), tplSignUp, &form)
  646. case models.IsErrEmailAlreadyUsed(err):
  647. ctx.Data["Err_Email"] = true
  648. ctx.RenderWithErr(ctx.Tr("form.email_been_used"), tplSignUp, &form)
  649. case models.IsErrNameReserved(err):
  650. ctx.Data["Err_UserName"] = true
  651. ctx.RenderWithErr(ctx.Tr("user.form.name_reserved", err.(models.ErrNameReserved).Name), tplSignUp, &form)
  652. case models.IsErrNamePatternNotAllowed(err):
  653. ctx.Data["Err_UserName"] = true
  654. ctx.RenderWithErr(ctx.Tr("user.form.name_pattern_not_allowed", err.(models.ErrNamePatternNotAllowed).Pattern), tplSignUp, &form)
  655. default:
  656. ctx.ServerError("CreateUser", err)
  657. }
  658. return
  659. }
  660. log.Trace("Account created: %s", u.Name)
  661. // Auto-set admin for the only user.
  662. if models.CountUsers() == 1 {
  663. u.IsAdmin = true
  664. u.IsActive = true
  665. u.SetLastLogin()
  666. if err := models.UpdateUserCols(u, "is_admin", "is_active", "last_login_unix"); err != nil {
  667. ctx.ServerError("UpdateUser", err)
  668. return
  669. }
  670. }
  671. // Send confirmation email, no need for social account.
  672. if setting.Service.RegisterEmailConfirm && u.ID > 1 {
  673. models.SendActivateAccountMail(ctx.Context, u)
  674. ctx.Data["IsSendRegisterMail"] = true
  675. ctx.Data["Email"] = u.Email
  676. ctx.Data["ActiveCodeLives"] = base.MinutesToFriendly(setting.Service.ActiveCodeLives, ctx.Locale.Language())
  677. ctx.HTML(200, TplActivate)
  678. if err := ctx.Cache.Put("MailResendLimit_"+u.LowerName, u.LowerName, 180); err != nil {
  679. log.Error(4, "Set cache(MailResendLimit) fail: %v", err)
  680. }
  681. return
  682. }
  683. ctx.Redirect(setting.AppSubURL + "/user/login")
  684. }
  685. // Activate render activate user page
  686. func Activate(ctx *context.Context) {
  687. code := ctx.Query("code")
  688. if len(code) == 0 {
  689. ctx.Data["IsActivatePage"] = true
  690. if ctx.User.IsActive {
  691. ctx.Error(404)
  692. return
  693. }
  694. // Resend confirmation email.
  695. if setting.Service.RegisterEmailConfirm {
  696. if ctx.Cache.IsExist("MailResendLimit_" + ctx.User.LowerName) {
  697. ctx.Data["ResendLimited"] = true
  698. } else {
  699. ctx.Data["ActiveCodeLives"] = base.MinutesToFriendly(setting.Service.ActiveCodeLives, ctx.Locale.Language())
  700. models.SendActivateAccountMail(ctx.Context, ctx.User)
  701. if err := ctx.Cache.Put("MailResendLimit_"+ctx.User.LowerName, ctx.User.LowerName, 180); err != nil {
  702. log.Error(4, "Set cache(MailResendLimit) fail: %v", err)
  703. }
  704. }
  705. } else {
  706. ctx.Data["ServiceNotEnabled"] = true
  707. }
  708. ctx.HTML(200, TplActivate)
  709. return
  710. }
  711. // Verify code.
  712. if user := models.VerifyUserActiveCode(code); user != nil {
  713. user.IsActive = true
  714. var err error
  715. if user.Rands, err = models.GetUserSalt(); err != nil {
  716. ctx.ServerError("UpdateUser", err)
  717. return
  718. }
  719. if err := models.UpdateUserCols(user, "is_active", "rands"); err != nil {
  720. if models.IsErrUserNotExist(err) {
  721. ctx.Error(404)
  722. } else {
  723. ctx.ServerError("UpdateUser", err)
  724. }
  725. return
  726. }
  727. log.Trace("User activated: %s", user.Name)
  728. ctx.Session.Set("uid", user.ID)
  729. ctx.Session.Set("uname", user.Name)
  730. ctx.Redirect(setting.AppSubURL + "/")
  731. return
  732. }
  733. ctx.Data["IsActivateFailed"] = true
  734. ctx.HTML(200, TplActivate)
  735. }
  736. // ActivateEmail render the activate email page
  737. func ActivateEmail(ctx *context.Context) {
  738. code := ctx.Query("code")
  739. emailStr := ctx.Query("email")
  740. // Verify code.
  741. if email := models.VerifyActiveEmailCode(code, emailStr); email != nil {
  742. if err := email.Activate(); err != nil {
  743. ctx.ServerError("ActivateEmail", err)
  744. }
  745. log.Trace("Email activated: %s", email.Email)
  746. ctx.Flash.Success(ctx.Tr("settings.add_email_success"))
  747. }
  748. ctx.Redirect(setting.AppSubURL + "/user/settings/email")
  749. return
  750. }
  751. // ForgotPasswd render the forget pasword page
  752. func ForgotPasswd(ctx *context.Context) {
  753. ctx.Data["Title"] = ctx.Tr("auth.forgot_password_title")
  754. if setting.MailService == nil {
  755. ctx.Data["IsResetDisable"] = true
  756. ctx.HTML(200, tplForgotPassword)
  757. return
  758. }
  759. email := ctx.Query("email")
  760. ctx.Data["Email"] = email
  761. ctx.Data["IsResetRequest"] = true
  762. ctx.HTML(200, tplForgotPassword)
  763. }
  764. // ForgotPasswdPost response for forget password request
  765. func ForgotPasswdPost(ctx *context.Context) {
  766. ctx.Data["Title"] = ctx.Tr("auth.forgot_password_title")
  767. if setting.MailService == nil {
  768. ctx.NotFound("ForgotPasswdPost", nil)
  769. return
  770. }
  771. ctx.Data["IsResetRequest"] = true
  772. email := ctx.Query("email")
  773. ctx.Data["Email"] = email
  774. u, err := models.GetUserByEmail(email)
  775. if err != nil {
  776. if models.IsErrUserNotExist(err) {
  777. ctx.Data["ResetPwdCodeLives"] = base.MinutesToFriendly(setting.Service.ResetPwdCodeLives, ctx.Locale.Language())
  778. ctx.Data["IsResetSent"] = true
  779. ctx.HTML(200, tplForgotPassword)
  780. return
  781. }
  782. ctx.ServerError("user.ResetPasswd(check existence)", err)
  783. return
  784. }
  785. if !u.IsLocal() && !u.IsOAuth2() {
  786. ctx.Data["Err_Email"] = true
  787. ctx.RenderWithErr(ctx.Tr("auth.non_local_account"), tplForgotPassword, nil)
  788. return
  789. }
  790. if ctx.Cache.IsExist("MailResendLimit_" + u.LowerName) {
  791. ctx.Data["ResendLimited"] = true
  792. ctx.HTML(200, tplForgotPassword)
  793. return
  794. }
  795. models.SendResetPasswordMail(ctx.Context, u)
  796. if err = ctx.Cache.Put("MailResendLimit_"+u.LowerName, u.LowerName, 180); err != nil {
  797. log.Error(4, "Set cache(MailResendLimit) fail: %v", err)
  798. }
  799. ctx.Data["ResetPwdCodeLives"] = base.MinutesToFriendly(setting.Service.ResetPwdCodeLives, ctx.Locale.Language())
  800. ctx.Data["IsResetSent"] = true
  801. ctx.HTML(200, tplForgotPassword)
  802. }
  803. // ResetPasswd render the reset password page
  804. func ResetPasswd(ctx *context.Context) {
  805. ctx.Data["Title"] = ctx.Tr("auth.reset_password")
  806. code := ctx.Query("code")
  807. if len(code) == 0 {
  808. ctx.Error(404)
  809. return
  810. }
  811. ctx.Data["Code"] = code
  812. ctx.Data["IsResetForm"] = true
  813. ctx.HTML(200, tplResetPassword)
  814. }
  815. // ResetPasswdPost response from reset password request
  816. func ResetPasswdPost(ctx *context.Context) {
  817. ctx.Data["Title"] = ctx.Tr("auth.reset_password")
  818. code := ctx.Query("code")
  819. if len(code) == 0 {
  820. ctx.Error(404)
  821. return
  822. }
  823. ctx.Data["Code"] = code
  824. if u := models.VerifyUserActiveCode(code); u != nil {
  825. // Validate password length.
  826. passwd := ctx.Query("password")
  827. if len(passwd) < setting.MinPasswordLength {
  828. ctx.Data["IsResetForm"] = true
  829. ctx.Data["Err_Password"] = true
  830. ctx.RenderWithErr(ctx.Tr("auth.password_too_short", setting.MinPasswordLength), tplResetPassword, nil)
  831. return
  832. }
  833. var err error
  834. if u.Rands, err = models.GetUserSalt(); err != nil {
  835. ctx.ServerError("UpdateUser", err)
  836. return
  837. }
  838. if u.Salt, err = models.GetUserSalt(); err != nil {
  839. ctx.ServerError("UpdateUser", err)
  840. return
  841. }
  842. u.HashPassword(passwd)
  843. if err := models.UpdateUserCols(u, "passwd", "rands", "salt"); err != nil {
  844. ctx.ServerError("UpdateUser", err)
  845. return
  846. }
  847. log.Trace("User password reset: %s", u.Name)
  848. ctx.Redirect(setting.AppSubURL + "/user/login")
  849. return
  850. }
  851. ctx.Data["IsResetFailed"] = true
  852. ctx.HTML(200, tplResetPassword)
  853. }