authenticate: fix branding for webauthn device registration page (#3572)

pomerium · Aug 24, 2022 · e9e52d8 · e9e52d8
1 parent ce818b3
commit e9e52d8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/authenticate/handlers.go b/authenticate/handlers.go
@@ -744,6 +744,7 @@ func (a *Authenticate) getWebauthnState(ctx context.Context) (*webauthn.State, e
 		SessionState:            ss,
 		SessionStore:            state.sessionStore,
 		RelyingParty:            state.webauthnRelyingParty,
+		BrandingOptions:         a.options.Load().BrandingOptions,
 	}, nil
 }
 

diff --git a/authenticate/handlers/webauthn/webauthn.go b/authenticate/handlers/webauthn/webauthn.go
@@ -55,6 +55,7 @@ type State struct {
 	SessionState            *sessions.State
 	SessionStore            sessions.SessionStore
 	SharedKey               []byte
+	BrandingOptions         httputil.BrandingOptions
 }
 
 // A StateProvider provides state for the handler.
@@ -398,11 +399,13 @@ func (h *Handler) handleView(w http.ResponseWriter, r *http.Request, state *Stat
 		return err
 	}
 
-	return ui.ServePage(w, r, "WebAuthnRegistration", map[string]interface{}{
+	m := map[string]interface{}{
 		"creationOptions": creationOptions,
 		"requestOptions":  requestOptions,
 		"selfUrl":         r.URL.String(),
-	})
+	}
+	httputil.AddBrandingOptionsToMap(m, state.BrandingOptions)
+	return ui.ServePage(w, r, "WebAuthnRegistration", m)
 }
 
 func (h *Handler) saveSessionAndRedirect(w http.ResponseWriter, r *http.Request, state *State, rawRedirectURI string) error {