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

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