IAM-13: oidc-provider-fuer-drittanwendungen (client-registrierung, authorization-code-flow, jwks ueber API-05-schluessel)

2026-08-28 23:25:45 +02:00
parent f7863fd4d2
commit 42466c7b40
6 changed files with 833 additions and 0 deletions
@@ -0,0 +1,294 @@
+package oidc
+
+import (
+	"context"
+	"crypto/ed25519"
+	"encoding/base64"
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"net/url"
+	"os"
+	"strings"
+	"testing"
+
+	"github.com/golang-jwt/jwt/v5"
+	"github.com/jackc/pgx/v5/pgxpool"
+
+	"gitea.perlbach24.de/scripte/nexarch/internal/auth"
+	"gitea.perlbach24.de/scripte/nexarch/internal/moduletrust"
+)
+
+// newProviderTestSetup baut Tenant-DB-Schema (users, oidc_clients,
+// oidc_auth_codes) direkt gegen TEST_ADMIN_DSN auf, analog zum Muster in
+// internal/tenant/lifecycle_test.go — braucht kein volles Tenant-Provisioning,
+// nur eine isolierte Datenbank fuer diesen Testlauf.
+func newProviderTestSetup(t *testing.T) (*Handler, *pgxpool.Pool, func()) {
+	t.Helper()
+	adminDSN := os.Getenv("TEST_ADMIN_DSN")
+	if adminDSN == "" {
+		t.Skip("TEST_ADMIN_DSN nicht gesetzt, Integrationstest uebersprungen")
+	}
+	ctx := context.Background()
+
+	pool, err := pgxpool.New(ctx, adminDSN)
+	if err != nil {
+		t.Fatalf("pool: %v", err)
+	}
+
+	if _, err := pool.Exec(ctx, `
+		CREATE EXTENSION IF NOT EXISTS pgcrypto;
+		CREATE TABLE IF NOT EXISTS users (
+			id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+			email TEXT NOT NULL UNIQUE,
+			name TEXT NOT NULL,
+			status TEXT NOT NULL DEFAULT 'active',
+			created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
+			updated_at TIMESTAMPTZ NOT NULL DEFAULT now()
+		);
+		CREATE TABLE IF NOT EXISTS oidc_clients (
+			id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+			name TEXT NOT NULL,
+			client_id TEXT NOT NULL UNIQUE,
+			secret_hash BYTEA NOT NULL,
+			redirect_uris TEXT[] NOT NULL,
+			allowed_scopes TEXT[] NOT NULL,
+			created_at TIMESTAMPTZ NOT NULL DEFAULT now()
+		);
+		CREATE TABLE IF NOT EXISTS oidc_auth_codes (
+			code_hash BYTEA PRIMARY KEY,
+			client_id TEXT NOT NULL REFERENCES oidc_clients(client_id),
+			user_id UUID NOT NULL REFERENCES users(id),
+			redirect_uri TEXT NOT NULL,
+			scopes TEXT[] NOT NULL,
+			created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
+			expires_at TIMESTAMPTZ NOT NULL,
+			used_at TIMESTAMPTZ
+		);
+	`); err != nil {
+		t.Fatalf("schema: %v", err)
+	}
+
+	keys, err := moduletrust.NewKeyManager()
+	if err != nil {
+		t.Fatalf("keymanager: %v", err)
+	}
+
+	clients := NewClientStore(pool)
+	authCodes := NewAuthCodeStore(pool)
+	handler := NewHandler(clients, authCodes, keys, "https://core.test.nexarch.example")
+
+	cleanup := func() {
+		ctx := context.Background()
+		_, _ = pool.Exec(ctx, `DROP TABLE IF EXISTS oidc_auth_codes, oidc_clients, users CASCADE`)
+		pool.Close()
+	}
+	return handler, pool, cleanup
+}
+
+func createTestUser(t *testing.T, pool *pgxpool.Pool, email string) string {
+	t.Helper()
+	var id string
+	err := pool.QueryRow(context.Background(), `
+		INSERT INTO users (email, name) VALUES ($1, $2) RETURNING id
+	`, email, "Test Nutzer").Scan(&id)
+	if err != nil {
+		t.Fatalf("testnutzer anlegen: %v", err)
+	}
+	return id
+}
+
+// Akzeptanzkriterium 1 (Client-Registrierung) + 2 (voller Flow) + 3 (JWKS-
+// Verifikation, Scope-Einschraenkung). Deckt Pruefungen 1-3 ab.
+func TestFullAuthorizationCodeFlow(t *testing.T) {
+	handler, pool, cleanup := newProviderTestSetup(t)
+	defer cleanup()
+	ctx := context.Background()
+	userID := createTestUser(t, pool, "nutzer@acme.example")
+
+	// Pruefung 1 / Akzeptanzkriterium 1: Client registriert sich.
+	clientID, clientSecret, err := handler.clients.Register(ctx, "Test-Drittanwendung",
+		[]string{"https://app.example.com/callback"},
+		[]string{"openid", "profile"}, // erlaubte Scopes: KEIN "admin"
+	)
+	if err != nil {
+		t.Fatalf("register: %v", err)
+	}
+
+	// Session-Cookie fuer den Authorize-Aufruf ausstellen, exakt wie ein
+	// echter Browser ihn haette — RequireAuth uebernimmt die Verifikation,
+	// dieselbe Middleware wie in Produktion (nicht simuliert).
+	issuer := auth.NewTokenIssuer("test-session-secret")
+	sessionToken, err := issuer.Issue(userID, "acme")
+	if err != nil {
+		t.Fatalf("session-token ausstellen: %v", err)
+	}
+
+	authorizeURL := "/oidc/authorize?" + url.Values{
+		"response_type": {"code"},
+		"client_id":     {clientID},
+		"redirect_uri":  {"https://app.example.com/callback"},
+		// admin wird angefragt, ist aber NICHT erlaubt -> muss herausgefiltert werden.
+		"scope": {"openid profile admin"},
+		"state": {"xyz123"},
+	}.Encode()
+
+	req := httptest.NewRequest(http.MethodGet, authorizeURL, nil)
+	req.AddCookie(&http.Cookie{Name: auth.CookieName, Value: sessionToken})
+	rec := httptest.NewRecorder()
+
+	protected := auth.RequireAuth(issuer, handler.Authorize)
+	protected(rec, req)
+
+	if rec.Code != http.StatusFound {
+		t.Fatalf("authorize: status = %d, want %d, body: %s", rec.Code, http.StatusFound, rec.Body.String())
+	}
+	loc, err := url.Parse(rec.Header().Get("Location"))
+	if err != nil {
+		t.Fatalf("redirect-location parsen: %v", err)
+	}
+	if loc.Query().Get("state") != "xyz123" {
+		t.Fatalf("state = %q, want xyz123 (muss unveraendert durchgereicht werden)", loc.Query().Get("state"))
+	}
+	code := loc.Query().Get("code")
+	if code == "" {
+		t.Fatal("erwartet gesetzten code-Parameter in der redirect-URL")
+	}
+
+	// Pruefung 1 / Akzeptanzkriterium 2: Code gegen Token tauschen.
+	tokenForm := url.Values{
+		"grant_type":    {"authorization_code"},
+		"code":          {code},
+		"redirect_uri":  {"https://app.example.com/callback"},
+		"client_id":     {clientID},
+		"client_secret": {clientSecret},
+	}
+	tokenReq := httptest.NewRequest(http.MethodPost, "/oidc/token", strings.NewReader(tokenForm.Encode()))
+	tokenReq.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+	tokenRec := httptest.NewRecorder()
+	handler.Token(tokenRec, tokenReq)
+
+	if tokenRec.Code != http.StatusOK {
+		t.Fatalf("token: status = %d, want 200, body: %s", tokenRec.Code, tokenRec.Body.String())
+	}
+
+	var tokenResp struct {
+		IDToken string `json:"id_token"`
+		Scope   string `json:"scope"`
+	}
+	if err := decodeJSON(tokenRec.Body.Bytes(), &tokenResp); err != nil {
+		t.Fatalf("token-response dekodieren: %v", err)
+	}
+
+	// Akzeptanzkriterium 3 / Pruefung 3: "admin" wurde herausgefiltert.
+	grantedScopes := strings.Fields(tokenResp.Scope)
+	for _, s := range grantedScopes {
+		if s == "admin" {
+			t.Fatal("nicht erlaubter scope 'admin' wurde dennoch gewaehrt")
+		}
+	}
+	if !containsAll(grantedScopes, "openid", "profile") {
+		t.Fatalf("erwartete scopes openid+profile, habe %v", grantedScopes)
+	}
+
+	// Pruefung 2 / Akzeptanzkriterium 3: ausgestelltes Token gegen JWKS verifizieren.
+	jwksRec := httptest.NewRecorder()
+	handler.JWKS(jwksRec, httptest.NewRequest(http.MethodGet, "/oidc/jwks.json", nil))
+
+	var jwks jwkSet
+	if err := decodeJSON(jwksRec.Body.Bytes(), &jwks); err != nil {
+		t.Fatalf("jwks dekodieren: %v", err)
+	}
+	if len(jwks.Keys) == 0 {
+		t.Fatal("jwks enthaelt keine schluessel")
+	}
+
+	parsedToken, err := jwt.ParseWithClaims(tokenResp.IDToken, &IDTokenClaims{}, func(tok *jwt.Token) (interface{}, error) {
+		kid, _ := tok.Header["kid"].(string)
+		for _, k := range jwks.Keys {
+			if k.Kid == kid {
+				pub, err := decodeBase64URLEd25519(k.X)
+				if err != nil {
+					return nil, err
+				}
+				return ed25519.PublicKey(pub), nil
+			}
+		}
+		return nil, jwt.ErrTokenUnverifiable
+	})
+	if err != nil || !parsedToken.Valid {
+		t.Fatalf("id-token gegen jwks verifizieren: %v (valid=%v)", err, parsedToken != nil && parsedToken.Valid)
+	}
+
+	claims := parsedToken.Claims.(*IDTokenClaims)
+	if claims.Subject != userID {
+		t.Fatalf("sub = %q, want %q", claims.Subject, userID)
+	}
+	if len(claims.Audience) != 1 || claims.Audience[0] != clientID {
+		t.Fatalf("aud = %v, want [%q]", claims.Audience, clientID)
+	}
+}
+
+// Pruefung 3 (dediziert): Client, der einen Scope anfragt, den er bei der
+// Registrierung nicht erhalten hat, bekommt ihn unter keinen Umstaenden.
+func TestGrantedScopes_NeverExceedsAllowed(t *testing.T) {
+	c := Client{AllowedScopes: []string{"openid", "profile"}}
+	granted := c.GrantedScopes([]string{"openid", "admin", "profile", "billing"})
+	if !containsAll(granted, "openid", "profile") || len(granted) != 2 {
+		t.Fatalf("granted = %v, want genau [openid profile]", granted)
+	}
+}
+
+func TestValidateRedirectURI_RejectsUnregistered(t *testing.T) {
+	c := Client{RedirectURIs: []string{"https://app.example.com/callback"}}
+	if err := c.ValidateRedirectURI("https://boese-seite.example.com/callback"); err == nil {
+		t.Fatal("erwartet fehler fuer nicht registrierte redirect_uri")
+	}
+	if err := c.ValidateRedirectURI("https://app.example.com/callback"); err != nil {
+		t.Fatalf("registrierte redirect_uri sollte akzeptiert werden: %v", err)
+	}
+}
+
+func TestAuthCodeStore_CannotBeConsumedTwice(t *testing.T) {
+	handler, pool, cleanup := newProviderTestSetup(t)
+	defer cleanup()
+	ctx := context.Background()
+	userID := createTestUser(t, pool, "einmalig@acme.example")
+	clientID, _, err := handler.clients.Register(ctx, "Einmal-Client", []string{"https://a.example/cb"}, []string{"openid"})
+	if err != nil {
+		t.Fatalf("register: %v", err)
+	}
+
+	code, err := handler.authCodes.Issue(ctx, clientID, userID, "https://a.example/cb", []string{"openid"})
+	if err != nil {
+		t.Fatalf("issue: %v", err)
+	}
+
+	if _, err := handler.authCodes.Consume(ctx, code, clientID, "https://a.example/cb"); err != nil {
+		t.Fatalf("erster consume sollte funktionieren: %v", err)
+	}
+	if _, err := handler.authCodes.Consume(ctx, code, clientID, "https://a.example/cb"); err == nil {
+		t.Fatal("zweiter consume desselben codes haette fehlschlagen muessen")
+	}
+}
+
+func containsAll(haystack []string, needles ...string) bool {
+	set := make(map[string]bool, len(haystack))
+	for _, h := range haystack {
+		set[h] = true
+	}
+	for _, n := range needles {
+		if !set[n] {
+			return false
+		}
+	}
+	return true
+}
+
+func decodeJSON(data []byte, v any) error {
+	return json.Unmarshal(data, v)
+}
+
+func decodeBase64URLEd25519(s string) ([]byte, error) {
+	return base64.RawURLEncoding.DecodeString(s)
+}