[發明專利]一種基于圖卷積網絡的源代碼漏洞檢測方法有效
| 申請號: | 202010168037.0 | 申請日: | 2020-03-11 |
| 公開(公告)號: | CN111460450B | 公開(公告)日: | 2023-02-10 |
| 發明(設計)人: | 孔維星;葉貴鑫;王煥廷;湯戰勇;房鼎益 | 申請(專利權)人: | 西北大學 |
| 主分類號: | G06F21/56 | 分類號: | G06F21/56;G06N3/0464;G06N3/08;G06N20/00 |
| 代理公司: | 西安恒泰知識產權代理事務所 61216 | 代理人: | 李鄭建 |
| 地址: | 710069 *** | 國省代碼: | 陜西;61 |
| 權利要求書: | 查看更多 | 說明書: | 查看更多 |
| 摘要: | |||
| 搜索關鍵詞: | 一種 基于 圖卷 網絡 源代碼 漏洞 檢測 方法 | ||
1.一種基于圖卷積網絡的源代碼漏洞自動檢測方法,包括訓練階段和測試階段,其特征在于:
在訓練階段,對訓練數據集進行數據獲取和數據預處理,數據預處理的過程是,首先判斷訓練數據集的源代碼是否有sink方法調用,對sink方法調用進行數據流分析,然后將與sink方法調用相關的可疑代碼構造為代碼屬性圖,并計算代碼屬性圖的鄰接矩陣與特征矩陣,并且給sink方法調用和代碼屬性圖加標簽,將代碼屬性圖中的鄰接矩陣和特征矩陣作為圖卷積網絡的輸入,訓練圖卷積網絡,進一步得到訓練好的網絡模型;
在測試階段,對測試數據集進行數據獲取和數據預處理,數據預處理的過程是,首先判斷測試數據集的源代碼是否有sink方法調用,對sink方法調用進行數據流分析,然后將與sink方法調用相關的可疑代碼構造為代碼屬性圖,并計算代碼屬性圖的鄰接矩陣與特征矩陣,將鄰接矩陣與特征矩陣輸入訓練好的圖卷積網絡模型,圖卷積網絡模型輸出代碼屬性圖的分類結果,表示代碼屬性圖是否有漏洞即對應的sink方法調用是否含有漏洞。
該專利技術資料僅供研究查看技術是否侵權等信息,商用須獲得專利權人授權。該專利全部權利屬于西北大學,未經西北大學許可,擅自商用是侵權行為。如果您想購買此專利、獲得商業授權和技術合作,請聯系【客服】
本文鏈接:http://www.szxzyx.cn/pat/books/202010168037.0/1.html,轉載請聲明來源鉆瓜專利網。





