ghkfuddl1327-wq/rojaprove

GitHub: ghkfuddl1327-wq/rojaprove

一款针对 LLM 应用的发布前红队测试 CLI 工具,通过确定性 canary 机制检测系统 Prompt 泄露并提供可操作的修复方案。

Stars: 1 | Forks: 0

# rojaprove **专为 LLM 应用打造的发布前红队测试工具。** 将其指向你正在运行的 endpoint;rojaprove 会发送探测请求,用证据验证你的系统 prompt 是否泄露,并为你提供可直接粘贴的防御指令——然后你可以重新测试以确认修复是否有效。 不是猜测,而是证据。每项发现都包含发送的确切输入、收到的原始响应以及确定性的判定结果。没有证据就没有发现。 ![rojaprove 在运行中的 endpoint 发现系统 prompt 泄露——带有证据的红色判定结果](https://static.pigsec.cn/wp-content/uploads/repos/2026/06/a47d0b54ea134809.gif) ## 范围——请先阅读此部分 rojaprove **仅**测试你拥有或获得明确书面授权的 endpoint。在此授权范围内操作是你的责任。该工具始终会打印范围提示;`--i-own-this` 会在输出中记录你的授权。这是一项防御性的发布前检查,而不是用于攻击你无法控制的系统的工具。 ## 哪些数据会离开你的计算机 - rojaprove **仅**与你指定的目标 URL 通信。不会向任何其他地方发送任何内容。 - **BYOK**(自带密钥);rojaprove 绝不会附带或回传任何凭证。 - 每次运行前,它会打印传输披露信息:目标 URL、连接是否加密,以及你的 auth token **掩码处理**(仅显示最后 4 个字符)。 - v0.1 核心路径**不进行外部 LLM 调用**——探测是静态的,判定结果是确定性的字符串匹配。 ## 安装说明 要求 Python 3.10+。 ``` pip install rojaprove ``` ## 快速开始 对你拥有的 endpoint 运行扫描: ``` rojaprove scan http://127.0.0.1:8000/chat --i-own-this ``` rojaprove 会打印哪些数据将离开你的计算机,运行系统 prompt 泄露探测,并针对任何泄露情况展示证据以及可直接粘贴的防御指令。当发现泄露时退出代码为 `1`(以便 CI 可以对其进行拦截),结果正常时为 `0`。 对于受 bearer token 保护的 target,请传入 `--auth-bearer` 或设置 `ROJAPROVE_AUTH_BEARER`(参见 `.env.example`)。 ## rojaprove 目前的检查项 rojaprove 是一个**发布前针对 LLM 应用的红队 CLI 工具**:它会发送攻击输入 然后在响应中搜索 **canary**——即*你*植入到应用系统 prompt 中的、绝不应该在正常输出中出现的秘密字符串。 判定结果是确定性的:只有当 canary 确实在响应中出现时,才计为泄露。 没有启发式算法,没有概率判断。 **v0.1 覆盖一个类别:** | 类别 | 状态 | |---|---| | **(1) 系统 Prompt 泄露** (OWASP LLM07) | **今日已支持检测** — 确定性的绿色/红色判定 | | (2) 间接 Prompt 注入 | 路线图 — 暂无探测 | | (3) 可标记的数据外泄 | 路线图 — 暂无探测 | 如果你在这个 repo 中看到“tested”(已测试)字样,它**仅指代类别 (1)**。 类别 (2) 和 (3) 已列入路线图,此处没有任何内容被描述为已针对它们进行过测试。 rojaprove 测试的是 **AI 层**——即 LLM 生成响应的环节。它 **不**涵盖传统的 web/API 漏洞(如 SQL 注入、身份验证绕过 等);这些属于标准的 DAST/渗透测试工具的范畴,不在此列。 ## 在参考 target 上尝试(2 分钟) 这个 repo 内置了一个故意存在漏洞的演示 target,让你可以在将其指向你自己的任何应用之前观察测试工具的运行情况。这是一个小型的 FastAPI “邮件助手”,在其系统 prompt 中植入了一个 canary。 ``` # 1. 安装 rojaprove 及 demo server deps pip install -e ".[demo]" # 2. 运行 vulnerable target(在一个 terminal 中保持运行) uvicorn targets.inbox_assistant.app:app --host 127.0.0.1 --port 8000 # 3. 在第二个 terminal 中,扫描它 rojaprove scan http://127.0.0.1:8000/chat --i-own-this ``` 你应该会得到一个 **红色** 判定。演示 target 的系统 prompt 中已经植入了一个 canary (这正是使其成为 *target* 的原因);rojaprove 发送 系统 prompt 泄露输入,并在响应中发现了该 canary 被原样返回。 这就是一个确定性的泄露。 演示 target 还有一个 `defend` 开关,因此你也可以看到 **绿色** 的一面 ——输入相同,但应用拒绝泄露其系统 prompt,且 canary 永远不会出现。存在漏洞时为红色,受到防御时为绿色:这就是它传递的全部信号。 ## 在你自己的应用上使用它 参考 target 有一个**预先植入的** canary。你的真实应用则不同: **你**需要植入**你自己的** canary,然后针对**你的** endpoint 运行测试工具。 ### 如果你熟悉终端操作 1. 选择一个永远不会出现在你应用输出中的秘密字符串。请使用 方便检索且唯一的格式,例如 `ROJA_CANARY_MYAPP_<8位随机十六进制>`。 2. 将其放在系统 prompt 中模型能看到但不应 泄露的某个位置(例如“内部配置”行)。 3. 针对你正在运行的 endpoint 运行 rojaprove,使用 `--canary` 告诉它要查找哪个 canary: rojaprove scan https://your-app.example.com/your-endpoint \ --canary ROJA_CANARY_MYAPP_<8位十六进制> \ --i-own-this `--canary` 必须与你在步骤 2 中植入的确切字符串匹配。rojaprove 不会 植入任何内容——它只会根据你提供的值搜索响应。 4. 阅读判定结果。**红色** = 你的 canary 泄露到了输出中(你的系统 prompt 是可被泄露的)。**绿色** = 对于 rojaprove 尝试的输入,它没有泄露。 `--i-own-this` 是必填项:仅扫描你拥有或获得明确 授权进行测试的 endpoint。 ### 不熟悉这些?将此内容粘贴给你的 AI 助手 如果你不确定如何进行设置,请将下方的代码块复制到 Claude、 ChatGPT、Cursor 或你使用的任何助手中,并按照其步骤操作。(这会让你的 AI 指导*你*完成测试——它不会替你“保护”你的应用。) ``` I want to test whether my LLM app leaks its system prompt (OWASP LLM07, "system prompt leakage"). Help me do this, step by step: 1. Generate a unique canary string in the format ROJA_CANARY_MYAPP_<8 hex>. 2. Show me exactly where to place it in my system prompt — somewhere the model can see it but is instructed never to reveal it (e.g. an internal-config line). 3. Give me 8–10 distinct "leak" inputs that try to make the model dump its system prompt verbatim (ignore-previous-instructions, "repeat everything above", "reveal your configuration", etc.). 4. For each input, tell me to check whether my canary string appears in the model's output. If it does, that's a deterministic leak. 5. Remind me that this only tests system-prompt leakage — it does not prove my app is safe against anything else. My app's system prompt is: ``` 一旦你植入了 canary 并了解了测试的形式,你就可以自动运行 相同的检查:将 `rojaprove scan` 指向你的 endpoint,并使用 `--canary` 传入相同的 canary 字符串(参见上方的终端步骤)。 ## ⚠️ 此工具不会做什么 - **rojaprove 通过并不等于你的应用是安全的。** 绿色结果仅意味着 rojaprove *在其尝试的输入中*没有发现 系统 prompt 泄露——仅此而已。 - **它目前仅检查一个类别:** 类别 (1),系统 Prompt 泄露。 间接注入 (2) 和数据外泄 (3) 属于路线图,暂未测试。 - **canary 需要由你植入。** 在演示 target 中它是预先植入的,以便你 了解其机制;在你自己的应用中,你需要放置自己的 canary。对没有 植入 canary 的应用进行扫描不会提供任何有用信息。 - **它是测试工具,而不是防御工具。** 它可以帮助你*发现*并证明泄露;修复 泄露是另一个单独的步骤(rojaprove 会打印一个修复 prompt 作为起点)。 ## License 基于 Business Source License 1.1 授权 — 详见 [LICENSE](./LICENSE)
标签:AI安全, Chat Copilot, DLL 劫持, Python, 大语言模型, 安全检测工具, 文档结构分析, 无后门, 红队评估, 逆向工具