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.

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