PentestGPT：A GPT-empowered penetration testing tool

阅读目录(Content)

一、What is PentestGPT？

PentestGPT是一种由大型语言模型（LLMs）赋能的渗透测试工具。它旨在自动化渗透测试过程。它基于ChatGPT构建，并以交互模式运行，以指导渗透测试人员在整体进展和具体操作上。

PentestGPT提供了一个统一的终端输入处理程序，并由三个主要组件支持：

处理程序是渗透测试工具的主要入口点。它允许渗透测试人员执行以下操作：

用户初始化所有会话。
用户通过以下几个步骤完成初始化任务。
1. 用户向ReasoningSession提供目标信息。
2. ReasoningSession根据目标信息生成任务树。
3. ReasoningSession决定下一步要做的事情，并将信息传递给GenerationSession。
4. GenerationSession生成用户执行的确切命令，并将其传递给用户。
进入主循环。用户可以选择以下几个步骤：
1. 向PentestGPT提供todo-list待办列表执行结果。
  1. 用户将工具的输出提供给ParsingSession。
  2. ParsingSession解析输出，并将信息传递给ReasoningSession。
  3. ReasoningSession根据信息更新任务树。
  4. 重复执行步骤3.2.1-3.2.3。
2. 询问待办事项。
  1. ReasoningSession分析任务树。它决定下一个待办事项，包括（1）自然语言描述和（2）要执行的确切命令。
  2. ReasoningSession将信息传递给GenerationSession进行进一步验证。
  3. GenerationSession生成用户执行的确切命令，并将其传递给用户。
3. 通过提供任意信息与PentestGPT讨论。
  1. 用户将信息提供给ParsingSession。
  2. ParsingSession解析信息：如果信息太长，进行摘要。否则，重新表述。
  3. ReasoningSession分析信息并更新任务树。
退出程序。

下面显示了一个流程图：

参考链接：

https://github.com/GreyDGL/PentestGPT/blob/main/PentestGPT_design.md 
https://github.com/GreyDGL/PentestGPT

历史操作过程（status db）需要以某种形式记录，并且LLM的token窗口能够允许每次请求时status db作为context。
- 值得注意的是，需要记录哪些信息，以什么样的格式记录，这部分需要领域专家进行人工定义，属于领域先验知识的部分。
任务必须能够被分解为多个子步骤。
- 最好通过领域专家通过few-shot的方式预先定义好任务步骤分解的套路，这部分也属于领域先验知识的部分。
涉及到工具调用和返回结果解析和具体垂直领域和具体的工具强相关，协议解析和信息格式提取是一个重工作量的领域。