[發明專利]漏洞修復方法、裝置、設備及可讀存儲介質

在審

• 申請號: 202211247035.6
• 申請日: 2022-10-12
• 公開（公告）號: CN115455433A
• 公開（公告）日: 2022-12-09
• 發明（設計）人: 鄭子彬;莫一凡;郭瀚陽
• 申請（專利權）人: 中山大學
• 主分類號: G06F21/57
• 分類號: G06F21/57
• 代理公司: 北京集佳知識產權代理有限公司 11227
• 代理人: 李偉賢

1.一種漏洞修復方法，其特征在于，包括：

獲取待修復智能合約的源代碼；

將所述待修復智能合約的源代碼進行序列化，得到目標語言序列；

利用預設的預測模型對所述目標語言序列進行處理，得到預測后的目標語言序列，其中，所述預測模型以對有漏洞的智能合約源代碼進行序列化得到的初始語言序列作為訓練樣本，以將所述初始語言序列中，漏洞對應的語言子序列替換為修復后的語言子序列，得到的語言序列作為樣本標簽訓練得到；

將所述預測后的目標語言序列進行反序列化，得到已修復的智能合約的源代碼。

2.根據權利要求1所述的漏洞修復方法，其特征在于，所述將所述待修復智能合約的源代碼進行序列化，得到目標語言序列，包括：

對所述待修復智能合約的源代碼進行詞法分析掃描，得到所述待修復智能合約的源代碼的標識符列表，所述標識符列表包括一個或多個標識符；

將所述標識符列表轉換成抽象語法樹AST；

利用預設的標識符字典中的標識符替換掉所述AST中與所述預設的標識符字典中的標識符對應的節點的標識符，得到替換后的AST，其中，所述預設的標識符字典中包括一個或多個與所述待修復智能合約的源代碼相關且使用頻率達到預設閾值的標識符；

將所述替換后的AST進行前序遍歷，得到目標語言序列。

3.根據權利要求1所述的漏洞修復方法，其特征在于，所述預測模型包括編碼器和解碼器，

所述利用預設的預測模型對所述目標語言序列進行處理，得到預測后的目標語言序列，包括：

利用預設的預測模型中的編碼器對所述目標語言序列進行編碼，得到編碼后的目標語言序列；

利用預設的預測模型中的解碼器對編碼后的目標語言序列進行解碼，得到預測后的目標語言序列。

4.根據權利要求3所述的漏洞修復方法，其特征在于，所述利用預設的預測模型中的解碼器對編碼后的目標語言序列進行解碼，得到預測后的目標語言序列，包括：

利用預設的預測模型中的解碼器對編碼后的目標語言序列進行解碼，得到所述目標語言序列的狀態序列；

利用預設的預測模型中的解碼器基于所述目標語言序列的狀態序列進行預測，得到預測后的目標語言序列。

5.根據權利要求1所述的漏洞修復方法，其特征在于，所述將所述預測后的目標語言序列進行反序列化，得到已修復的智能合約的源代碼，包括：

將所述預測后的目標語言序列轉化為AST；

將所述AST轉換為源代碼，得到已修復的智能合約的源代碼。

6.一種漏洞修復裝置，其特征在于，包括：

代碼獲取單元，用于獲取待修復智能合約的源代碼；

序列化單元，用于將所述待修復智能合約的源代碼進行序列化，得到目標語言序列；

預測單元，用于利用預設的預測模型對所述目標語言序列進行處理，得到預測后的目標語言序列，其中，所述預測模型以對有漏洞的智能合約源代碼進行序列化得到的初始語言序列作為訓練樣本，以將所述初始語言序列中，漏洞對應的語言子序列替換為修復后的語言子序列，得到的語言序列作為樣本標簽訓練得到；

反序列化單元，用于將所述預測后的目標語言序列進行反序列化，得到已修復的智能合約的源代碼。

7.根據權利要求6所述的漏洞修復裝置，其特征在于，所述序列化單元的執行過程，包括：

對所述待修復智能合約的源代碼進行詞法分析掃描，得到所述待修復智能合約的源代碼的標識符列表，所述標識符列表包括一個或多個標識符；

將所述標識符列表轉換成抽象語法樹AST；

利用預設的標識符字典中的標識符替換掉所述AST中與所述預設的標識符字典中的標識符對應的節點的標識符，得到替換后的AST，其中，所述預設的標識符字典中包括一個或多個與所述待修復智能合約的源代碼相關且使用頻率達到預設閾值的標識符；

將所述替換后的AST進行前序遍歷，得到目標語言序列。