feat: v0.4.0 四阶段迭代 — 安全加固 + 工程基线 + 架构重构 + 双 Provider 扩展
P0 安全修复: - API Key 加密存储(safeStorage 密钥链,版本化前缀,历史明文平滑兼容) - 间接提示注入防护(SecurityScanHook 工具结果深扫描,网络工具脱敏/本地工具警示分级) - error:report IPC 断链修复(渲染进程错误上报落 electron-log + 审计) - abort 信号贯通工具层(run_command/dev-tools 子进程随会话中断终止) - run_command 沙箱加固(cd 系统目录/敏感文件读取拦截 + chcp 前缀剥离防解析退化) - .env 真实生效(dotenv 回退加载,应用内配置优先) P1 工程基础: - ESLint 9 flat config + 全部 34 条存量 warnings 清零(零容忍基线) - 测试基线 118 用例 11 文件(token/文件防护/权限/沙箱/注入/命令/引擎/注册表/审计链/摘要分层) - test:electron 双模式(ELECTRON_RUN_AS_NODE 跑 Electron ABI,SQLite 套件全执行) - SessionRecorder 多会话隔离 + 9 种 TRACE 事件补全(含最终轮 iteration_end) - Provider 故障转移(重试耗尽/不可重试一次性切换 fallback + 前端通知) - MCP 真就绪(等待全部连接完成再广播 tools:ready) - SLO/HealthChecker 真实接入(60s 巡检 + 托盘状态) - CONFIG_DEFAULTS 单一来源(消除 SEED 双源漂移) P2 架构升级: - handlers.ts 1940 行拆分为 13 个 IPC 域模块(防重入注册 + 多窗口广播) - AgentEngineManager 每会话独立引擎(LRU 30 + adapter 工厂隔离 abort 信号) - TaskOrchestrator EngineProvider 改造 + abortByParent 联动中断 SubAgent - 会话摘要分层上下文(session_summaries 滚动摘要 + 截断游标清理防因果污染) - 消息编辑重发/重新生成(truncateAfter IPC + store 动作 + UI) - Markdown 导出 / WebSearch 并行抓取(并发 3)/ 记忆 TF 缓存 / 版本构建期注入 P3 能力扩展: - OpenAI Adapter(o 系列推理模型 reasoning_effort/max_completion_tokens) - Anthropic Adapter(原生 Messages API:tool_use 块/角色合并/thinking budget/图片 base64/SSE 事件机) - 设置页/Onboarding 六 Provider 全链路接入
This commit is contained in:
@@ -0,0 +1,93 @@
|
||||
/**
|
||||
* PolicyEngine 单元测试(P1-14 测试基线)
|
||||
* 覆盖:默认策略、deniedPatterns 深度扫描、频率限制、通配符策略
|
||||
*/
|
||||
|
||||
import { describe, it, expect } from 'vitest';
|
||||
import { PolicyEngine, DEFAULT_POLICIES } from '../permissions';
|
||||
|
||||
describe('PolicyEngine 默认策略', () => {
|
||||
it('所有内置工具均有策略配置', () => {
|
||||
const knownTools = [
|
||||
'read_file', 'write_file', 'list_directory', 'search_files', 'delete_file',
|
||||
'file_move', 'file_info', 'file_editor', 'code_search', 'diff_viewer',
|
||||
'web_search', 'web_fetch', 'web_browser', 'http_request',
|
||||
'memory_store', 'memory_search', 'run_command', 'task_manager',
|
||||
'delegate_task', 'git_status', 'git_diff', 'git_log', 'git_commit',
|
||||
'lint_code', 'run_tests', 'project_info', 'think', 'view_image',
|
||||
];
|
||||
for (const tool of knownTools) {
|
||||
expect(DEFAULT_POLICIES.some((p) => p.toolName === tool)).toBe(true);
|
||||
}
|
||||
});
|
||||
|
||||
it('未配置策略的工具被拒绝(fail-closed)', () => {
|
||||
const engine = new PolicyEngine();
|
||||
const result = engine.checkAuthorization('unknown_tool_xyz', {});
|
||||
expect(result.authorized).toBe(false);
|
||||
expect(result.reason).toContain('No policy configured');
|
||||
});
|
||||
|
||||
it('read_file 默认放行', () => {
|
||||
const engine = new PolicyEngine();
|
||||
expect(engine.checkAuthorization('read_file', { file_path: 'a.ts' }).authorized).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
describe('deniedPatterns 深度扫描', () => {
|
||||
it('read_file 访问 /etc/passwd 被拒绝', () => {
|
||||
const engine = new PolicyEngine();
|
||||
const result = engine.checkAuthorization('read_file', { file_path: '/etc/passwd' });
|
||||
expect(result.authorized).toBe(false);
|
||||
});
|
||||
|
||||
it('嵌套对象中的危险路径被拒绝(deepScanStrings)', () => {
|
||||
const engine = new PolicyEngine();
|
||||
const result = engine.checkAuthorization('read_file', {
|
||||
nested: { deep: { path: '/etc/passwd' } },
|
||||
});
|
||||
expect(result.authorized).toBe(false);
|
||||
});
|
||||
|
||||
it('run_command 参数中包含 MEMORY.md 被拒绝', () => {
|
||||
const engine = new PolicyEngine();
|
||||
const result = engine.checkAuthorization('run_command', { command: 'cat MEMORY.md' });
|
||||
expect(result.authorized).toBe(false);
|
||||
});
|
||||
|
||||
it('正常路径不误判', () => {
|
||||
const engine = new PolicyEngine();
|
||||
expect(engine.checkAuthorization('read_file', { file_path: 'src/main.ts' }).authorized).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
describe('频率限制(滑动窗口)', () => {
|
||||
it('超过 maxFrequency 后被限流', () => {
|
||||
const engine = new PolicyEngine();
|
||||
// web_search 默认 maxFrequency: 10
|
||||
for (let i = 0; i < 10; i++) {
|
||||
expect(engine.checkAuthorization('web_search', { query: 'x' }).authorized).toBe(true);
|
||||
engine.recordCall('web_search');
|
||||
}
|
||||
const blocked = engine.checkAuthorization('web_search', { query: 'x' });
|
||||
expect(blocked.authorized).toBe(false);
|
||||
expect(blocked.reason).toContain('Rate limit exceeded');
|
||||
});
|
||||
|
||||
it('限流只影响对应工具', () => {
|
||||
const engine = new PolicyEngine();
|
||||
for (let i = 0; i < 10; i++) engine.recordCall('web_search');
|
||||
// read_file 无频率限制
|
||||
expect(engine.checkAuthorization('read_file', {}).authorized).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
describe('通配符策略(mcp_*)', () => {
|
||||
it('MCP 动态工具命中 mcp_* 通配策略', () => {
|
||||
const engine = new PolicyEngine();
|
||||
const result = engine.checkAuthorization('mcp_filesystem_read_file', {});
|
||||
// mcp_* 策略:EXTERNAL_ACTION + requireConfirmation
|
||||
expect(result.authorized).toBe(true);
|
||||
expect(result.requiresConfirmation).toBe(true);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user