一种渗透测试的上下文管理方法、计算机设备及存储介质

本申请属于人工智能，具体涉及一种渗透测试的上下文管理方法、计算机设备及存储介质。

背景技术：

1、大模型驱动的自动化渗透测试智能体在近年来取得显著进展，其能够自主执行信息侦察、漏洞利用、权限提升及横向移动等多阶段攻击任务。然而，在长时间、多阶段的渗透测试场景中，智能体面临一个关键瓶颈：失败分支的累积导致上下文腐烂与决策质量退化。

2、具体而言，存在以下三方面问题。第一，失败尝试的累积导致上下文膨胀与关键信息埋没。在渗透测试过程中，智能体需要执行大量试错性操作，如目录爆破、端口扫描、漏洞利用尝试等，这些失败分支在短期内是必要的探索行为，但长期累积后会迅速填满大模型的上下文窗口；更为严重的是，大语言模型存在“lost in the middle”现象，当相关信息位于长上下文的中间位置时，模型的信息召回准确率显著下降，早期的侦查结果往往被淹没在上下文的中间区域而无法被智能体有效利用。第二，失败尝试中的线索价值与信息密度存在深层矛盾，尽管失败尝试本身不直接导致攻击成功，其输出中往往包含对后续决策至关重要的侦查线索，例如一次失败的端口连接可确认防火墙过滤规则、一次依赖缺失的错误回显可揭示目标环境的具体版本信息，然而现有上下文管理方法面临两难困境：压缩粒度过大导致线索丢失，压缩粒度过小则导致上下文膨胀。第三，渗透测试作为长程任务具有天然的分支探索结构，在每个阶段智能体通常需要从同一锚点出发尝试多个不同攻击方向，每个方向构成一条独立分支且只有少数分支能通向下一锚点，这对上下文管理提出了能够感知渗透测试分支结构、按分支单元进行信息提炼与保留的能力要求。

技术实现思路

1、本申请实施例的目的是提供一种渗透测试的上下文管理方法、计算机设备及存储介质，以解决失败分支累积导致的上下文腐烂与决策质量退化问题。

2、为了解决上述技术问题，本申请是这样实现的：

3、第一方面，本申请实施例提供了一种渗透测试的上下文管理方法，该方法包括：

4、在渗透测试的主循环中，调用大模型进行推理决策，输出下一步的行动类型，所述行动类型包括执行渗透测试工具或调用上下文管理工具；

5、当所述行动类型为执行渗透测试工具时，调用对应的渗透测试工具面向目标系统执行操作并捕获输出，同时以当前分支标识和命令字符串为复合键将所述输出存入外部存储模块；

6、当所述行动类型为调用上下文管理工具时，根据所述上下文管理工具的类型执行相应的上下文管理操作；

7、其中，所述上下文管理工具包括失败分支摘要化工具，调用所述失败分支摘要化工具时，将当前分支下的全部原始命令与输出记录整合为包含原始命令列表、关键发现列表和分支概述的结构化摘要，并以所述结构化摘要替换当前分支原有的所有原始上下文记录。

8、第二方面，本申请实施例提供了一种计算机设备，该计算机设备包括处理器、存储器及存储在存储器上并可在处理器上运行的程序或指令，程序或指令被处理器执行时实现如第一方面所述的方法的步骤。

9、第三方面，本申请实施例提供了一种计算机可读存储介质，计算机可读存储介质上存储程序或指令，程序或指令被处理器执行时实现如第一方面所述的方法的步骤。

10、第四方面，本申请实施例还提供了一种计算机程序产品，包括计算机程序，该计算机程序被处理器执行时实现如第一方面所述的方法的步骤。

11、在本申请实施例中，通过以分支为基本压缩单元对失败分支进行结构化摘要，在保留侦查线索价值的同时大幅缩减上下文占用；通过外部存储模块持久化保存原始输出并配合按需检索机制，实现了渐进式信息披露，使智能体在常态下仅需阅读精简摘要即可保持全局感知，在需要时又能精确还原历史原始输出进行细粒度分析，从而在上下文占用与线索保留之间取得了有效平衡。

1.一种渗透测试的上下文管理方法，应用于部署有智能体的设备，其特征在于，渗透测试的上下文管理方法包括：

2.根据权利要求1所述的渗透测试的上下文管理方法，其特征在于，所述上下文管理工具还包括锚点设置工具和原始输出检索工具；

3.根据权利要求2所述的渗透测试的上下文管理方法，其特征在于，所述声明新锚点并将锚点记录追加至上下文窗口，包括：

4.根据权利要求1所述的渗透测试的上下文管理方法，其特征在于，所述调用所述失败分支摘要化工具的触发方式包括以下至少一种：

5.根据权利要求4所述的渗透测试的上下文管理方法，其特征在于，所述智能体自主触发的优先级高于所述系统强制触发的优先级。

6.根据权利要求1所述的渗透测试的上下文管理方法，其特征在于，生成所述原始命令列表，包括：

7.根据权利要求6所述的渗透测试的上下文管理方法，其特征在于，所述关键发现列表包括以下至少一种：

8.根据权利要求1所述的渗透测试的上下文管理方法，其特征在于，所述调用对应的渗透测试工具面向目标系统执行操作并捕获输出之后，所述方法还包括：

9.一种计算机设备，其特征在于，包括处理器，存储器及存储在所述存储器上并可在所述处理器上运行的程序或指令，所述程序或指令被所述处理器执行时实现如权利要求1-8中任一项所述的渗透测试的上下文管理方法的步骤。

10.一种可读存储介质，其特征在于，所述可读存储介质上存储程序或指令，所述程序或指令被处理器执行时实现如权利要求1-8中任一项所述的渗透测试的上下文管理方法的步骤。