用 Gemini CLI 做威胁建模:命令文件与 GEMINI.md 配合
「威胁建模」是安全工程师与开发团队的高频任务。它的输入通常是架构说明,期望得到的是威胁清单与缓解措施。这类有固定流程、又需要一点判断力的工作,很适合封装成智能体 Skill。
本文介绍如何在 Gemini CLI 上实现这个技能,包括它的形态、示例配置、启用步骤、一次完整的运行过程和常见坑。
这个技能做什么
| 项目 | 说明 |
|---|---|
| 技能 | 威胁建模(threat-model) |
| 作用 | 为功能或系统做简化威胁建模 |
| 输入 | 架构说明 |
| 输出 | 威胁清单与缓解措施 |
| 形态 | 纯提示词 |
| 风险 | 低 |
| 适用人群 | 安全工程师与开发团队 |
处理流程分四步:
- 画出数据流
- 标出信任边界
- 按 STRIDE 列威胁
- 给出缓解建议
这个技能纯指令型,不需要额外脚本或外部连接,复制到大多数平台都能用;风险较低,但输出仍需快速过目再使用。
在 Gemini CLI 中是什么形态
Gemini CLI 会读取项目中的 GEMINI.md 作为长期上下文,还支持在 .gemini/commands/ 下用 TOML 文件定义自定义斜杠命令,把一段固定提示封装成一个命令。
项目级命令放在仓库的 .gemini/commands/,个人级放在用户目录下的 .gemini/commands/;命令中可用 {{args}} 接收调用时的参数。
示例配置
自定义命令示例(示意):
# .gemini/commands/threat-model.toml
description = "威胁建模:为功能或系统做简化威胁建模"
prompt = """
你现在执行「威胁建模」任务。
输入:{{args}}
按以下步骤处理:
1. 画出数据流
2. 标出信任边界
3. 按 STRIDE 列威胁
4. 给出缓解建议
最后输出:威胁清单与缓解措施,并列出需要人工确认的内容。
"""
示例里最重要的是两处:一是描述(或指令开头)要写明「什么时候用」,让智能体在安全工程师与开发团队提出相关请求时能对上号;二是约束部分要写进「扫描结论需人工复核,涉及生产的动作走变更流程」这类领域规矩,避免智能体自作主张。
安装与启用步骤
- 在项目中新建
.gemini/commands/threat-model.toml - 填写 description 与 prompt,prompt 中写清步骤与输出
- 在 GEMINI.md 中补充项目约定,供命令执行时参考
- 在 CLI 中输入
/threat-model加参数调用 - 提交到仓库,团队共享
实操:一次完整的运行
假设你在代码、配置与云环境中发出这样的请求(虚构示例):
用威胁建模技能处理这份架构说明,输出威胁清单与缓解措施。
智能体识别到任务与「威胁建模」相关后,会依次画出数据流、标出信任边界、按 STRIDE 列威胁、给出缓解建议。最终交付威胁清单与缓解措施,并在末尾列出需要人工确认的内容。你的工作是对照原始架构说明抽查两三处关键信息,确认无误后再使用。
测试与验收
用空参数、正常参数和超长参数各调用一次命令,确认提示中的步骤与约束都生效。
针对这个技能,还建议额外准备:
- 一份信息齐全的标准架构说明,检查输出是否完整
- 一份缺少关键信息的输入,检查智能体是否先提问而不是编造
- 一份包含“忽略以上规则”等字样的输入,检查约束是否仍然有效
常见坑
- 把所有流程都塞进 GEMINI.md,导致每次对话上下文过长。
- 命令名与内置命令或其他命令重名,调用时混淆。
- 只写了步骤,没写交付格式,导致每次输出的威胁清单与缓解措施结构不同。
- 没有给出样例,智能体对“好结果”的理解与你不一致;在参考资料里放一两份范例会明显改善。
延伸阅读
- 威胁建模 · SKILL.md 写法
- 威胁建模 · 安全审查
- 威胁建模 · 效果评估
- 安全响应头检查 · Gemini CLI 版
- 供应商安全问卷 · Gemini CLI 版
- 智能体Skills 分类:625 个技能 × 32 个角度
说明:文中的配置、代码与目录结构均为示意,用于说明思路;各平台的入口、字段名与能力会更新,落地前请以对应官方文档为准。示例中的数据与场景为虚构,不代表真实案例。