harden dashboard API contracts and proxy errors

This commit is contained in:
Marco0300
2026-09-03 23:20:47 +02:00
parent 594da00240
commit 00bd49a894
4 changed files with 138 additions and 10 deletions
+19 -7
View File
@@ -3,6 +3,7 @@
from __future__ import annotations
import http.client
import json
import os
from http.server import SimpleHTTPRequestHandler, ThreadingHTTPServer
from urllib.parse import urlsplit
@@ -15,20 +16,29 @@ UPSTREAM_PORT = int(os.environ.get("API_UPSTREAM_PORT", "8000"))
class ProxyStaticHandler(SimpleHTTPRequestHandler):
proxy_api = True
def _send_json_error(self, status: int, code: str) -> None:
payload = json.dumps({"error": code}, separators=(",", ":")).encode()
self.send_response(status)
self.send_header("Content-Type", "application/json; charset=utf-8")
self.send_header("Cache-Control", "no-store, private")
self.send_header("Content-Length", str(len(payload)))
self.end_headers()
self.wfile.write(payload)
def _proxy_request(self) -> None:
parsed = urlsplit(self.path)
if parsed.path == "/api" or parsed.path.startswith("/api/"):
target = self.path
else:
self.send_error(404)
self._send_json_error(404, "not_found")
return
try:
length = int(self.headers.get("Content-Length", "0"))
except ValueError:
self.send_error(400, "invalid content length")
self._send_json_error(400, "invalid_content_length")
return
if length > MAX_PROXY_BODY:
self.send_error(413, "request body too large")
self._send_json_error(413, "request_body_too_large")
return
body = self.rfile.read(length) if length else None
headers = {
@@ -37,13 +47,14 @@ class ProxyStaticHandler(SimpleHTTPRequestHandler):
if key.lower() in {"accept", "content-type", "cookie", "user-agent", "x-request-id"}
}
headers["Host"] = f"{UPSTREAM_HOST}:{UPSTREAM_PORT}"
connection = http.client.HTTPConnection(UPSTREAM_HOST, UPSTREAM_PORT, timeout=15)
connection = None
try:
connection = http.client.HTTPConnection(UPSTREAM_HOST, UPSTREAM_PORT, timeout=15)
connection.request(self.command, target, body=body, headers=headers)
response = connection.getresponse()
payload = response.read(MAX_PROXY_BODY + 1)
if len(payload) > MAX_PROXY_BODY:
self.send_error(502, "upstream response too large")
self._send_json_error(502, "upstream_response_too_large")
return
self.send_response(response.status, response.reason)
for key, value in response.getheaders():
@@ -52,9 +63,10 @@ class ProxyStaticHandler(SimpleHTTPRequestHandler):
self.end_headers()
self.wfile.write(payload)
except (OSError, http.client.HTTPException) as exc:
self.send_error(502, f"api upstream unavailable: {exc}")
self._send_json_error(502, "api_upstream_unavailable")
finally:
connection.close()
if connection is not None:
connection.close()
def do_GET(self):
if self.path.startswith("/api"):