PentAGI - AI 渗透测试工具详解

PentAGI - AI 渗透测试工具详解

本文档基于源码（v0.1.1）与公开资料整理，GitHub: https://github.com/vxcontrol/pentagi

什么是 PentAGI？

PentAGI(Penetration testing Artificial General Intelligence) 是由VXControl开发的一个完全自主的 AI 渗透测试系统。它通过多代理（Multi-Agent）AI 架构来自动化执行渗透测试任务，将复杂的安全评估任务分解为子任务，由专门的 AI Agents 执行并生成综合漏洞报告。

核心定位

Slogan:“Fully autonomous AI Agent that performs complicated penetration testing tasks using terminal, browser, editor, and external search system.”

核心特性

🛡️ 安全隔离

• 所有操作在沙箱 Docker 环境中执行
• 完全隔离，确保测试过程安全可控
• 自动清理容器，支持重复测试

🤖 全自主运行

• AI 驱动的自动化渗透测试工作流
• Agent 自动规划攻击路径
• 遇到防火墙等阻碍时自动调整策略
• 记住成功的攻击方法供后续使用

🔧 预装渗透测试工具（Kali Linux）

PentAGI 使用Kali Linux镜像作为渗透测试的默认环境，工具并非动态安装，而是预装在镜像中。

默认镜像：

vxcontrol/kali-linux:latest

工具数量：

• Kali Linux 预装600+渗透测试工具（不仅 20+）
• Agent 直接调用，无需动态安装

常用工具列表：

架构示意：

用户发起渗透测试任务 │ ▼ ┌─────────────────────────────────────┐ │ 后端动态创建容器 │ │ (pentagi-terminal-1, -2, ...) │ └─────────────────────────────────────┘ │ ▼ 使用默认镜像 vxcontrol/kali-linux ┌─────────────────────────────────────┐ │ Kali Linux 容器 (600+ 工具已安装) │ │ ├─ nmap │ │ ├─ sqlmap │ │ ├─ metasploit │ │ └─ ... │ └─────────────────────────────────────┘ │ ▼ Agent 直接调用工具命令 ┌─────────────────────────────────────┐ │ 执行渗透测试 │ └─────────────────────────────────────┘

可自定义镜像：
如需使用其他镜像（如 Parrot Security），可配置：

DOCKER_DEFAULT_IMAGE_FOR_PENTEST=parrotsec/security:latest

🧠 智能记忆系统

• 长期记忆: 存储研究结果和成功的攻击方法
• 向量存储: PostgreSQL + pgvector 实现语义搜索
• 知识图谱: Neo4j + Graphiti 跟踪实体关系

📊 现代界面

• React + TypeScript 构建的 Web UI
• 直观的系统管理和监控界面
• 实时任务状态追踪

🔌 全功能 API

• REST API
• GraphQL API
• Bearer Token 认证
• 支持自动化和集成

系统架构

技术栈

┌─────────────────────────────────────────────────────┐ │ Frontend UI │ │ React + TypeScript Web UI │ └─────────────────────────────────────────────────────┘ │ ▼ ┌─────────────────────────────────────────────────────┐ │ Backend API │ │ Go REST/GraphQL APIs │ │ (Bearer Token 认证) │ └─────────────────────────────────────────────────────┘ │ │ │ │ ▼ ▼ ▼ ▼ ┌────────────┐ ┌────────────┐ ┌────────────┐ ┌────────────┐ │ Vector │ │ Task │ │ Knowledge │ │ Docker │ │ Store │ │ Queue │ │ Graph │ │ Runtime │ │ (pgvector) │ │ (Async) │ │ (Neo4j) │ │ (Sandbox) │ └────────────┘ └────────────┘ └────────────┘ └────────────┘

核心组件

13 种 AI Agent 类型

PentAGI 采用多代理架构，包含 13 种专业 Agent：

LLM Provider 支持

支持多种大语言模型 Provider：

每个 Agent 类型可以配置不同的模型，实现专业分工。

工作流程

PentAGI 渗透测试流程

┌──────────────────────────────────────────────────────────────┐ │ 1. 目标指定 │ │ 用户输入目标 (URL/IP) │ └──────────────────────────────────────────────────────────────┘