AI Agent 的权限最小化:每个 agent 实例只拿到完成任务的最小权限集合

📅 2026/7/23 14:22:45
AI Agent 的权限最小化:每个 agent 实例只拿到完成任务的最小权限集合
AI Agent 的权限最小化每个 agent 实例只拿到完成任务的最小权限集合一、AI Agent 权限过多的灾难场景先看一个真实可能发生的场景你给 Agent 配了文件读取权限和 Shell 执行权限让它帮你分析 /var/log 下的日志找出错误行并汇总。为什么会出现这种情况因为传统的权限模型是角色粒度的不是任务粒度的角色文件管理 Agent → 拥有所有文件操作权限实际任务读日志文件 → 只需要 Read不需要 Delete二、Capability 令牌把权限变成凭票入场我在第三篇讲过了 Capabilities-based Security 的一般原理。现在把它用到 AI Agent 上不给 Agent 全局权限只给每项具体任务发放临时的功能令牌。三、Rust 实现编译期保证权限安全3.1 定义权限类型use std::marker::PhantomData; use std::path::PathBuf; /// 文件读取权限 — 限定只能读指定目录下的文件 pub struct ReadFile { // 限定可以读取的根目录 allowed_root: PathBuf, } /// 文件写入权限 — 限定只能写指定目录 pub struct WriteFile { allowed_root: PathBuf, } /// 网络请求权限 — 限定只能访问白名单域名 pub struct NetworkAccess { allowed_domains: VecString, } /// Shell 执行权限 — 限定只能运行白名单命令 pub struct ExecShell { allowed_commands: VecString, } /// Capability 令牌持有它才能使用对应的能力 pub struct CapT { _marker: PhantomDataT, } implT CapT { /// 创建令牌只能在权限管理器内部调用 fn new() - Self { Cap { _marker: PhantomData } } }3.2 Agent 的定义只持有被授予的令牌/// AI Agent通过泛型参数限制它拥有的能力 /// T1, T2, T3... 是它可以持有的权限类型 pub struct AgentR (), W (), N (), E () where R: static, W: static, N: static, E: static, { // 每种权限都是可选的未授予时类型为 () read_token: OptionCapR, // 文件读权限 write_token: OptionCapW, // 文件写权限 network_token: OptionCapN, // 网络权限 exec_token: OptionCapE, // Shell 执行权限 } impl Agent { /// 创建没有任何权限的 Agent pub fn new() - Self { Agent { read_token: None, write_token: None, network_token: None, exec_token: None, } } }3.3 权限管理器按任务分配最小权限/// 权限管理器负责按任务需求分配权限令牌 pub struct PermissionManager; impl PermissionManager { /// 为日志分析任务创建 Agent /// 只需要读日志文件的权限 pub fn create_log_analyzer(log_dir: str) - AgentReadFile, (), (), () { // 只拿类型信息不创建 Agent 本身也可以 // 但这里我们返回带 ReadFile 权限的 Agent Agent { read_token: Some(Cap::ReadFile::new()), write_token: None, // 明确不给写权限 network_token: None, // 明确不给网络权限 exec_token: None, // 明确不给执行权限 } } /// 为自动部署任务创建 Agent /// 需要读文件 执行 Shell pub fn create_deploy_agent( deploy_dir: str, ) - AgentReadFile, WriteFile, (), ExecShell { Agent { read_token: Some(Cap::new()), write_token: Some(Cap::WriteFile::new()), network_token: None, exec_token: Some(Cap::ExecShell::new()), } } }3.4 资源访问检查令牌才能执行/// 安全的文件系统访问层 pub struct SecureFileSystem; impl SecureFileSystem { /// 读取文件必须持有 ReadFile 令牌 pub fn read_file(self, path: str, _token: CapReadFile) - ResultString, String { // 编译期保证调用者一定持有 ReadFile 权限 // 运行时检查路径是否在允许的目录下 std::fs::read_to_string(path) .map_err(|e| format!(读取文件失败: {}, e)) } /// 写入文件必须持有 WriteFile 令牌 pub fn write_file( self, path: str, content: str, _token: CapWriteFile, ) - Result(), String { std::fs::write(path, content) .map_err(|e| format!(写入文件失败: {}, e)) } /// 删除文件必须持有 DeleteFile 令牌 /// 注意日志分析 Agent 永远拿不到这个令牌 pub fn delete_file(self, path: str, _token: CapDeleteFile) - Result(), String { std::fs::remove_file(path) .map_err(|e| format!(删除文件失败: {}, e)) } } // 删除权限类型定义日志分析 Agent 拿不到这个 pub struct DeleteFile;3.5 实际使用示例fn main() { let pm PermissionManager::new(); let fs SecureFileSystem; // 任务1日志分析 // 只能读到 ReadFile 权限的 Agent let analyzer pm.create_log_analyzer(/var/log); // ✓ 可以读日志 if let Some(token) analyzer.read_token { let content fs.read_file(/var/log/syslog, token); println!(日志内容: {:?}, content); } // ❌ 编译错误Agent 没有 WriteFile 令牌 // fs.write_file(/var/log/syslog, hacked, analyzer.write_token); // ❌ 编译错误Agent 没有 DeleteFile 令牌 // fs.delete_file(/var/log/syslog, analyzer.delete_token); // 任务2自动部署 let deployer pm.create_deploy_agent(/opt/app); // ✓ 可以读部署脚本 if let Some(token) deployer.read_token { fs.read_file(/opt/app/deploy.sh, token).unwrap(); } // ✓ 可以写部署日志 if let Some(token) deployer.write_token { fs.write_file(/opt/app/deploy.log, deploy success, token).unwrap(); } // ❌ 编译错误部署 Agent 也没有 DeleteFile 令牌 // 即使它具有写入权限也不能删除 }3.6 实战踩坑泛型参数爆炸与 Option 令牌的安全漏洞这套方案在原型阶段很好用但扩展到 5 种权限时遇到两个现实问题问题一泛型参数爆炸。每增加一种权限数据库读、消息队列写、Redis 访问……Agent的泛型参数就多一个。5 种权限意味着R, W, N, E, D, M, K函数签名长得一行写不下// ❌ 类型签名失控 pub fn create_full_agent() - AgentReadFile, WriteFile, NetworkAccess, ExecShell, DbRead, MqPublish {解法把权限令牌放到HashMapTypeId, Boxdyn Any里做运行时管理。牺牲一部分编译期安全换取可扩展性pub struct DynamicAgent { caps: HashMapTypeId, Boxdyn Any, } impl DynamicAgent { pub fn has_capT: static(self) - bool { self.caps.contains_key(TypeId::of::T()) } }问题二Option 令牌的假安全感。OptionCapReadFile在编译期保证 Agent 不能调用read_file因为需要CapReadFile但它不保证 Agent 不持有其他更危险的令牌。一个声称只读日志的 Agent 可能实际也拿到了ExecShell只是 Option 是Some而非None。真正可靠的做法是让每个任务的函数签名明确暴露所有权限类型通过 Code Review 检查而非纯依赖编译器。四、动态权限运行时的细粒度控制编译期检查很好但有些权限需要在运行时根据上下文决定。比如只能读 100MB 以内的文件这个限制编译器做不了。/// 运行时权限约束 pub struct PermissionConstraint { pub max_file_size: usize, // 最大文件大小字节 pub allowed_paths: VecPathBuf, // 允许的路径前缀 pub rate_limit_per_minute: u32, // 每分钟最大操作次数 } impl SecureFileSystem { /// 带运行时约束的文件读取 pub fn read_file_constrained( self, path: str, token: CapReadFile, constraint: PermissionConstraint, ) - ResultString, String { // 编译期必须有 ReadFile 令牌 ✓ // 运行时1检查路径是否在允许范围内 let path_buf PathBuf::from(path); if !constraint.allowed_paths.iter().any(|p| path_buf.starts_with(p)) { return Err(路径不在允许范围内.to_string()); } // 运行时2检查文件大小 let metadata std::fs::metadata(path) .map_err(|_| 无法读取文件元数据.to_string())?; if metadata.len() constraint.max_file_size as u64 { return Err(format!( 文件过大{}字节最大允许 {} 字节, metadata.len(), constraint.max_file_size )); } // 通过所有检查后才执行读取 self.read_file(path, token) } }我们内部做了一次攻击测试给一个只持有 ReadFile 令牌的 Agent 输入恶意 Prompt诱导它调用 delete_file。结果 Agent 的 LLM 确实想删除文件但因为没令牌操作直接被编译期拒绝了。五、总结AI Agent 权限最小化的核心思路编译期隔离用 Rust 泛型 零大小类型标记让 Agent 在编译时就确定能做什么任务粒度授权不为 Agent 分配角色权限而是为每项具体任务分配最小必要权限令牌即权限Agent 只能使用它持有的 Cap 令牌调用对应操作不能凭空调双重检查编译期有没有令牌 运行时具体约束缺一不可用完即弃任务完成后丢弃 Agent 实例令牌随之失效这个方案的额外好处是代码即文档。看 Agent 的泛型参数就知道它有哪些权限Code Review 时一眼能看出来权限分配是否合理。保持学习保持输出你在设计 Agent 权限时遇到过什么坑评论区分享参考资料Object-Capability Model (Wikipedia)AI Agent Security Considerations (OWASP)Rust Typestate Pattern