thzxx
e4d81d8247
本次升级基于完整代码审查，修复 Critical/High/Medium/Low 四级共 96 项问题， 并通过返工审计修复 10 项遗留问题，tsc 双端类型检查零错误。 Critical (10/10 完成): - C-4: command.ts 接入 shell-quote 进行 token-level 注入检测，替代原有正则匹配 可防御 r"m" -rf /、$'rm'、$(echo rm) 等字符串拼接绕过 High (11/11 完成): - 竞态保护、Promise.allSettled、AbortController 资源泄漏、IPC 参数校验等 Medium (55/55 完成): - 事务保护、敏感数据脱敏、枚举校验、MUI v9 Stack prop 迁移、 React 组件 cancelled 标志、类型收窄等 Low (20/20 完成): - 辅助方法提取（flushToolCallBuffer/scoreAndPushMemory/tryAddColumn 等） - nanoid 统一替代 Date.now()+Math.random() - confirm() 替换为 MUI Dialog、useMemo 缓存、魔法数字命名化等 返工审计修复 (10/10 完成): - L-11: LogsSettings 残留的原生 confirm()/alert() 全部替换为 MUI Dialog/Alert - M-53: MemoryViewer handleSearch 独立 ref，修复 searching 状态卡死 - M-42: 脱敏短值（length <= 4）泄露修复 - M-47: tasks:update 补全 title/description 类型校验 - L-9: ollama.adapter 非流式路径 nanoid 统一 - M-45: audit:query limit 策略与 memory:listAll 一致化 - SettingsModal handleConfirmRemove 补全 try/catch + loadServers cleanup - L-15: CommandPalette useMemo 补全 sessions 响应式依赖 - useAgentStream 事件类型补全 seq/timestamp 字段 新增依赖: shell-quote + @types/shell-quote 版本号: 0.3.0 -> 0.3.1
109 lines
4.1 KiB
TypeScript
109 lines
4.1 KiB
TypeScript
/**
|
||
* File Guard — 受保护文件守卫
|
||
*
|
||
* 确保工作空间根目录的 MEMORY.md 只能由系统内部（WorkspaceService）管理，
|
||
* 任何工具（read_file / write_file / search_files / run_command 等）均禁止直接读写。
|
||
*
|
||
* 注意：仅保护工作空间根目录的 MEMORY.md，
|
||
* 子目录或其他位置的同名文件不受限制。
|
||
*/
|
||
|
||
import { resolve, sep } from 'path';
|
||
import { realpathSync } from 'fs';
|
||
|
||
/**
|
||
* 受保护文件名列表（工作空间根目录）
|
||
*/
|
||
const PROTECTED_FILES = ['MEMORY.md'];
|
||
|
||
/**
|
||
* 检查目标路径是否为工作空间根目录的受保护文件
|
||
*
|
||
* @param filePath 用户传入的文件路径（绝对或相对）
|
||
* @param workspacePath 当前工作空间根路径
|
||
* @returns true 如果路径指向受保护文件
|
||
*/
|
||
export function isProtectedWorkspaceFile(
|
||
filePath: string,
|
||
workspacePath: string,
|
||
): boolean {
|
||
const resolved = resolve(workspacePath, filePath);
|
||
const workspaceRoot = resolve(workspacePath);
|
||
|
||
for (const protectedName of PROTECTED_FILES) {
|
||
const protectedPath = resolve(workspaceRoot, protectedName);
|
||
if (resolved === protectedPath) {
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
/**
|
||
* 检查路径是否在工作空间内（防止路径遍历攻击）
|
||
*
|
||
* 修复前缀碰撞漏洞：`/home/user/app-evil` 不应被误判为在 `/home/user/app` 内。
|
||
*
|
||
* M-22 修复: 添加 realpathSync 二次校验防止符号链接逃逸
|
||
* 攻击场景：工作空间内创建符号链接 `ln -s /etc/passwd workspace/leak.txt`，
|
||
* 字符串校验会通过（leak.txt 在 workspace 内），但实际读取的是 /etc/passwd。
|
||
*
|
||
* 注意：realpathSync 在路径不存在时会抛 ENOENT，此时降级为字符串校验
|
||
* （write_file 的目标文件可能尚不存在，无法 realpath）。
|
||
*
|
||
* @see project_memory.md — sandbox validatePath must perform realpathSync secondary check
|
||
* @param filePath 用户传入的文件路径
|
||
* @param workspacePath 当前工作空间根路径
|
||
* @returns true 如果路径在工作空间内
|
||
*/
|
||
export function isPathWithinWorkspace(
|
||
filePath: string,
|
||
workspacePath: string,
|
||
): boolean {
|
||
const resolved = resolve(workspacePath, filePath);
|
||
const workspaceRoot = resolve(workspacePath);
|
||
|
||
// 第一层：字符串前缀校验（快速路径）
|
||
const stringCheck = resolved === workspaceRoot || resolved.startsWith(workspaceRoot + sep);
|
||
if (!stringCheck) return false;
|
||
|
||
// 第二层：realpathSync 二次校验（防范符号链接逃逸）
|
||
// 仅对实际存在的路径做 realpath 校验；不存在的路径（如 write_file 目标）降级为字符串校验
|
||
try {
|
||
const realResolved = realpathSync(resolved);
|
||
const realWorkspaceRoot = realpathSync(workspaceRoot);
|
||
return realResolved === realWorkspaceRoot || realResolved.startsWith(realWorkspaceRoot + sep);
|
||
} catch {
|
||
// 路径不存在（ENOENT）或 realpath 失败 → 降级为字符串校验结果
|
||
return stringCheck;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 检查命令字符串是否尝试访问工作空间根目录的受保护文件
|
||
*
|
||
* 用于 run_command 工具的命令校验。
|
||
* 仅匹配直接引用的 MEMORY.md（前面是命令起始/空白/引号/分号/管道），
|
||
* 不拦截子目录路径中的同名文件（如 subdir/MEMORY.md 或 subdir\MEMORY.md）。
|
||
*
|
||
* 注意：run_command 的工作目录固定为 workspacePath，因此裸引用 MEMORY.md
|
||
* 等价于工作空间根目录的 MEMORY.md。
|
||
*
|
||
* @param command Shell 命令字符串
|
||
* @returns true 如果命令直接引用了受保护文件名
|
||
*/
|
||
export function commandTouchesProtectedFile(command: string): boolean {
|
||
const lowerCmd = command.toLowerCase();
|
||
for (const protectedName of PROTECTED_FILES) {
|
||
const lowerName = protectedName.toLowerCase();
|
||
// 前面是起始/空白/引号/分号/管道/&/>；后面是结束/空白/引号/分号/管道/&/</>
|
||
// 这样 subdir/MEMORY.md 和 subdir\MEMORY.md 不会被匹配（前面是 / 或 \）
|
||
const escaped = lowerName.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
|
||
const regex = new RegExp(`(?:^|[\\s"'|;&>])${escaped}(?:$|[\\s"'|;&<])`, 'i');
|
||
if (regex.test(lowerCmd)) {
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|