AbasSec/AUTO_Pen-Powered-by-Zenith-Security-AI

GitHub: AbasSec/AUTO_Pen-Powered-by-Zenith-Security-AI

AUTO_Pen 是一个AI驱动的自主渗透测试框架,用于自动化红队评估和漏洞发现。

Stars: 2 | Forks: 0

# AUTO_Pen — 由 Zenith Security AI 驱动 ### 自主AI驱动的渗透测试框架 — v3.2.0 ## 概述 **AUTO_Pen** 是一款为专业安全评估、漏洞赏金任务和对抗性模拟构建的全栈自主渗透测试引擎。它用一个活跃的、能够推理的AI团队取代了静态的扫描工作流,该团队能够映射攻击面、链接利用链、验证发现结果,并撰写可直接发布的报告——所有操作均通过一个交互式命令行完成。 **Zenith AI 三人组**(战略家·执行者·审计员)驱动每一次任务:一个智能体规划任务,一个执行任务,另一个在发现结果呈现给操作员之前,验证其准确性和合规性。 **超过243个专业引擎模块**覆盖完整的攻击链——从被动侦察到后渗透阶段,包括云/SaaS边界、汽车协议、供应链分析以及映射到MITRE ATT&CK的报告生成。 ## 架构 ``` ┌─────────────────────────────────────────────────────────┐ │ OPERATOR SHELL │ │ Interactive CLI · Conversational NL │ └────────────────────────┬────────────────────────────────┘ │ ┌────────────────────────▼────────────────────────────────┐ │ ZENITH AI TRIAD │ │ │ │ ┌─────────────┐ ┌─────────────┐ ┌─────────────────┐ │ │ │ Strategist │ │ Operative │ │ Auditor │ │ │ │ Mission │◄─► Execution │◄─► Validation & │ │ │ │ Planning │ │ & Exploit │ │ Compliance │ │ │ └─────────────┘ └─────────────┘ └─────────────────┘ │ └────────────────────────┬────────────────────────────────┘ │ ┌────────────────────────▼────────────────────────────────┐ │ ENGINE LAYER (243+ modules) │ │ │ │ Recon → Exploitation → Post-Exploit → Reporting │ │ Attack Graph · OAST · WAF Bypass · Chain Executor │ └────────────────────────┬────────────────────────────────┘ │ ┌────────────────────────▼────────────────────────────────┐ │ DATA & PERSISTENCE LAYER │ │ SQLite · Attack Path Memory · Neural Memory │ │ Data Governance · PII Redaction · Schema Migration │ └─────────────────────────────────────────────────────────┘ ``` ## 系统要求 | 组件 | 最低要求 | |---|---| | 操作系统 | Linux (Kali, Ubuntu 22.04+, Parrot) | | Python | 3.11+ | | 浏览器引擎 | Playwright / Chromium (用于视觉侦察) | | AI 后端 | NVIDIA NIM, DeepSeek, Gemini, OpenAI 或本地 Ollama | ## 安装 ``` git clone https://github.com/AbasSec/AUTO_Pen-Powered-by-Zenith-Security-AI.git cd AUTO_Pen-Powered-by-Zenith-Security-AI pip install -r requirements.txt playwright install chromium --with-deps ``` ## 配置 ### AI 供应商 AUTO 支持云端和本地模型后端。在 `.env` 文件中设置凭证: ``` # NVIDIA NIM (默认 — 访问超过400个模型) NVIDIA_API_KEY=your_key_here # 替代方案 DEEPSEEK_API_KEY=your_key_here GEMINI_API_KEY=your_key_here OPENAI_API_KEY=your_key_here # 可选 SUDO_PASSWORD=your_password_here ``` ### 默认AI团队 (`config/config.yaml`) | 角色 | 模型 | 供应商 | |---|---|---| | 战略家 | `abacusai/dracarys-llama-3.1-70b-instruct` | NVIDIA NIM | | 执行者 | `qwen/qwen3-coder-480b-a35b-instruct` | NVIDIA NIM | | 审计员 | `mistralai/mistral-large-3-675b-instruct-2512` | NVIDIA NIM | 使用 `select-model` 命令可在运行时切换供应商或模型——AUTO 会热重载AI核心,无需重启。 ## 启动 ``` python3 auto.py ``` 命令行启动时会先进行供应商连接性检查,加载AI团队,然后让您进入 Zenith 提示符。 ## 核心功能 ### 侦察与开源情报 - 通过 crt.sh、Wayback Machine 和 GitHub Dorking 进行子域名枚举 - 基于响应特征和HTTP头的技术指纹识别 - 存档URL挖掘和API接口恢复 - DNS/邮件配置审计(SPF, DMARC, DKIM) - SSL/TLS版本和证书链分析 - 资产聚类和攻击面覆盖评分 ### Web应用利用 - **SQLi** — 基于错误、布尔盲注、时间盲注、联合查询注入,支持 MySQL, PostgreSQL, Oracle, MSSQL, MongoDB, SQLite - **XSS** — DOM型、反射型、存储型、上下文感知的载荷变异 - **SSRF / SSTI / XXE / RCE / 路径遍历 / CORS / CSRF** - **IDOR** — 顺序型、UUID型、哈希型;水平与垂直权限提升 - **认证与会话** — JWT算法混淆、弱密钥暴力破解、会话熵分析、会话固定 - **SAML攻击** — 签名包装、XML注入 - **参数挖掘** — 通过差异响应发现隐藏参数 - **目录列表和敏感文件发现** - **内容欺骗和同站脚本** ### API安全 (OWASP API Top 10) - REST、GraphQL、gRPC 和 WebSocket 端点发现与模糊测试 - BOLA / BFLA 检测 - 批量赋值利用 - API架构偏移检测(实时 vs. 文档规范) - API规范导入(OpenAPI/Swagger, Postman) ### 高级与专业模块 - **WAF绕过** — 编码链、头部注入、请求分片 - **竞态条件利用** — 基于时间的并发请求规划器 - **业务逻辑滥用** — 多步骤逻辑流程颠覆 - **盲OAST关联** — 带外交互跟踪 - **防御感知利用** — 检测WAF/IDS存在并自适应载荷 - **供应链分析** — 依赖与制品风险映射 - **云/SaaS边界测试** — 跨租户隔离与配置错误 - **汽车协议审计** — CAN总线和嵌入式系统评估 - **凭证与密钥扫描** — 硬编码密钥、令牌和高熵字符串 - **访问控制利用** — 权限边界跨越 ### AI驱动的自动化 - **多智能体OODA循环** — 规划 → 执行 → 验证 → 迭代,无需操作员输入 - **自适应学习团队** — 智能体在任务中共享发现并更新策略 - **攻击链执行器** — 将独立漏洞链接成权限提升路径 - **攻击图可视化** — 可导出的可利用路径图(SVG/JSON格式) - **利用链构建器** — 组合多步骤概念验证链 - **载荷变异引擎** — 上下文感知、优化规避的载荷变体 - **置信度门控** — 阻止低置信度操作进入自主执行 - **自主质量评分** — 对照真实情况基准评估AI决策质量 ### 报告与合规 - **赏金报告撰写器** — 符合特定项目要求、可直接提交的报告 - **CVSS v3.1 计算器** — 根据发现元数据自动评分 - **MITRE ATT&CK映射** — 每个发现都带有技术标签;支持Navigator JSON导出 - **合规映射器** — 将发现映射到PCI-DSS, ISO 27001, OWASP Top 10 - **SBOM生成器** — 目标依赖项的软件物料清单 - **PoC导出器** — 最小化、可复现的概念验证包 - **时间线导出** — 按时间顺序排列的任务叙述 - **自然语言摘要** — 从原始发现生成自然语言执行摘要 ### 基础设施 - **C2管理器** — 基于Flask的指挥控制仪表板,带反向TCP监听器 - **计划任务** — 无头化、基于cron驱动的自主扫描 - **持续守护进程模式** — 后台持久化扫描循环 - **插件系统** — 运行时热加载自定义Python模块 - **数据治理** — PII脱敏、证据保留策略、清除调度器 - **架构迁移** — Alembic风格的版本化SQLite迁移 - **OWASP LLM Top 10 防护** — 对所有AI调用进行输入/输出消毒 ## 命令参考 ### 任务命令 | 命令 | 描述 | |---|---| | `scan ` | 被动侦察 — 子域名枚举、技术指纹、DNS、SSL | | `hack ` | 完全自主利用 — OODA循环驱动 | | `breakthrough ` | 高级绕过合成 — 视觉探测 + JWT + 逻辑 | | `surgical ` | 高速、低噪声的敏感文件和端点探测 | | `study ` | 深度研究 — 获取技术文章并加载至神经记忆 | | `study tool ` | 阅读man页面和帮助输出;教会AI CLI语法 | | `weaponize ` | 从GitHub查找、克隆并集成外部工具 | | `build ` | AI按需编写并集成新的Python模块 | | `debug ` | AI诊断并修补失败或缓慢的模块 | | `add module ` | 将自定义模块注册到执行引擎 | ### 会话与管理 | 命令 | 描述 | |---|---| | `show findings` | 显示所有已发现漏洞的表格及其严重性 | | `c2 start` | 启动C2服务器和反向TCP监听器 | | `c2 status` | 活跃会话、捕获的载荷、节点图 | | `team on` / `team off` | 切换多智能体协作推理 | | `agent ` | 设置活跃智能体:`strategist`, `operative`, `auditor` | | `select-model` | 交互式模型/供应商向导 — 热重载AI核心 | | `api-keys` | 显示活跃供应商密钥状态 | | `api-keys export` | 显示原始密钥值 | | `sudo-pass` | 缓存sudo密码以用于自动化工具安装 | | `reset` | 清除会话状态和神经上下文 | | `setup` | 首次安装的配置指南 | | `exit` | 保存会话状态并关闭 | ### 自然语言 任何不匹配命令的输入都将直接路由到当前活跃的AI智能体。可以使用自然语言询问战术问题、分析目标或引导正在进行的任务。 ``` How can I bypass the rate limit on /api/login? What vulnerabilities did you find in the auth flow? Generate a CVSS score for the IDOR you found. ``` ## AI团队配置 三个智能体可以单独配置: ``` # 配置文件 config/config.yaml ai: team_mode: true agents: strategist: provider: nvidia model: abacusai/dracarys-llama-3.1-70b-instruct operative: provider: nvidia model: qwen/qwen3-coder-480b-a35b-instruct auditor: provider: nvidia model: mistralai/mistral-large-3-675b-instruct-2512 ``` 可以将任意智能体切换到不同的供应商(`deepseek`, `gemini`, `openai`, `ollama`)或模型,无需重启。`select-model` 命令可以交互式地引导完成此操作。 ## 隐蔽模式 默认启用隐蔽模式。可在 `config/config.yaml` 中配置: ``` stealth: enabled: true header_spoofing: true user_agent_rotation: true request_jitter: true performance: delay_min: 0.5 delay_max: 2.0 concurrency: 5 stealth_mode: delay_min: 2.0 delay_max: 5.0 concurrency: 10 ``` ## 输出 报告默认以Markdown格式写入 `reports/` 目录: ``` reports/auto_report_{target}_{date}_{time}.md ``` 每个发现包括:标题、严重性、CVSS评分、MITRE技术ID、复现步骤和修复建议。在会话期间使用 `show findings` 可获取实时摘要。 ## 法律声明 AUTO_Pen **仅用于授权的安全测试**。 - 仅可对您拥有或获得明确书面授权进行测试的系统使用。 - 用于漏洞赏金计划时,必须遵守该计划的范围和规则。 - 未经授权的使用是非法且不道德的。 作者对滥用行为不承担任何责任。构建此工具是为了提升安全性——而非破坏安全。 *由 AbasSec 构建 · Zenith Security AI · v3.2.0*
标签:AES-256, AI风险缓解, AI驱动, Cloudflare, kill chain, MITRE ATT&CK, PyRIT, 专业安全测试, 反取证, 多智能体系统, 大型语言模型, 安全评估, 对抗模拟, 汽车安全, 漏洞评估, 特征检测, 网络安全, 自主系统, 逆向工具, 隐私保护