traffic_ops/traffic_ops_golang/login/login.go (598 lines of code) (raw):
package login
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
import (
"bytes"
"context"
"crypto/rand"
"database/sql"
"encoding/base64"
"encoding/json"
"errors"
"fmt"
"html/template"
"net/http"
"net/url"
"path/filepath"
"time"
"github.com/apache/trafficcontrol/v8/lib/go-log"
"github.com/apache/trafficcontrol/v8/lib/go-rfc"
"github.com/apache/trafficcontrol/v8/lib/go-tc"
"github.com/apache/trafficcontrol/v8/traffic_ops/traffic_ops_golang/api"
"github.com/apache/trafficcontrol/v8/traffic_ops/traffic_ops_golang/auth"
"github.com/apache/trafficcontrol/v8/traffic_ops/traffic_ops_golang/config"
"github.com/apache/trafficcontrol/v8/traffic_ops/traffic_ops_golang/tocookie"
"github.com/jmoiron/sqlx"
"github.com/lestrrat-go/jwx/jwa"
"github.com/lestrrat-go/jwx/jwk"
"github.com/lestrrat-go/jwx/jwt"
)
type emailFormatter struct {
From rfc.EmailAddress
To rfc.EmailAddress
InstanceName string
ResetURL string
Token string
}
const instanceNameQuery = `
SELECT value
FROM parameter
WHERE name='tm.instance_name' AND
config_file=$1
`
const userQueryByEmail = `SELECT EXISTS(SELECT * FROM tm_user WHERE email=$1)`
const setTokenQuery = `UPDATE tm_user SET token=$1 WHERE email=$2`
// UpdateLoginTimeQuery is meant to only update the last_authenticated field once per minute in order to avoid row-locking when the same user logs in frequently.
const UpdateLoginTimeQuery = `UPDATE tm_user SET last_authenticated = NOW() WHERE username=$1 AND (last_authenticated IS NULL OR last_authenticated < NOW() - INTERVAL '1 MINUTE')`
const defaultCookieDuration = 6 * time.Hour
var resetPasswordEmailTemplate = template.Must(template.New("Password Reset Email").Parse("From: {{.From.Address.Address}}\r" + `
To: {{.To.Address.Address}}` + "\r" + `
Content-Type: text/html` + "\r" + `
Subject: {{.InstanceName}} Password Reset Request` + "\r\n\r" + `
<!DOCTYPE html>
<html lang="en">
<head>
<title>{{.InstanceName}} Password Reset Request</title>
<meta charset="utf-8"/>
<style>
.button_link {
display: block;
width: 130px;
height: 35px;
background: #2682AF;
padding: 5px;
text-align: center;
border-radius: 5px;
color: white;
font-weight: bold;
text-decoration: none;
cursor: pointer;
}
</style>
</head>
<body>
<main>
<p>Someone has requested to change your password for the {{.InstanceName}}. If you requested this change, please click the link below and change your password. Otherwise, you can disregard this email.</p>
<p><a class="button_link" target="_blank" href="{{.ResetURL}}?token={{.Token}}">Click to Reset Your Password</a></p>
</main>
<footer>
<p>Thank you,<br/>
The {{.InstanceName}} Team</p>
</footer>
</body>
</html>
`))
func clientCertAuthentication(w http.ResponseWriter, r *http.Request, db *sqlx.DB, cfg config.Config, dbCtx context.Context, cancelTx context.CancelFunc, form *auth.PasswordForm, authenticated bool) bool {
// No certs provided by the client. Skip to form authentication
if r.TLS == nil || len(r.TLS.PeerCertificates) == 0 {
return false
}
// If no configuration is set, skip to form auth
if cfg.ClientCertAuth == nil || len(cfg.ClientCertAuth.RootCertsDir) == 0 {
return false
}
// Perform certificate verification to ensure it is valid against Root CAs
err := auth.VerifyClientCertificate(r, cfg.ClientCertAuth.RootCertsDir, cfg.Insecure)
if err != nil {
log.Warnf("client cert auth: error attempting to verify client provided TLS certificate. err: %s\n", err)
return false
}
// Client provided a verified certificate. Extract UID value.
if username, err := auth.ParseClientCertificateUID(r.TLS.PeerCertificates[0]); err != nil {
log.Errorf("parsing client certificate: %s\n", err)
return false
} else {
form.Username = username
}
// Check if user exists locally (TODB) and has a role.
var blockingErr error
authenticated, err, blockingErr = auth.CheckLocalUserIsAllowed(form.Username, db, dbCtx)
if blockingErr != nil {
api.HandleErr(w, r, nil, http.StatusServiceUnavailable, nil, fmt.Errorf("error checking local user has role: %s", blockingErr.Error()))
return false
}
if err != nil {
log.Warnf("client cert auth: checking local user: %s\n", err)
}
// Check LDAP if enabled
if !authenticated && cfg.LDAPEnabled {
_, authenticated, err = auth.LookupUserDN(form.Username, cfg.ConfigLDAP)
if err != nil {
log.Warnf("Client Cert Auth: checking ldap user: %s\n", err)
}
}
return authenticated
}
// LoginHandler first attempts to verify and parse user information from an optionally
// provided client TLS certificate. If it fails at any point, it will fall back and
// continue with the standard submitted form authentication.
func LoginHandler(db *sqlx.DB, cfg config.Config) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
defer r.Body.Close()
authenticated := false
form := auth.PasswordForm{}
var resp tc.Alerts
dbCtx, cancelTx := context.WithTimeout(r.Context(), time.Duration(cfg.DBQueryTimeoutSeconds)*time.Second)
defer cancelTx()
// Attempt to perform client certificate authentication. If fails, goto standard form auth. If the
// certificate was verified, has a UID, and the UID matches an existing user we consider this to
// be a successful login.
authenticated = clientCertAuthentication(w, r, db, cfg, dbCtx, cancelTx, &form, authenticated)
// Failed certificate-based auth, perform standard form auth
if !authenticated {
log.Infof("user %s could not be successfully authenticated using client certificates", form.Username)
// Perform form authentication
if err := json.NewDecoder(r.Body).Decode(&form); err != nil {
api.HandleErr(w, r, nil, http.StatusBadRequest, err, nil)
return
}
if form.Username == "" || form.Password == "" {
api.HandleErr(w, r, nil, http.StatusBadRequest, errors.New("username and password are required"), nil)
return
}
// Check if user exists and has a role
userAllowed, err, blockingErr := auth.CheckLocalUserIsAllowed(form.Username, db, dbCtx)
if blockingErr != nil {
api.HandleErr(w, r, nil, http.StatusServiceUnavailable, nil, fmt.Errorf("error checking local user has role: %s", blockingErr.Error()))
return
}
if err != nil {
log.Errorf("checking local user: %s\n", err)
}
// User w/ role does not exist, return unauthorized
if !userAllowed {
resp = tc.CreateAlerts(tc.ErrorLevel, "Invalid username or password.")
w.WriteHeader(http.StatusUnauthorized)
api.WriteRespRaw(w, r, resp)
return
}
// Check local DB or LDAP
authenticated, err, blockingErr = auth.CheckLocalUserPassword(form, db, dbCtx)
if blockingErr != nil {
api.HandleErr(w, r, nil, http.StatusServiceUnavailable, nil, fmt.Errorf("error checking local user password: %s", blockingErr.Error()))
return
}
if err != nil {
log.Errorf("checking local user password: %s\n", err)
}
if authenticated {
log.Infof("user %s successfully authenticated using username/ password", form.Username)
} else if cfg.LDAPEnabled {
var ldapErr error
authenticated, ldapErr = auth.CheckLDAPUser(form, cfg.ConfigLDAP)
if ldapErr != nil {
log.Infof("user %s could not be successfully authenticated using LDAP", form.Username)
log.Errorf("checking ldap user: %s\n", ldapErr.Error())
} else {
log.Infof("user %s successfully authenticated using LDAP", form.Username)
}
}
} else {
log.Infof("user %s successfully authenticated using client certificates", form.Username)
}
// Failed to authenticate in either local DB or LDAP, return unauthorized
if !authenticated {
log.Infof("user %s could not be successfully authenticated using username/ password", form.Username)
resp = tc.CreateAlerts(tc.ErrorLevel, "Invalid username or password.")
w.WriteHeader(http.StatusUnauthorized)
api.WriteRespRaw(w, r, resp)
return
}
// Successful authentication, write cookie and return
httpCookie := tocookie.GetCookie(form.Username, defaultCookieDuration, cfg.Secrets[0])
http.SetCookie(w, httpCookie)
var jwtToken jwt.Token
var jwtSigned []byte
jwtBuilder := jwt.NewBuilder()
emptyConf := config.CdniConf{}
if cfg.Cdni != nil && *cfg.Cdni != emptyConf {
ucdn, err := auth.GetUserUcdn(form, db, dbCtx)
if err != nil {
// log but do not error out since this is optional in the JWT for CDNi integration
log.Errorf("getting ucdn for user %s: %v", form.Username, err)
}
jwtBuilder.Claim(jwt.IssuerKey, ucdn)
jwtBuilder.Claim(jwt.AudienceKey, cfg.Cdni.DCdnId)
}
jwtBuilder.Claim(jwt.ExpirationKey, httpCookie.Expires.Unix())
jwtBuilder.Claim(api.MojoCookie, httpCookie.Value)
jwtToken, err := jwtBuilder.Build()
if err != nil {
api.HandleErr(w, r, nil, http.StatusInternalServerError, nil, fmt.Errorf("building token: %s", err))
return
}
jwtSigned, err = jwt.Sign(jwtToken, jwa.HS256, []byte(cfg.Secrets[0]))
if err != nil {
api.HandleErr(w, r, nil, http.StatusInternalServerError, nil, err)
return
}
http.SetCookie(w, &http.Cookie{
Name: rfc.AccessToken,
Value: string(jwtSigned),
Path: "/",
MaxAge: httpCookie.MaxAge,
Expires: httpCookie.Expires,
HttpOnly: true, // prevents the cookie being accessed by Javascript. DO NOT remove, security vulnerability
})
// If all's well until here, then update last authenticated time
tx, txErr := db.BeginTx(dbCtx, nil)
if txErr != nil {
api.HandleErr(w, r, tx, http.StatusInternalServerError, nil, fmt.Errorf("beginning transaction: %w", txErr))
return
}
defer func() {
if err := tx.Commit(); err != nil && err != sql.ErrTxDone {
log.Errorf("committing transaction: %s", err)
}
}()
_, dbErr := tx.Exec(UpdateLoginTimeQuery, form.Username)
if dbErr != nil {
log.Errorf("unable to update authentication time for a given user: %s\n", dbErr.Error())
}
resp = tc.CreateAlerts(tc.SuccessLevel, "Successfully logged in.")
w.WriteHeader(http.StatusOK)
api.WriteRespRaw(w, r, resp)
}
}
func TokenLoginHandler(db *sqlx.DB, cfg config.Config) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
defer r.Body.Close()
var t tc.UserToken
if err := json.NewDecoder(r.Body).Decode(&t); err != nil {
log.Infof("user could not be successfully authenticated using token")
api.HandleErr(w, r, nil, http.StatusBadRequest, fmt.Errorf("Invalid request: %v", err), nil)
return
}
tokenMatches, username, err := auth.CheckLocalUserToken(t.Token, db, time.Duration(cfg.DBQueryTimeoutSeconds)*time.Second)
if err != nil {
sysErr := fmt.Errorf("Checking token: %v", err)
errCode := http.StatusInternalServerError
log.Infof("user could not be successfully authenticated using token")
api.HandleErr(w, r, nil, errCode, nil, sysErr)
return
} else if !tokenMatches {
userErr := errors.New("Invalid token. Please contact your administrator.")
errCode := http.StatusUnauthorized
log.Infof("user could not be successfully authenticated using token")
api.HandleErr(w, r, nil, errCode, userErr, nil)
return
}
httpCookie := tocookie.GetCookie(username, defaultCookieDuration, cfg.Secrets[0])
http.SetCookie(w, httpCookie)
respBts, err := json.Marshal(tc.CreateAlerts(tc.SuccessLevel, "Successfully logged in."))
if err != nil {
sysErr := fmt.Errorf("Marshaling response: %v", err)
errCode := http.StatusInternalServerError
log.Infof("user could not be successfully authenticated using token")
api.HandleErr(w, r, nil, errCode, nil, sysErr)
return
}
_, dbErr := db.Exec(UpdateLoginTimeQuery, username)
if dbErr != nil {
dbErr = fmt.Errorf("unable to update authentication time for user '%s': %w", username, dbErr)
log.Infof("user %s could not be successfully authenticated using token", username)
api.HandleErr(w, r, nil, http.StatusInternalServerError, nil, dbErr)
return
}
w.Header().Set(rfc.ContentType, rfc.ApplicationJSON)
api.WriteAndLogErr(w, r, append(respBts, '\n'))
// TODO: afaik, Perl never clears these tokens. They should be reset to NULL on login, I think.
}
}
type whitelist struct {
urls []string
}
func (w *whitelist) IsAllowed(u string) bool {
for _, listing := range w.urls {
if listing == "" {
continue
}
urlParsed, err := url.Parse(u)
if err != nil {
return false
}
matched, err := filepath.Match(listing, urlParsed.Hostname())
if err != nil {
return false
}
if matched {
return true
}
}
return false
}
type jwksFetch struct {
ar *jwk.AutoRefresh
wl jwk.Whitelist
}
func (f *jwksFetch) Fetch(u string) (jwk.Set, error) {
// Note: all calls to jwk.AutoRefresh should be conccurency-safe
if !f.ar.IsRegistered(u) {
f.ar.Configure(u, jwk.WithFetchWhitelist(f.wl))
}
return f.ar.Fetch(context.TODO(), u)
}
var jwksFetcher *jwksFetch
// OauthLoginHandler accepts a JSON web token previously obtained from an OAuth provider, decodes it, validates it, authorizes the user against the database, and returns the login result as either an error or success message
func OauthLoginHandler(db *sqlx.DB, cfg config.Config) http.HandlerFunc {
// The jwk.AutoRefresh and jwk.Whitelist objects only get created once.
// They are shared between all handlers
// Note: This assumes two things:
// 1) that the cfg.ConfigTrafficOpsGolang.WhitelistedOAuthUrls is not updated once it has been initialized
// 2) OauthLoginHandler is not called conccurently
if jwksFetcher == nil {
ar := jwk.NewAutoRefresh(context.TODO())
wl := &whitelist{urls: cfg.ConfigTrafficOpsGolang.WhitelistedOAuthUrls}
jwksFetcher = &jwksFetch{
ar: ar,
wl: wl,
}
}
return func(w http.ResponseWriter, r *http.Request) {
defer r.Body.Close()
resp := struct {
tc.Alerts
}{}
form := auth.PasswordForm{}
parameters := struct {
AuthCodeTokenUrl string `json:"authCodeTokenUrl"`
Code string `json:"code"`
ClientId string `json:"clientId"`
RedirectUri string `json:"redirectUri"`
}{}
if err := json.NewDecoder(r.Body).Decode(¶meters); err != nil {
log.Infof("user %s could not be successfully authenticated using SSO", form.Username)
api.HandleErr(w, r, nil, http.StatusBadRequest, err, nil)
return
}
matched, err := VerifyUrlOnWhiteList(parameters.AuthCodeTokenUrl, cfg.ConfigTrafficOpsGolang.WhitelistedOAuthUrls)
if err != nil {
log.Infof("user %s could not be successfully authenticated using SSO", form.Username)
api.HandleErr(w, r, nil, http.StatusInternalServerError, nil, err)
return
}
if !matched {
log.Infof("user %s could not be successfully authenticated using SSO", form.Username)
api.HandleErr(w, r, nil, http.StatusForbidden, nil, errors.New("Key URL from token is not included in the whitelisted urls. Received: "+parameters.AuthCodeTokenUrl))
return
}
data := url.Values{}
data.Add("code", parameters.Code)
data.Add("client_id", parameters.ClientId)
data.Add("grant_type", "authorization_code") // Required by RFC6749 section 4.1.3
data.Add("redirect_uri", parameters.RedirectUri)
req, err := http.NewRequest(http.MethodPost, parameters.AuthCodeTokenUrl, bytes.NewBufferString(data.Encode()))
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
if cfg.OAuthClientSecret != "" {
req.Header.Set("Authorization", "Basic "+base64.StdEncoding.EncodeToString([]byte(parameters.ClientId+":"+cfg.OAuthClientSecret))) // per RFC6749 section 2.3.1
}
if err != nil {
log.Infof("user %s could not be successfully authenticated using SSO", form.Username)
api.HandleErr(w, r, nil, http.StatusInternalServerError, nil, fmt.Errorf("obtaining token using code from oauth provider: %w", err))
return
}
client := http.Client{
Timeout: 30 * time.Second,
}
response, err := client.Do(req)
if err != nil {
log.Infof("user %s could not be successfully authenticated using SSO", form.Username)
api.HandleErr(w, r, nil, http.StatusInternalServerError, nil, fmt.Errorf("getting an http client: %w", err))
return
}
defer response.Body.Close()
buf := new(bytes.Buffer)
buf.ReadFrom(response.Body)
encodedToken := ""
var result map[string]interface{}
if err := json.Unmarshal(buf.Bytes(), &result); err != nil {
log.Warnf("Error parsing JSON response from OAuth: %s", err)
encodedToken = buf.String()
} else if _, ok := result[rfc.IDToken]; !ok {
sysErr := fmt.Errorf("Missing access token in response: %s\n", buf.String())
usrErr := errors.New("Bad response from OAuth2.0 provider")
api.HandleErr(w, r, nil, http.StatusBadGateway, usrErr, sysErr)
return
} else {
switch t := result[rfc.IDToken].(type) {
case string:
encodedToken = result[rfc.IDToken].(string)
default:
sysErr := fmt.Errorf("Incorrect type of access_token! Expected 'string', got '%v'\n", t)
usrErr := errors.New("Bad response from OAuth2.0 provider")
log.Infof("user %s could not be successfully authenticated using SSO", form.Username)
api.HandleErr(w, r, nil, http.StatusBadGateway, usrErr, sysErr)
return
}
}
if encodedToken == "" {
log.Infof("user %s could not be successfully authenticated using SSO", form.Username)
api.HandleErr(w, r, nil, http.StatusBadRequest, errors.New("Token not found in request but is required"), nil)
return
}
var decodedToken jwt.Token
if decodedToken, err = jwt.Parse(
[]byte(encodedToken),
jwt.WithVerifyAuto(true),
jwt.WithJWKSetFetcher(jwksFetcher),
); err != nil {
if decodedToken, err = jwt.Parse(
[]byte(encodedToken),
jwt.WithVerifyAuto(false),
jwt.WithJWKSetFetcher(jwksFetcher),
); err != nil {
log.Infof("user %s could not be successfully authenticated using SSO", form.Username)
api.HandleErr(w, r, nil, http.StatusInternalServerError, nil, fmt.Errorf("error decoding token with message: %w", err))
return
}
}
var userIDInterface interface{}
var userID string
var ok bool
if cfg.OAuthUserAttribute != "" {
attributes := decodedToken.PrivateClaims()
if userIDInterface, ok = attributes[cfg.OAuthUserAttribute]; !ok {
log.Infof("user %s could not be successfully authenticated using SSO", form.Username)
api.HandleErr(w, r, nil, http.StatusInternalServerError, nil, fmt.Errorf("Non-existent OAuth attribute : %s", cfg.OAuthUserAttribute))
return
}
userID = userIDInterface.(string)
} else {
userID = decodedToken.Subject()
}
form.Username = userID
dbCtx, cancelTx := context.WithTimeout(r.Context(), time.Duration(cfg.DBQueryTimeoutSeconds)*time.Second)
defer cancelTx()
userAllowed, err, blockingErr := auth.CheckLocalUserIsAllowed(form.Username, db, dbCtx)
if blockingErr != nil {
log.Infof("user %s could not be successfully authenticated using SSO", form.Username)
api.HandleErr(w, r, nil, http.StatusServiceUnavailable, nil, fmt.Errorf("error checking local user password: %s\n", blockingErr.Error()))
return
}
if err != nil {
log.Errorf("checking local user: %s\n", err)
}
if userAllowed {
_, dbErr := db.Exec(UpdateLoginTimeQuery, form.Username)
if dbErr != nil {
log.Infof("user %s could not be successfully authenticated using SSO", form.Username)
dbErr = fmt.Errorf("unable to update authentication time for user '%s': %w", form.Username, dbErr)
api.HandleErr(w, r, nil, http.StatusInternalServerError, nil, dbErr)
return
}
httpCookie := tocookie.GetCookie(userID, defaultCookieDuration, cfg.Secrets[0])
http.SetCookie(w, httpCookie)
resp = struct {
tc.Alerts
}{tc.CreateAlerts(tc.SuccessLevel, "Successfully logged in.")}
log.Infof("user %s successfully authenticated using SSO", form.Username)
} else {
resp = struct {
tc.Alerts
}{tc.CreateAlerts(tc.ErrorLevel, "Invalid username or password.")}
log.Infof("user %s could not be successfully authenticated using SSO", form.Username)
}
respBts, err := json.Marshal(resp)
if err != nil {
api.HandleErr(w, r, nil, http.StatusInternalServerError, nil, fmt.Errorf("encoding response: %w", err))
return
}
w.Header().Set(rfc.ContentType, rfc.ApplicationJSON)
if !userAllowed {
w.WriteHeader(http.StatusForbidden)
}
fmt.Fprintf(w, "%s", respBts)
}
}
func VerifyUrlOnWhiteList(urlString string, whiteListedUrls []string) (bool, error) {
for _, listing := range whiteListedUrls {
if listing == "" {
continue
}
urlParsed, err := url.Parse(urlString)
if err != nil {
return false, err
}
matched, err := filepath.Match(listing, urlParsed.Hostname())
if err != nil {
return false, err
}
if matched {
return true, nil
}
}
return false, nil
}
func generateToken() (string, error) {
var t = make([]byte, 16)
_, err := rand.Read(t)
if err != nil {
return "", err
}
t[6] = (t[6] & 0x0f) | 0x40
t[8] = (t[8] & 0x3f) | 0x80
return fmt.Sprintf("%x-%x-%x-%x-%x", t[0:4], t[4:6], t[6:8], t[8:10], t[10:]), nil
}
func setToken(addr rfc.EmailAddress, tx *sql.Tx) (string, error) {
token, err := generateToken()
if err != nil {
return "", err
}
if _, err = tx.Exec(setTokenQuery, token, addr.Address.Address); err != nil {
return "", err
}
return token, nil
}
func createMsg(addr rfc.EmailAddress, t string, db *sqlx.DB, c config.ConfigPortal) ([]byte, error) {
var instanceName string
row := db.QueryRow(instanceNameQuery, tc.GlobalConfigFileName)
if err := row.Scan(&instanceName); err != nil {
return nil, err
}
f := emailFormatter{
From: c.EmailFrom,
To: addr,
Token: t,
InstanceName: instanceName,
ResetURL: c.BaseURL.String() + c.PasswdResetPath,
}
var tmpl bytes.Buffer
if err := resetPasswordEmailTemplate.Execute(&tmpl, &f); err != nil {
return nil, err
}
return tmpl.Bytes(), nil
}
func ResetPassword(db *sqlx.DB, cfg config.Config) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
var userErr, sysErr error
var errCode int
tx, err := db.Begin()
if err != nil {
sysErr = fmt.Errorf("Beginning transaction: %v", err)
errCode = http.StatusInternalServerError
api.HandleErr(w, r, tx, errCode, nil, sysErr)
return
}
defer r.Body.Close()
var req tc.UserPasswordResetRequest
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
userErr = fmt.Errorf("Malformed request: %v", err)
errCode = http.StatusBadRequest
api.HandleErr(w, r, tx, errCode, userErr, nil)
return
}
row := tx.QueryRow(userQueryByEmail, req.Email.Address.Address)
var userExists bool
if err := row.Scan(&userExists); err != nil {
sysErr = fmt.Errorf("Checking for existence of user with email '%s': %v", req.Email.String(), err)
errCode = http.StatusInternalServerError
api.HandleErr(w, r, tx, errCode, nil, sysErr)
return
}
if !userExists {
// TODO: consider concealing database state from unauthenticated parties;
// this should maybe just return a 2XX w/ success message at this point?
userErr = fmt.Errorf("No account with the email address '%s' was found!", req.Email.Address.Address)
errCode = http.StatusNotFound
api.HandleErr(w, r, tx, errCode, userErr, nil)
return
}
token, err := setToken(req.Email, tx)
if err != nil {
sysErr = fmt.Errorf("Failed to generate and insert UUID: %v", err)
errCode = http.StatusInternalServerError
api.HandleErr(w, r, tx, errCode, nil, sysErr)
return
}
tx.Commit()
msg, err := createMsg(req.Email, token, db, cfg.ConfigPortal)
if err != nil {
sysErr = fmt.Errorf("Failed to create email message: %v", err)
errCode = http.StatusInternalServerError
api.HandleErr(w, r, nil, errCode, nil, sysErr)
return
}
log.Debugf("Sending password reset email to %s", req.Email)
if errCode, userErr, sysErr = api.SendMail(req.Email, msg, &cfg); userErr != nil || sysErr != nil {
api.HandleErr(w, r, nil, errCode, userErr, sysErr)
return
}
alerts := tc.CreateAlerts(tc.SuccessLevel, "Password reset email sent")
respBts, err := json.Marshal(alerts)
if err != nil {
userErr = errors.New("Email was sent, but an error occurred afterward")
sysErr = fmt.Errorf("Marshaling response: %v", err)
errCode = http.StatusInternalServerError
api.HandleErr(w, r, nil, errCode, userErr, sysErr)
return
}
w.Header().Set(rfc.ContentType, rfc.ApplicationJSON)
api.WriteAndLogErr(w, r, append(respBts, '\n'))
}
}