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

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