feat: v0.7.2 安全收口 · 断链接线 · 观测补洞 — 230 用例扩充与全量回归
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:
@@ -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 POST(SDK _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→command,sse/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) {
|
||||
|
||||
Reference in New Issue
Block a user