feat: Phase 3.1 - SQS JWT validation against Authentik JWKS
Implements gateway-level JWT validation for SQS requests: - Validates JWT signature against Authentik JWKS - Verifies claims: iss, aud, exp, nbf (with 60s skew) - Checks 'permissions' claim for sqs:read/sqs:write/wildcard - Returns 403 with error details on validation failure - JWKS caching with 15min TTL and auto-refresh on key rotation Architecture: - SQS: Gateway validates JWT (kmsvc code unverified) - MinIO, Temporal: Native JWT support (pass-through) - Memory, IAM: Service-owned JWT validation Integration tests added: - Reject requests without Authorization header (403) - Accept requests with valid JWT from Authentik - Pass through Authorization header unchanged for other services Uses github.com/MicahParks/keyfunc/v2 for JWKS handling: - Automatic refresh every 15 minutes - On-demand refresh if kid not found - Handles RS256 signatures
This commit is contained in:
@@ -0,0 +1,141 @@
|
||||
package auth
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/MicahParks/keyfunc/v2"
|
||||
"github.com/golang-jwt/jwt/v5"
|
||||
)
|
||||
|
||||
// Validator validates JWTs against Authentik JWKS.
|
||||
type Validator struct {
|
||||
issuer string
|
||||
audience string
|
||||
jwks *keyfunc.JWKS
|
||||
}
|
||||
|
||||
// NewValidator creates a new JWT validator for a service.
|
||||
func NewValidator(issuer, audience, jwksURL string) *Validator {
|
||||
// Create JWKS from URL with automatic refresh
|
||||
options := keyfunc.Options{
|
||||
Ctx: context.Background(),
|
||||
RefreshInterval: 15 * time.Minute,
|
||||
RefreshRateLimit: 5 * time.Minute,
|
||||
RefreshTimeout: 10 * time.Second,
|
||||
RefreshErrorHandler: func(err error) {
|
||||
// Log refresh errors but don't fail
|
||||
fmt.Printf("JWKS refresh error for %s: %v\n", issuer, err)
|
||||
},
|
||||
}
|
||||
|
||||
jwks, err := keyfunc.Get(jwksURL, options)
|
||||
if err != nil {
|
||||
panic(fmt.Sprintf("failed to fetch JWKS from %s: %v", jwksURL, err))
|
||||
}
|
||||
|
||||
return &Validator{
|
||||
issuer: issuer,
|
||||
audience: audience,
|
||||
jwks: jwks,
|
||||
}
|
||||
}
|
||||
|
||||
// ValidateBearerToken extracts and validates the Bearer token from Authorization header.
|
||||
// Returns claims on success, error message on failure.
|
||||
func (v *Validator) ValidateBearerToken(authHeader string) (jwt.MapClaims, error) {
|
||||
if authHeader == "" {
|
||||
return nil, fmt.Errorf("missing Authorization header")
|
||||
}
|
||||
|
||||
// Extract token from "Bearer <token>"
|
||||
tokenString := ""
|
||||
if len(authHeader) > 7 && authHeader[:7] == "Bearer " {
|
||||
tokenString = authHeader[7:]
|
||||
} else {
|
||||
return nil, fmt.Errorf("invalid Authorization header format")
|
||||
}
|
||||
|
||||
// Parse and validate
|
||||
claims := jwt.MapClaims{}
|
||||
token, err := jwt.ParseWithClaims(tokenString, claims, v.jwks.Keyfunc)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("token validation failed: %v", err)
|
||||
}
|
||||
|
||||
if !token.Valid {
|
||||
return nil, fmt.Errorf("token is invalid")
|
||||
}
|
||||
|
||||
// Verify required claims
|
||||
now := time.Now()
|
||||
const skew = 60 * time.Second
|
||||
|
||||
// Check exp
|
||||
if exp, ok := claims["exp"].(float64); ok {
|
||||
if time.Now().After(time.Unix(int64(exp), 0).Add(skew)) {
|
||||
return nil, fmt.Errorf("token expired")
|
||||
}
|
||||
}
|
||||
|
||||
// Check nbf (not before)
|
||||
if nbf, ok := claims["nbf"].(float64); ok {
|
||||
if now.Before(time.Unix(int64(nbf), 0).Add(-skew)) {
|
||||
return nil, fmt.Errorf("token not yet valid")
|
||||
}
|
||||
}
|
||||
|
||||
// Check iss (issuer)
|
||||
if iss, ok := claims["iss"].(string); !ok || iss != v.issuer {
|
||||
return nil, fmt.Errorf("invalid issuer: expected %s, got %s", v.issuer, iss)
|
||||
}
|
||||
|
||||
// Check aud (audience)
|
||||
if aud, ok := claims["aud"].(string); !ok || aud != v.audience {
|
||||
return nil, fmt.Errorf("invalid audience: expected %s, got %s", v.audience, aud)
|
||||
}
|
||||
|
||||
return claims, nil
|
||||
}
|
||||
|
||||
// CheckPermissions checks if claims contain required permission(s).
|
||||
// Returns true if any required permission is found or wildcard "*" exists.
|
||||
func (v *Validator) CheckPermissions(claims jwt.MapClaims, required ...string) bool {
|
||||
permsIface, ok := claims["permissions"]
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
|
||||
perms, ok := permsIface.([]interface{})
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
|
||||
for _, perm := range perms {
|
||||
permStr, ok := perm.(string)
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
if permStr == "*" {
|
||||
return true
|
||||
}
|
||||
for _, req := range required {
|
||||
if permStr == req {
|
||||
return true
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
// DecodeToken decodes JWT payload without verification (for debugging/testing).
|
||||
func DecodeToken(tokenString string) (jwt.MapClaims, error) {
|
||||
claims := jwt.MapClaims{}
|
||||
_, _, err := new(jwt.Parser).ParseUnverified(tokenString, claims)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return claims, nil
|
||||
}
|
||||
Reference in New Issue
Block a user