選択できるのは25トピックまでです。 トピックは、先頭が英数字で、英数字とダッシュ('-')を使用した35文字以内のものにしてください。

auth.go 24KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802
  1. // Copyright 2014 The Gogs Authors. All rights reserved.
  2. // Copyright 2018 The Gitea Authors. All rights reserved.
  3. // Use of this source code is governed by a MIT-style
  4. // license that can be found in the LICENSE file.
  5. package auth
  6. import (
  7. "fmt"
  8. "net/http"
  9. "strings"
  10. "code.gitea.io/gitea/models/auth"
  11. "code.gitea.io/gitea/models/db"
  12. user_model "code.gitea.io/gitea/models/user"
  13. "code.gitea.io/gitea/modules/base"
  14. "code.gitea.io/gitea/modules/context"
  15. "code.gitea.io/gitea/modules/eventsource"
  16. "code.gitea.io/gitea/modules/hcaptcha"
  17. "code.gitea.io/gitea/modules/log"
  18. "code.gitea.io/gitea/modules/password"
  19. "code.gitea.io/gitea/modules/recaptcha"
  20. "code.gitea.io/gitea/modules/session"
  21. "code.gitea.io/gitea/modules/setting"
  22. "code.gitea.io/gitea/modules/timeutil"
  23. "code.gitea.io/gitea/modules/web"
  24. "code.gitea.io/gitea/modules/web/middleware"
  25. "code.gitea.io/gitea/routers/utils"
  26. auth_service "code.gitea.io/gitea/services/auth"
  27. "code.gitea.io/gitea/services/auth/source/oauth2"
  28. "code.gitea.io/gitea/services/externalaccount"
  29. "code.gitea.io/gitea/services/forms"
  30. "code.gitea.io/gitea/services/mailer"
  31. "github.com/markbates/goth"
  32. )
  33. const (
  34. // tplSignIn template for sign in page
  35. tplSignIn base.TplName = "user/auth/signin"
  36. // tplSignUp template path for sign up page
  37. tplSignUp base.TplName = "user/auth/signup"
  38. // TplActivate template path for activate user
  39. TplActivate base.TplName = "user/auth/activate"
  40. )
  41. // AutoSignIn reads cookie and try to auto-login.
  42. func AutoSignIn(ctx *context.Context) (bool, error) {
  43. if !db.HasEngine {
  44. return false, nil
  45. }
  46. uname := ctx.GetCookie(setting.CookieUserName)
  47. if len(uname) == 0 {
  48. return false, nil
  49. }
  50. isSucceed := false
  51. defer func() {
  52. if !isSucceed {
  53. log.Trace("auto-login cookie cleared: %s", uname)
  54. ctx.DeleteCookie(setting.CookieUserName)
  55. ctx.DeleteCookie(setting.CookieRememberName)
  56. }
  57. }()
  58. u, err := user_model.GetUserByName(uname)
  59. if err != nil {
  60. if !user_model.IsErrUserNotExist(err) {
  61. return false, fmt.Errorf("GetUserByName: %v", err)
  62. }
  63. return false, nil
  64. }
  65. if val, ok := ctx.GetSuperSecureCookie(
  66. base.EncodeMD5(u.Rands+u.Passwd), setting.CookieRememberName); !ok || val != u.Name {
  67. return false, nil
  68. }
  69. isSucceed = true
  70. if _, err := session.RegenerateSession(ctx.Resp, ctx.Req); err != nil {
  71. return false, fmt.Errorf("unable to RegenerateSession: Error: %w", err)
  72. }
  73. // Set session IDs
  74. if err := ctx.Session.Set("uid", u.ID); err != nil {
  75. return false, err
  76. }
  77. if err := ctx.Session.Set("uname", u.Name); err != nil {
  78. return false, err
  79. }
  80. if err := ctx.Session.Release(); err != nil {
  81. return false, err
  82. }
  83. if err := resetLocale(ctx, u); err != nil {
  84. return false, err
  85. }
  86. middleware.DeleteCSRFCookie(ctx.Resp)
  87. return true, nil
  88. }
  89. func resetLocale(ctx *context.Context, u *user_model.User) error {
  90. // Language setting of the user overwrites the one previously set
  91. // If the user does not have a locale set, we save the current one.
  92. if len(u.Language) == 0 {
  93. u.Language = ctx.Locale.Language()
  94. if err := user_model.UpdateUserCols(ctx, u, "language"); err != nil {
  95. return err
  96. }
  97. }
  98. middleware.SetLocaleCookie(ctx.Resp, u.Language, 0)
  99. if ctx.Locale.Language() != u.Language {
  100. ctx.Locale = middleware.Locale(ctx.Resp, ctx.Req)
  101. }
  102. return nil
  103. }
  104. func checkAutoLogin(ctx *context.Context) bool {
  105. // Check auto-login
  106. isSucceed, err := AutoSignIn(ctx)
  107. if err != nil {
  108. ctx.ServerError("AutoSignIn", err)
  109. return true
  110. }
  111. redirectTo := ctx.FormString("redirect_to")
  112. if len(redirectTo) > 0 {
  113. middleware.SetRedirectToCookie(ctx.Resp, redirectTo)
  114. } else {
  115. redirectTo = ctx.GetCookie("redirect_to")
  116. }
  117. if isSucceed {
  118. middleware.DeleteRedirectToCookie(ctx.Resp)
  119. ctx.RedirectToFirst(redirectTo, setting.AppSubURL+string(setting.LandingPageURL))
  120. return true
  121. }
  122. return false
  123. }
  124. // SignIn render sign in page
  125. func SignIn(ctx *context.Context) {
  126. ctx.Data["Title"] = ctx.Tr("sign_in")
  127. // Check auto-login
  128. if checkAutoLogin(ctx) {
  129. return
  130. }
  131. orderedOAuth2Names, oauth2Providers, err := oauth2.GetActiveOAuth2Providers()
  132. if err != nil {
  133. ctx.ServerError("UserSignIn", err)
  134. return
  135. }
  136. ctx.Data["OrderedOAuth2Names"] = orderedOAuth2Names
  137. ctx.Data["OAuth2Providers"] = oauth2Providers
  138. ctx.Data["Title"] = ctx.Tr("sign_in")
  139. ctx.Data["SignInLink"] = setting.AppSubURL + "/user/login"
  140. ctx.Data["PageIsSignIn"] = true
  141. ctx.Data["PageIsLogin"] = true
  142. ctx.Data["EnableSSPI"] = auth.IsSSPIEnabled()
  143. ctx.HTML(http.StatusOK, tplSignIn)
  144. }
  145. // SignInPost response for sign in request
  146. func SignInPost(ctx *context.Context) {
  147. ctx.Data["Title"] = ctx.Tr("sign_in")
  148. orderedOAuth2Names, oauth2Providers, err := oauth2.GetActiveOAuth2Providers()
  149. if err != nil {
  150. ctx.ServerError("UserSignIn", err)
  151. return
  152. }
  153. ctx.Data["OrderedOAuth2Names"] = orderedOAuth2Names
  154. ctx.Data["OAuth2Providers"] = oauth2Providers
  155. ctx.Data["Title"] = ctx.Tr("sign_in")
  156. ctx.Data["SignInLink"] = setting.AppSubURL + "/user/login"
  157. ctx.Data["PageIsSignIn"] = true
  158. ctx.Data["PageIsLogin"] = true
  159. ctx.Data["EnableSSPI"] = auth.IsSSPIEnabled()
  160. if ctx.HasError() {
  161. ctx.HTML(http.StatusOK, tplSignIn)
  162. return
  163. }
  164. form := web.GetForm(ctx).(*forms.SignInForm)
  165. u, source, err := auth_service.UserSignIn(form.UserName, form.Password)
  166. if err != nil {
  167. if user_model.IsErrUserNotExist(err) || user_model.IsErrEmailAddressNotExist(err) {
  168. ctx.RenderWithErr(ctx.Tr("form.username_password_incorrect"), tplSignIn, &form)
  169. log.Info("Failed authentication attempt for %s from %s: %v", form.UserName, ctx.RemoteAddr(), err)
  170. } else if user_model.IsErrEmailAlreadyUsed(err) {
  171. ctx.RenderWithErr(ctx.Tr("form.email_been_used"), tplSignIn, &form)
  172. log.Info("Failed authentication attempt for %s from %s: %v", form.UserName, ctx.RemoteAddr(), err)
  173. } else if user_model.IsErrUserProhibitLogin(err) {
  174. log.Info("Failed authentication attempt for %s from %s: %v", form.UserName, ctx.RemoteAddr(), err)
  175. ctx.Data["Title"] = ctx.Tr("auth.prohibit_login")
  176. ctx.HTML(http.StatusOK, "user/auth/prohibit_login")
  177. } else if user_model.IsErrUserInactive(err) {
  178. if setting.Service.RegisterEmailConfirm {
  179. ctx.Data["Title"] = ctx.Tr("auth.active_your_account")
  180. ctx.HTML(http.StatusOK, TplActivate)
  181. } else {
  182. log.Info("Failed authentication attempt for %s from %s: %v", form.UserName, ctx.RemoteAddr(), err)
  183. ctx.Data["Title"] = ctx.Tr("auth.prohibit_login")
  184. ctx.HTML(http.StatusOK, "user/auth/prohibit_login")
  185. }
  186. } else {
  187. ctx.ServerError("UserSignIn", err)
  188. }
  189. return
  190. }
  191. // Now handle 2FA:
  192. // First of all if the source can skip local two fa we're done
  193. if skipper, ok := source.Cfg.(auth_service.LocalTwoFASkipper); ok && skipper.IsSkipLocalTwoFA() {
  194. handleSignIn(ctx, u, form.Remember)
  195. return
  196. }
  197. // If this user is enrolled in 2FA TOTP, we can't sign the user in just yet.
  198. // Instead, redirect them to the 2FA authentication page.
  199. hasTOTPtwofa, err := auth.HasTwoFactorByUID(u.ID)
  200. if err != nil {
  201. ctx.ServerError("UserSignIn", err)
  202. return
  203. }
  204. // Check if the user has webauthn registration
  205. hasWebAuthnTwofa, err := auth.HasWebAuthnRegistrationsByUID(u.ID)
  206. if err != nil {
  207. ctx.ServerError("UserSignIn", err)
  208. return
  209. }
  210. if !hasTOTPtwofa && !hasWebAuthnTwofa {
  211. // No two factor auth configured we can sign in the user
  212. handleSignIn(ctx, u, form.Remember)
  213. return
  214. }
  215. if _, err := session.RegenerateSession(ctx.Resp, ctx.Req); err != nil {
  216. ctx.ServerError("UserSignIn: Unable to set regenerate session", err)
  217. return
  218. }
  219. // User will need to use 2FA TOTP or WebAuthn, save data
  220. if err := ctx.Session.Set("twofaUid", u.ID); err != nil {
  221. ctx.ServerError("UserSignIn: Unable to set twofaUid in session", err)
  222. return
  223. }
  224. if err := ctx.Session.Set("twofaRemember", form.Remember); err != nil {
  225. ctx.ServerError("UserSignIn: Unable to set twofaRemember in session", err)
  226. return
  227. }
  228. if hasTOTPtwofa {
  229. // User will need to use U2F, save data
  230. if err := ctx.Session.Set("totpEnrolled", u.ID); err != nil {
  231. ctx.ServerError("UserSignIn: Unable to set WebAuthn Enrolled in session", err)
  232. return
  233. }
  234. }
  235. if err := ctx.Session.Release(); err != nil {
  236. ctx.ServerError("UserSignIn: Unable to save session", err)
  237. return
  238. }
  239. // If we have U2F redirect there first
  240. if hasWebAuthnTwofa {
  241. ctx.Redirect(setting.AppSubURL + "/user/webauthn")
  242. return
  243. }
  244. // Fallback to 2FA
  245. ctx.Redirect(setting.AppSubURL + "/user/two_factor")
  246. }
  247. // This handles the final part of the sign-in process of the user.
  248. func handleSignIn(ctx *context.Context, u *user_model.User, remember bool) {
  249. redirect := handleSignInFull(ctx, u, remember, true)
  250. if ctx.Written() {
  251. return
  252. }
  253. ctx.Redirect(redirect)
  254. }
  255. func handleSignInFull(ctx *context.Context, u *user_model.User, remember, obeyRedirect bool) string {
  256. if remember {
  257. days := 86400 * setting.LogInRememberDays
  258. ctx.SetCookie(setting.CookieUserName, u.Name, days)
  259. ctx.SetSuperSecureCookie(base.EncodeMD5(u.Rands+u.Passwd),
  260. setting.CookieRememberName, u.Name, days)
  261. }
  262. if _, err := session.RegenerateSession(ctx.Resp, ctx.Req); err != nil {
  263. ctx.ServerError("RegenerateSession", err)
  264. return setting.AppSubURL + "/"
  265. }
  266. // Delete the openid, 2fa and linkaccount data
  267. _ = ctx.Session.Delete("openid_verified_uri")
  268. _ = ctx.Session.Delete("openid_signin_remember")
  269. _ = ctx.Session.Delete("openid_determined_email")
  270. _ = ctx.Session.Delete("openid_determined_username")
  271. _ = ctx.Session.Delete("twofaUid")
  272. _ = ctx.Session.Delete("twofaRemember")
  273. _ = ctx.Session.Delete("u2fChallenge")
  274. _ = ctx.Session.Delete("linkAccount")
  275. if err := ctx.Session.Set("uid", u.ID); err != nil {
  276. log.Error("Error setting uid %d in session: %v", u.ID, err)
  277. }
  278. if err := ctx.Session.Set("uname", u.Name); err != nil {
  279. log.Error("Error setting uname %s session: %v", u.Name, err)
  280. }
  281. if err := ctx.Session.Release(); err != nil {
  282. log.Error("Unable to store session: %v", err)
  283. }
  284. // Language setting of the user overwrites the one previously set
  285. // If the user does not have a locale set, we save the current one.
  286. if len(u.Language) == 0 {
  287. u.Language = ctx.Locale.Language()
  288. if err := user_model.UpdateUserCols(ctx, u, "language"); err != nil {
  289. ctx.ServerError("UpdateUserCols Language", fmt.Errorf("Error updating user language [user: %d, locale: %s]", u.ID, u.Language))
  290. return setting.AppSubURL + "/"
  291. }
  292. }
  293. middleware.SetLocaleCookie(ctx.Resp, u.Language, 0)
  294. if ctx.Locale.Language() != u.Language {
  295. ctx.Locale = middleware.Locale(ctx.Resp, ctx.Req)
  296. }
  297. // Clear whatever CSRF cookie has right now, force to generate a new one
  298. middleware.DeleteCSRFCookie(ctx.Resp)
  299. // Register last login
  300. u.SetLastLogin()
  301. if err := user_model.UpdateUserCols(ctx, u, "last_login_unix"); err != nil {
  302. ctx.ServerError("UpdateUserCols", err)
  303. return setting.AppSubURL + "/"
  304. }
  305. if redirectTo := ctx.GetCookie("redirect_to"); len(redirectTo) > 0 && !utils.IsExternalURL(redirectTo) {
  306. middleware.DeleteRedirectToCookie(ctx.Resp)
  307. if obeyRedirect {
  308. ctx.RedirectToFirst(redirectTo)
  309. }
  310. return redirectTo
  311. }
  312. if obeyRedirect {
  313. ctx.Redirect(setting.AppSubURL + "/")
  314. }
  315. return setting.AppSubURL + "/"
  316. }
  317. func getUserName(gothUser *goth.User) string {
  318. switch setting.OAuth2Client.Username {
  319. case setting.OAuth2UsernameEmail:
  320. return strings.Split(gothUser.Email, "@")[0]
  321. case setting.OAuth2UsernameNickname:
  322. return gothUser.NickName
  323. default: // OAuth2UsernameUserid
  324. return gothUser.UserID
  325. }
  326. }
  327. // HandleSignOut resets the session and sets the cookies
  328. func HandleSignOut(ctx *context.Context) {
  329. _ = ctx.Session.Flush()
  330. _ = ctx.Session.Destroy(ctx.Resp, ctx.Req)
  331. ctx.DeleteCookie(setting.CookieUserName)
  332. ctx.DeleteCookie(setting.CookieRememberName)
  333. middleware.DeleteCSRFCookie(ctx.Resp)
  334. middleware.DeleteLocaleCookie(ctx.Resp)
  335. middleware.DeleteRedirectToCookie(ctx.Resp)
  336. }
  337. // SignOut sign out from login status
  338. func SignOut(ctx *context.Context) {
  339. if ctx.Doer != nil {
  340. eventsource.GetManager().SendMessageBlocking(ctx.Doer.ID, &eventsource.Event{
  341. Name: "logout",
  342. Data: ctx.Session.ID(),
  343. })
  344. }
  345. HandleSignOut(ctx)
  346. ctx.Redirect(setting.AppSubURL + "/")
  347. }
  348. // SignUp render the register page
  349. func SignUp(ctx *context.Context) {
  350. ctx.Data["Title"] = ctx.Tr("sign_up")
  351. ctx.Data["SignUpLink"] = setting.AppSubURL + "/user/sign_up"
  352. ctx.Data["EnableCaptcha"] = setting.Service.EnableCaptcha
  353. ctx.Data["RecaptchaURL"] = setting.Service.RecaptchaURL
  354. ctx.Data["Captcha"] = context.GetImageCaptcha()
  355. ctx.Data["CaptchaType"] = setting.Service.CaptchaType
  356. ctx.Data["RecaptchaSitekey"] = setting.Service.RecaptchaSitekey
  357. ctx.Data["HcaptchaSitekey"] = setting.Service.HcaptchaSitekey
  358. ctx.Data["PageIsSignUp"] = true
  359. // Show Disabled Registration message if DisableRegistration or AllowOnlyExternalRegistration options are true
  360. ctx.Data["DisableRegistration"] = setting.Service.DisableRegistration || setting.Service.AllowOnlyExternalRegistration
  361. ctx.HTML(http.StatusOK, tplSignUp)
  362. }
  363. // SignUpPost response for sign up information submission
  364. func SignUpPost(ctx *context.Context) {
  365. form := web.GetForm(ctx).(*forms.RegisterForm)
  366. ctx.Data["Title"] = ctx.Tr("sign_up")
  367. ctx.Data["SignUpLink"] = setting.AppSubURL + "/user/sign_up"
  368. ctx.Data["EnableCaptcha"] = setting.Service.EnableCaptcha
  369. ctx.Data["RecaptchaURL"] = setting.Service.RecaptchaURL
  370. ctx.Data["Captcha"] = context.GetImageCaptcha()
  371. ctx.Data["CaptchaType"] = setting.Service.CaptchaType
  372. ctx.Data["RecaptchaSitekey"] = setting.Service.RecaptchaSitekey
  373. ctx.Data["HcaptchaSitekey"] = setting.Service.HcaptchaSitekey
  374. ctx.Data["PageIsSignUp"] = true
  375. // Permission denied if DisableRegistration or AllowOnlyExternalRegistration options are true
  376. if setting.Service.DisableRegistration || setting.Service.AllowOnlyExternalRegistration {
  377. ctx.Error(http.StatusForbidden)
  378. return
  379. }
  380. if ctx.HasError() {
  381. ctx.HTML(http.StatusOK, tplSignUp)
  382. return
  383. }
  384. if setting.Service.EnableCaptcha {
  385. var valid bool
  386. var err error
  387. switch setting.Service.CaptchaType {
  388. case setting.ImageCaptcha:
  389. valid = context.GetImageCaptcha().VerifyReq(ctx.Req)
  390. case setting.ReCaptcha:
  391. valid, err = recaptcha.Verify(ctx, form.GRecaptchaResponse)
  392. case setting.HCaptcha:
  393. valid, err = hcaptcha.Verify(ctx, form.HcaptchaResponse)
  394. default:
  395. ctx.ServerError("Unknown Captcha Type", fmt.Errorf("Unknown Captcha Type: %s", setting.Service.CaptchaType))
  396. return
  397. }
  398. if err != nil {
  399. log.Debug("%s", err.Error())
  400. }
  401. if !valid {
  402. ctx.Data["Err_Captcha"] = true
  403. ctx.RenderWithErr(ctx.Tr("form.captcha_incorrect"), tplSignUp, &form)
  404. return
  405. }
  406. }
  407. if !form.IsEmailDomainAllowed() {
  408. ctx.RenderWithErr(ctx.Tr("auth.email_domain_blacklisted"), tplSignUp, &form)
  409. return
  410. }
  411. if form.Password != form.Retype {
  412. ctx.Data["Err_Password"] = true
  413. ctx.RenderWithErr(ctx.Tr("form.password_not_match"), tplSignUp, &form)
  414. return
  415. }
  416. if len(form.Password) < setting.MinPasswordLength {
  417. ctx.Data["Err_Password"] = true
  418. ctx.RenderWithErr(ctx.Tr("auth.password_too_short", setting.MinPasswordLength), tplSignUp, &form)
  419. return
  420. }
  421. if !password.IsComplexEnough(form.Password) {
  422. ctx.Data["Err_Password"] = true
  423. ctx.RenderWithErr(password.BuildComplexityError(ctx), tplSignUp, &form)
  424. return
  425. }
  426. pwned, err := password.IsPwned(ctx, form.Password)
  427. if pwned {
  428. errMsg := ctx.Tr("auth.password_pwned")
  429. if err != nil {
  430. log.Error(err.Error())
  431. errMsg = ctx.Tr("auth.password_pwned_err")
  432. }
  433. ctx.Data["Err_Password"] = true
  434. ctx.RenderWithErr(errMsg, tplSignUp, &form)
  435. return
  436. }
  437. u := &user_model.User{
  438. Name: form.UserName,
  439. Email: form.Email,
  440. Passwd: form.Password,
  441. }
  442. if !createAndHandleCreatedUser(ctx, tplSignUp, form, u, nil, nil, false) {
  443. // error already handled
  444. return
  445. }
  446. ctx.Flash.Success(ctx.Tr("auth.sign_up_successful"))
  447. handleSignIn(ctx, u, false)
  448. }
  449. // createAndHandleCreatedUser calls createUserInContext and
  450. // then handleUserCreated.
  451. func createAndHandleCreatedUser(ctx *context.Context, tpl base.TplName, form interface{}, u *user_model.User, overwrites *user_model.CreateUserOverwriteOptions, gothUser *goth.User, allowLink bool) bool {
  452. if !createUserInContext(ctx, tpl, form, u, overwrites, gothUser, allowLink) {
  453. return false
  454. }
  455. return handleUserCreated(ctx, u, gothUser)
  456. }
  457. // createUserInContext creates a user and handles errors within a given context.
  458. // Optionally a template can be specified.
  459. func createUserInContext(ctx *context.Context, tpl base.TplName, form interface{}, u *user_model.User, overwrites *user_model.CreateUserOverwriteOptions, gothUser *goth.User, allowLink bool) (ok bool) {
  460. if err := user_model.CreateUser(u, overwrites); err != nil {
  461. if allowLink && (user_model.IsErrUserAlreadyExist(err) || user_model.IsErrEmailAlreadyUsed(err)) {
  462. if setting.OAuth2Client.AccountLinking == setting.OAuth2AccountLinkingAuto {
  463. var user *user_model.User
  464. user = &user_model.User{Name: u.Name}
  465. hasUser, err := user_model.GetUser(user)
  466. if !hasUser || err != nil {
  467. user = &user_model.User{Email: u.Email}
  468. hasUser, err = user_model.GetUser(user)
  469. if !hasUser || err != nil {
  470. ctx.ServerError("UserLinkAccount", err)
  471. return
  472. }
  473. }
  474. // TODO: probably we should respect 'remember' user's choice...
  475. linkAccount(ctx, user, *gothUser, true)
  476. return // user is already created here, all redirects are handled
  477. } else if setting.OAuth2Client.AccountLinking == setting.OAuth2AccountLinkingLogin {
  478. showLinkingLogin(ctx, *gothUser)
  479. return // user will be created only after linking login
  480. }
  481. }
  482. // handle error without template
  483. if len(tpl) == 0 {
  484. ctx.ServerError("CreateUser", err)
  485. return
  486. }
  487. // handle error with template
  488. switch {
  489. case user_model.IsErrUserAlreadyExist(err):
  490. ctx.Data["Err_UserName"] = true
  491. ctx.RenderWithErr(ctx.Tr("form.username_been_taken"), tpl, form)
  492. case user_model.IsErrEmailAlreadyUsed(err):
  493. ctx.Data["Err_Email"] = true
  494. ctx.RenderWithErr(ctx.Tr("form.email_been_used"), tpl, form)
  495. case user_model.IsErrEmailCharIsNotSupported(err):
  496. ctx.Data["Err_Email"] = true
  497. ctx.RenderWithErr(ctx.Tr("form.email_invalid"), tpl, form)
  498. case user_model.IsErrEmailInvalid(err):
  499. ctx.Data["Err_Email"] = true
  500. ctx.RenderWithErr(ctx.Tr("form.email_invalid"), tpl, form)
  501. case db.IsErrNameReserved(err):
  502. ctx.Data["Err_UserName"] = true
  503. ctx.RenderWithErr(ctx.Tr("user.form.name_reserved", err.(db.ErrNameReserved).Name), tpl, form)
  504. case db.IsErrNamePatternNotAllowed(err):
  505. ctx.Data["Err_UserName"] = true
  506. ctx.RenderWithErr(ctx.Tr("user.form.name_pattern_not_allowed", err.(db.ErrNamePatternNotAllowed).Pattern), tpl, form)
  507. case db.IsErrNameCharsNotAllowed(err):
  508. ctx.Data["Err_UserName"] = true
  509. ctx.RenderWithErr(ctx.Tr("user.form.name_chars_not_allowed", err.(db.ErrNameCharsNotAllowed).Name), tpl, form)
  510. default:
  511. ctx.ServerError("CreateUser", err)
  512. }
  513. return
  514. }
  515. log.Trace("Account created: %s", u.Name)
  516. return true
  517. }
  518. // handleUserCreated does additional steps after a new user is created.
  519. // It auto-sets admin for the only user, updates the optional external user and
  520. // sends a confirmation email if required.
  521. func handleUserCreated(ctx *context.Context, u *user_model.User, gothUser *goth.User) (ok bool) {
  522. // Auto-set admin for the only user.
  523. if user_model.CountUsers(nil) == 1 {
  524. u.IsAdmin = true
  525. u.IsActive = true
  526. u.SetLastLogin()
  527. if err := user_model.UpdateUserCols(ctx, u, "is_admin", "is_active", "last_login_unix"); err != nil {
  528. ctx.ServerError("UpdateUser", err)
  529. return
  530. }
  531. }
  532. // update external user information
  533. if gothUser != nil {
  534. if err := externalaccount.UpdateExternalUser(u, *gothUser); err != nil {
  535. log.Error("UpdateExternalUser failed: %v", err)
  536. }
  537. }
  538. // Send confirmation email
  539. if !u.IsActive && u.ID > 1 {
  540. if setting.Service.RegisterManualConfirm {
  541. ctx.Data["ManualActivationOnly"] = true
  542. ctx.HTML(http.StatusOK, TplActivate)
  543. return
  544. }
  545. mailer.SendActivateAccountMail(ctx.Locale, u)
  546. ctx.Data["IsSendRegisterMail"] = true
  547. ctx.Data["Email"] = u.Email
  548. ctx.Data["ActiveCodeLives"] = timeutil.MinutesToFriendly(setting.Service.ActiveCodeLives, ctx.Locale.Language())
  549. ctx.HTML(http.StatusOK, TplActivate)
  550. if err := ctx.Cache.Put("MailResendLimit_"+u.LowerName, u.LowerName, 180); err != nil {
  551. log.Error("Set cache(MailResendLimit) fail: %v", err)
  552. }
  553. return
  554. }
  555. return true
  556. }
  557. // Activate render activate user page
  558. func Activate(ctx *context.Context) {
  559. code := ctx.FormString("code")
  560. if len(code) == 0 {
  561. ctx.Data["IsActivatePage"] = true
  562. if ctx.Doer == nil || ctx.Doer.IsActive {
  563. ctx.NotFound("invalid user", nil)
  564. return
  565. }
  566. // Resend confirmation email.
  567. if setting.Service.RegisterEmailConfirm {
  568. if ctx.Cache.IsExist("MailResendLimit_" + ctx.Doer.LowerName) {
  569. ctx.Data["ResendLimited"] = true
  570. } else {
  571. ctx.Data["ActiveCodeLives"] = timeutil.MinutesToFriendly(setting.Service.ActiveCodeLives, ctx.Locale.Language())
  572. mailer.SendActivateAccountMail(ctx.Locale, ctx.Doer)
  573. if err := ctx.Cache.Put("MailResendLimit_"+ctx.Doer.LowerName, ctx.Doer.LowerName, 180); err != nil {
  574. log.Error("Set cache(MailResendLimit) fail: %v", err)
  575. }
  576. }
  577. } else {
  578. ctx.Data["ServiceNotEnabled"] = true
  579. }
  580. ctx.HTML(http.StatusOK, TplActivate)
  581. return
  582. }
  583. user := user_model.VerifyUserActiveCode(code)
  584. // if code is wrong
  585. if user == nil {
  586. ctx.Data["IsActivateFailed"] = true
  587. ctx.HTML(http.StatusOK, TplActivate)
  588. return
  589. }
  590. // if account is local account, verify password
  591. if user.LoginSource == 0 {
  592. ctx.Data["Code"] = code
  593. ctx.Data["NeedsPassword"] = true
  594. ctx.HTML(http.StatusOK, TplActivate)
  595. return
  596. }
  597. handleAccountActivation(ctx, user)
  598. }
  599. // ActivatePost handles account activation with password check
  600. func ActivatePost(ctx *context.Context) {
  601. code := ctx.FormString("code")
  602. if len(code) == 0 {
  603. ctx.Redirect(setting.AppSubURL + "/user/activate")
  604. return
  605. }
  606. user := user_model.VerifyUserActiveCode(code)
  607. // if code is wrong
  608. if user == nil {
  609. ctx.Data["IsActivateFailed"] = true
  610. ctx.HTML(http.StatusOK, TplActivate)
  611. return
  612. }
  613. // if account is local account, verify password
  614. if user.LoginSource == 0 {
  615. password := ctx.FormString("password")
  616. if len(password) == 0 {
  617. ctx.Data["Code"] = code
  618. ctx.Data["NeedsPassword"] = true
  619. ctx.HTML(http.StatusOK, TplActivate)
  620. return
  621. }
  622. if !user.ValidatePassword(password) {
  623. ctx.Data["IsActivateFailed"] = true
  624. ctx.HTML(http.StatusOK, TplActivate)
  625. return
  626. }
  627. }
  628. handleAccountActivation(ctx, user)
  629. }
  630. func handleAccountActivation(ctx *context.Context, user *user_model.User) {
  631. user.IsActive = true
  632. var err error
  633. if user.Rands, err = user_model.GetUserSalt(); err != nil {
  634. ctx.ServerError("UpdateUser", err)
  635. return
  636. }
  637. if err := user_model.UpdateUserCols(ctx, user, "is_active", "rands"); err != nil {
  638. if user_model.IsErrUserNotExist(err) {
  639. ctx.NotFound("UpdateUserCols", err)
  640. } else {
  641. ctx.ServerError("UpdateUser", err)
  642. }
  643. return
  644. }
  645. if err := user_model.ActivateUserEmail(user.ID, user.Email, true); err != nil {
  646. log.Error("Unable to activate email for user: %-v with email: %s: %v", user, user.Email, err)
  647. ctx.ServerError("ActivateUserEmail", err)
  648. return
  649. }
  650. log.Trace("User activated: %s", user.Name)
  651. if _, err := session.RegenerateSession(ctx.Resp, ctx.Req); err != nil {
  652. log.Error("Unable to regenerate session for user: %-v with email: %s: %v", user, user.Email, err)
  653. ctx.ServerError("ActivateUserEmail", err)
  654. return
  655. }
  656. if err := ctx.Session.Set("uid", user.ID); err != nil {
  657. log.Error("Error setting uid in session[%s]: %v", ctx.Session.ID(), err)
  658. }
  659. if err := ctx.Session.Set("uname", user.Name); err != nil {
  660. log.Error("Error setting uname in session[%s]: %v", ctx.Session.ID(), err)
  661. }
  662. if err := ctx.Session.Release(); err != nil {
  663. log.Error("Error storing session[%s]: %v", ctx.Session.ID(), err)
  664. }
  665. if err := resetLocale(ctx, user); err != nil {
  666. ctx.ServerError("resetLocale", err)
  667. return
  668. }
  669. ctx.Flash.Success(ctx.Tr("auth.account_activated"))
  670. ctx.Redirect(setting.AppSubURL + "/")
  671. }
  672. // ActivateEmail render the activate email page
  673. func ActivateEmail(ctx *context.Context) {
  674. code := ctx.FormString("code")
  675. emailStr := ctx.FormString("email")
  676. // Verify code.
  677. if email := user_model.VerifyActiveEmailCode(code, emailStr); email != nil {
  678. if err := user_model.ActivateEmail(email); err != nil {
  679. ctx.ServerError("ActivateEmail", err)
  680. }
  681. log.Trace("Email activated: %s", email.Email)
  682. ctx.Flash.Success(ctx.Tr("settings.add_email_success"))
  683. if u, err := user_model.GetUserByID(email.UID); err != nil {
  684. log.Warn("GetUserByID: %d", email.UID)
  685. } else {
  686. // Allow user to validate more emails
  687. _ = ctx.Cache.Delete("MailResendLimit_" + u.LowerName)
  688. }
  689. }
  690. // FIXME: e-mail verification does not require the user to be logged in,
  691. // so this could be redirecting to the login page.
  692. // Should users be logged in automatically here? (consider 2FA requirements, etc.)
  693. ctx.Redirect(setting.AppSubURL + "/user/settings/account")
  694. }