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

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