Add cast error handling

auth0 · Sep 1, 2022 · d482642 · d482642
1 parent f8fbc53
commit d482642
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/examples/http-example/main.go b/examples/http-example/main.go
@@ -28,7 +28,11 @@ func (c *CustomClaimsExample) Validate(ctx context.Context) error {
 }
 
 var handler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-	claims, _ := r.Context().Value(jwtmiddleware.ContextKey{}).(*validator.ValidatedClaims)
+	claims, ok := r.Context().Value(jwtmiddleware.ContextKey{}).(*validator.ValidatedClaims)
+	if !ok {
+		http.Error(w, "failed to get validated claims", http.StatusInternalServerError)
+		return
+	}
 
 	payload, err := json.Marshal(claims)
 	if err != nil {

diff --git a/examples/http-jwks-example/main.go b/examples/http-jwks-example/main.go
@@ -13,7 +13,11 @@ import (
 )
 
 var handler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-	claims, _ := r.Context().Value(jwtmiddleware.ContextKey{}).(*validator.ValidatedClaims)
+	claims, ok := r.Context().Value(jwtmiddleware.ContextKey{}).(*validator.ValidatedClaims)
+	if !ok {
+		http.Error(w, "failed to get validated claims", http.StatusInternalServerError)
+		return
+	}
 
 	payload, err := json.Marshal(claims)
 	if err != nil {