C-6: ConnectorRuntime enforces capability_mask per operation.
READ-only ACs cannot invoke MUTATE operations (wipe, lock, retire).
C-7: AC validated against database (exists, active, not expired)
before connector invocation.
C-9: Delegated AC capability bounded by delegator's capability.
C-10: Command counter uses atomic SQL increment with limit check.
M-23: expire_stale() uses same atomic SQL pattern.
H-1: Sensitive credential fields hidden from repr/logs via repr=False.
H-2: Stub backend requires ALLOW_STUB_CREDENTIALS=true to activate.
H-3: Kerberos backend raises CredentialResolutionError instead of
returning stub ticket.
H-4: Chronicle INTENT emitted before execution, RESULT after.
H-5: device_id validated as UUID before Graph API URL interpolation.
H-8: ConnectorRuntime enforces governance for all connector invocations.
Signed-off-by: Tyler King <tking@guildhouse.dev>
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Callable
|
|
|
|
from starlette.authentication import (
|
|
AuthCredentials,
|
|
AuthenticationBackend,
|
|
AuthenticationError,
|
|
UnauthenticatedUser,
|
|
)
|
|
from starlette.requests import HTTPConnection
|
|
from starlette.responses import PlainTextResponse, Response
|
|
from starlette.types import ASGIApp, Receive, Scope, Send
|
|
|
|
|
|
class AuthenticationMiddleware:
|
|
def __init__(
|
|
self,
|
|
app: ASGIApp,
|
|
backend: AuthenticationBackend,
|
|
on_error: Callable[[HTTPConnection, AuthenticationError], Response] | None = None,
|
|
) -> None:
|
|
self.app = app
|
|
self.backend = backend
|
|
self.on_error: Callable[[HTTPConnection, AuthenticationError], Response] = (
|
|
on_error if on_error is not None else self.default_on_error
|
|
)
|
|
|
|
async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
|
|
if scope["type"] not in ["http", "websocket"]:
|
|
await self.app(scope, receive, send)
|
|
return
|
|
|
|
conn = HTTPConnection(scope)
|
|
try:
|
|
auth_result = await self.backend.authenticate(conn)
|
|
except AuthenticationError as exc:
|
|
response = self.on_error(conn, exc)
|
|
if scope["type"] == "websocket":
|
|
await send({"type": "websocket.close", "code": 1000})
|
|
else:
|
|
await response(scope, receive, send)
|
|
return
|
|
|
|
if auth_result is None:
|
|
auth_result = AuthCredentials(), UnauthenticatedUser()
|
|
scope["auth"], scope["user"] = auth_result
|
|
await self.app(scope, receive, send)
|
|
|
|
@staticmethod
|
|
def default_on_error(conn: HTTPConnection, exc: Exception) -> Response:
|
|
return PlainTextResponse(str(exc), status_code=400)
|