fix(auth): add PKCE support for public OIDC client
This commit is contained in:
parent
9922349351
commit
743f745cbf
1 changed files with 37 additions and 16 deletions
|
|
@ -1,6 +1,7 @@
|
||||||
package auth
|
package auth
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"crypto/sha256"
|
||||||
"encoding/base64"
|
"encoding/base64"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
|
@ -38,6 +39,11 @@ func (h *OIDCHandler) LoginRedirect(w http.ResponseWriter, r *http.Request) {
|
||||||
state := generateToken(16)
|
state := generateToken(16)
|
||||||
nonce := generateToken(16)
|
nonce := generateToken(16)
|
||||||
|
|
||||||
|
// PKCE: generate code verifier and challenge
|
||||||
|
verifier := generateToken(32)
|
||||||
|
challenge := pkceChallenge(verifier)
|
||||||
|
|
||||||
|
// Store state + verifier in cookies
|
||||||
http.SetCookie(w, &http.Cookie{
|
http.SetCookie(w, &http.Cookie{
|
||||||
Name: "oidc_state",
|
Name: "oidc_state",
|
||||||
Value: state,
|
Value: state,
|
||||||
|
|
@ -46,14 +52,23 @@ func (h *OIDCHandler) LoginRedirect(w http.ResponseWriter, r *http.Request) {
|
||||||
HttpOnly: true,
|
HttpOnly: true,
|
||||||
SameSite: http.SameSiteLaxMode,
|
SameSite: http.SameSiteLaxMode,
|
||||||
})
|
})
|
||||||
|
http.SetCookie(w, &http.Cookie{
|
||||||
|
Name: "oidc_verifier",
|
||||||
|
Value: verifier,
|
||||||
|
Path: "/",
|
||||||
|
MaxAge: 300,
|
||||||
|
HttpOnly: true,
|
||||||
|
SameSite: http.SameSiteLaxMode,
|
||||||
|
})
|
||||||
|
|
||||||
authURL := fmt.Sprintf(
|
authURL := fmt.Sprintf(
|
||||||
"%s/api/oidc/authorize?response_type=code&client_id=%s&redirect_uri=%s&scope=openid+profile+email&state=%s&nonce=%s",
|
"%s/api/oidc/authorize?response_type=code&client_id=%s&redirect_uri=%s&scope=openid+profile+email&state=%s&nonce=%s&code_challenge=%s&code_challenge_method=S256",
|
||||||
h.config.IssuerURL,
|
h.config.IssuerURL,
|
||||||
url.QueryEscape(h.config.ClientID),
|
url.QueryEscape(h.config.ClientID),
|
||||||
url.QueryEscape(h.config.RedirectURL),
|
url.QueryEscape(h.config.RedirectURL),
|
||||||
state,
|
state,
|
||||||
nonce,
|
nonce,
|
||||||
|
challenge,
|
||||||
)
|
)
|
||||||
|
|
||||||
http.Redirect(w, r, authURL, http.StatusFound)
|
http.Redirect(w, r, authURL, http.StatusFound)
|
||||||
|
|
@ -78,14 +93,16 @@ func (h *OIDCHandler) Callback(w http.ResponseWriter, r *http.Request) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Clear the state cookie
|
// Get PKCE verifier from cookie
|
||||||
http.SetCookie(w, &http.Cookie{
|
verifierCookie, _ := r.Cookie("oidc_verifier")
|
||||||
Name: "oidc_state",
|
verifier := ""
|
||||||
Value: "",
|
if verifierCookie != nil {
|
||||||
Path: "/",
|
verifier = verifierCookie.Value
|
||||||
MaxAge: -1,
|
}
|
||||||
HttpOnly: true,
|
|
||||||
})
|
// Clear state cookies
|
||||||
|
http.SetCookie(w, &http.Cookie{Name: "oidc_state", Value: "", Path: "/", MaxAge: -1, HttpOnly: true})
|
||||||
|
http.SetCookie(w, &http.Cookie{Name: "oidc_verifier", Value: "", Path: "/", MaxAge: -1, HttpOnly: true})
|
||||||
|
|
||||||
// Get code from URL query (GET) or form body (POST)
|
// Get code from URL query (GET) or form body (POST)
|
||||||
code := r.URL.Query().Get("code")
|
code := r.URL.Query().Get("code")
|
||||||
|
|
@ -97,8 +114,8 @@ func (h *OIDCHandler) Callback(w http.ResponseWriter, r *http.Request) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Exchange code for tokens
|
// Exchange code for tokens (with PKCE verifier)
|
||||||
username, err := h.exchangeCode(code)
|
username, err := h.exchangeCode(code, verifier)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
http.Error(w, "token exchange failed: "+err.Error(), http.StatusInternalServerError)
|
http.Error(w, "token exchange failed: "+err.Error(), http.StatusInternalServerError)
|
||||||
return
|
return
|
||||||
|
|
@ -123,9 +140,8 @@ func (h *OIDCHandler) Callback(w http.ResponseWriter, r *http.Request) {
|
||||||
http.Redirect(w, r, "/", http.StatusFound)
|
http.Redirect(w, r, "/", http.StatusFound)
|
||||||
}
|
}
|
||||||
|
|
||||||
// exchangeCode exchanges an OIDC authorization code for an ID token
|
// exchangeCode exchanges an OIDC authorization code for an ID token.
|
||||||
// and extracts the username (sub claim) from it.
|
func (h *OIDCHandler) exchangeCode(code, verifier string) (string, error) {
|
||||||
func (h *OIDCHandler) exchangeCode(code string) (string, error) {
|
|
||||||
tokenURL := h.config.IssuerURL + "/api/oidc/token"
|
tokenURL := h.config.IssuerURL + "/api/oidc/token"
|
||||||
|
|
||||||
data := url.Values{
|
data := url.Values{
|
||||||
|
|
@ -133,7 +149,7 @@ func (h *OIDCHandler) exchangeCode(code string) (string, error) {
|
||||||
"code": {code},
|
"code": {code},
|
||||||
"redirect_uri": {h.config.RedirectURL},
|
"redirect_uri": {h.config.RedirectURL},
|
||||||
"client_id": {h.config.ClientID},
|
"client_id": {h.config.ClientID},
|
||||||
"client_secret": {h.config.ClientSecret},
|
"code_verifier": {verifier},
|
||||||
}
|
}
|
||||||
|
|
||||||
resp, err := http.PostForm(tokenURL, data)
|
resp, err := http.PostForm(tokenURL, data)
|
||||||
|
|
@ -158,7 +174,6 @@ func (h *OIDCHandler) exchangeCode(code string) (string, error) {
|
||||||
return "", fmt.Errorf("no id_token in response")
|
return "", fmt.Errorf("no id_token in response")
|
||||||
}
|
}
|
||||||
|
|
||||||
// Decode JWT payload (without signature verification for now)
|
|
||||||
username, err := decodeJWTSub(tokenResp.IDToken)
|
username, err := decodeJWTSub(tokenResp.IDToken)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", fmt.Errorf("decode id_token: %w", err)
|
return "", fmt.Errorf("decode id_token: %w", err)
|
||||||
|
|
@ -193,6 +208,12 @@ func decodeJWTSub(token string) (string, error) {
|
||||||
return claims.Sub, nil
|
return claims.Sub, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// pkceChallenge creates a PKCE S256 challenge from a verifier.
|
||||||
|
func pkceChallenge(verifier string) string {
|
||||||
|
h := sha256.Sum256([]byte(verifier))
|
||||||
|
return base64.RawURLEncoding.EncodeToString(h[:])
|
||||||
|
}
|
||||||
|
|
||||||
// AuthGateMiddleware protects routes behind OIDC authentication.
|
// AuthGateMiddleware protects routes behind OIDC authentication.
|
||||||
func (h *OIDCHandler) AuthGateMiddleware(next http.Handler) http.Handler {
|
func (h *OIDCHandler) AuthGateMiddleware(next http.Handler) http.Handler {
|
||||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue