BugTraceAI-CLI
AI驱动的自动化渗透测试框架,6阶段流水线融合LLM分析与真实攻击验证
加载项目详情…
本应用为开源项目,仅供学习研究,请遵守其开源协议。
AI驱动的自动化渗透测试框架,6阶段流水线融合LLM分析与真实攻击验证
加载项目详情…
本应用为开源项目,仅供学习研究,请遵守其开源协议。
2024年,AI 大模型开始在渗透测试领域崭露头角——但大多数"AI 安全扫描器"本质上是将漏洞描述词塞给 GPT,让它猜测"这个网站有没有漏洞"。这本质上是用 AI 幻想替代真实攻击,误报率居高不下,难以真正用于实战。
BugTraceAI-CLI 正是针对这一痛点诞生的。它的核心理念只有一句话:「AI 负责想,工具负责干」——用 AI 生成漏洞假设,用 SQLMap、Playwright、GoSpider 等成熟工具完成真实攻击和验证。

图1:BugTraceAI-CLI 项目 Logo
BugTraceAI-CLI 采用经典的六阶段渗透测试流水线,每个阶段独立并行运行:
| 阶段 | 名称 | 核心能力 |
|---|---|---|
| Phase 1 | Recon(侦察) | GoSpider 爬取、参数发现、SPA 路由推断 |
| Phase 2 | Discovery(发现) | 6 种 AI 人格并行分析 + Nuclei CVE 扫描 |
| Phase 3 | Strategy(策略) | 思考整合 Agent、去重、优先级路由 |
| Phase 4 | Exploitation(利用) | 15 个专业 Agent 负责不同漏洞类型 |
| Phase 5 | Validation(验证) | Chrome DevTools Protocol + 视觉 AI 截图验证 |
| Phase 6 | Reporting(报告) | JSON/MD/HTML 多格式输出 |
Phase 2 的多人格分析是亮点:同时调度「赏金猎人」「代码审计员」「渗透测试员」「安全研究员」「红队操作员」5 个不同视角的 AI Agent,必须 4/5 达成共识才进入下一阶段,再由第 6 个"怀疑者"人格做最终过滤。这种设计极大降低了 AI 幻觉导致的误报。
后端核心:Python 3.10+,asyncio + aiohttp 提供异步并发,FastAPI + Uvicorn 承载 REST API。数据库层使用 SQLModel(SQLAlchemy + Pydantic),SQLite WAL 模式保证并发写入安全。
AI 层:通过 OpenRouter 统一接入 Gemini、Claude、DeepSeek、Qwen 等多个 LLM,支持自动降级(LLM Shifting)。本地嵌入模型使用 BAAI/bge-small-en-v1.5,支持跨扫描语义搜索。PyTorch CPU 版本确保 embedding 计算不依赖 GPU。
浏览器自动化:XSS 验证使用 Playwright 在真实 Chromium 中执行 JS 载荷;最终验证阶段通过 Chrome DevTools Protocol(CDP)进行底层截图分析,Vision AI 识别漏洞特征。
渗透工具集成:GoSpider(自源码编译)、Nuclei 3.3.7(预下载二进制)、SQLMap(pip 安装),全部内置于 Docker 镜像,无需宿主机安装任何安全工具。
AI 助手集成:原生支持 Model Context Protocol(MCP),暴露扫描引擎为 MCP Tools,可被 Claude Code、Cursor、OpenClaw 等 AI 助手直接调用。

图2:ClawdBot —— BugTraceAI 的 MCP 集成形象
Docker 部署极为简洁:docker-compose up -d 一键启动两个容器——API 服务(8000 端口)和 MCP 服务(8001 端口)。install.sh 安装向导支持自动端口检测,即使 8000 被占用也能自动寻找下一个可用端口。镜像采用多阶段构建,最终镜像合理分层,SQLMap 和 PyTorch CPU 均内置,无需额外依赖。
AGPL-3.0 协议限制:与许多安全工具采用 BSD/Apache 许可证不同,BugTraceAI-CLI 使用 AGPL-3.0,这意味着任何将修改后的版本作为网络服务对外提供的行为,都必须开源。这对于商业安全公司而言是一道门槛。
API Key 依赖:扫描能力依赖 OpenRouter API Key,虽然有免费 tier,但正式使用需要付费,否则速度和模型选择都受限。
无 GUI:目前仅提供 CLI 和 REST API,适合自动化集成,不适合非技术用户直接上手。
BugTraceAI-CLI 代表了 AI 安全工具的一种演进方向——不把 AI 当成万能漏洞检测器,而是作为智能路由器和分析师,将推理能力与确定性工具结合。它的 15 个专业 Agent 架构、多人格共识机制、与 MCP 的深度集成,都为下一代渗透测试框架提供了值得参考的设计思路。
在 GitHub 上已获得 162 Stars,标签涵盖 ai-agents、bugbounty、cybersecurity、hacking、webpentesting,是目前 AI + 渗透测试交叉领域中最具工程完整性的开源项目之一。