[发明专利]终端推理框架漏洞挖掘方法、装置、设备及存储介质

权利要求书

1.一种终端推理框架漏洞挖掘方法，其特征在于，所述终端推理框架漏洞挖掘方法包括：

对待测框架的源码文档进行特征检测，得到算子特征库；

根据所述算子特征库和预设初始模型种子库构造模型种子库；

对所述模型种子库进行变异操作得到待测框架漏洞测试用例；

通过所述待测框架测试用例对所述待测框架进行漏洞测试。

2.如权利要求1所述的方法，其特征在于，所述对所述模型种子库进行变异操作得到待测框架测试用例，包括：

获取模型种子库中的区块信息和/或条目信息；

检测所述算子特征库中的约束信息；

根据所述约束信息和区块信息和/或条目信息生成待测框架测试用例。

3.如权利要求2所述的方法，其特征在于，所述根据所述约束信息和区块信息和/或条目信息生成待测框架测试用例，包括：

根据所述约束信息生成解析框架；

根据所述解析框架对区块信息和/或条目信息进行逻辑变异得到变异信息；

根据所述变异信息生成待测框架测试用例。

4.如权利要求2所述的方法，其特征在于，所述区块信息包括子图信息以及张量信息，所述条目信息包括维度信息以及属性信息。

5.如权利要求1所述的方法，其特征在于，所述根据所述算子特征库和预设初始模型种子库构造模型种子库，包括：

根据所述预设初始模型种子库得到初始单算子模型文件；

根据所述算子特征库获取特征权重信息；

根据所述特征权重信息和预设初始模型种子库构造新算子模型文件；

根据所述初始单算子模型文件和新算子模型文件生成模型种子库。

6.如权利要求5所述的方法，其特征在于，所述根据所述特征权重信息和预设初始模型种子库构造新算子模型文件，包括：

根据所述预设初始模型种子库构造第一新算子模型文件；

根据所述特征权重信息和预设初始模型种子库构造第二新算子模型文件；

根据所述第一新算子模型文件和第二新算子模型文件得到新算子模型文件。

7.如权利要求1所述的方法，其特征在于，所述对待测框架的源码文档进行特征检测，得到算子特征库，包括：

根据待测框架的源码文档测得到文件信息和算子特征；

根据预设权重算法获得所述算子特征的权重值；

根据所述文件信息、算子特征的权重值和算子特征建立算子特征库。

8.一种终端推理框架漏洞挖掘装置，其特征在于，所述终端推理框架漏洞挖掘装置包括：

特征提取模块，用于对待测框架的源码文档进行特征检测，得到算子特征库；

筛选模块，用于根据所述算子特征库和预设初始模型种子库构造模型种子库；

变异模块，用于对所述模型种子库进行变异操作得到待测框架漏洞测试用例；

测试模块，用于通过所述待测框架测试用例对所述待测框架进行漏洞测试。

9.一种终端推理框架漏洞挖掘设备，其特征在于，所述设备包括：存储器、处理器及存储在所述存储器上并可在所述处理器上运行的终端推理框架漏洞挖掘程序，所述终端推理框架漏洞挖掘程序配置为实现如权利要求1至7中任一项所述的终端推理框架漏洞挖掘方法。

10.一种存储介质，其特征在于，所述存储介质上存储有终端推理框架漏洞挖掘程序，所述终端推理框架漏洞挖掘程序被处理器执行时实现如权利要求1至7任一项所述的终端推理框架漏洞挖掘方法。