利索能及
我要发布
收藏
专利号: 2024115150109
申请人: 浙江工业大学
专利类型:发明专利
专利状态:已下证
更新日期:2026-07-29
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种基于多智能体的大语言模型自动渗透测试方法,其特征在于,所述基于多智能体的大语言模型自动渗透测试方法,包括:构建渗透测试解决方案知识库和渗透测试工具库;

建立五种大模型智能体,包括知识检索智能体、任务规划智能体、任务调度智能体、工具调用智能体和任务总结智能体;

接收输入的任务描述,调用所述知识检索智能体,根据任务描述在所述渗透测试解决方案知识库检索得到知识语料块;

调用所述任务规划智能体,将任务描述和知识语料块转化为第一提示词,根据第一提示词得到渗透测试步骤列表;

对于渗透测试步骤列表中的每个渗透测试步骤,调用任务调度智能体,将当前渗透测试步骤以及当前渗透测试步骤之前的所有渗透测试步骤的工具输出结果转化为第二提示词,根据第二提示词得到当前渗透测试步骤对应的操作类型,若操作类型为工具调用,则调用工具调用智能体;若操作类型为执行总结,则调用任务总结智能体;

所述工具调用智能体根据当前渗透测试步骤在所述渗透测试工具库中检索对应的程序接口,结合当前渗透测试步骤、上一步渗透测试步骤、上一步渗透测试步骤的执行结果转化为第三提示词,根据第三提示词得到当前渗透测试步骤对应的工具输出结果,并将所述工具输出结果输入任务调度智能体;

所述任务总结智能体将任务描述和所有渗透测试步骤的执行结果转化为第四提示词,根据第四提示词输出自动渗透测试结果。

2.根据权利要求1所述的基于多智能体的大语言模型自动渗透测试方法,其特征在于,所述渗透测试解决方案知识库,构建过程如下:通过网络爬取的方式获取渗透测试专业领域知识语料;

根据主题对渗透测试专业领域知识语料进行划分,得到知识语料块,每一个知识语料块描述一次渗透测试的过程,记为 ,其中 表示第 个知识语料块,表示第 个知识语料块的任务描述, 表示第 个知识语料块的解决方案步骤描述;

使用文本向量模型把每个知识语料块 的任务描述 转化为语义向量,将语义向量存储在向量数据库中,并建立索引,完成渗透测试解决方案知识库的构建。

3.根据权利要求1所述的基于多智能体的大语言模型自动渗透测试方法,其特征在于,所述渗透测试工具库,构建过程如下:根据渗透测试的需求在攻击机上安装相应的渗透测试工具;

为每一渗透测试工具的每种功能提供一个统一形式的程序接口;

为每个程序接口提供文字描述,记为 ,其中,

为第 个程序接口, 为第 个程序接口的接口调用格式, 为第 个程序接口的功能描述, 为第 个程序接口的参数列表描述, 为第 个程序接口的接口返回描述;

使用文本向量模型把每个程序接口 的功能描述 转化为语义向量,将语义向量存储在向量数据库中,并建立索引,完成渗透测试工具库的构建。

4.根据权利要求1所述的基于多智能体的大语言模型自动渗透测试方法,其特征在于,所述根据第一提示词得到渗透测试步骤列表,包括:将第一提示词作为输入,所述任务规划智能体返回解决方案步骤描述;

解析解决方案步骤描述得到渗透测试步骤列表,记为 ,其中 表示第 个渗透测试步骤的描述, 。

5.根据权利要求1所述的基于多智能体的大语言模型自动渗透测试方法,其特征在于,所述将当前渗透测试步骤以及当前渗透测试步骤之前的所有渗透测试步骤的工具输出结果转化为第二提示词,包括:获取工具调用智能体返回的第 步渗透测试步骤的工具输出结果 ,并与之前步渗透测试步骤的工具输出结果组合,得到前 步渗透测试步骤的执行轨迹;

将执行轨迹 与第 步渗透测试步骤的描述 结合,得到第 步渗透测试步骤的执行结果 ;

基于当前渗透测试步骤的描述 、第 步渗透测试步骤的描述 和第 步渗透测试步骤的执行结果 转化为第二提示词。

6.根据权利要求1所述的基于多智能体的大语言模型自动渗透测试方法,其特征在于,所述操作类型还包括执行停止,若操作类型为执行停止,则任务调度智能体根据第 步渗透测试步骤的工具输出结果 生成错误描述,并输出错误描述。

7.根据权利要求6所述的基于多智能体的大语言模型自动渗透测试方法,其特征在于,所述第二提示词中包含操作类型判断依据,所述操作类型判断依据为:如果上一步渗透测试步骤不是渗透测试步骤列表中的最后一个渗透测试步骤且上一步渗透测试步骤对应的工具输出结果包含执行正常,则当前渗透测试步骤的操作类型为工具调用;

如果上一步渗透测试步骤是渗透测试步骤列表中的最后一个渗透测试步骤且上一步渗透测试步骤对应的工具输出结果包含执行正常,则当前渗透测试步骤的操作类型为执行总结;

如果上一步渗透测试步骤对应的工具输出结果包含执行异常,则当前渗透测试步骤的操作类型为执行停止。

8.根据权利要求1所述的基于多智能体的大语言模型自动渗透测试方法,其特征在于,所述结合当前渗透测试步骤、上一步渗透测试步骤、上一步渗透测试步骤的执行结果转化为第三提示词,包含:获取检索得到的程序接口对应的接口调用格式、功能描述和参数列表描述;

根据当前渗透测试步骤的描述、上一步渗透测试步骤的描述、上一步渗透测试步骤的执行结果、接口调用格式、功能描述和参数列表描述转化为第三提示词。