feat: v0.7.2 安全收口 · 断链接线 · 观测补洞 — 230 用例扩充与全量回归
CI / 类型检查 + Lint + 单元测试 (push) Failing after 5m43s
CI / 全量测试 (Electron ABI) (push) Failing after 5m25s
CI / 产物编译验证 (push) Successful in 10m1s

P1 修复面收口: /clear 全链路根治(前端清空联动 DB messages+摘要游标+TRACE 快照,
IPC 语义改"操作完成"; 流式中拒绝); web_browser open 补 SSRF 校验(Chromium 旁路关闭,
与 web_fetch/http_request 同源 validateSSRF); MCP 工具结果纳入注入扫描(mcp_* 前缀
按网络来源同级 full 模式, 收敛 resolveScanMode 单点); Trace 落库/入 store 双重瘦身
(tool_result base64/超长字段剥离, metadata 防 MB 级膨胀); 文本附件 512KB 闸门
(file.slice 首段读取+truncated 标志随消息持久化+主进程附件提示感知截断);
单实例锁(requestSingleInstanceLock + second-instance 聚焦已有窗口)

P2 安全纵深: ConfirmationHook 多窗口化(确认请求/超时提示改全窗口广播,
getAllWindows 空时回退 mainWindow, fail-closed 判定升级双通道); mcp_servers.headers
全链路接线(safeParseHeaders 容错解析+SSE/StreamableHTTP requestInit 注入+IPC 逐项
校验+设置页 JSON 输入, 远程 MCP 鉴权头可用)

P3 断链接线: llm:listModels IPC(六家 adapter 动态模型发现首次接线, 配置完整性
前置校验); Ollama pullModel IPC+设置页下载卡片(进度/取消/能力徽标, v0.7.0 死代码
激活); 后台会话运行指示(sessionRunStates 图+Sidebar 状态点, 多会话并发可见);
IR 卫生(移除 THINKING_START/END 死枚举, constraints 标注预留)

P4 质量与文档: i18n 第二阶段(确认弹框/侧栏/状态栏/AgentMonitor/终止原因出层,
外观设置 zh-CN/en-US 切换, ui.locale 持久化, 渲染时求值规避异步注册); README/D1
文档对齐(http_request 风险等级/用例数/实现状态注记); 版本号 0.7.2

测试: 507 → 737 用例(+230, 11 个新文件)。覆盖补齐: context-builder/consolidator/
orchestrator/workspace.service/session-recorder/config-layering/secure-config/
network-proxy + IPC mcp/tasks/memory/app/data 域 + 渲染层 store 与流事件管线纯函数。
测试驱动修复: workspace.appendMemory 中文分区 \b 词边界失效(JS \b 不含 CJK),
固化条目恒追加文件末尾产生重复分区头 → (?=\n|$) 前瞻断言根治

回归: typecheck 双端 0 错误; ESLint 0/0; 系统 Node 687 通过 50 跳过;
Electron ABI 全量 737/737 零跳过
This commit is contained in:
2026-08-30 00:09:25 +08:00
parent 36e4b28d93
commit 26169b7be4
55 changed files with 5673 additions and 434 deletions
+64 -4
View File
@@ -137,6 +137,38 @@ export function buildSafeEnv(): Record<string, string> {
return env;
}
/**
* v0.7.2 P2-8: 安全解析 MCP headers 列(JSON 对象,键值均为字符串)。
*
* 数据库中存储为 JSON 字符串(mcp_servers.headers 列自建表起即存在,
* 此前从未被读写)。解析规则:
* - 空/null → undefined(匿名连接)
* - 非对象 / 数组 / 含非字符串键值 → 丢弃该条目并 WARN(保持可用性优先,
* 不因单列损坏阻断整个 server 连接)
*/
/** @visibleForTesting 纯函数,供安全表测直接断言 */
export function safeParseHeaders(
raw: string | null | undefined,
): Record<string, string> | undefined {
if (!raw) return undefined;
try {
const parsed: unknown = JSON.parse(raw);
if (!parsed || typeof parsed !== 'object' || Array.isArray(parsed)) return undefined;
const out: Record<string, string> = {};
for (const [k, v] of Object.entries(parsed as Record<string, unknown>)) {
if (typeof v === 'string') {
out[k] = v;
} else {
log.warn(`MCP headers entry "${k}" dropped: value is not a string`);
}
}
return Object.keys(out).length > 0 ? out : undefined;
} catch {
log.warn(`MCP headers parse failed, ignoring: ${raw.slice(0, 100)}`);
return undefined;
}
}
// ===== 类型定义 =====
export type MCPServerStatus = 'connecting' | 'connected' | 'disconnected' | 'error';
@@ -149,6 +181,13 @@ export interface MCPServerConfig {
command?: string;
args?: string[];
url?: string;
/**
* v0.7.2 P2-8: 远程传输(sse / streamable-http)的自定义请求头。
* 用于 Bearer/Basic 鉴权或网关路由头;仅 HTTP 传输生效(stdio 走 buildSafeEnv)。
* 通过 requestInit 注入 —— MCP SDK 的 SSE/StreamableHTTP 客户端在
* SSE GET 流与 JSON-RPC POST 中统一合并 _commonHeaders(含 requestInit.headers)。
*/
headers?: Record<string, string>;
enabled: boolean;
}
@@ -278,6 +317,7 @@ export class MCPManager {
command: string | null;
args: string | null;
url: string | null;
headers: string | null;
}>;
const connectWithTimeout = (config: MCPServerConfig): Promise<unknown> =>
@@ -296,6 +336,8 @@ export class MCPManager {
command: row.command ?? undefined,
args: row.args ? safeParseArgs(row.args) : undefined,
url: row.url ?? undefined,
// v0.7.2 P2-8: headers 列接线(此前建表即存在但从未读写)
headers: safeParseHeaders(row.headers),
enabled: true,
};
return connectWithTimeout(config).catch((err) => {
@@ -342,10 +384,22 @@ export class MCPManager {
});
} else if (config.transport === 'sse' && config.url) {
// SSE 模式(远程 HTTP,旧式传输,保留向后兼容)
transport = new SSEClientTransport(new URL(config.url));
// v0.7.2 P2-8: 注入自定义请求头(非空时)—— requestInit.headers 同时
// 覆盖 SSE GET 流与 JSON-RPC POSTSDK _commonHeaders 统一合并点)
const headerInit = {
...(config.headers && Object.keys(config.headers).length > 0
? { requestInit: { headers: config.headers } }
: {}),
};
transport = new SSEClientTransport(new URL(config.url), headerInit);
} else if (config.transport === 'streamable-http' && config.url) {
// v0.4.1: streamable HTTP 模式(MCP 当前主流远程传输)
transport = new StreamableHTTPClientTransport(new URL(config.url));
const headerInit = {
...(config.headers && Object.keys(config.headers).length > 0
? { requestInit: { headers: config.headers } }
: {}),
};
transport = new StreamableHTTPClientTransport(new URL(config.url), headerInit);
} else {
throw new Error(
`Unsupported transport "${config.transport}". ` +
@@ -468,6 +522,7 @@ export class MCPManager {
command: string | null;
args: string | null;
url: string | null;
headers: string | null;
}
| undefined;
if (row) {
@@ -478,6 +533,7 @@ export class MCPManager {
command: row.command ?? undefined,
args: row.args ? safeParseArgs(row.args) : undefined,
url: row.url ?? undefined,
headers: safeParseHeaders(row.headers),
enabled: true,
});
}
@@ -490,6 +546,7 @@ export class MCPManager {
* 添加新的 MCP Server
*
* v0.4.1: 校验 transport 与对应字段匹配(stdio→commandsse/streamable-http→url
* v0.7.2 P2-8: headers 持久化(JSON 列;仅远程传输消费,stdio 忽略)
*/
async addServer(config: Omit<MCPServerConfig, 'id'>): Promise<void> {
const db = this.getDB();
@@ -505,8 +562,8 @@ export class MCPManager {
db.prepare(
`
INSERT INTO mcp_servers (id, name, transport, command, args, url, enabled)
VALUES (?, ?, ?, ?, ?, ?, 1)
INSERT INTO mcp_servers (id, name, transport, command, args, url, headers, enabled)
VALUES (?, ?, ?, ?, ?, ?, ?, 1)
`,
).run(
id,
@@ -515,6 +572,9 @@ export class MCPManager {
config.command ?? null,
config.args ? JSON.stringify(config.args) : null,
config.url ?? null,
config.headers && Object.keys(config.headers).length > 0
? JSON.stringify(config.headers)
: null,
);
if (config.enabled !== false) {