fix: implement vault key decryption on login + fix token refresh via cookie

Three bugs fixed:

1. Phase 30 (auth.ts): After SRP login the encrypted_key_set was returned
   from the server but the vault key and RSA private key were never unwrapped
   with the AUK. keyStore.getVaultKey() was always null, causing Tier 1
   config-backup diffs to crash with a TypeError.
   Fix: unwrap vault key and private key using crypto.subtle.unwrapKey after
   successful SRP verification. Non-fatal: warns to console if decryption
   fails so login always succeeds.

2. Token refresh (auth.py): The /refresh endpoint required refresh_token in
   the request body, but the frontend never stored or sent it. After the 15-
   minute access token TTL, all authenticated API calls would fail silently
   because the interceptor sent an empty body and received 422 (not 401),
   so the retry loop never fired.
   Fix: login/srpVerify now set an httpOnly refresh_token cookie scoped to
   /api/auth/refresh. The refresh endpoint now accepts the token from either
   cookie (preferred) or body (legacy). Logout clears both cookies.
   RefreshRequest.refresh_token is now Optional to allow empty-body calls.

3. Silent token rotation: the /refresh endpoint now also rotates the refresh
   token cookie on each use (issues a fresh token), reducing the window for
   stolen refresh token replay.
This commit is contained in:
Cog
2026-03-11 21:56:08 -05:00
committed by Jason Staack
parent d0548bec86
commit 57e754bb27
3 changed files with 98 additions and 7 deletions

View File

@@ -75,6 +75,10 @@ router = APIRouter(prefix="/auth", tags=["auth"])
ACCESS_TOKEN_COOKIE = "access_token"
ACCESS_TOKEN_MAX_AGE = 15 * 60 # 15 minutes in seconds
# Refresh token cookie settings (httpOnly, longer-lived)
REFRESH_TOKEN_COOKIE = "refresh_token"
REFRESH_TOKEN_MAX_AGE = 7 * 24 * 60 * 60 # 7 days in seconds
# Cookie Secure flag requires HTTPS. Safari strictly enforces this —
# it silently drops Secure cookies over plain HTTP, unlike Chrome
# which exempts localhost. Auto-detect from CORS origins: if all
@@ -239,7 +243,7 @@ async def srp_verify_endpoint(
)
await db.commit()
# Set cookie (same as existing login)
# Set access token cookie
response.set_cookie(
key=ACCESS_TOKEN_COOKIE,
value=access_token,
@@ -248,6 +252,16 @@ async def srp_verify_endpoint(
secure=_COOKIE_SECURE,
samesite="lax",
)
# Set refresh token cookie (httpOnly, scoped to refresh endpoint)
response.set_cookie(
key=REFRESH_TOKEN_COOKIE,
value=refresh_token,
max_age=REFRESH_TOKEN_MAX_AGE,
httponly=True,
secure=_COOKIE_SECURE,
samesite="lax",
path="/api/auth/refresh",
)
# Fetch encrypted key set
key_set = await get_user_key_set(db, user.id)
@@ -360,6 +374,18 @@ async def login(
secure=_COOKIE_SECURE,
samesite="lax",
)
# Also set refresh token as httpOnly cookie so auto-refresh works
# without the frontend needing to persist the token in JS memory.
if not user.must_upgrade_auth:
response.set_cookie(
key=REFRESH_TOKEN_COOKIE,
value=refresh,
max_age=REFRESH_TOKEN_MAX_AGE,
httponly=True,
secure=_COOKIE_SECURE,
samesite="lax",
path="/api/auth/refresh", # scope cookie to refresh endpoint only
)
# Update last_login
await db.execute(
@@ -400,17 +426,29 @@ async def login(
@limiter.limit("10/minute")
async def refresh_token(
request: StarletteRequest,
body: RefreshRequest,
response: Response,
body: Optional[RefreshRequest] = None,
response: Response = None,
db: AsyncSession = Depends(get_admin_db),
redis: aioredis.Redis = Depends(get_redis),
refresh_token_cookie: Optional[str] = Cookie(default=None, alias="refresh_token"),
) -> TokenResponse:
"""
Exchange a valid refresh token for a new access token.
Accepts the refresh token either in the JSON body (legacy) or as an
httpOnly cookie named 'refresh_token' (preferred — set automatically at login).
Rate limited to 10 requests per minute per IP.
"""
# Resolve token: body takes precedence over cookie
raw_token = (body.refresh_token if body and body.refresh_token else None) or refresh_token_cookie
if not raw_token:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail="No refresh token provided",
)
# Validate refresh token
payload = verify_token(body.refresh_token, expected_type="refresh")
payload = verify_token(raw_token, expected_type="refresh")
user_id_str = payload.get("sub")
if not user_id_str:
@@ -453,7 +491,7 @@ async def refresh_token(
)
new_refresh_token = create_refresh_token(user_id=user.id)
# Update cookie
# Rotate access token cookie
response.set_cookie(
key=ACCESS_TOKEN_COOKIE,
value=new_access_token,
@@ -462,6 +500,16 @@ async def refresh_token(
secure=_COOKIE_SECURE,
samesite="lax",
)
# Rotate refresh token cookie (silent token rotation)
response.set_cookie(
key=REFRESH_TOKEN_COOKIE,
value=new_refresh_token,
max_age=REFRESH_TOKEN_MAX_AGE,
httponly=True,
secure=_COOKIE_SECURE,
samesite="lax",
path="/api/auth/refresh",
)
return TokenResponse(
access_token=new_access_token,
@@ -501,6 +549,13 @@ async def logout(
secure=_COOKIE_SECURE,
samesite="lax",
)
response.delete_cookie(
key=REFRESH_TOKEN_COOKIE,
httponly=True,
secure=_COOKIE_SECURE,
samesite="lax",
path="/api/auth/refresh",
)
@router.post("/change-password", response_model=MessageResponse, summary="Change password for authenticated user")

View File

@@ -19,7 +19,7 @@ class TokenResponse(BaseModel):
class RefreshRequest(BaseModel):
refresh_token: str
refresh_token: Optional[str] = None # Optional: also accepted via httpOnly cookie
class UserMeResponse(BaseModel):

View File

@@ -145,7 +145,43 @@ export const useAuth = create<AuthState>((set, get) => ({
// 8. Store AUK and unlock key set
keyStore.setAUK(auk)
// TODO (Phase 30): Decrypt encrypted_key_set with AUK to get vault key
// Decrypt encrypted_key_set with AUK to get vault key + RSA private key.
// Non-fatal: if decryption fails (e.g. corrupted key set, wrong AUK) we log
// a warning and continue. Server-side Transit encryption still works; only
// Tier 1 (client-side) encrypted data will be inaccessible until re-auth.
if (result.encrypted_key_set) {
const ks = result.encrypted_key_set
try {
const b64 = (s: string) => Uint8Array.from(atob(s), (c) => c.charCodeAt(0))
// Unwrap vault key (AES-256-GCM) using AUK
const vaultKey = await crypto.subtle.unwrapKey(
'raw',
b64(ks.encrypted_vault_key),
auk,
{ name: 'AES-GCM', iv: b64(ks.vault_key_nonce) },
{ name: 'AES-GCM', length: 256 },
false, // non-extractable
['encrypt', 'decrypt'],
)
keyStore.setVaultKey(vaultKey)
// Unwrap RSA-OAEP private key using AUK
const privateKey = await crypto.subtle.unwrapKey(
'pkcs8',
b64(ks.encrypted_private_key),
auk,
{ name: 'AES-GCM', iv: b64(ks.private_key_nonce) },
{ name: 'RSA-OAEP', hash: 'SHA-256' },
false, // non-extractable
['decrypt'],
)
keyStore.setPrivateKey(privateKey)
} catch (e) {
console.warn('[auth] key set decryption failed (Tier 1 data will be inaccessible):', e)
}
}
// 9. Store Secret Key in IndexedDB for future logins on this device
await keyStore.storeSecretKey(email, secretKeyBytes)