77 lines
2.3 KiB
Go
77 lines
2.3 KiB
Go
package application
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gorilla/securecookie"
|
|
"goauthentik.io/internal/outpost/proxyv2/constants"
|
|
)
|
|
|
|
func (a *Application) handleRedirect(rw http.ResponseWriter, r *http.Request) {
|
|
newState := base64.RawStdEncoding.EncodeToString(securecookie.GenerateRandomKey(32))
|
|
s, err := a.sessions.Get(r, constants.SeesionName)
|
|
if err != nil {
|
|
s.Values[constants.SessionOAuthState] = []string{}
|
|
}
|
|
state, ok := s.Values[constants.SessionOAuthState].([]string)
|
|
if !ok {
|
|
s.Values[constants.SessionOAuthState] = []string{}
|
|
state = []string{}
|
|
}
|
|
s.Values[constants.SessionOAuthState] = append(state, newState)
|
|
err = s.Save(r, rw)
|
|
if err != nil {
|
|
a.log.WithError(err).Warning("failed to save session")
|
|
}
|
|
if loop, ok := s.Values[constants.SessionLoopDetection]; ok {
|
|
if loop.(int) > 10 {
|
|
rw.WriteHeader(http.StatusBadRequest)
|
|
a.ErrorPage(rw, r, "Detected redirect loop, make sure /akprox is accessible without authentication.")
|
|
return
|
|
}
|
|
}
|
|
http.Redirect(rw, r, a.oauthConfig.AuthCodeURL(newState), http.StatusFound)
|
|
}
|
|
|
|
func (a *Application) handleCallback(rw http.ResponseWriter, r *http.Request) {
|
|
s, _ := a.sessions.Get(r, constants.SeesionName)
|
|
state, ok := s.Values[constants.SessionOAuthState]
|
|
if !ok {
|
|
a.log.Warning("No state saved in session")
|
|
http.Redirect(rw, r, a.proxyConfig.ExternalHost, http.StatusFound)
|
|
return
|
|
}
|
|
claims, err := a.redeemCallback(r, state.([]string))
|
|
if err != nil {
|
|
a.log.WithError(err).Warning("failed to redeem code")
|
|
rw.WriteHeader(400)
|
|
// To prevent the user from just refreshing and cause more errors, delete
|
|
// the state from the session
|
|
delete(s.Values, constants.SessionOAuthState)
|
|
err := s.Save(r, rw)
|
|
if err != nil {
|
|
a.log.WithError(err).Warning("failed to save session")
|
|
rw.WriteHeader(400)
|
|
return
|
|
}
|
|
return
|
|
}
|
|
s.Options.MaxAge = int(time.Until(time.Unix(int64(claims.Exp), 0)).Seconds())
|
|
s.Values[constants.SessionClaims] = &claims
|
|
err = s.Save(r, rw)
|
|
if err != nil {
|
|
a.log.WithError(err).Warning("failed to save session")
|
|
rw.WriteHeader(400)
|
|
return
|
|
}
|
|
redirect := a.proxyConfig.ExternalHost
|
|
redirectR, ok := s.Values[constants.SessionRedirect]
|
|
if ok {
|
|
a.log.WithField("redirect", redirectR).Trace("got final redirect from session")
|
|
redirect = redirectR.(string)
|
|
}
|
|
http.Redirect(rw, r, redirect, http.StatusFound)
|
|
}
|