Phase 2: Fix high-priority security and stability issues (H1-H7)

H1: Sanitize error endpoint context (strip tokens/passwords, limit depth/size)
H2: Rate limiter IP spoofing fix (trusted proxy CIDR check for X-Forwarded-For)
H3: CSRF middleware uses Redis singleton instead of per-request connection
H4: WebSocket origin verification added to both kommunikation and ai_ui_control
H5: Storage path traversal protection, get_url() returns relative URL not filesystem path
H6: Security headers middleware (HSTS, X-Content-Type-Options, X-Frame-Options, CSP, Referrer-Policy)
H7: Forward-repair migration 0045 for databases that ran original 0021/0027

Also: add trusted_proxy_cidrs to config, add verify_ws_origin to auth
This commit is contained in:
Agent Zero
2026-07-26 20:49:15 +02:00
parent 5ec1fc9b05
commit 604a2b7648
10 changed files with 360 additions and 38 deletions
+17
View File
@@ -91,6 +91,23 @@ def hash_token(token: str) -> str:
return hashlib.sha256(token.encode()).hexdigest()
def verify_ws_origin(websocket) -> bool:
"""Verify that the WebSocket upgrade request comes from an allowed origin.
Checks the Origin header against the configured CORS origins.
Returns True if the origin is allowed or if no CORS restriction is configured.
"""
from app.config import get_settings
settings = get_settings()
allowed_origins = settings.cors_origin_list
if not allowed_origins:
return True
origin = websocket.headers.get("origin", "")
if not origin:
return True # Non-browser clients don't send Origin
return origin in allowed_origins
async def create_session(
db: AsyncSession,
redis: aioredis.Redis,