AI工程语言

Rust 实现最小权限命令执行器

2026-04-15 #安全#Sandbox#Rust

执行器应使用参数数组、允许列表和资源限制。

问题背景

生成式模型擅长处理模糊语言和提出候选方案,却不天然保证状态一致、权限正确或动作可恢复。讨论“Rust 实现最小权限命令执行器”时,重点不应只是跑通演示,而是系统失败后是否知道发生了什么、能否安全继续,以及结果能否重复验证。

设计方案

解析 Program 与 Args,校验目录,禁用 shell 展开并限制环境和输出。

在“受限命令执行器”进入实现前,先标注输入来源、调用身份和数据等级;输出通过结构与领域校验后才允许进入下一步。涉及外部副作用时,执行记录必须包含稳定业务键和脱敏证据。

实施产物:受限命令执行器

命令使用允许列表且不经过 shell;参数逐项传递;设置工作目录、环境白名单、资源与超时;捕获截断输出。先在测试环境保存一条完整运行证据,再接入真实写操作;每次变更都必须能定位到配置、代码和数据版本。

1
2
3
4
5
6
7
#[derive(Debug, Deserialize)]
#[serde(deny_unknown_fields)]
struct RustSandboxRunnerConfig { timeout_ms: u64, max_in_flight: usize }

async fn rust_sandbox_runner(cfg: &RustSandboxRunnerConfig) -> Result<(), ToolError> {
tokio::time::timeout(Duration::from_millis(cfg.timeout_ms), execute(cfg)).await??
}

示例只表达边界,不是可直接复制的完整框架配置。落地时应使用当前依赖版本的官方 API,并把超时、身份和审计接到现有基础设施。

实施步骤

  1. 选一个可回放的小任务,写清输入、成功条件和禁止发生的副作用。
  2. 实现“受限命令执行器”,把模型建议与真正执行隔离。
  3. 人工走通成功、拒绝和恢复三条路径,保存 run_id 与关键事件。
  4. 加入自动化回归,再用小比例非关键流量观察指标。
  5. 达到门槛后逐步放量;任一安全约束失败立即停止新任务并回退版本。

失败演练与验收

首要反例是:拼接命令字符串造成注入。不要只验证顺利完成;还要注入超时、重复请求、取消、无权限输入和依赖返回畸形数据。

本篇的硬性验收是:用分号、路径穿越、超长输出和超时样本验证全部受控。同时记录 拒绝次数、超时回收率、输出截断数。验收脚本应按结构化字段判定,不以“回答看起来合理”代替测试。

上线与回滚

“受限命令执行器”上线前要固定模型、提示、规则、工具 schema 与数据版本,并用影子或灰度流量比较旧版。若 拒绝次数、超时回收率、输出截断数 越过门槛,先停止新 Run,再处理在途任务,恢复旧组合后逐项核对未决副作用。

适用边界

当场景是“不执行外部进程”时,这套设计的维护成本通常高于收益。权限、金额、删除和一致性约束始终交给确定性代码,不能交给模型自行判断。

命令模型拒绝 Shell 字符串

字段 约束
program 来自允许列表的绝对路径/工具 ID
args 独立 argv,逐项长度与格式校验
cwd 解析后必须在任务根目录内
env 从空白集合加入允许键
stdin/stdout 有界、分离、标记截断
deadline 墙钟时间 + 终止宽限期

执行器不调用 sh -c,因此分号、管道和命令替换只是普通参数。程序 ID 映射由可信配置维护,不能让模型传任意绝对路径。cwd 和文件参数在打开后验证真实目标,防止 symlink 与 .. 逃逸。

子进程进入独立进程组/作业对象,超时先请求终止,宽限期后清理整个树。stdout/stderr 并发读取到有界 ring/buffer,达到限制后继续排空或终止进程,避免子进程因管道写满死锁。返回 exit code、signal、timed_out、truncated 与输出摘要。

属性测试生成 shell 元字符、Unicode、极长 argv 和路径变体;集成测试生成孙进程、无限输出与忽略终止信号。底层仍应运行在 OS 沙箱、资源配额和网络默认拒绝环境中,参数校验不是隔离替代品。

第二轮实践复核

围绕“受限命令执行器”建立一个不接触生产数据的最小项目,固定运行时、框架、模型、提示、规则和工具 schema 版本。正文代码块若缺少完整依赖、导入或领域类型,应视为架构伪代码;只有放入示例仓库并通过下述命令的代码才标记为可运行。

  • 项目验证:cargo test --all-targets && cargo clippy -- -D warnings
  • 回归夹具:把输入、结构化输出、工具轨迹和副作用记录为 JSON,用固定版本执行正常、拒绝、超时、取消与重复请求五组回归样本。
  • 必留证据:run_id、版本组合、终止原因、动作幂等键,以及 拒绝次数、超时回收率、输出截断数。
  • 通过条件:未批准写操作为零副作用;相同幂等键重复执行只产生一个业务结果;超时或恢复后状态可解释。
  • 失败处理:先停新任务,再核对在途动作;回滚模型、提示、规则、工具与状态迁移的完整组合,不能只换回模型。

复用顺序

  1. 为“Rust 实现最小权限命令执行器”建立确定性假工具,先跑契约与失败用例,不接触真实凭据。
  2. 围绕“受限命令执行器”接入只读能力,核对权限裁剪、脱敏字段和 Trace 完整性。
  3. 需要写能力时先输出 dry-run 差异,再绑定审批结果与一次性动作哈希。
  4. 以本文验收指标比较影子流量;达到门槛才放量,越线立即执行文中的组合回滚。

实践与资料复核日期:2026-08-22;框架 API 以锁定版本的官方文档为准。

参考资料

框架 API 会持续演进,复制示例前应再次确认当前版本。可持续的 AI 工程不依赖某个模型偶尔表现出色,而依赖清晰契约、可靠 Harness 与可重复验证。

评论
分享

评论