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 43KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411
  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 user
  6. import (
  7. "errors"
  8. "fmt"
  9. "net/http"
  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/recaptcha"
  18. "code.gitea.io/gitea/modules/setting"
  19. "code.gitea.io/gitea/modules/timeutil"
  20. "code.gitea.io/gitea/modules/util"
  21. "github.com/go-macaron/captcha"
  22. "github.com/markbates/goth"
  23. "github.com/tstranex/u2f"
  24. )
  25. const (
  26. // tplMustChangePassword template for updating a user's password
  27. tplMustChangePassword = "user/auth/change_passwd"
  28. // tplSignIn template for sign in page
  29. tplSignIn base.TplName = "user/auth/signin"
  30. // tplSignUp template path for sign up page
  31. tplSignUp base.TplName = "user/auth/signup"
  32. // TplActivate template path for activate user
  33. TplActivate base.TplName = "user/auth/activate"
  34. tplForgotPassword base.TplName = "user/auth/forgot_passwd"
  35. tplResetPassword base.TplName = "user/auth/reset_passwd"
  36. tplTwofa base.TplName = "user/auth/twofa"
  37. tplTwofaScratch base.TplName = "user/auth/twofa_scratch"
  38. tplLinkAccount base.TplName = "user/auth/link_account"
  39. tplU2F base.TplName = "user/auth/u2f"
  40. )
  41. // AutoSignIn reads cookie and try to auto-login.
  42. func AutoSignIn(ctx *context.Context) (bool, error) {
  43. if !models.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.SetCookie(setting.CookieUserName, "", -1, setting.AppSubURL, setting.SessionConfig.Domain, setting.SessionConfig.Secure, true)
  55. ctx.SetCookie(setting.CookieRememberName, "", -1, setting.AppSubURL, setting.SessionConfig.Domain, setting.SessionConfig.Secure, true)
  56. }
  57. }()
  58. u, err := models.GetUserByName(uname)
  59. if err != nil {
  60. if !models.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. err = ctx.Session.Set("uid", u.ID)
  71. if err != nil {
  72. return false, err
  73. }
  74. err = ctx.Session.Set("uname", u.Name)
  75. if err != nil {
  76. return false, err
  77. }
  78. ctx.SetCookie(setting.CSRFCookieName, "", -1, setting.AppSubURL, setting.SessionConfig.Domain, setting.SessionConfig.Secure, true)
  79. return true, nil
  80. }
  81. func checkAutoLogin(ctx *context.Context) bool {
  82. // Check auto-login.
  83. isSucceed, err := AutoSignIn(ctx)
  84. if err != nil {
  85. ctx.ServerError("AutoSignIn", err)
  86. return true
  87. }
  88. redirectTo := ctx.Query("redirect_to")
  89. if len(redirectTo) > 0 {
  90. ctx.SetCookie("redirect_to", redirectTo, 0, setting.AppSubURL, "", setting.SessionConfig.Secure, true)
  91. } else {
  92. redirectTo = ctx.GetCookie("redirect_to")
  93. }
  94. if isSucceed {
  95. ctx.SetCookie("redirect_to", "", -1, setting.AppSubURL, "", setting.SessionConfig.Secure, true)
  96. ctx.RedirectToFirst(redirectTo, setting.AppSubURL+string(setting.LandingPageURL))
  97. return true
  98. }
  99. return false
  100. }
  101. // SignIn render sign in page
  102. func SignIn(ctx *context.Context) {
  103. ctx.Data["Title"] = ctx.Tr("sign_in")
  104. // Check auto-login.
  105. if checkAutoLogin(ctx) {
  106. return
  107. }
  108. orderedOAuth2Names, oauth2Providers, err := models.GetActiveOAuth2Providers()
  109. if err != nil {
  110. ctx.ServerError("UserSignIn", err)
  111. return
  112. }
  113. ctx.Data["OrderedOAuth2Names"] = orderedOAuth2Names
  114. ctx.Data["OAuth2Providers"] = oauth2Providers
  115. ctx.Data["Title"] = ctx.Tr("sign_in")
  116. ctx.Data["SignInLink"] = setting.AppSubURL + "/user/login"
  117. ctx.Data["PageIsSignIn"] = true
  118. ctx.Data["PageIsLogin"] = true
  119. ctx.HTML(200, tplSignIn)
  120. }
  121. // SignInPost response for sign in request
  122. func SignInPost(ctx *context.Context, form auth.SignInForm) {
  123. ctx.Data["Title"] = ctx.Tr("sign_in")
  124. orderedOAuth2Names, oauth2Providers, err := models.GetActiveOAuth2Providers()
  125. if err != nil {
  126. ctx.ServerError("UserSignIn", err)
  127. return
  128. }
  129. ctx.Data["OrderedOAuth2Names"] = orderedOAuth2Names
  130. ctx.Data["OAuth2Providers"] = oauth2Providers
  131. ctx.Data["Title"] = ctx.Tr("sign_in")
  132. ctx.Data["SignInLink"] = setting.AppSubURL + "/user/login"
  133. ctx.Data["PageIsSignIn"] = true
  134. ctx.Data["PageIsLogin"] = true
  135. if ctx.HasError() {
  136. ctx.HTML(200, tplSignIn)
  137. return
  138. }
  139. u, err := models.UserSignIn(form.UserName, form.Password)
  140. if err != nil {
  141. if models.IsErrUserNotExist(err) {
  142. ctx.RenderWithErr(ctx.Tr("form.username_password_incorrect"), tplSignIn, &form)
  143. log.Info("Failed authentication attempt for %s from %s", form.UserName, ctx.RemoteAddr())
  144. } else if models.IsErrEmailAlreadyUsed(err) {
  145. ctx.RenderWithErr(ctx.Tr("form.email_been_used"), tplSignIn, &form)
  146. log.Info("Failed authentication attempt for %s from %s", form.UserName, ctx.RemoteAddr())
  147. } else if models.IsErrUserProhibitLogin(err) {
  148. log.Info("Failed authentication attempt for %s from %s", form.UserName, ctx.RemoteAddr())
  149. ctx.Data["Title"] = ctx.Tr("auth.prohibit_login")
  150. ctx.HTML(200, "user/auth/prohibit_login")
  151. } else if models.IsErrUserInactive(err) {
  152. if setting.Service.RegisterEmailConfirm {
  153. ctx.Data["Title"] = ctx.Tr("auth.active_your_account")
  154. ctx.HTML(200, TplActivate)
  155. } else {
  156. log.Info("Failed authentication attempt for %s from %s", form.UserName, ctx.RemoteAddr())
  157. ctx.Data["Title"] = ctx.Tr("auth.prohibit_login")
  158. ctx.HTML(200, "user/auth/prohibit_login")
  159. }
  160. } else {
  161. ctx.ServerError("UserSignIn", err)
  162. }
  163. return
  164. }
  165. // If this user is enrolled in 2FA, we can't sign the user in just yet.
  166. // Instead, redirect them to the 2FA authentication page.
  167. _, err = models.GetTwoFactorByUID(u.ID)
  168. if err != nil {
  169. if models.IsErrTwoFactorNotEnrolled(err) {
  170. handleSignIn(ctx, u, form.Remember)
  171. } else {
  172. ctx.ServerError("UserSignIn", err)
  173. }
  174. return
  175. }
  176. // User needs to use 2FA, save data and redirect to 2FA page.
  177. err = ctx.Session.Set("twofaUid", u.ID)
  178. if err != nil {
  179. ctx.ServerError("UserSignIn", err)
  180. return
  181. }
  182. err = ctx.Session.Set("twofaRemember", form.Remember)
  183. if err != nil {
  184. ctx.ServerError("UserSignIn", err)
  185. return
  186. }
  187. regs, err := models.GetU2FRegistrationsByUID(u.ID)
  188. if err == nil && len(regs) > 0 {
  189. ctx.Redirect(setting.AppSubURL + "/user/u2f")
  190. return
  191. }
  192. ctx.Redirect(setting.AppSubURL + "/user/two_factor")
  193. }
  194. // TwoFactor shows the user a two-factor authentication page.
  195. func TwoFactor(ctx *context.Context) {
  196. ctx.Data["Title"] = ctx.Tr("twofa")
  197. // Check auto-login.
  198. if checkAutoLogin(ctx) {
  199. return
  200. }
  201. // Ensure user is in a 2FA session.
  202. if ctx.Session.Get("twofaUid") == nil {
  203. ctx.ServerError("UserSignIn", errors.New("not in 2FA session"))
  204. return
  205. }
  206. ctx.HTML(200, tplTwofa)
  207. }
  208. // TwoFactorPost validates a user's two-factor authentication token.
  209. func TwoFactorPost(ctx *context.Context, form auth.TwoFactorAuthForm) {
  210. ctx.Data["Title"] = ctx.Tr("twofa")
  211. // Ensure user is in a 2FA session.
  212. idSess := ctx.Session.Get("twofaUid")
  213. if idSess == nil {
  214. ctx.ServerError("UserSignIn", errors.New("not in 2FA session"))
  215. return
  216. }
  217. id := idSess.(int64)
  218. twofa, err := models.GetTwoFactorByUID(id)
  219. if err != nil {
  220. ctx.ServerError("UserSignIn", err)
  221. return
  222. }
  223. // Validate the passcode with the stored TOTP secret.
  224. ok, err := twofa.ValidateTOTP(form.Passcode)
  225. if err != nil {
  226. ctx.ServerError("UserSignIn", err)
  227. return
  228. }
  229. if ok && twofa.LastUsedPasscode != form.Passcode {
  230. remember := ctx.Session.Get("twofaRemember").(bool)
  231. u, err := models.GetUserByID(id)
  232. if err != nil {
  233. ctx.ServerError("UserSignIn", err)
  234. return
  235. }
  236. if ctx.Session.Get("linkAccount") != nil {
  237. gothUser := ctx.Session.Get("linkAccountGothUser")
  238. if gothUser == nil {
  239. ctx.ServerError("UserSignIn", errors.New("not in LinkAccount session"))
  240. return
  241. }
  242. err = models.LinkAccountToUser(u, gothUser.(goth.User))
  243. if err != nil {
  244. ctx.ServerError("UserSignIn", err)
  245. return
  246. }
  247. }
  248. twofa.LastUsedPasscode = form.Passcode
  249. if err = models.UpdateTwoFactor(twofa); err != nil {
  250. ctx.ServerError("UserSignIn", err)
  251. return
  252. }
  253. handleSignIn(ctx, u, remember)
  254. return
  255. }
  256. ctx.RenderWithErr(ctx.Tr("auth.twofa_passcode_incorrect"), tplTwofa, auth.TwoFactorAuthForm{})
  257. }
  258. // TwoFactorScratch shows the scratch code form for two-factor authentication.
  259. func TwoFactorScratch(ctx *context.Context) {
  260. ctx.Data["Title"] = ctx.Tr("twofa_scratch")
  261. // Check auto-login.
  262. if checkAutoLogin(ctx) {
  263. return
  264. }
  265. // Ensure user is in a 2FA session.
  266. if ctx.Session.Get("twofaUid") == nil {
  267. ctx.ServerError("UserSignIn", errors.New("not in 2FA session"))
  268. return
  269. }
  270. ctx.HTML(200, tplTwofaScratch)
  271. }
  272. // TwoFactorScratchPost validates and invalidates a user's two-factor scratch token.
  273. func TwoFactorScratchPost(ctx *context.Context, form auth.TwoFactorScratchAuthForm) {
  274. ctx.Data["Title"] = ctx.Tr("twofa_scratch")
  275. // Ensure user is in a 2FA session.
  276. idSess := ctx.Session.Get("twofaUid")
  277. if idSess == nil {
  278. ctx.ServerError("UserSignIn", errors.New("not in 2FA session"))
  279. return
  280. }
  281. id := idSess.(int64)
  282. twofa, err := models.GetTwoFactorByUID(id)
  283. if err != nil {
  284. ctx.ServerError("UserSignIn", err)
  285. return
  286. }
  287. // Validate the passcode with the stored TOTP secret.
  288. if twofa.VerifyScratchToken(form.Token) {
  289. // Invalidate the scratch token.
  290. _, err = twofa.GenerateScratchToken()
  291. if err != nil {
  292. ctx.ServerError("UserSignIn", err)
  293. return
  294. }
  295. if err = models.UpdateTwoFactor(twofa); err != nil {
  296. ctx.ServerError("UserSignIn", err)
  297. return
  298. }
  299. remember := ctx.Session.Get("twofaRemember").(bool)
  300. u, err := models.GetUserByID(id)
  301. if err != nil {
  302. ctx.ServerError("UserSignIn", err)
  303. return
  304. }
  305. handleSignInFull(ctx, u, remember, false)
  306. ctx.Flash.Info(ctx.Tr("auth.twofa_scratch_used"))
  307. ctx.Redirect(setting.AppSubURL + "/user/settings/security")
  308. return
  309. }
  310. ctx.RenderWithErr(ctx.Tr("auth.twofa_scratch_token_incorrect"), tplTwofaScratch, auth.TwoFactorScratchAuthForm{})
  311. }
  312. // U2F shows the U2F login page
  313. func U2F(ctx *context.Context) {
  314. ctx.Data["Title"] = ctx.Tr("twofa")
  315. ctx.Data["RequireU2F"] = true
  316. // Check auto-login.
  317. if checkAutoLogin(ctx) {
  318. return
  319. }
  320. // Ensure user is in a 2FA session.
  321. if ctx.Session.Get("twofaUid") == nil {
  322. ctx.ServerError("UserSignIn", errors.New("not in U2F session"))
  323. return
  324. }
  325. ctx.HTML(200, tplU2F)
  326. }
  327. // U2FChallenge submits a sign challenge to the browser
  328. func U2FChallenge(ctx *context.Context) {
  329. // Ensure user is in a U2F session.
  330. idSess := ctx.Session.Get("twofaUid")
  331. if idSess == nil {
  332. ctx.ServerError("UserSignIn", errors.New("not in U2F session"))
  333. return
  334. }
  335. id := idSess.(int64)
  336. regs, err := models.GetU2FRegistrationsByUID(id)
  337. if err != nil {
  338. ctx.ServerError("UserSignIn", err)
  339. return
  340. }
  341. if len(regs) == 0 {
  342. ctx.ServerError("UserSignIn", errors.New("no device registered"))
  343. return
  344. }
  345. challenge, err := u2f.NewChallenge(setting.U2F.AppID, setting.U2F.TrustedFacets)
  346. if err != nil {
  347. ctx.ServerError("u2f.NewChallenge", err)
  348. return
  349. }
  350. if err = ctx.Session.Set("u2fChallenge", challenge); err != nil {
  351. ctx.ServerError("UserSignIn", err)
  352. return
  353. }
  354. ctx.JSON(200, challenge.SignRequest(regs.ToRegistrations()))
  355. }
  356. // U2FSign authenticates the user by signResp
  357. func U2FSign(ctx *context.Context, signResp u2f.SignResponse) {
  358. challSess := ctx.Session.Get("u2fChallenge")
  359. idSess := ctx.Session.Get("twofaUid")
  360. if challSess == nil || idSess == nil {
  361. ctx.ServerError("UserSignIn", errors.New("not in U2F session"))
  362. return
  363. }
  364. challenge := challSess.(*u2f.Challenge)
  365. id := idSess.(int64)
  366. regs, err := models.GetU2FRegistrationsByUID(id)
  367. if err != nil {
  368. ctx.ServerError("UserSignIn", err)
  369. return
  370. }
  371. for _, reg := range regs {
  372. r, err := reg.Parse()
  373. if err != nil {
  374. log.Fatal("parsing u2f registration: %v", err)
  375. continue
  376. }
  377. newCounter, authErr := r.Authenticate(signResp, *challenge, reg.Counter)
  378. if authErr == nil {
  379. reg.Counter = newCounter
  380. user, err := models.GetUserByID(id)
  381. if err != nil {
  382. ctx.ServerError("UserSignIn", err)
  383. return
  384. }
  385. remember := ctx.Session.Get("twofaRemember").(bool)
  386. if err := reg.UpdateCounter(); err != nil {
  387. ctx.ServerError("UserSignIn", err)
  388. return
  389. }
  390. if ctx.Session.Get("linkAccount") != nil {
  391. gothUser := ctx.Session.Get("linkAccountGothUser")
  392. if gothUser == nil {
  393. ctx.ServerError("UserSignIn", errors.New("not in LinkAccount session"))
  394. return
  395. }
  396. err = models.LinkAccountToUser(user, gothUser.(goth.User))
  397. if err != nil {
  398. ctx.ServerError("UserSignIn", err)
  399. return
  400. }
  401. }
  402. redirect := handleSignInFull(ctx, user, remember, false)
  403. if redirect == "" {
  404. redirect = setting.AppSubURL + "/"
  405. }
  406. ctx.PlainText(200, []byte(redirect))
  407. return
  408. }
  409. }
  410. ctx.Error(401)
  411. }
  412. // This handles the final part of the sign-in process of the user.
  413. func handleSignIn(ctx *context.Context, u *models.User, remember bool) {
  414. handleSignInFull(ctx, u, remember, true)
  415. }
  416. func handleSignInFull(ctx *context.Context, u *models.User, remember bool, obeyRedirect bool) string {
  417. if remember {
  418. days := 86400 * setting.LogInRememberDays
  419. ctx.SetCookie(setting.CookieUserName, u.Name, days, setting.AppSubURL, setting.SessionConfig.Domain, setting.SessionConfig.Secure, true)
  420. ctx.SetSuperSecureCookie(base.EncodeMD5(u.Rands+u.Passwd),
  421. setting.CookieRememberName, u.Name, days, setting.AppSubURL, setting.SessionConfig.Domain, setting.SessionConfig.Secure, true)
  422. }
  423. _ = ctx.Session.Delete("openid_verified_uri")
  424. _ = ctx.Session.Delete("openid_signin_remember")
  425. _ = ctx.Session.Delete("openid_determined_email")
  426. _ = ctx.Session.Delete("openid_determined_username")
  427. _ = ctx.Session.Delete("twofaUid")
  428. _ = ctx.Session.Delete("twofaRemember")
  429. _ = ctx.Session.Delete("u2fChallenge")
  430. _ = ctx.Session.Delete("linkAccount")
  431. err := ctx.Session.Set("uid", u.ID)
  432. if err != nil {
  433. log.Error(fmt.Sprintf("Error setting session: %v", err))
  434. }
  435. err = ctx.Session.Set("uname", u.Name)
  436. if err != nil {
  437. log.Error(fmt.Sprintf("Error setting session: %v", err))
  438. }
  439. // Language setting of the user overwrites the one previously set
  440. // If the user does not have a locale set, we save the current one.
  441. if len(u.Language) == 0 {
  442. u.Language = ctx.Locale.Language()
  443. if err := models.UpdateUserCols(u, "language"); err != nil {
  444. log.Error(fmt.Sprintf("Error updating user language [user: %d, locale: %s]", u.ID, u.Language))
  445. return setting.AppSubURL + "/"
  446. }
  447. }
  448. ctx.SetCookie("lang", u.Language, nil, setting.AppSubURL, setting.SessionConfig.Domain, setting.SessionConfig.Secure, true)
  449. // Clear whatever CSRF has right now, force to generate a new one
  450. ctx.SetCookie(setting.CSRFCookieName, "", -1, setting.AppSubURL, setting.SessionConfig.Domain, setting.SessionConfig.Secure, true)
  451. // Register last login
  452. u.SetLastLogin()
  453. if err := models.UpdateUserCols(u, "last_login_unix"); err != nil {
  454. ctx.ServerError("UpdateUserCols", err)
  455. return setting.AppSubURL + "/"
  456. }
  457. if redirectTo := ctx.GetCookie("redirect_to"); len(redirectTo) > 0 && !util.IsExternalURL(redirectTo) {
  458. ctx.SetCookie("redirect_to", "", -1, setting.AppSubURL, "", setting.SessionConfig.Secure, true)
  459. if obeyRedirect {
  460. ctx.RedirectToFirst(redirectTo)
  461. }
  462. return redirectTo
  463. }
  464. if obeyRedirect {
  465. ctx.Redirect(setting.AppSubURL + "/")
  466. }
  467. return setting.AppSubURL + "/"
  468. }
  469. // SignInOAuth handles the OAuth2 login buttons
  470. func SignInOAuth(ctx *context.Context) {
  471. provider := ctx.Params(":provider")
  472. loginSource, err := models.GetActiveOAuth2LoginSourceByName(provider)
  473. if err != nil {
  474. ctx.ServerError("SignIn", err)
  475. return
  476. }
  477. // try to do a direct callback flow, so we don't authenticate the user again but use the valid accesstoken to get the user
  478. user, gothUser, err := oAuth2UserLoginCallback(loginSource, ctx.Req.Request, ctx.Resp)
  479. if err == nil && user != nil {
  480. // we got the user without going through the whole OAuth2 authentication flow again
  481. handleOAuth2SignIn(user, gothUser, ctx, err)
  482. return
  483. }
  484. err = oauth2.Auth(loginSource.Name, ctx.Req.Request, ctx.Resp)
  485. if err != nil {
  486. ctx.ServerError("SignIn", err)
  487. }
  488. // redirect is done in oauth2.Auth
  489. }
  490. // SignInOAuthCallback handles the callback from the given provider
  491. func SignInOAuthCallback(ctx *context.Context) {
  492. provider := ctx.Params(":provider")
  493. // first look if the provider is still active
  494. loginSource, err := models.GetActiveOAuth2LoginSourceByName(provider)
  495. if err != nil {
  496. ctx.ServerError("SignIn", err)
  497. return
  498. }
  499. if loginSource == nil {
  500. ctx.ServerError("SignIn", errors.New("No valid provider found, check configured callback url in provider"))
  501. return
  502. }
  503. u, gothUser, err := oAuth2UserLoginCallback(loginSource, ctx.Req.Request, ctx.Resp)
  504. handleOAuth2SignIn(u, gothUser, ctx, err)
  505. }
  506. func handleOAuth2SignIn(u *models.User, gothUser goth.User, ctx *context.Context, err error) {
  507. if err != nil {
  508. ctx.ServerError("UserSignIn", err)
  509. return
  510. }
  511. if u == nil {
  512. // no existing user is found, request attach or new account
  513. err = ctx.Session.Set("linkAccountGothUser", gothUser)
  514. if err != nil {
  515. log.Error(fmt.Sprintf("Error setting session: %v", err))
  516. }
  517. ctx.Redirect(setting.AppSubURL + "/user/link_account")
  518. return
  519. }
  520. // If this user is enrolled in 2FA, we can't sign the user in just yet.
  521. // Instead, redirect them to the 2FA authentication page.
  522. _, err = models.GetTwoFactorByUID(u.ID)
  523. if err != nil {
  524. if models.IsErrTwoFactorNotEnrolled(err) {
  525. err = ctx.Session.Set("uid", u.ID)
  526. if err != nil {
  527. log.Error(fmt.Sprintf("Error setting session: %v", err))
  528. }
  529. err = ctx.Session.Set("uname", u.Name)
  530. if err != nil {
  531. log.Error(fmt.Sprintf("Error setting session: %v", err))
  532. }
  533. // Clear whatever CSRF has right now, force to generate a new one
  534. ctx.SetCookie(setting.CSRFCookieName, "", -1, setting.AppSubURL, setting.SessionConfig.Domain, setting.SessionConfig.Secure, true)
  535. // Register last login
  536. u.SetLastLogin()
  537. if err := models.UpdateUserCols(u, "last_login_unix"); err != nil {
  538. ctx.ServerError("UpdateUserCols", err)
  539. return
  540. }
  541. if redirectTo := ctx.GetCookie("redirect_to"); len(redirectTo) > 0 {
  542. ctx.SetCookie("redirect_to", "", -1, setting.AppSubURL, "", setting.SessionConfig.Secure, true)
  543. ctx.RedirectToFirst(redirectTo)
  544. return
  545. }
  546. ctx.Redirect(setting.AppSubURL + "/")
  547. } else {
  548. ctx.ServerError("UserSignIn", err)
  549. }
  550. return
  551. }
  552. // User needs to use 2FA, save data and redirect to 2FA page.
  553. err = ctx.Session.Set("twofaUid", u.ID)
  554. if err != nil {
  555. log.Error(fmt.Sprintf("Error setting session: %v", err))
  556. }
  557. err = ctx.Session.Set("twofaRemember", false)
  558. if err != nil {
  559. log.Error(fmt.Sprintf("Error setting session: %v", err))
  560. }
  561. // If U2F is enrolled -> Redirect to U2F instead
  562. regs, err := models.GetU2FRegistrationsByUID(u.ID)
  563. if err == nil && len(regs) > 0 {
  564. ctx.Redirect(setting.AppSubURL + "/user/u2f")
  565. return
  566. }
  567. ctx.Redirect(setting.AppSubURL + "/user/two_factor")
  568. }
  569. // OAuth2UserLoginCallback attempts to handle the callback from the OAuth2 provider and if successful
  570. // login the user
  571. func oAuth2UserLoginCallback(loginSource *models.LoginSource, request *http.Request, response http.ResponseWriter) (*models.User, goth.User, error) {
  572. gothUser, err := oauth2.ProviderCallback(loginSource.Name, request, response)
  573. if err != nil {
  574. return nil, goth.User{}, err
  575. }
  576. user := &models.User{
  577. LoginName: gothUser.UserID,
  578. LoginType: models.LoginOAuth2,
  579. LoginSource: loginSource.ID,
  580. }
  581. hasUser, err := models.GetUser(user)
  582. if err != nil {
  583. return nil, goth.User{}, err
  584. }
  585. if hasUser {
  586. return user, goth.User{}, nil
  587. }
  588. // search in external linked users
  589. externalLoginUser := &models.ExternalLoginUser{
  590. ExternalID: gothUser.UserID,
  591. LoginSourceID: loginSource.ID,
  592. }
  593. hasUser, err = models.GetExternalLogin(externalLoginUser)
  594. if err != nil {
  595. return nil, goth.User{}, err
  596. }
  597. if hasUser {
  598. user, err = models.GetUserByID(externalLoginUser.UserID)
  599. return user, goth.User{}, err
  600. }
  601. // no user found to login
  602. return nil, gothUser, nil
  603. }
  604. // LinkAccount shows the page where the user can decide to login or create a new account
  605. func LinkAccount(ctx *context.Context) {
  606. ctx.Data["DisablePassword"] = !setting.Service.RequireExternalRegistrationCaptcha || setting.Service.AllowOnlyExternalRegistration
  607. ctx.Data["Title"] = ctx.Tr("link_account")
  608. ctx.Data["LinkAccountMode"] = true
  609. ctx.Data["EnableCaptcha"] = setting.Service.EnableCaptcha && setting.Service.RequireExternalRegistrationCaptcha
  610. ctx.Data["CaptchaType"] = setting.Service.CaptchaType
  611. ctx.Data["RecaptchaURL"] = setting.Service.RecaptchaURL
  612. ctx.Data["RecaptchaSitekey"] = setting.Service.RecaptchaSitekey
  613. ctx.Data["DisableRegistration"] = setting.Service.DisableRegistration
  614. ctx.Data["ShowRegistrationButton"] = false
  615. // use this to set the right link into the signIn and signUp templates in the link_account template
  616. ctx.Data["SignInLink"] = setting.AppSubURL + "/user/link_account_signin"
  617. ctx.Data["SignUpLink"] = setting.AppSubURL + "/user/link_account_signup"
  618. gothUser := ctx.Session.Get("linkAccountGothUser")
  619. if gothUser == nil {
  620. ctx.ServerError("UserSignIn", errors.New("not in LinkAccount session"))
  621. return
  622. }
  623. uname := gothUser.(goth.User).NickName
  624. email := gothUser.(goth.User).Email
  625. ctx.Data["user_name"] = uname
  626. ctx.Data["email"] = email
  627. if len(email) != 0 {
  628. u, err := models.GetUserByEmail(email)
  629. if err != nil && !models.IsErrUserNotExist(err) {
  630. ctx.ServerError("UserSignIn", err)
  631. return
  632. }
  633. if u != nil {
  634. ctx.Data["user_exists"] = true
  635. }
  636. } else if len(uname) != 0 {
  637. u, err := models.GetUserByName(uname)
  638. if err != nil && !models.IsErrUserNotExist(err) {
  639. ctx.ServerError("UserSignIn", err)
  640. return
  641. }
  642. if u != nil {
  643. ctx.Data["user_exists"] = true
  644. }
  645. }
  646. ctx.HTML(200, tplLinkAccount)
  647. }
  648. // LinkAccountPostSignIn handle the coupling of external account with another account using signIn
  649. func LinkAccountPostSignIn(ctx *context.Context, signInForm auth.SignInForm) {
  650. ctx.Data["DisablePassword"] = setting.Service.AllowOnlyExternalRegistration
  651. ctx.Data["Title"] = ctx.Tr("link_account")
  652. ctx.Data["LinkAccountMode"] = true
  653. ctx.Data["LinkAccountModeSignIn"] = true
  654. ctx.Data["EnableCaptcha"] = setting.Service.EnableCaptcha && setting.Service.RequireExternalRegistrationCaptcha
  655. ctx.Data["RecaptchaURL"] = setting.Service.RecaptchaURL
  656. ctx.Data["CaptchaType"] = setting.Service.CaptchaType
  657. ctx.Data["RecaptchaSitekey"] = setting.Service.RecaptchaSitekey
  658. ctx.Data["DisableRegistration"] = setting.Service.DisableRegistration
  659. ctx.Data["ShowRegistrationButton"] = false
  660. // use this to set the right link into the signIn and signUp templates in the link_account template
  661. ctx.Data["SignInLink"] = setting.AppSubURL + "/user/link_account_signin"
  662. ctx.Data["SignUpLink"] = setting.AppSubURL + "/user/link_account_signup"
  663. gothUser := ctx.Session.Get("linkAccountGothUser")
  664. if gothUser == nil {
  665. ctx.ServerError("UserSignIn", errors.New("not in LinkAccount session"))
  666. return
  667. }
  668. if ctx.HasError() {
  669. ctx.HTML(200, tplLinkAccount)
  670. return
  671. }
  672. u, err := models.UserSignIn(signInForm.UserName, signInForm.Password)
  673. if err != nil {
  674. if models.IsErrUserNotExist(err) {
  675. ctx.RenderWithErr(ctx.Tr("form.username_password_incorrect"), tplLinkAccount, &signInForm)
  676. } else {
  677. ctx.ServerError("UserLinkAccount", err)
  678. }
  679. return
  680. }
  681. // If this user is enrolled in 2FA, we can't sign the user in just yet.
  682. // Instead, redirect them to the 2FA authentication page.
  683. _, err = models.GetTwoFactorByUID(u.ID)
  684. if err != nil {
  685. if models.IsErrTwoFactorNotEnrolled(err) {
  686. err = models.LinkAccountToUser(u, gothUser.(goth.User))
  687. if err != nil {
  688. ctx.ServerError("UserLinkAccount", err)
  689. } else {
  690. handleSignIn(ctx, u, signInForm.Remember)
  691. }
  692. } else {
  693. ctx.ServerError("UserLinkAccount", err)
  694. }
  695. return
  696. }
  697. // User needs to use 2FA, save data and redirect to 2FA page.
  698. err = ctx.Session.Set("twofaUid", u.ID)
  699. if err != nil {
  700. log.Error(fmt.Sprintf("Error setting session: %v", err))
  701. }
  702. err = ctx.Session.Set("twofaRemember", signInForm.Remember)
  703. if err != nil {
  704. log.Error(fmt.Sprintf("Error setting session: %v", err))
  705. }
  706. err = ctx.Session.Set("linkAccount", true)
  707. if err != nil {
  708. log.Error(fmt.Sprintf("Error setting session: %v", err))
  709. }
  710. // If U2F is enrolled -> Redirect to U2F instead
  711. regs, err := models.GetU2FRegistrationsByUID(u.ID)
  712. if err == nil && len(regs) > 0 {
  713. ctx.Redirect(setting.AppSubURL + "/user/u2f")
  714. return
  715. }
  716. ctx.Redirect(setting.AppSubURL + "/user/two_factor")
  717. }
  718. // LinkAccountPostRegister handle the creation of a new account for an external account using signUp
  719. func LinkAccountPostRegister(ctx *context.Context, cpt *captcha.Captcha, form auth.RegisterForm) {
  720. // TODO Make insecure passwords optional for local accounts also,
  721. // once email-based Second-Factor Auth is available
  722. ctx.Data["DisablePassword"] = !setting.Service.RequireExternalRegistrationCaptcha || setting.Service.AllowOnlyExternalRegistration
  723. ctx.Data["Title"] = ctx.Tr("link_account")
  724. ctx.Data["LinkAccountMode"] = true
  725. ctx.Data["LinkAccountModeRegister"] = true
  726. ctx.Data["EnableCaptcha"] = setting.Service.EnableCaptcha && setting.Service.RequireExternalRegistrationCaptcha
  727. ctx.Data["RecaptchaURL"] = setting.Service.RecaptchaURL
  728. ctx.Data["CaptchaType"] = setting.Service.CaptchaType
  729. ctx.Data["RecaptchaSitekey"] = setting.Service.RecaptchaSitekey
  730. ctx.Data["DisableRegistration"] = setting.Service.DisableRegistration
  731. ctx.Data["ShowRegistrationButton"] = false
  732. // use this to set the right link into the signIn and signUp templates in the link_account template
  733. ctx.Data["SignInLink"] = setting.AppSubURL + "/user/link_account_signin"
  734. ctx.Data["SignUpLink"] = setting.AppSubURL + "/user/link_account_signup"
  735. gothUser := ctx.Session.Get("linkAccountGothUser")
  736. if gothUser == nil {
  737. ctx.ServerError("UserSignUp", errors.New("not in LinkAccount session"))
  738. return
  739. }
  740. if ctx.HasError() {
  741. ctx.HTML(200, tplLinkAccount)
  742. return
  743. }
  744. if setting.Service.DisableRegistration {
  745. ctx.Error(403)
  746. return
  747. }
  748. if setting.Service.EnableCaptcha && setting.Service.RequireExternalRegistrationCaptcha {
  749. var valid bool
  750. switch setting.Service.CaptchaType {
  751. case setting.ImageCaptcha:
  752. valid = cpt.VerifyReq(ctx.Req)
  753. case setting.ReCaptcha:
  754. valid, _ = recaptcha.Verify(form.GRecaptchaResponse)
  755. default:
  756. ctx.ServerError("Unknown Captcha Type", fmt.Errorf("Unknown Captcha Type: %s", setting.Service.CaptchaType))
  757. return
  758. }
  759. if !valid {
  760. ctx.Data["Err_Captcha"] = true
  761. ctx.RenderWithErr(ctx.Tr("form.captcha_incorrect"), tplLinkAccount, &form)
  762. return
  763. }
  764. }
  765. if setting.Service.AllowOnlyExternalRegistration || !setting.Service.RequireExternalRegistrationPassword {
  766. // In models.User an empty password is classed as not set, so we set form.Password to empty.
  767. // Eventually the database should be changed to indicate "Second Factor"-enabled accounts
  768. // (accounts that do not introduce the security vulnerabilities of a password).
  769. // If a user decides to circumvent second-factor security, and purposefully create a password,
  770. // they can still do so using the "Recover Account" option.
  771. form.Password = ""
  772. } else {
  773. if (len(strings.TrimSpace(form.Password)) > 0 || len(strings.TrimSpace(form.Retype)) > 0) && form.Password != form.Retype {
  774. ctx.Data["Err_Password"] = true
  775. ctx.RenderWithErr(ctx.Tr("form.password_not_match"), tplLinkAccount, &form)
  776. return
  777. }
  778. if len(strings.TrimSpace(form.Password)) > 0 && len(form.Password) < setting.MinPasswordLength {
  779. ctx.Data["Err_Password"] = true
  780. ctx.RenderWithErr(ctx.Tr("auth.password_too_short", setting.MinPasswordLength), tplLinkAccount, &form)
  781. return
  782. }
  783. }
  784. loginSource, err := models.GetActiveOAuth2LoginSourceByName(gothUser.(goth.User).Provider)
  785. if err != nil {
  786. ctx.ServerError("CreateUser", err)
  787. }
  788. u := &models.User{
  789. Name: form.UserName,
  790. Email: form.Email,
  791. Passwd: form.Password,
  792. IsActive: !setting.Service.RegisterEmailConfirm,
  793. LoginType: models.LoginOAuth2,
  794. LoginSource: loginSource.ID,
  795. LoginName: gothUser.(goth.User).UserID,
  796. }
  797. if err := models.CreateUser(u); err != nil {
  798. switch {
  799. case models.IsErrUserAlreadyExist(err):
  800. ctx.Data["Err_UserName"] = true
  801. ctx.RenderWithErr(ctx.Tr("form.username_been_taken"), tplLinkAccount, &form)
  802. case models.IsErrEmailAlreadyUsed(err):
  803. ctx.Data["Err_Email"] = true
  804. ctx.RenderWithErr(ctx.Tr("form.email_been_used"), tplLinkAccount, &form)
  805. case models.IsErrNameReserved(err):
  806. ctx.Data["Err_UserName"] = true
  807. ctx.RenderWithErr(ctx.Tr("user.form.name_reserved", err.(models.ErrNameReserved).Name), tplLinkAccount, &form)
  808. case models.IsErrNamePatternNotAllowed(err):
  809. ctx.Data["Err_UserName"] = true
  810. ctx.RenderWithErr(ctx.Tr("user.form.name_pattern_not_allowed", err.(models.ErrNamePatternNotAllowed).Pattern), tplLinkAccount, &form)
  811. default:
  812. ctx.ServerError("CreateUser", err)
  813. }
  814. return
  815. }
  816. log.Trace("Account created: %s", u.Name)
  817. // Auto-set admin for the only user.
  818. if models.CountUsers() == 1 {
  819. u.IsAdmin = true
  820. u.IsActive = true
  821. u.SetLastLogin()
  822. if err := models.UpdateUserCols(u, "is_admin", "is_active", "last_login_unix"); err != nil {
  823. ctx.ServerError("UpdateUser", err)
  824. return
  825. }
  826. }
  827. // Send confirmation email
  828. if setting.Service.RegisterEmailConfirm && u.ID > 1 {
  829. models.SendActivateAccountMail(ctx.Context, u)
  830. ctx.Data["IsSendRegisterMail"] = true
  831. ctx.Data["Email"] = u.Email
  832. ctx.Data["ActiveCodeLives"] = timeutil.MinutesToFriendly(setting.Service.ActiveCodeLives, ctx.Locale.Language())
  833. ctx.HTML(200, TplActivate)
  834. if err := ctx.Cache.Put("MailResendLimit_"+u.LowerName, u.LowerName, 180); err != nil {
  835. log.Error("Set cache(MailResendLimit) fail: %v", err)
  836. }
  837. return
  838. }
  839. ctx.Redirect(setting.AppSubURL + "/user/login")
  840. }
  841. func handleSignOut(ctx *context.Context) {
  842. _ = ctx.Session.Delete("uid")
  843. _ = ctx.Session.Delete("uname")
  844. _ = ctx.Session.Delete("socialId")
  845. _ = ctx.Session.Delete("socialName")
  846. _ = ctx.Session.Delete("socialEmail")
  847. ctx.SetCookie(setting.CookieUserName, "", -1, setting.AppSubURL, setting.SessionConfig.Domain, setting.SessionConfig.Secure, true)
  848. ctx.SetCookie(setting.CookieRememberName, "", -1, setting.AppSubURL, setting.SessionConfig.Domain, setting.SessionConfig.Secure, true)
  849. ctx.SetCookie(setting.CSRFCookieName, "", -1, setting.AppSubURL, setting.SessionConfig.Domain, setting.SessionConfig.Secure, true)
  850. 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.
  851. }
  852. // SignOut sign out from login status
  853. func SignOut(ctx *context.Context) {
  854. handleSignOut(ctx)
  855. ctx.Redirect(setting.AppSubURL + "/")
  856. }
  857. // SignUp render the register page
  858. func SignUp(ctx *context.Context) {
  859. ctx.Data["Title"] = ctx.Tr("sign_up")
  860. ctx.Data["SignUpLink"] = setting.AppSubURL + "/user/sign_up"
  861. ctx.Data["EnableCaptcha"] = setting.Service.EnableCaptcha
  862. ctx.Data["RecaptchaURL"] = setting.Service.RecaptchaURL
  863. ctx.Data["CaptchaType"] = setting.Service.CaptchaType
  864. ctx.Data["RecaptchaSitekey"] = setting.Service.RecaptchaSitekey
  865. ctx.Data["DisableRegistration"] = setting.Service.DisableRegistration
  866. ctx.HTML(200, tplSignUp)
  867. }
  868. // SignUpPost response for sign up information submission
  869. func SignUpPost(ctx *context.Context, cpt *captcha.Captcha, form auth.RegisterForm) {
  870. ctx.Data["Title"] = ctx.Tr("sign_up")
  871. ctx.Data["SignUpLink"] = setting.AppSubURL + "/user/sign_up"
  872. ctx.Data["EnableCaptcha"] = setting.Service.EnableCaptcha
  873. ctx.Data["RecaptchaURL"] = setting.Service.RecaptchaURL
  874. ctx.Data["CaptchaType"] = setting.Service.CaptchaType
  875. ctx.Data["RecaptchaSitekey"] = setting.Service.RecaptchaSitekey
  876. //Permission denied if DisableRegistration or AllowOnlyExternalRegistration options are true
  877. if !setting.Service.ShowRegistrationButton {
  878. ctx.Error(403)
  879. return
  880. }
  881. if ctx.HasError() {
  882. ctx.HTML(200, tplSignUp)
  883. return
  884. }
  885. if setting.Service.EnableCaptcha {
  886. var valid bool
  887. switch setting.Service.CaptchaType {
  888. case setting.ImageCaptcha:
  889. valid = cpt.VerifyReq(ctx.Req)
  890. case setting.ReCaptcha:
  891. valid, _ = recaptcha.Verify(form.GRecaptchaResponse)
  892. default:
  893. ctx.ServerError("Unknown Captcha Type", fmt.Errorf("Unknown Captcha Type: %s", setting.Service.CaptchaType))
  894. return
  895. }
  896. if !valid {
  897. ctx.Data["Err_Captcha"] = true
  898. ctx.RenderWithErr(ctx.Tr("form.captcha_incorrect"), tplSignUp, &form)
  899. return
  900. }
  901. }
  902. if !form.IsEmailDomainWhitelisted() {
  903. ctx.RenderWithErr(ctx.Tr("auth.email_domain_blacklisted"), tplSignUp, &form)
  904. return
  905. }
  906. if form.Password != form.Retype {
  907. ctx.Data["Err_Password"] = true
  908. ctx.RenderWithErr(ctx.Tr("form.password_not_match"), tplSignUp, &form)
  909. return
  910. }
  911. if len(form.Password) < setting.MinPasswordLength {
  912. ctx.Data["Err_Password"] = true
  913. ctx.RenderWithErr(ctx.Tr("auth.password_too_short", setting.MinPasswordLength), tplSignUp, &form)
  914. return
  915. }
  916. u := &models.User{
  917. Name: form.UserName,
  918. Email: form.Email,
  919. Passwd: form.Password,
  920. IsActive: !setting.Service.RegisterEmailConfirm,
  921. }
  922. if err := models.CreateUser(u); err != nil {
  923. switch {
  924. case models.IsErrUserAlreadyExist(err):
  925. ctx.Data["Err_UserName"] = true
  926. ctx.RenderWithErr(ctx.Tr("form.username_been_taken"), tplSignUp, &form)
  927. case models.IsErrEmailAlreadyUsed(err):
  928. ctx.Data["Err_Email"] = true
  929. ctx.RenderWithErr(ctx.Tr("form.email_been_used"), tplSignUp, &form)
  930. case models.IsErrNameReserved(err):
  931. ctx.Data["Err_UserName"] = true
  932. ctx.RenderWithErr(ctx.Tr("user.form.name_reserved", err.(models.ErrNameReserved).Name), tplSignUp, &form)
  933. case models.IsErrNamePatternNotAllowed(err):
  934. ctx.Data["Err_UserName"] = true
  935. ctx.RenderWithErr(ctx.Tr("user.form.name_pattern_not_allowed", err.(models.ErrNamePatternNotAllowed).Pattern), tplSignUp, &form)
  936. default:
  937. ctx.ServerError("CreateUser", err)
  938. }
  939. return
  940. }
  941. log.Trace("Account created: %s", u.Name)
  942. // Auto-set admin for the only user.
  943. if models.CountUsers() == 1 {
  944. u.IsAdmin = true
  945. u.IsActive = true
  946. u.SetLastLogin()
  947. if err := models.UpdateUserCols(u, "is_admin", "is_active", "last_login_unix"); err != nil {
  948. ctx.ServerError("UpdateUser", err)
  949. return
  950. }
  951. }
  952. // Send confirmation email, no need for social account.
  953. if setting.Service.RegisterEmailConfirm && u.ID > 1 {
  954. models.SendActivateAccountMail(ctx.Context, u)
  955. ctx.Data["IsSendRegisterMail"] = true
  956. ctx.Data["Email"] = u.Email
  957. ctx.Data["ActiveCodeLives"] = timeutil.MinutesToFriendly(setting.Service.ActiveCodeLives, ctx.Locale.Language())
  958. ctx.HTML(200, TplActivate)
  959. if err := ctx.Cache.Put("MailResendLimit_"+u.LowerName, u.LowerName, 180); err != nil {
  960. log.Error("Set cache(MailResendLimit) fail: %v", err)
  961. }
  962. return
  963. }
  964. ctx.Flash.Success(ctx.Tr("auth.sign_up_successful"))
  965. handleSignInFull(ctx, u, false, true)
  966. }
  967. // Activate render activate user page
  968. func Activate(ctx *context.Context) {
  969. code := ctx.Query("code")
  970. if len(code) == 0 {
  971. ctx.Data["IsActivatePage"] = true
  972. if ctx.User.IsActive {
  973. ctx.Error(404)
  974. return
  975. }
  976. // Resend confirmation email.
  977. if setting.Service.RegisterEmailConfirm {
  978. if ctx.Cache.IsExist("MailResendLimit_" + ctx.User.LowerName) {
  979. ctx.Data["ResendLimited"] = true
  980. } else {
  981. ctx.Data["ActiveCodeLives"] = timeutil.MinutesToFriendly(setting.Service.ActiveCodeLives, ctx.Locale.Language())
  982. models.SendActivateAccountMail(ctx.Context, ctx.User)
  983. if err := ctx.Cache.Put("MailResendLimit_"+ctx.User.LowerName, ctx.User.LowerName, 180); err != nil {
  984. log.Error("Set cache(MailResendLimit) fail: %v", err)
  985. }
  986. }
  987. } else {
  988. ctx.Data["ServiceNotEnabled"] = true
  989. }
  990. ctx.HTML(200, TplActivate)
  991. return
  992. }
  993. // Verify code.
  994. if user := models.VerifyUserActiveCode(code); user != nil {
  995. user.IsActive = true
  996. var err error
  997. if user.Rands, err = models.GetUserSalt(); err != nil {
  998. ctx.ServerError("UpdateUser", err)
  999. return
  1000. }
  1001. if err := models.UpdateUserCols(user, "is_active", "rands"); err != nil {
  1002. if models.IsErrUserNotExist(err) {
  1003. ctx.Error(404)
  1004. } else {
  1005. ctx.ServerError("UpdateUser", err)
  1006. }
  1007. return
  1008. }
  1009. log.Trace("User activated: %s", user.Name)
  1010. err = ctx.Session.Set("uid", user.ID)
  1011. if err != nil {
  1012. log.Error(fmt.Sprintf("Error setting session: %v", err))
  1013. }
  1014. err = ctx.Session.Set("uname", user.Name)
  1015. if err != nil {
  1016. log.Error(fmt.Sprintf("Error setting session: %v", err))
  1017. }
  1018. ctx.Flash.Success(ctx.Tr("auth.account_activated"))
  1019. ctx.Redirect(setting.AppSubURL + "/")
  1020. return
  1021. }
  1022. ctx.Data["IsActivateFailed"] = true
  1023. ctx.HTML(200, TplActivate)
  1024. }
  1025. // ActivateEmail render the activate email page
  1026. func ActivateEmail(ctx *context.Context) {
  1027. code := ctx.Query("code")
  1028. emailStr := ctx.Query("email")
  1029. // Verify code.
  1030. if email := models.VerifyActiveEmailCode(code, emailStr); email != nil {
  1031. if err := email.Activate(); err != nil {
  1032. ctx.ServerError("ActivateEmail", err)
  1033. }
  1034. log.Trace("Email activated: %s", email.Email)
  1035. ctx.Flash.Success(ctx.Tr("settings.add_email_success"))
  1036. }
  1037. ctx.Redirect(setting.AppSubURL + "/user/settings/email")
  1038. }
  1039. // ForgotPasswd render the forget pasword page
  1040. func ForgotPasswd(ctx *context.Context) {
  1041. ctx.Data["Title"] = ctx.Tr("auth.forgot_password_title")
  1042. if setting.MailService == nil {
  1043. ctx.Data["IsResetDisable"] = true
  1044. ctx.HTML(200, tplForgotPassword)
  1045. return
  1046. }
  1047. email := ctx.Query("email")
  1048. ctx.Data["Email"] = email
  1049. ctx.Data["IsResetRequest"] = true
  1050. ctx.HTML(200, tplForgotPassword)
  1051. }
  1052. // ForgotPasswdPost response for forget password request
  1053. func ForgotPasswdPost(ctx *context.Context) {
  1054. ctx.Data["Title"] = ctx.Tr("auth.forgot_password_title")
  1055. if setting.MailService == nil {
  1056. ctx.NotFound("ForgotPasswdPost", nil)
  1057. return
  1058. }
  1059. ctx.Data["IsResetRequest"] = true
  1060. email := ctx.Query("email")
  1061. ctx.Data["Email"] = email
  1062. u, err := models.GetUserByEmail(email)
  1063. if err != nil {
  1064. if models.IsErrUserNotExist(err) {
  1065. ctx.Data["ResetPwdCodeLives"] = timeutil.MinutesToFriendly(setting.Service.ResetPwdCodeLives, ctx.Locale.Language())
  1066. ctx.Data["IsResetSent"] = true
  1067. ctx.HTML(200, tplForgotPassword)
  1068. return
  1069. }
  1070. ctx.ServerError("user.ResetPasswd(check existence)", err)
  1071. return
  1072. }
  1073. if !u.IsLocal() && !u.IsOAuth2() {
  1074. ctx.Data["Err_Email"] = true
  1075. ctx.RenderWithErr(ctx.Tr("auth.non_local_account"), tplForgotPassword, nil)
  1076. return
  1077. }
  1078. if ctx.Cache.IsExist("MailResendLimit_" + u.LowerName) {
  1079. ctx.Data["ResendLimited"] = true
  1080. ctx.HTML(200, tplForgotPassword)
  1081. return
  1082. }
  1083. models.SendResetPasswordMail(ctx.Context, u)
  1084. if err = ctx.Cache.Put("MailResendLimit_"+u.LowerName, u.LowerName, 180); err != nil {
  1085. log.Error("Set cache(MailResendLimit) fail: %v", err)
  1086. }
  1087. ctx.Data["ResetPwdCodeLives"] = timeutil.MinutesToFriendly(setting.Service.ResetPwdCodeLives, ctx.Locale.Language())
  1088. ctx.Data["IsResetSent"] = true
  1089. ctx.HTML(200, tplForgotPassword)
  1090. }
  1091. func commonResetPassword(ctx *context.Context) *models.User {
  1092. code := ctx.Query("code")
  1093. ctx.Data["Title"] = ctx.Tr("auth.reset_password")
  1094. ctx.Data["Code"] = code
  1095. if nil != ctx.User {
  1096. ctx.Data["user_signed_in"] = true
  1097. }
  1098. if len(code) == 0 {
  1099. ctx.Flash.Error(ctx.Tr("auth.invalid_code"))
  1100. return nil
  1101. }
  1102. // Fail early, don't frustrate the user
  1103. u := models.VerifyUserActiveCode(code)
  1104. if u == nil {
  1105. ctx.Flash.Error(ctx.Tr("auth.invalid_code"))
  1106. return nil
  1107. }
  1108. // Show the user that they are affecting the account that they intended to
  1109. ctx.Data["user_email"] = u.Email
  1110. if nil != ctx.User && u.ID != ctx.User.ID {
  1111. ctx.Flash.Error(ctx.Tr("auth.reset_password_wrong_user", ctx.User.Email, u.Email))
  1112. return nil
  1113. }
  1114. return u
  1115. }
  1116. // ResetPasswd render the account recovery page
  1117. func ResetPasswd(ctx *context.Context) {
  1118. ctx.Data["IsResetForm"] = true
  1119. commonResetPassword(ctx)
  1120. ctx.HTML(200, tplResetPassword)
  1121. }
  1122. // ResetPasswdPost response from account recovery request
  1123. func ResetPasswdPost(ctx *context.Context) {
  1124. u := commonResetPassword(ctx)
  1125. if u == nil {
  1126. // Flash error has been set
  1127. ctx.HTML(200, tplResetPassword)
  1128. return
  1129. }
  1130. // Validate password length.
  1131. passwd := ctx.Query("password")
  1132. if len(passwd) < setting.MinPasswordLength {
  1133. ctx.Data["IsResetForm"] = true
  1134. ctx.Data["Err_Password"] = true
  1135. ctx.RenderWithErr(ctx.Tr("auth.password_too_short", setting.MinPasswordLength), tplResetPassword, nil)
  1136. return
  1137. }
  1138. var err error
  1139. if u.Rands, err = models.GetUserSalt(); err != nil {
  1140. ctx.ServerError("UpdateUser", err)
  1141. return
  1142. }
  1143. if u.Salt, err = models.GetUserSalt(); err != nil {
  1144. ctx.ServerError("UpdateUser", err)
  1145. return
  1146. }
  1147. u.HashPassword(passwd)
  1148. u.MustChangePassword = false
  1149. if err := models.UpdateUserCols(u, "must_change_password", "passwd", "rands", "salt"); err != nil {
  1150. ctx.ServerError("UpdateUser", err)
  1151. return
  1152. }
  1153. log.Trace("User password reset: %s", u.Name)
  1154. ctx.Data["IsResetFailed"] = true
  1155. remember := len(ctx.Query("remember")) != 0
  1156. handleSignInFull(ctx, u, remember, true)
  1157. }
  1158. // MustChangePassword renders the page to change a user's password
  1159. func MustChangePassword(ctx *context.Context) {
  1160. ctx.Data["Title"] = ctx.Tr("auth.must_change_password")
  1161. ctx.Data["ChangePasscodeLink"] = setting.AppSubURL + "/user/settings/change_password"
  1162. ctx.HTML(200, tplMustChangePassword)
  1163. }
  1164. // MustChangePasswordPost response for updating a user's password after his/her
  1165. // account was created by an admin
  1166. func MustChangePasswordPost(ctx *context.Context, cpt *captcha.Captcha, form auth.MustChangePasswordForm) {
  1167. ctx.Data["Title"] = ctx.Tr("auth.must_change_password")
  1168. ctx.Data["ChangePasscodeLink"] = setting.AppSubURL + "/user/settings/change_password"
  1169. if ctx.HasError() {
  1170. ctx.HTML(200, tplMustChangePassword)
  1171. return
  1172. }
  1173. u := ctx.User
  1174. // Make sure only requests for users who are eligible to change their password via
  1175. // this method passes through
  1176. if !u.MustChangePassword {
  1177. ctx.ServerError("MustUpdatePassword", errors.New("cannot update password.. Please visit the settings page"))
  1178. return
  1179. }
  1180. if form.Password != form.Retype {
  1181. ctx.Data["Err_Password"] = true
  1182. ctx.RenderWithErr(ctx.Tr("form.password_not_match"), tplMustChangePassword, &form)
  1183. return
  1184. }
  1185. if len(form.Password) < setting.MinPasswordLength {
  1186. ctx.Data["Err_Password"] = true
  1187. ctx.RenderWithErr(ctx.Tr("auth.password_too_short", setting.MinPasswordLength), tplMustChangePassword, &form)
  1188. return
  1189. }
  1190. var err error
  1191. if u.Salt, err = models.GetUserSalt(); err != nil {
  1192. ctx.ServerError("UpdateUser", err)
  1193. return
  1194. }
  1195. u.HashPassword(form.Password)
  1196. u.MustChangePassword = false
  1197. if err := models.UpdateUserCols(u, "must_change_password", "passwd", "salt"); err != nil {
  1198. ctx.ServerError("UpdateUser", err)
  1199. return
  1200. }
  1201. ctx.Flash.Success(ctx.Tr("settings.change_password_success"))
  1202. log.Trace("User updated password: %s", u.Name)
  1203. if redirectTo := ctx.GetCookie("redirect_to"); len(redirectTo) > 0 && !util.IsExternalURL(redirectTo) {
  1204. ctx.SetCookie("redirect_to", "", -1, setting.AppSubURL)
  1205. ctx.RedirectToFirst(redirectTo)
  1206. return
  1207. }
  1208. ctx.Redirect(setting.AppSubURL + "/")
  1209. }