Compare commits
1
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
ac070c160a |
@@ -0,0 +1,37 @@
|
||||
package lockout
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"gitea.perlbach24.de/scripte/nexarch/internal/auth"
|
||||
)
|
||||
|
||||
var ErrAccountLocked = errors.New("lockout: konto ist vorübergehend gesperrt")
|
||||
|
||||
// GuardedLogin komponiert IAM-02s LoginService mit dem Lockout-Zustand,
|
||||
// OHNE LoginService selbst zu veraendern (kein Umbau der Vorbedingung):
|
||||
// vor jedem Login-Versuch wird die Sperre geprueft, nach jedem Versuch wird
|
||||
// Erfolg/Fehlschlag im geteilten Postgres-Zustand vermerkt.
|
||||
func GuardedLogin(ctx context.Context, store *Store, login *auth.LoginService, email, password string) (string, error) {
|
||||
locked, _, err := store.IsLocked(ctx, email)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
if locked {
|
||||
return "", ErrAccountLocked
|
||||
}
|
||||
|
||||
token, err := login.Login(ctx, email, password)
|
||||
if err != nil {
|
||||
if _, _, lockErr := store.RecordFailure(ctx, email); lockErr != nil {
|
||||
return "", lockErr
|
||||
}
|
||||
return "", err
|
||||
}
|
||||
|
||||
if err := store.RecordSuccess(ctx, email); err != nil {
|
||||
return "", err
|
||||
}
|
||||
return token, nil
|
||||
}
|
||||
@@ -0,0 +1,103 @@
|
||||
// Package lockout implementiert Core IAM-07: Account-Lockout nach
|
||||
// Fehlversuchen und Login-Rate-Limiting mit geteiltem, externem
|
||||
// (Postgres-basiertem) Zustand — kein In-Process-Zaehler, der bei
|
||||
// Mehrinstanzbetrieb aushebelbar waere (bekannter Fehler aus archivdms
|
||||
// internal/auth/ratelimit.go).
|
||||
package lockout
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
)
|
||||
|
||||
// DefaultMaxFailedAttempts/DefaultLockoutDuration sind explizit benannte
|
||||
// Defaults, ueberschreibbar via WithPolicy.
|
||||
const (
|
||||
DefaultMaxFailedAttempts = 5
|
||||
DefaultLockoutDuration = 15 * time.Minute
|
||||
)
|
||||
|
||||
type Store struct {
|
||||
pool *pgxpool.Pool
|
||||
maxFailed int
|
||||
lockoutDuration time.Duration
|
||||
}
|
||||
|
||||
func NewStore(pool *pgxpool.Pool) *Store {
|
||||
return &Store{pool: pool, maxFailed: DefaultMaxFailedAttempts, lockoutDuration: DefaultLockoutDuration}
|
||||
}
|
||||
|
||||
func (s *Store) WithPolicy(maxFailed int, lockoutDuration time.Duration) *Store {
|
||||
return &Store{pool: s.pool, maxFailed: maxFailed, lockoutDuration: lockoutDuration}
|
||||
}
|
||||
|
||||
// IsLocked prueft, ob ein Konto aktuell gesperrt ist. Eine abgelaufene
|
||||
// Sperre gilt automatisch als nicht mehr gesperrt (Akzeptanzkriterium 3) —
|
||||
// es ist keine explizite Entsperr-Aktion noetig, der Zeitvergleich reicht.
|
||||
func (s *Store) IsLocked(ctx context.Context, email string) (locked bool, lockedUntil time.Time, err error) {
|
||||
err = s.pool.QueryRow(ctx, `
|
||||
SELECT locked_until FROM login_attempts WHERE email = $1 AND locked_until IS NOT NULL
|
||||
`, email).Scan(&lockedUntil)
|
||||
if err != nil {
|
||||
return false, time.Time{}, nil // kein Datensatz oder kein Lock -> nicht gesperrt
|
||||
}
|
||||
return time.Now().Before(lockedUntil), lockedUntil, nil
|
||||
}
|
||||
|
||||
// RecordFailure erhoeht den Fehlversuchszaehler ATOMAR (UPSERT) und sperrt
|
||||
// das Konto, sobald die Schwelle erreicht ist (Akzeptanzkriterium 1 / 2).
|
||||
// Der Zustand liegt ausschliesslich in Postgres, mehrere Core-Instanzen
|
||||
// teilen sich denselben Zaehler (Akzeptanzkriterium 2).
|
||||
func (s *Store) RecordFailure(ctx context.Context, email string) (locked bool, lockedUntil time.Time, err error) {
|
||||
var failedCount int
|
||||
var lockedUntilPtr *time.Time
|
||||
|
||||
err = s.pool.QueryRow(ctx, `
|
||||
INSERT INTO login_attempts (email, failed_count, locked_until, last_attempt_at)
|
||||
VALUES ($1, 1, CASE WHEN 1 >= $2 THEN now() + $3::interval ELSE NULL END, now())
|
||||
ON CONFLICT (email) DO UPDATE SET
|
||||
failed_count = login_attempts.failed_count + 1,
|
||||
last_attempt_at = now(),
|
||||
locked_until = CASE
|
||||
WHEN login_attempts.failed_count + 1 >= $2 THEN now() + $3::interval
|
||||
ELSE login_attempts.locked_until
|
||||
END
|
||||
RETURNING failed_count, locked_until
|
||||
`, email, s.maxFailed, fmt.Sprintf("%f seconds", s.lockoutDuration.Seconds())).Scan(&failedCount, &lockedUntilPtr)
|
||||
if err != nil {
|
||||
return false, time.Time{}, fmt.Errorf("fehlversuch erfassen: %w", err)
|
||||
}
|
||||
|
||||
if lockedUntilPtr != nil {
|
||||
return time.Now().Before(*lockedUntilPtr), *lockedUntilPtr, nil
|
||||
}
|
||||
return false, time.Time{}, nil
|
||||
}
|
||||
|
||||
// RecordSuccess setzt den Fehlversuchszaehler nach erfolgreichem Login zurueck.
|
||||
func (s *Store) RecordSuccess(ctx context.Context, email string) error {
|
||||
_, err := s.pool.Exec(ctx, `
|
||||
INSERT INTO login_attempts (email, failed_count, locked_until, last_attempt_at)
|
||||
VALUES ($1, 0, NULL, now())
|
||||
ON CONFLICT (email) DO UPDATE SET failed_count = 0, locked_until = NULL, last_attempt_at = now()
|
||||
`, email)
|
||||
if err != nil {
|
||||
return fmt.Errorf("erfolgreichen login erfassen: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Unlock entsperrt ein Konto durch Administratoreingriff, unabhaengig von
|
||||
// der Sperrzeit (Akzeptanzkriterium 3).
|
||||
func (s *Store) Unlock(ctx context.Context, email string) error {
|
||||
_, err := s.pool.Exec(ctx, `
|
||||
UPDATE login_attempts SET failed_count = 0, locked_until = NULL WHERE email = $1
|
||||
`, email)
|
||||
if err != nil {
|
||||
return fmt.Errorf("konto entsperren: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
@@ -0,0 +1,233 @@
|
||||
package lockout
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
|
||||
"gitea.perlbach24.de/scripte/nexarch/internal/auth"
|
||||
"gitea.perlbach24.de/scripte/nexarch/internal/user"
|
||||
)
|
||||
|
||||
const schema = `
|
||||
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', password_hash TEXT NOT NULL DEFAULT '',
|
||||
created_at TIMESTAMPTZ NOT NULL DEFAULT now(), updated_at TIMESTAMPTZ NOT NULL DEFAULT now()
|
||||
);
|
||||
CREATE TABLE IF NOT EXISTS login_attempts (
|
||||
email TEXT PRIMARY KEY, failed_count INT NOT NULL DEFAULT 0,
|
||||
locked_until TIMESTAMPTZ, last_attempt_at TIMESTAMPTZ NOT NULL DEFAULT now()
|
||||
);`
|
||||
|
||||
func setupTest(t *testing.T) (*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, schema); err != nil {
|
||||
t.Fatalf("schema: %v", err)
|
||||
}
|
||||
cleanup := func() { pool.Close() }
|
||||
return pool, cleanup
|
||||
}
|
||||
|
||||
func uniqueEmail() string {
|
||||
return fmt.Sprintf("lockout-test-%d@example.com", time.Now().UnixNano())
|
||||
}
|
||||
|
||||
// Akzeptanzkriterium 1 + Pruefung 2: Sperre greift nach definierten
|
||||
// Fehlversuchen zuverlaessig.
|
||||
func TestRecordFailure_LocksAfterThreshold(t *testing.T) {
|
||||
pool, cleanup := setupTest(t)
|
||||
defer cleanup()
|
||||
ctx := context.Background()
|
||||
store := NewStore(pool).WithPolicy(3, time.Hour)
|
||||
email := uniqueEmail()
|
||||
|
||||
for i := 0; i < 2; i++ {
|
||||
locked, _, err := store.RecordFailure(ctx, email)
|
||||
if err != nil {
|
||||
t.Fatalf("record failure %d: %v", i, err)
|
||||
}
|
||||
if locked {
|
||||
t.Fatalf("sollte nach %d fehlversuchen noch nicht gesperrt sein", i+1)
|
||||
}
|
||||
}
|
||||
|
||||
locked, lockedUntil, err := store.RecordFailure(ctx, email)
|
||||
if err != nil {
|
||||
t.Fatalf("record failure 3: %v", err)
|
||||
}
|
||||
if !locked {
|
||||
t.Fatal("erwartet gesperrt nach 3 fehlversuchen (schwelle=3)")
|
||||
}
|
||||
if !lockedUntil.After(time.Now()) {
|
||||
t.Fatal("lockedUntil sollte in der zukunft liegen")
|
||||
}
|
||||
|
||||
isLocked, _, err := store.IsLocked(ctx, email)
|
||||
if err != nil {
|
||||
t.Fatalf("is locked: %v", err)
|
||||
}
|
||||
if !isLocked {
|
||||
t.Fatal("IsLocked sollte true liefern")
|
||||
}
|
||||
}
|
||||
|
||||
// Akzeptanzkriterium 2 + Pruefung 1: zwei Dienstinstanzen teilen sich
|
||||
// denselben Zaehler (Mehrinstanz-Test) — hier durch zwei unabhaengige
|
||||
// pgxpool.Pool-Verbindungen zur selben Datenbank simuliert.
|
||||
func TestRecordFailure_SharedAcrossInstances(t *testing.T) {
|
||||
_, cleanup := setupTest(t)
|
||||
defer cleanup()
|
||||
ctx := context.Background()
|
||||
email := uniqueEmail()
|
||||
|
||||
adminDSN := os.Getenv("TEST_ADMIN_DSN")
|
||||
instanceA, err := pgxpool.New(ctx, adminDSN)
|
||||
if err != nil {
|
||||
t.Fatalf("instance a: %v", err)
|
||||
}
|
||||
defer instanceA.Close()
|
||||
instanceB, err := pgxpool.New(ctx, adminDSN)
|
||||
if err != nil {
|
||||
t.Fatalf("instance b: %v", err)
|
||||
}
|
||||
defer instanceB.Close()
|
||||
|
||||
storeA := NewStore(instanceA).WithPolicy(4, time.Hour)
|
||||
storeB := NewStore(instanceB).WithPolicy(4, time.Hour)
|
||||
|
||||
// Zwei fehlversuche ueber "instanz A", zwei ueber "instanz B" — beide
|
||||
// muessen sich denselben Zaehler teilen und gemeinsam die Schwelle erreichen.
|
||||
if _, _, err := storeA.RecordFailure(ctx, email); err != nil {
|
||||
t.Fatalf("a1: %v", err)
|
||||
}
|
||||
if _, _, err := storeB.RecordFailure(ctx, email); err != nil {
|
||||
t.Fatalf("b1: %v", err)
|
||||
}
|
||||
if _, _, err := storeA.RecordFailure(ctx, email); err != nil {
|
||||
t.Fatalf("a2: %v", err)
|
||||
}
|
||||
locked, _, err := storeB.RecordFailure(ctx, email)
|
||||
if err != nil {
|
||||
t.Fatalf("b2: %v", err)
|
||||
}
|
||||
if !locked {
|
||||
t.Fatal("erwartet gesperrt nach insgesamt 4 fehlversuchen ueber beide instanzen hinweg")
|
||||
}
|
||||
}
|
||||
|
||||
// Akzeptanzkriterium 3 + Pruefung 3: automatische Entsperrung nach Ablauf
|
||||
// der Sperrzeit, Zeitversatz automatisiert getestet.
|
||||
func TestIsLocked_AutoUnlocksAfterExpiry(t *testing.T) {
|
||||
pool, cleanup := setupTest(t)
|
||||
defer cleanup()
|
||||
ctx := context.Background()
|
||||
store := NewStore(pool).WithPolicy(1, 50*time.Millisecond)
|
||||
email := uniqueEmail()
|
||||
|
||||
locked, _, err := store.RecordFailure(ctx, email)
|
||||
if err != nil {
|
||||
t.Fatalf("record failure: %v", err)
|
||||
}
|
||||
if !locked {
|
||||
t.Fatal("erwartet sofort gesperrt bei schwelle=1")
|
||||
}
|
||||
|
||||
isLocked, _, err := store.IsLocked(ctx, email)
|
||||
if err != nil {
|
||||
t.Fatalf("is locked (vor ablauf): %v", err)
|
||||
}
|
||||
if !isLocked {
|
||||
t.Fatal("erwartet gesperrt vor ablauf der sperrzeit")
|
||||
}
|
||||
|
||||
time.Sleep(70 * time.Millisecond)
|
||||
|
||||
isLocked, _, err = store.IsLocked(ctx, email)
|
||||
if err != nil {
|
||||
t.Fatalf("is locked (nach ablauf): %v", err)
|
||||
}
|
||||
if isLocked {
|
||||
t.Fatal("erwartet automatisch entsperrt nach ablauf der sperrzeit")
|
||||
}
|
||||
}
|
||||
|
||||
// Akzeptanzkriterium 3: Entsperrung durch Administratoreingriff.
|
||||
func TestUnlock_ClearsLockImmediately(t *testing.T) {
|
||||
pool, cleanup := setupTest(t)
|
||||
defer cleanup()
|
||||
ctx := context.Background()
|
||||
store := NewStore(pool).WithPolicy(1, time.Hour)
|
||||
email := uniqueEmail()
|
||||
|
||||
if _, _, err := store.RecordFailure(ctx, email); err != nil {
|
||||
t.Fatalf("record failure: %v", err)
|
||||
}
|
||||
locked, _, _ := store.IsLocked(ctx, email)
|
||||
if !locked {
|
||||
t.Fatal("erwartet gesperrt vor admin-eingriff")
|
||||
}
|
||||
|
||||
if err := store.Unlock(ctx, email); err != nil {
|
||||
t.Fatalf("unlock: %v", err)
|
||||
}
|
||||
locked, _, err := store.IsLocked(ctx, email)
|
||||
if err != nil {
|
||||
t.Fatalf("is locked nach unlock: %v", err)
|
||||
}
|
||||
if locked {
|
||||
t.Fatal("erwartet sofort entsperrt nach admin-eingriff, unabhaengig von der sperrzeit")
|
||||
}
|
||||
}
|
||||
|
||||
// Integrationstest: GuardedLogin komponiert IAM-02 LoginService mit dem
|
||||
// Lockout-Zustand, ohne LoginService selbst zu veraendern.
|
||||
func TestGuardedLogin_LocksAfterRepeatedFailures(t *testing.T) {
|
||||
pool, cleanup := setupTest(t)
|
||||
defer cleanup()
|
||||
ctx := context.Background()
|
||||
|
||||
email := uniqueEmail()
|
||||
userStore := user.NewTenantUserStore(pool)
|
||||
u, err := userStore.Create(ctx, email, "Lockout Test")
|
||||
if err != nil {
|
||||
t.Fatalf("create user: %v", err)
|
||||
}
|
||||
hash, err := auth.HashPassword("korrektes-passwort")
|
||||
if err != nil {
|
||||
t.Fatalf("hash: %v", err)
|
||||
}
|
||||
if err := userStore.SetPasswordHash(ctx, u.ID, hash); err != nil {
|
||||
t.Fatalf("set password: %v", err)
|
||||
}
|
||||
|
||||
issuer := auth.NewTokenIssuer("test-secret-nur-fuer-tests")
|
||||
loginService := auth.NewLoginService(userStore, issuer, "acme")
|
||||
lockoutStore := NewStore(pool).WithPolicy(3, time.Hour)
|
||||
|
||||
for i := 0; i < 3; i++ {
|
||||
if _, err := GuardedLogin(ctx, lockoutStore, loginService, email, "falsches-passwort"); !errors.Is(err, auth.ErrInvalidCredentials) {
|
||||
t.Fatalf("versuch %d: erwartet ErrInvalidCredentials, habe %v", i, err)
|
||||
}
|
||||
}
|
||||
|
||||
// Selbst das RICHTIGE Passwort wird jetzt abgewiesen, weil das Konto gesperrt ist.
|
||||
if _, err := GuardedLogin(ctx, lockoutStore, loginService, email, "korrektes-passwort"); !errors.Is(err, ErrAccountLocked) {
|
||||
t.Fatalf("erwartet ErrAccountLocked trotz korrektem passwort, habe %v", err)
|
||||
}
|
||||
}
|
||||
@@ -1,151 +0,0 @@
|
||||
// Package serviceaccount implementiert Core IAM-09: Service-Accounts als
|
||||
// eigenstaendige Identitaeten (getrennt von menschlichen Benutzern, siehe
|
||||
// internal/user) sowie API-Token mit Scopes und optionaler Ablaufzeit zur
|
||||
// programmatischen Authentifizierung.
|
||||
package serviceaccount
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/rand"
|
||||
"crypto/sha256"
|
||||
"encoding/hex"
|
||||
"errors"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/jackc/pgx/v5"
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrTokenInvalid = errors.New("serviceaccount: token ungueltig, widerrufen oder abgelaufen")
|
||||
ErrScopeInsufficient = errors.New("serviceaccount: token hat nicht den erforderlichen scope")
|
||||
ErrServiceAccountNotFound = errors.New("serviceaccount: nicht gefunden")
|
||||
)
|
||||
|
||||
type ServiceAccount struct {
|
||||
ID string
|
||||
Name string
|
||||
}
|
||||
|
||||
type Token struct {
|
||||
ID string
|
||||
ServiceAccountID string
|
||||
Scopes []string
|
||||
ExpiresAt *time.Time
|
||||
}
|
||||
|
||||
type Store struct {
|
||||
pool *pgxpool.Pool
|
||||
}
|
||||
|
||||
func NewStore(pool *pgxpool.Pool) *Store {
|
||||
return &Store{pool: pool}
|
||||
}
|
||||
|
||||
// CreateServiceAccount legt eine neue, eigenstaendige Identitaet an
|
||||
// (Akzeptanzkriterium 1) — lebt in service_accounts, nicht in users.
|
||||
func (s *Store) CreateServiceAccount(ctx context.Context, name string) (ServiceAccount, error) {
|
||||
if name == "" {
|
||||
return ServiceAccount{}, errors.New("serviceaccount: name darf nicht leer sein")
|
||||
}
|
||||
var sa ServiceAccount
|
||||
sa.Name = name
|
||||
if err := s.pool.QueryRow(ctx, `
|
||||
INSERT INTO service_accounts (name) VALUES ($1) RETURNING id
|
||||
`, name).Scan(&sa.ID); err != nil {
|
||||
return ServiceAccount{}, fmt.Errorf("service-account anlegen: %w", err)
|
||||
}
|
||||
return sa, nil
|
||||
}
|
||||
|
||||
// IssueToken erzeugt ein neues API-Token fuer einen Service-Account.
|
||||
// scopes beschraenkt, wofuer das Token verwendet werden darf; ttl ist
|
||||
// optional (nil = kein Ablauf) — beides Akzeptanzkriterium 2. Der
|
||||
// Klartext-Token wird NUR hier zurueckgegeben, gespeichert wird ausschliesslich
|
||||
// der Hash (Akzeptanzkriterium 1 / Pruefung 1).
|
||||
func (s *Store) IssueToken(ctx context.Context, serviceAccountID string, scopes []string, ttl *time.Duration) (id, plaintext string, err error) {
|
||||
plaintext, err = randomToken()
|
||||
if err != nil {
|
||||
return "", "", fmt.Errorf("token erzeugen: %w", err)
|
||||
}
|
||||
hash := hashToken(plaintext)
|
||||
if scopes == nil {
|
||||
scopes = []string{}
|
||||
}
|
||||
|
||||
var expiresAt *time.Time
|
||||
if ttl != nil {
|
||||
t := time.Now().Add(*ttl)
|
||||
expiresAt = &t
|
||||
}
|
||||
|
||||
err = s.pool.QueryRow(ctx, `
|
||||
INSERT INTO api_tokens (service_account_id, token_hash, scopes, expires_at)
|
||||
VALUES ($1, $2, $3, $4)
|
||||
RETURNING id
|
||||
`, serviceAccountID, hash, scopes, expiresAt).Scan(&id)
|
||||
if err != nil {
|
||||
return "", "", fmt.Errorf("token speichern: %w", err)
|
||||
}
|
||||
return id, plaintext, nil
|
||||
}
|
||||
|
||||
// Verify prueft ein Token gegen einen benoetigten Scope. Widerruf und Ablauf
|
||||
// werden bei JEDEM Aufruf direkt gegen die Datenbank geprueft — es gibt
|
||||
// keinen Cache, ein widerrufenes Token wird beim naechsten Request sofort
|
||||
// abgewiesen (Akzeptanzkriterium 3). requiredScope == "" ueberspringt die
|
||||
// Scope-Pruefung.
|
||||
func (s *Store) Verify(ctx context.Context, tokenString, requiredScope string) (Token, error) {
|
||||
hash := hashToken(tokenString)
|
||||
|
||||
var tok Token
|
||||
err := s.pool.QueryRow(ctx, `
|
||||
SELECT id, service_account_id, scopes, expires_at
|
||||
FROM api_tokens
|
||||
WHERE token_hash = $1 AND revoked_at IS NULL AND (expires_at IS NULL OR expires_at > now())
|
||||
`, hash).Scan(&tok.ID, &tok.ServiceAccountID, &tok.Scopes, &tok.ExpiresAt)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return Token{}, ErrTokenInvalid
|
||||
}
|
||||
return Token{}, fmt.Errorf("token pruefen: %w", err)
|
||||
}
|
||||
|
||||
if requiredScope != "" && !hasScope(tok.Scopes, requiredScope) {
|
||||
return Token{}, ErrScopeInsufficient
|
||||
}
|
||||
return tok, nil
|
||||
}
|
||||
|
||||
// Revoke widerruft ein Token sofort — Verify schlaegt ab dem naechsten
|
||||
// Aufruf fehl (Akzeptanzkriterium 3 / Pruefung 2).
|
||||
func (s *Store) Revoke(ctx context.Context, tokenID string) error {
|
||||
_, err := s.pool.Exec(ctx, `UPDATE api_tokens SET revoked_at = now() WHERE id = $1`, tokenID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("token widerrufen: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func hasScope(scopes []string, want string) bool {
|
||||
for _, s := range scopes {
|
||||
if s == want {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func randomToken() (string, error) {
|
||||
buf := make([]byte, 32)
|
||||
if _, err := rand.Read(buf); err != nil {
|
||||
return "", err
|
||||
}
|
||||
return hex.EncodeToString(buf), nil
|
||||
}
|
||||
|
||||
func hashToken(token string) []byte {
|
||||
sum := sha256.Sum256([]byte(token))
|
||||
return sum[:]
|
||||
}
|
||||
@@ -1,163 +0,0 @@
|
||||
package serviceaccount
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"os"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
)
|
||||
|
||||
func setupTest(t *testing.T) (*Store, *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 TABLE IF NOT EXISTS service_accounts (
|
||||
id UUID PRIMARY KEY DEFAULT gen_random_uuid(), name TEXT NOT NULL,
|
||||
status TEXT NOT NULL DEFAULT 'active', created_at TIMESTAMPTZ NOT NULL DEFAULT now()
|
||||
);
|
||||
CREATE TABLE IF NOT EXISTS api_tokens (
|
||||
id UUID PRIMARY KEY DEFAULT gen_random_uuid(), service_account_id UUID NOT NULL REFERENCES service_accounts(id),
|
||||
token_hash BYTEA NOT NULL UNIQUE, scopes TEXT[] NOT NULL DEFAULT '{}',
|
||||
expires_at TIMESTAMPTZ, revoked_at TIMESTAMPTZ, created_at TIMESTAMPTZ NOT NULL DEFAULT now()
|
||||
);
|
||||
`); err != nil {
|
||||
t.Fatalf("schema: %v", err)
|
||||
}
|
||||
|
||||
cleanup := func() { pool.Close() }
|
||||
return NewStore(pool), pool, cleanup
|
||||
}
|
||||
|
||||
// Akzeptanzkriterium 1: Service-Account anlegen, eigenes gehashtes Token erhalten.
|
||||
// Akzeptanzkriterium 1 + Pruefung 1: Token ausschliesslich gehasht gespeichert.
|
||||
func TestIssueToken_StoresOnlyHash(t *testing.T) {
|
||||
store, pool, cleanup := setupTest(t)
|
||||
defer cleanup()
|
||||
ctx := context.Background()
|
||||
|
||||
sa, err := store.CreateServiceAccount(ctx, "ci-pipeline")
|
||||
if err != nil {
|
||||
t.Fatalf("create service account: %v", err)
|
||||
}
|
||||
|
||||
tokenID, plaintext, err := store.IssueToken(ctx, sa.ID, []string{"read"}, nil)
|
||||
if err != nil {
|
||||
t.Fatalf("issue token: %v", err)
|
||||
}
|
||||
if plaintext == "" {
|
||||
t.Fatal("erwartet nicht-leeren klartext-token")
|
||||
}
|
||||
|
||||
var storedHash []byte
|
||||
if err := pool.QueryRow(ctx, `SELECT token_hash FROM api_tokens WHERE id = $1`, tokenID).Scan(&storedHash); err != nil {
|
||||
t.Fatalf("hash lesen: %v", err)
|
||||
}
|
||||
if string(storedHash) == plaintext {
|
||||
t.Fatal("token_hash enthaelt den klartext-token — verstoss gegen akzeptanzkriterium 1")
|
||||
}
|
||||
if len(storedHash) != 32 { // sha256
|
||||
t.Fatalf("erwartet 32-byte sha256-hash, habe %d bytes", len(storedHash))
|
||||
}
|
||||
|
||||
tok, err := store.Verify(ctx, plaintext, "")
|
||||
if err != nil {
|
||||
t.Fatalf("verify: %v", err)
|
||||
}
|
||||
if tok.ServiceAccountID != sa.ID {
|
||||
t.Fatalf("service account id = %q, want %q", tok.ServiceAccountID, sa.ID)
|
||||
}
|
||||
}
|
||||
|
||||
// Akzeptanzkriterium 2 + Pruefung 3: Scope-Verletzung wird abgewiesen.
|
||||
func TestVerify_RejectsInsufficientScope(t *testing.T) {
|
||||
store, _, cleanup := setupTest(t)
|
||||
defer cleanup()
|
||||
ctx := context.Background()
|
||||
|
||||
sa, err := store.CreateServiceAccount(ctx, "readonly-bot")
|
||||
if err != nil {
|
||||
t.Fatalf("create service account: %v", err)
|
||||
}
|
||||
_, plaintext, err := store.IssueToken(ctx, sa.ID, []string{"read"}, nil)
|
||||
if err != nil {
|
||||
t.Fatalf("issue token: %v", err)
|
||||
}
|
||||
|
||||
if _, err := store.Verify(ctx, plaintext, "read"); err != nil {
|
||||
t.Fatalf("erwartet erfolg fuer passenden scope: %v", err)
|
||||
}
|
||||
if _, err := store.Verify(ctx, plaintext, "write"); !errors.Is(err, ErrScopeInsufficient) {
|
||||
t.Fatalf("erwartet ErrScopeInsufficient fuer fehlenden scope, habe %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Akzeptanzkriterium 2: optionale zeitliche Befristung wird durchgesetzt.
|
||||
func TestVerify_RejectsExpiredToken(t *testing.T) {
|
||||
store, _, cleanup := setupTest(t)
|
||||
defer cleanup()
|
||||
ctx := context.Background()
|
||||
|
||||
sa, err := store.CreateServiceAccount(ctx, "temp-bot")
|
||||
if err != nil {
|
||||
t.Fatalf("create service account: %v", err)
|
||||
}
|
||||
ttl := -1 * time.Second // sofort abgelaufen
|
||||
_, plaintext, err := store.IssueToken(ctx, sa.ID, nil, &ttl)
|
||||
if err != nil {
|
||||
t.Fatalf("issue token: %v", err)
|
||||
}
|
||||
|
||||
if _, err := store.Verify(ctx, plaintext, ""); !errors.Is(err, ErrTokenInvalid) {
|
||||
t.Fatalf("erwartet ErrTokenInvalid bei abgelaufenem token, habe %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Akzeptanzkriterium 3 + Pruefung 2: widerrufenes Token wird sofort abgewiesen.
|
||||
func TestRevoke_TakesEffectImmediately(t *testing.T) {
|
||||
store, _, cleanup := setupTest(t)
|
||||
defer cleanup()
|
||||
ctx := context.Background()
|
||||
|
||||
sa, err := store.CreateServiceAccount(ctx, "revoke-bot")
|
||||
if err != nil {
|
||||
t.Fatalf("create service account: %v", err)
|
||||
}
|
||||
tokenID, plaintext, err := store.IssueToken(ctx, sa.ID, nil, nil)
|
||||
if err != nil {
|
||||
t.Fatalf("issue token: %v", err)
|
||||
}
|
||||
|
||||
if _, err := store.Verify(ctx, plaintext, ""); err != nil {
|
||||
t.Fatalf("verify vor widerruf: %v", err)
|
||||
}
|
||||
|
||||
if err := store.Revoke(ctx, tokenID); err != nil {
|
||||
t.Fatalf("revoke: %v", err)
|
||||
}
|
||||
|
||||
if _, err := store.Verify(ctx, plaintext, ""); !errors.Is(err, ErrTokenInvalid) {
|
||||
t.Fatalf("erwartet ErrTokenInvalid sofort nach widerruf, habe %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestVerify_RejectsUnknownToken(t *testing.T) {
|
||||
store, _, cleanup := setupTest(t)
|
||||
defer cleanup()
|
||||
ctx := context.Background()
|
||||
|
||||
if _, err := store.Verify(ctx, "niemals-ausgestelltes-token", ""); !errors.Is(err, ErrTokenInvalid) {
|
||||
t.Fatalf("erwartet ErrTokenInvalid, habe %v", err)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1 @@
|
||||
DROP TABLE IF EXISTS login_attempts;
|
||||
@@ -0,0 +1,10 @@
|
||||
-- Account-Lockout & Login-Rate-Limiting (IAM-07, siehe
|
||||
-- core-kanban/tickets/IAM-07.md). Geteilter, externer Zustand in Postgres —
|
||||
-- kein In-Process-Zaehler (bekannter Fehler aus archivdms
|
||||
-- internal/auth/ratelimit.go), gilt korrekt ueber mehrere Core-Instanzen.
|
||||
CREATE TABLE login_attempts (
|
||||
email TEXT PRIMARY KEY,
|
||||
failed_count INT NOT NULL DEFAULT 0,
|
||||
locked_until TIMESTAMPTZ,
|
||||
last_attempt_at TIMESTAMPTZ NOT NULL DEFAULT now()
|
||||
);
|
||||
@@ -1,2 +0,0 @@
|
||||
DROP TABLE IF EXISTS api_tokens;
|
||||
DROP TABLE IF EXISTS service_accounts;
|
||||
@@ -1,20 +0,0 @@
|
||||
-- Service-Accounts & API-Token (IAM-09, siehe core-kanban/tickets/IAM-09.md).
|
||||
-- Service-Accounts sind eine eigenstaendige Identitaetsklasse neben
|
||||
-- menschlichen Benutzern (users), nicht dieselbe Tabelle (Zitadel-Vorbild).
|
||||
CREATE TABLE service_accounts (
|
||||
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||
name TEXT NOT NULL,
|
||||
status TEXT NOT NULL DEFAULT 'active',
|
||||
created_at TIMESTAMPTZ NOT NULL DEFAULT now()
|
||||
);
|
||||
|
||||
-- token_hash enthaelt NIEMALS den Klartext-Token, nur dessen SHA-256-Hash.
|
||||
CREATE TABLE api_tokens (
|
||||
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||
service_account_id UUID NOT NULL REFERENCES service_accounts(id),
|
||||
token_hash BYTEA NOT NULL UNIQUE,
|
||||
scopes TEXT[] NOT NULL DEFAULT '{}',
|
||||
expires_at TIMESTAMPTZ,
|
||||
revoked_at TIMESTAMPTZ,
|
||||
created_at TIMESTAMPTZ NOT NULL DEFAULT now()
|
||||
);
|
||||
Reference in New Issue
Block a user