[发明专利]一种基于WGAN模型的模糊检测种子集合生成方法及生成器

说明书

本发明公开了一种基于WGAN模型的模糊检测种子集合生成器，包括：训练集合获取模块，具有基于突变算法的模糊检测工具，以普通输入作为种子检测多个输入格式相同的程序，将发现程序新路径或触发程序异常情况的输入作为WGAN模型的训练集合；二进制转矩阵模块，将训练集合由二进制形式转换成矩阵形式；WGAN模块，包含WGAN模型，以矩阵形式的训练集合为真实数据集，将WGAN模型训练至收敛，通过训练后的WGAN模型生成矩阵形式的模糊检测种子集合；矩阵转二进制模块，将模糊检测种子集合由矩阵形式转换为二进制形式。本发明还公开了基于模糊检测种子集合生成器来生成模糊检测种子集合的方法，该方法获取的模糊检测种子集合能大大提高模糊检测的漏洞挖掘效率。

技术领域

本发明涉及软件模糊检测技术领域，尤其涉及一种基于WGAN模型的模糊检测种子集合生成方法及生成器。

背景技术

随着计算机技术和互联网信息技术的发展，各式各样的软件与系统被开发出来，软件的功能也变得越来越丰富。然而由于开发人员水平参差不齐、开发人员之间沟通不足、架构设计不合理等原因，软件中往往存在着会导致程序崩溃、甚至会受不法分子利用的漏洞。由于软件漏洞总是不可避免地存在，因而漏洞检测受到广泛关注。传统的漏洞检测方案是由富有经验的安全人员人工检查代码，然而该方案任务繁重，成本高，效率较低，检测漏洞的召回率也较低，大型系统难以用该方案完整测试。新兴的一种漏洞检测方案是由机器学习算法去检查代码，返回可能存在漏洞的代码位置。然而机器学习算法检测软件漏洞的方案存在着诸多限制与不足：该方案误报率较高；只能做到缩小检测范围的作用，最终仍需要安全人员人工检查是否存在漏洞；该方案的机器学习模型效果取决于训练集合是否完备、训练次数是否合理，实际过程中存在机器学习模型在训练集合上表现优良但在使用中表现不尽人意、训练过拟合欠拟合等情况。

有别于以上提到的漏洞检测方案，模糊测试是一种自动化的软件漏洞检测方法。模糊测试可以分成两大类型：(1)基于规范生成的模糊测试；(2)基于突变算法的模糊测试。基于规范生成的模糊测试根据人工提供的规范随机生成具有一定格式的种子文件，模糊测试将生成的种子输入被测软件，并监视被测软件是否出现预期外的异常情况。基于突变算法的模糊测试根据突变算法对人工输入的种子集合进行位变换、组合交叉等处理，获取到大量随机输入集合。模糊测试将集合输入被测软件，并监视软件是否出现预期外的异常情况。当模糊测试发现软件的新路径或者软件出现异常情况，它会将发现该新路径或触发该异常情况的输入文件保存在本地，并认为该输入文件为有价值的输入。模糊测试会将发现新路径或触发异常情况的输入文件加入种子集合，并以这些输入作为父代基因，基于突变算法生成后续输入。由于模糊测试的输入生成具有随机性，它并不会猜测哪个输入数据会导致程序崩溃，所以它的软件漏洞挖掘具有一定的盲目性。但是模糊检测是自动化的工具，借助当前计算机强大的计算能力，它将尽可能多的杂乱数据输入程序中，而记录下的发现新路径或触发程序崩溃的输入也为后续的漏洞挖掘提供了一定的启发性，效率高于上述提到的漏洞检测方案。其中AFL是受到广泛认可的基于突变算法的模糊测试工具之一，它漏洞挖掘速率极高。

研究表明，选择高质量的种子集合有助于提高基于突变算法的模糊测试的效率。而且高质量的种子集合具有可迁移性，对于使用同一种输入格式的程序A和B，如果某一种子集合可以提高模糊测试对于程序A的漏洞挖掘效率，那么该种子集合也能提高对于程序B的漏洞挖掘效率。然而当前并没有快速的产生高质量种子集合的方法，虽然存在一些启发式的种子选择方法，但是这些方法效果有限，并没有明显优于随机选择种子集合。所以，学术界与工业界都缺少合适的获取高质量种子集合的方法。

发明内容

本发明提供了一种基于WGAN模型的模糊检测种子集合生成器，该生成器可以生成任意输入格式的模糊检测种子集合，获取的模糊检测种子集合能大大提高基于突变算法的模糊检测的漏洞挖掘效率。

本发明提供了如下技术方案：

一种基于WGAN模型的模糊检测种子集合生成器，包括：