[发明专利]一种软件源代码缺陷自动修复方法及系统

权利要求书

1.一种软件源代码缺陷自动修复方法，其特征在于，包括：

提取源代码的缺陷相关CPG子图；

使用经过训练的缺陷修复模型基于所述CPG子图得到缺陷修复结果；其中，所述缺陷修复模型由一个图编码器、n个序列解码器和一个语法纠正解码器组成，所述n个序列解码器的输入端分别与所述图编码器的输出端连接，所述语法纠正解码器的输入端与所述n个序列解码器的输出端连接；其中，n ≥ 2；

所述缺陷修复模型的训练数据包括：1）包含漏洞的源代码片段；2）修复代码；3）修复代码的行数；4）修复代码中包含的语法相关符号序列；

其中，训练所述缺陷修复模型时，所述语法纠正解码器的输入包括所述训练数据的语法相关符号序列和所述序列解码器输出的修复代码序列。

2.根据权利要求1所述的软件源代码缺陷自动修复方法，其特征在于，所述图编码器采用GGNN网络，每个所述序列解码器均采用LSTM网络，所述语法纠正解码器采用LSTM网络。

3.根据权利要求1所述的软件源代码缺陷自动修复方法，其特征在于，所述提取源代码的缺陷相关CPG子图，包括：

获取所述源代码的CPG；

提取所述CPG中与缺陷相关的节点和边，以合成所述CPG子图。

4. 根据权利要求3所述的软件源代码缺陷自动修复方法，其特征在于，所述提取源代码的缺陷相关CPG子图，还包括：

将合成的所述CPG子图的每个节点中的漏洞特征转换为符号表示；以及

将所述符号表示转换为向量表示。

5.根据权利要求4所述的软件源代码缺陷自动修复方法，其特征在于，将基于源代码的每个词转换得到的向量的值相加作为整个源代码的向量表示输入所述缺陷修复模型。

6.一种软件源代码缺陷自动修复系统，其特征在于，包括：

子图提取模块，用于提取源代码的缺陷相关CPG子图；

缺陷修复模块，使用经过训练的缺陷修复模型基于所述CPG子图得到缺陷修复结果；其中，所述缺陷修复模型由一个图编码器、n个序列解码器和一个语法纠正解码器组成，所述n个序列解码器的输入端分别与所述图编码器的输出端连接，所述语法纠正解码器的输入端与所述n个序列解码器的输出端连接；其中，n ≥ 2；

所述缺陷修复模型的训练数据包括：1）包含漏洞的源代码片段；2）修复代码；3）修复代码的行数；4）修复代码中包含的语法相关符号序列；

其中，训练所述缺陷修复模型时，所述语法纠正解码器的输入包括所述训练数据的语法相关符号序列和所述序列解码器输出的修复代码序列。

7.一种电子设备，其特征在于，包括：存储器，用于存储计算机程序；处理器，与所述存储器数据耦合，用于在执行所述计算机程序时，实现根据权利要求1-5任一项所述的软件源代码缺陷自动修复方法。

8.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质上存储有计算机程序，所述计算机程序被处理器执行时，实现根据权利要求1-5任一项所述的软件源代码缺陷自动修复方法。