mirror of
https://github.com/ThearchyHelios/AgentLab.git
synced 2026-11-15 07:51:57 +00:00
ThearchyHelios
a05efbf27a
以前服务端一个安全头都不发，任何站点都能把 AgentLab 嵌进 iframe：同站门户（Lax 会话 cookie 会随请求带上）和关登录的部署 都有被点击劫持的可能，也没有「允许谁嵌」的白名单可配。 - 新增 app/web_security.py，纯 ASGI 中间件，main.py 里加在最外层：CORS 预检、登录检查自己回的 401 / 403 也带上； WebSocket 不经过它；接口自己设过的同名头（文件下载的 nosniff）不覆盖、不重复 - frame-ancestors 缺省只有 'self'，这时另加 X-Frame-Options: SAMEORIGIN；AGENTLAB_FRAME_ANCESTORS 配了同站门户的白名单时 只发 CSP（XFO 写不了白名单，两个都发会让只认 XFO 的老浏览器把门户也拦掉） - 分享页 /s 只认 AGENTLAB_EMBED_ORIGINS、Referrer-Policy 用 no-referrer。分享链接第 3 波才加，这一波先把配置读进来、 头的规则写好 - 严格的 script-src 先只发 Content-Security-Policy-Report-Only：mount_web 启动时给构建产物 index.html 里的内联主题 脚本算 sha256，只在页面响应上发；接口返回的 HTML（导出的报告自带 meta CSP）不发、不动。观察一个版本没有误报再强制执行 - HSTS 交给终结 https 的反向代理 - 两个白名单只收「协议://主机[:端口]」（可以 *. 开头），带路径、裸 *、没写协议的启动时就报错；deploy/agentlab.env 写明 两个变量和缺省行为 测试 tests/test_web_security.py：页面、/api、中间件自己的 401、CORS 预检都带头；白名单生效时不发 XFO；/s 的规则； 上报策略只在页面上、哈希对得上；已有的头不重复；WebSocket 不受影响；白名单写错启动报错；子进程起真实的 app.main 核对 环境变量接线。
91 lines
5.4 KiB
Python
91 lines
5.4 KiB
Python
"""安全响应头（04 细案 C1）：页面不能被别的站点套进 iframe 诱导点击，浏览器不按内容猜类型，链接不把完整地址带给别的站点。
|
||
|
||
以前服务端一个安全头都不发：任何站点都能把 AgentLab 嵌进 iframe。同站的门户页面（SameSite=Lax 的会话 cookie 会随请求
|
||
带上）和关登录的部署，都有被点击劫持的可能；也没有一条「允许谁嵌」的白名单可配。
|
||
|
||
纯 ASGI 中间件（同 auth/middleware.py，main.py 里加在最外层）：在 http.response.start 上补头，WebSocket 不经过它。
|
||
所有 HTTP 响应（页面、/api、登录检查自己回的 401 / 403、CORS 预检）都补：
|
||
- X-Content-Type-Options: nosniff；
|
||
- Referrer-Policy: strict-origin-when-cross-origin（分享页 /s 用 no-referrer：分享的 key 在地址的 # 后面，本来不随
|
||
Referer 走，这里再收一道）；
|
||
- Content-Security-Policy: frame-ancestors。缺省只有 'self'；AGENTLAB_FRAME_ANCESTORS 列出允许嵌整站的同站门户；分享页
|
||
/s 只认 AGENTLAB_EMBED_ORIGINS（分享链接第 3 波才加，路由在那时挂，头的规则先在这里定好）；
|
||
- 只有 frame-ancestors 是 'self' 时另加 X-Frame-Options: SAMEORIGIN，给不认 CSP 的老浏览器。XFO 写不了白名单，有白名单
|
||
时只发 CSP：两个都发的话，只认 XFO 的老浏览器会把白名单里的门户也拦掉（认 CSP 的浏览器以 frame-ancestors 为准）。
|
||
接口自己设过的同名头（文件下载的 nosniff）不覆盖、不重复。
|
||
|
||
严格的 script-src 先只上报、不拦（Content-Security-Policy-Report-Only）：页面里有一段首帧前定主题的内联脚本，强制执行
|
||
的话写错一个哈希页面就会先闪一帧深色，构建会不会产出别的内联内容也没核实。所以 mount_web（app/web.py）启动时给构建产物
|
||
index.html 里的内联脚本算 sha256（script_policy），只在页面这一个响应上发；观察一个版本没有误报，再改成强制执行。没配
|
||
上报地址：违规只在浏览器的控制台里看得到。接口返回的 HTML（导出的报告、可验证报告）文件里自带 meta CSP，不发它，也不动。
|
||
|
||
HSTS 交给终结 https 的反向代理：这里不知道对外是不是 https、要不要连子域名一起管，发错了浏览器会按 max-age 记很久，
|
||
撤不回来。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import base64
|
||
import hashlib
|
||
import re
|
||
from typing import Any
|
||
|
||
#: 分享页的路径（/s#<分享 key>）：只认 /s 这一段，/settings、/studio 不算
|
||
SHARE_PATH = "/s"
|
||
_SELF = "'self'"
|
||
_DEFAULT_REFERRER = b"strict-origin-when-cross-origin"
|
||
_SHARE_REFERRER = b"no-referrer"
|
||
#: 页面里不带 src 的 <script>：要算哈希的内联脚本
|
||
_INLINE_SCRIPT = re.compile(r"<script\b(?![^>]*\bsrc\s*=)[^>]*>(.*?)</script\s*>", re.IGNORECASE | re.DOTALL)
|
||
REPORT_ONLY_HEADER = "Content-Security-Policy-Report-Only"
|
||
|
||
|
||
def is_share_path(path: str) -> bool:
|
||
return path == SHARE_PATH or path.startswith(SHARE_PATH + "/")
|
||
|
||
|
||
def frame_policy(origins: list[str]) -> tuple[bytes, bool]:
|
||
"""(CSP 的值, 要不要另加 X-Frame-Options)。origins 是白名单（config 里校验过、去掉了 'self'）。"""
|
||
sources = [_SELF, *[o for o in origins if o != _SELF]]
|
||
return f"frame-ancestors {' '.join(sources)}".encode(), len(sources) == 1
|
||
|
||
|
||
def script_policy(html: str) -> str:
|
||
"""页面的严格 script-src：本站的脚本文件（'self'）加上页面里每段内联脚本的 sha256（按原文的 UTF-8 字节算，同浏览器）。"""
|
||
hashes = [base64.b64encode(hashlib.sha256(body.encode()).digest()).decode()
|
||
for body in _INLINE_SCRIPT.findall(html)]
|
||
return " ".join(["script-src", _SELF, *(f"'sha256-{h}'" for h in dict.fromkeys(hashes))])
|
||
|
||
|
||
class SecurityHeaders:
|
||
"""frame_ancestors / embed_origins 不传时取 core/config 的 AGENTLAB_FRAME_ANCESTORS、AGENTLAB_EMBED_ORIGINS（测试传别的）。"""
|
||
|
||
def __init__(self, app: Any, *, frame_ancestors: list[str] | None = None,
|
||
embed_origins: list[str] | None = None) -> None:
|
||
from app.core.config import settings
|
||
|
||
self.app = app
|
||
self.site = frame_policy(settings.frame_ancestors if frame_ancestors is None else frame_ancestors)
|
||
self.share = frame_policy(settings.embed_origins if embed_origins is None else embed_origins)
|
||
|
||
async def __call__(self, scope: dict[str, Any], receive: Any, send: Any) -> None:
|
||
if scope["type"] != "http":
|
||
await self.app(scope, receive, send)
|
||
return
|
||
share = is_share_path(scope.get("path") or "")
|
||
frame, xfo = self.share if share else self.site
|
||
wanted = [(b"x-content-type-options", b"nosniff"),
|
||
(b"referrer-policy", _SHARE_REFERRER if share else _DEFAULT_REFERRER),
|
||
(b"content-security-policy", frame)]
|
||
if xfo:
|
||
wanted.append((b"x-frame-options", b"SAMEORIGIN"))
|
||
|
||
async def send_with_headers(message: dict[str, Any]) -> None:
|
||
if message["type"] == "http.response.start":
|
||
headers = list(message.get("headers") or [])
|
||
present = {name.lower() for name, _ in headers}
|
||
headers += [(name, value) for name, value in wanted if name not in present]
|
||
message = {**message, "headers": headers}
|
||
await send(message)
|
||
|
||
await self.app(scope, receive, send_with_headers)
|