一种基于混合检索增强框架的漏洞利用方案生成方法与流程

本发明涉及网络安全，具体而言，涉及一种基于混合检索增强框架的漏洞利用方案生成方法。

背景技术：

1、在当今数字化转型加速推进的时代，网络空间已成为国家关键基础设施运行、经济社会发展和人们日常生活不可或缺的重要组成部分。然而，伴随而来的是网络安全威胁的日益严峻和复杂多变。据权威机构统计，全球范围内的网络攻击事件数量呈指数级增长，攻击手段也愈发多样化和智能化，从传统的病毒、木马到高级持续性威胁（apt）、勒索软件等，对企业和个人的信息安全构成了严重威胁。在这样的背景下，渗透测试作为一种主动发现和修复系统安全漏洞的重要手段，其重要性愈发凸显。而漏洞利用方案的生成则是渗透测试过程中的关键环节，直接影响到测试的效果和效率。

2、传统的漏洞利用方案生成高度依赖专业人员的自身经验，需要手动进行目标环境分析、漏洞发现与匹配、利用方案生成等工作。在当下网络环境日趋复杂，漏洞数量呈爆炸式增长的背景下，人工进行漏洞利用的方式正逐渐变得低效，其时间与人力成本逐渐增加，且难以应对大规模、复杂多变的网络环境。虽然大语言模型（llm）的出现，为漏洞利用方案的生成带来了新的思路，大幅降低了对人工经验的依赖。然而，通用大模型对于漏洞利用领域的专业知识的储备往往存在不足，在处理漏洞类型、漏洞利用条件等专业知识时，容易理解错误而生成无效的利用方案。此外，大模型的知识更新往往具有滞后性，对于新出现的漏洞利用需求无法及时进行适配，因此生成的方案容易由于缺乏新知识而失去应用价值。

技术实现思路

1、本发明的目的在于提供一种基于混合检索增强框架的漏洞利用方案生成方法，其能够通过实时检索外部的权威知识库，为模型动态补充漏洞利用相关的专业知识、最新漏洞情报与典型案例。

2、本发明的技术方案为：

3、本申请提供一种基于混合检索增强框架的漏洞利用方案生成方法，其包括以下步骤：

4、s1、获取多源异构的漏洞知识数据，并对多源异构的漏洞知识数据进行提取和整合，基于整合结果利用neo4j图数据库构建漏洞知识库；

5、s2、将自然语言查询内容进行向量嵌入，转化为结构化的查询向量，同时将漏洞知识库中的数据转化为可供查询的查询向量；

6、s3、通过计算向量间的距离以进行相似度比较，根据相似度比较结果构建混合检索增强机制，并通过混合检索增强机制对检索到的知识进行综合衡量以输出相应知识及最佳图节点；

7、s4、采用智能体框架和本地部署模型对步骤s3输出的相应知识进行图关系扩展，获取最佳图节点的邻近关系，构建检索增强智能体框架，生成漏洞利用方案。

8、进一步地，上述多源异构的漏洞知识数据包括：att&ck、capec、cve、cwe和exploit-db五类数据源。

9、进一步地，上述对多源异构的漏洞知识数据进行提取和整合包括：针对五类数据源分别构建数据提取程序，从中提取12种实体与17种关系，统一使用“实体-属性-关系结构”进行统一格式存储。

10、进一步地，上述实体包含实体种类以及对应的实体id，上述属性包含存储实体的详细描述，上述关系结构定义为“起始点-关系-终止点”结构。

11、进一步地，步骤s3包括：

12、通过计算向量间的距离获取相似度分数，将相似度分数进行对数化，并采用反比例归一化公式进行归一化处理；

13、根据归一化处理结果进行相似度比较，根据相似度比较结果及候选图节点本身在图中的重要性，构建“假设文档嵌入+语义相似度+图节点重要性+rag重排序”的混合检索机制；

14、通过混合检索增强机制对检索结果进行综合衡量以输出相应知识及最佳图节点。

15、进一步地，步骤s3还包括：采用假设文档嵌入技术和rag重排技术对检索结果进行优化；其中，上述假设文档嵌入技术包括：针对较为简短、关键词较少的查询，使用通用llm生成一个回答作为“假设文档”，以这份假设文档作为新的查询，去匹配真实文档；上述rag重排技术包括对初步检索到的候选内容，通过重排序模型进行二次排序，根据二次排序结果选取高价值节点。

16、进一步地，步骤s4中，上述智能体框架为轻量智能体框架smolagents；上述本地部署模型为qwen3.5-35b-a3b模型。

17、相对于现有技术，本发明至少具有如下优点或有益效果：

18、（1）本发明提供的一种基于混合检索增强框架的漏洞利用方案生成方法，通过构建综合性漏洞知识库，为大模型实时补充漏洞利用领域的专业知识，有效避免了通用大模型因知识储备不足而生成无效利用方案的问题，提高了漏洞利用方案的准确性和实用性；

19、（2）本发明将分散在att&ck、capec、cve、cwe、exploit-db等多个数据源的漏洞知识进行有效整合，构建了统一的漏洞知识图谱，解决了现有漏洞利用知识分散、格式不统一的问题，为漏洞利用方案生成提供了高质量的知识支撑；

20、（3）本发明采用“假设文档嵌入+语义相似度+图节点重要性+rag重排序”的混合检索机制，能够对检索到的知识进行综合衡量，筛选出最具价值的图节点，显著提高了检索的精准度和效率，为智能体生成高质量漏洞利用方案提供了可靠的数据基础；

21、（4）采用smolagents轻量化智能体框架和qwen3.5-35b-a3b小参数量高性能模型，在保证系统性能的同时，降低了系统部署和运行的硬件要求，使得本发明能够在消费级设备上运行，具有更广泛的应用前景；

22、（5）通过图关系扩展技术，能够从检索到的节点出发，构建完整的漏洞生成方案，不仅提高了方案的全面性和系统性，还为渗透测试人员提供了更具指导性的操作建议。

1.一种基于混合检索增强框架的漏洞利用方案生成方法，其特征在于，包括以下步骤：

2.如权利要求1所述的一种基于混合检索增强框架的漏洞利用方案生成方法，其特征在于，所述多源异构的漏洞知识数据包括：att&ck、capec、cve、cwe和exploit-db五类数据源。

3.如权利要求2所述的一种基于混合检索增强框架的漏洞利用方案生成方法，其特征在于，所述对多源异构的漏洞知识数据进行提取和整合包括：针对五类数据源分别构建数据提取程序，从中提取实体与关系，统一使用“实体-属性-关系结构”进行统一格式存储。

4.如权利要求3所述的一种基于混合检索增强框架的漏洞利用方案生成方法，其特征在于，所述实体包含实体种类以及对应的实体id，所述属性包含存储实体的详细描述，所述关系结构定义为“起始点-关系-终止点”结构。

5.如权利要求4所述的一种基于混合检索增强框架的漏洞利用方案生成方法，其特征在于，步骤s3包括：

6.如权利要求5所述的一种基于混合检索增强框架的漏洞利用方案生成方法，其特征在于，还包括：采用假设文档嵌入技术和rag重排技术对检索结果进行优化；其中，所述假设文档嵌入技术包括：针对较为简短、关键词较少的查询，使用通用llm生成一个回答作为“假设文档”，以这份假设文档作为新的查询，去匹配真实文档；所述rag重排技术包括对初步检索到的候选内容，通过重排序模型进行二次排序，根据二次排序结果选取高价值节点。

7.如权利要求1所述的一种基于混合检索增强框架的漏洞利用方案生成方法，其特征在于，步骤s4中，所述智能体框架为轻量智能体框架smolagents；所述本地部署模型为qwen3.5-35b-a3b模型。