PentestGPT

GitHub
12.7k 2.2k 中等 1 次阅读 今天MIT语言模型开发框架
AI 解读 由 AI 自动生成,仅供参考

PentestGPT 是一款基于大语言模型驱动的自动化渗透测试智能体框架,旨在让网络安全测试变得更加高效与智能化。它利用 AI 强大的推理能力,自动执行从信息收集到漏洞利用的完整渗透测试流程,并能实时展示解题思路与操作步骤。

该工具主要解决了传统渗透测试中高度依赖人工经验、操作繁琐且难以复现的痛点。通过引入自主代理(Agentic)机制,PentestGPT 能够独立规划测试路径,支持会话持久化以随时中断和恢复任务,并在预装安全工具的 Docker 隔离环境中运行,确保了测试过程的可复现性与安全性。无论是复杂的 CTF 夺旗赛挑战,还是涵盖 Web、密码学、逆向工程等多类别的真实漏洞场景,它都能提供实时的反馈与辅助。

PentestGPT 特别适合网络安全研究人员、渗透测试工程师以及热衷于安全技术的开发者使用。其独特的技术亮点在于构建了完整的智能体工作流,不仅支持多种主流大模型接入,还采用了模块化架构以便扩展。作为发表于 USENIX Security 2024 的研究成果,它将前沿的 AI 技术与实战安全需求紧密结合,为用户提供了一个强大且透明的自动化安全测试助手。

使用场景

某网络安全团队正在对一家金融客户的 Web 系统进行年度渗透测试,需在紧迫的工期内挖掘深层逻辑漏洞并完成合规报告。

没有 PentestGPT 时

  • 人工枚举效率低:测试人员需手动编写脚本或依赖固定规则工具扫描,面对复杂的业务逻辑漏洞(如越权访问)时,极易因思维定势而遗漏关键路径。
  • 知识断层与协作难:初级分析师遇到陌生的加密算法或逆向难题时,需中断工作去查阅文档或等待资深专家支援,导致测试流程频繁卡顿。
  • 报告整理耗时:测试过程中产生的大量零散命令、截图和临时结论难以自动归档,后期撰写报告时需花费数小时重新梳理攻击链条。
  • 环境复现成本高:每次切换测试目标或工具版本时,需手动配置依赖库和 Docker 环境,容易因环境差异导致测试结果不可复现。

使用 PentestGPT 后

  • 智能推理破局:PentestGPT 利用大模型的先进推理能力,自主分析业务逻辑并生成定制化攻击载荷,成功发现了人工扫描忽略的隐蔽越权漏洞。
  • 实时引导与赋能:遇到技术瓶颈时,PentestGPT 提供实时的“直播式”步骤指引,相当于每位成员身边都有一位资深专家在辅助决策,大幅降低了对特定人员的依赖。
  • 会话持久化自动记录:借助会话保持功能,PentestGPT 自动保存完整的攻击路径和操作上下文,测试结束后可直接导出结构化的中间数据,报告编写时间缩短 70%。
  • 开箱即用的隔离环境:基于 Docker 优先的架构,PentestGPT 一键启动预装全套安全工具的隔离环境,确保了测试过程的可复现性与系统安全性。

PentestGPT 将传统渗透测试从“人力密集型”作业升级为"AI 驱动的自主化”流程,显著提升了漏洞挖掘的深度与交付效率。

运行环境要求

操作系统
  • Linux
  • macOS
  • Windows
GPU

非必需(支持本地 LLM 时取决于所选模型,云端 API 模式无需 GPU)

内存

未说明(建议宿主机运行本地大模型时具备充足内存,Docker 容器内运行依赖宿主资源)

依赖
notes1. 必须安装 Docker 并配置好环境,工具采用'Docker First'策略,所有运行均在隔离的 Docker 容器中进行。 2. 需要配置 LLM 提供商(如 Anthropic API Key、OpenRouter 或本地大模型服务)。 3. 若使用本地大模型(如 Ollama, LM Studio),需确保宿主机服务开启并正确配置网络(使用 host.docker.internal)。 4. 安装时需使用 `--recurse-submodules` 参数克隆仓库以下载基准测试套件。
python3.x (容器内环境,具体版本未明确,需安装 python3)
Docker
Make
PentestGPT hero image

快速开始

贡献者 分支数 星标数 问题数 MIT 许可证 Discord 社区


PentestGPT

基于 AI 的自主渗透测试代理
发表于 USENIX Security 2024

官方网站:pentestgpt.com »

研究论文 · 报告 Bug · 请求功能

GreyDGL%2FPentestGPT | Trendshift


演示

安装

安装演示

在 YouTube 上观看

PentestGPT 实战

PentestGPT 演示

在 YouTube 上观看


v1.0 新特性(智能体升级)

  • 自主智能体 - 用于智能、自主渗透测试的智能体流程
  • 会话持久化 - 保存并恢复渗透测试会话
  • 以 Docker 优先 - 隔离且可复现的环境,预装安全工具

进行中:支持 OpenAI、Gemini 等多个 LLM 提供商


功能

  • AI 驱动的挑战解决者 - 利用 LLM 的高级推理能力执行渗透测试和 CTF 挑战
  • 实时步骤跟踪 - 智能体在解决挑战时实时记录每一步
  • 多类别支持 - Web、密码学、逆向工程、取证、PWN、权限提升
  • 实时反馈 - 观看 AI 工作并获取活动更新
  • 可扩展架构 - 清洁、模块化的设计,为未来增强做好准备

快速入门

先决条件

安装

# 克隆并构建
git clone --recurse-submodules https://github.com/GreyDGL/PentestGPT.git
cd PentestGPT
make install

# 配置认证(仅首次)
make config

# 连接到容器
make connect

注意--recurse-submodules 标志会下载基准测试套件。如果你之前未使用该标志克隆过,请运行:git submodule update --init --recursive

尝试基准测试

cd benchmark/standalone-xbow-benchmark-runner
python3 run_benchmarks.py --range 1-1 --pattern-flag

详细用法请参阅 基准测试文档

命令参考

命令 描述
make install 构建 Docker 镜像
make config 配置 API 密钥(首次设置)
make connect 连接到容器(主入口)
make stop 停止容器(配置保留)
make clean-docker 移除所有内容,包括配置

使用方法

# 交互式 TUI 模式(默认)
pentestgpt --target 10.10.11.234

# 非交互模式
pentestgpt --target 10.10.11.100 --non-interactive

# 带有挑战背景
pentestgpt --target 10.10.11.50 --instruction "WordPress 网站,重点关注插件漏洞"

快捷键F1 帮助 | Ctrl+P 暂停/继续 | Ctrl+Q 退出


使用本地 LLM

PentestGPT 支持将请求路由到您主机上运行的本地 LLM 服务器(LM Studio、Ollama、text-generation-webui 等)。

先决条件

  • 本地 LLM 服务器,具备与 OpenAI 兼容的 API 端点
    • LM Studio:启用服务器模式(默认端口 1234)
    • Ollama:运行 ollama serve(默认端口 11434)

设置

# 配置 PentestGPT 使用本地 LLM
make config
# 选择选项 4:本地 LLM

# 在主机上启动您的本地 LLM 服务器
# 然后连接到容器
make connect

自定义模型

编辑 scripts/ccr-config-template.json 进行自定义:

  • localLLM.api_base_url:您的 LLM 服务器 URL(默认:host.docker.internal:1234
  • localLLM.models:您服务器上可用的模型名称
  • 路由器部分:指定哪些模型负责哪些操作
路由 目的 默认模型
default 通用任务 openai/gpt-oss-20b
background 后台操作 openai/gpt-oss-20b
think 重推理任务 qwen/qwen3-coder-30b
longContext 处理大上下文 qwen/qwen3-coder-30b
webSearch 网络搜索操作 openai/gpt-oss-20b

故障排除

  • 连接被拒绝:确保您的 LLM 服务器正在运行,并监听配置的端口
  • Docker 网络:使用 host.docker.internal(而非 localhost)从 Docker 中访问主机服务
  • 检查 CCR 日志:在容器内运行 cat /tmp/ccr.log

遥测

PentestGPT 会收集匿名使用数据,以帮助改进工具。这些数据会被发送至我们的 Langfuse 项目,包括:

  • 会话元数据(目标类型、持续时间、完成状态)
  • 工具执行模式(使用了哪些工具,但不包含实际命令)
  • 标志检测事件(是否发现了标志,但不包含标志内容)

不会收集任何敏感数据 - 命令输出、凭据或实际标志值都不会被传输。

选择退出

# 通过命令行参数
pentestgpt --target 10.10.11.234 --no-telemetry

# 通过环境变量
export LANGFUSE_ENABLED=false

基准测试

PentestGPT 包含 104 个 XBOW 验证基准,用于全面的测试和评估。

cd benchmark/standalone-xbow-benchmark-runner

python3 run_benchmarks.py --range 1-10 --pattern-flag   # 运行基准测试 1-10
python3 run_benchmarks.py --all --pattern-flag          # 运行全部 104 个基准测试
python3 run_benchmarks.py --retry-failed                # 重试失败的基准测试
python3 run_benchmarks.py --dry-run --range 1-5         # 预览而不执行

性能亮点

PentestGPT 在 XBOW 验证套件上取得了 86.5% 的成功率(90/104 个基准):

  • 成本:平均每次成功基准测试为 $1.11,中位数为 $0.42
  • 时间:平均每次成功基准测试耗时 6.1 分钟,中位数为 3.3 分钟
  • 按难度划分的成功率
    • 1 级:91.1%
    • 2 级:74.5%
    • 3 级:62.5%

有关详细的基准测试结果、分析以及自动化测试说明,请参阅 基准测试文档


旧版

支持 OpenAI、Gemini、Deepseek 和 Ollama 的多模型旧版本(v0.15)已归档至 legacy/

cd legacy && pip install -e . && pentestgpt --reasoning gpt-4o

引用

如果您在研究中使用 PentestGPT,请引用我们的论文:

@inproceedings{299699,
  author = {Gelei Deng and Yi Liu and Víctor Mayoral-Vilches and Peng Liu and Yuekang Li and Yuan Xu and Tianwei Zhang and Yang Liu and Martin Pinzger and Stefan Rass},
  title = {{PentestGPT}: Evaluating and Harnessing Large Language Models for Automated Penetration Testing},
  booktitle = {第33届 USENIX 安全研讨会(USENIX Security 24)},
  year = {2024},
  isbn = {978-1-939133-44-1},
  address = {费城, PA},
  pages = {847--864},
  url = {https://www.usenix.org/conference/usenixsecurity24/presentation/deng},
  publisher = {USENIX 协会},
  month = aug
}

许可证

本项目采用 MIT 许可证进行分发。更多信息请参阅 LICENSE.md 文件。

免责声明:此工具仅用于教育目的及授权的安全测试。作者不鼓励任何非法使用。请自行承担使用风险。


致谢

(返回顶部)

版本历史

v1.0.02025/12/24
v0.14.02024/05/15
v0.13.02024/05/14
v0.9.12023/07/25
v0.82023/05/12
v0.7.02023/05/01
v0.6.02023/04/30
v0.5.02023/04/26
v0.4.02023/04/22
v0.3.02023/04/14
v0.2.02023/04/12

常见问题

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|1周前
Agent开发框架图像

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|2周前
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

160.8k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

109.2k|★★☆☆☆|昨天
开发框架图像Agent

gemini-cli

gemini-cli 是一款由谷歌推出的开源 AI 命令行工具,它将强大的 Gemini 大模型能力直接集成到用户的终端环境中。对于习惯在命令行工作的开发者而言,它提供了一条从输入提示词到获取模型响应的最短路径,无需切换窗口即可享受智能辅助。 这款工具主要解决了开发过程中频繁上下文切换的痛点,让用户能在熟悉的终端界面内直接完成代码理解、生成、调试以及自动化运维任务。无论是查询大型代码库、根据草图生成应用,还是执行复杂的 Git 操作,gemini-cli 都能通过自然语言指令高效处理。 它特别适合广大软件工程师、DevOps 人员及技术研究人员使用。其核心亮点包括支持高达 100 万 token 的超长上下文窗口,具备出色的逻辑推理能力;内置 Google 搜索、文件操作及 Shell 命令执行等实用工具;更独特的是,它支持 MCP(模型上下文协议),允许用户灵活扩展自定义集成,连接如图像生成等外部能力。此外,个人谷歌账号即可享受免费的额度支持,且项目基于 Apache 2.0 协议完全开源,是提升终端工作效率的理想助手。

100.8k|★★☆☆☆|1周前
插件Agent图像

markitdown

MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器

93.4k|★★☆☆☆|1周前
插件开发框架