基于代碼嵌入的二進制代碼相似性分析方法
網(wǎng)絡(luò)安全與數(shù)據(jù)治理 2023年3期
熊敏，薛吟興，徐云
（1.中國科學(xué)技術(shù)大學(xué)計算機科學(xué)與技術(shù)學(xué)院，安徽合肥230026； 2.安徽省高性能計算重點實驗室，安徽合肥230026）
摘要： 代碼嵌入利用神經(jīng)網(wǎng)絡(luò)模型將二進制函數(shù)的代碼表示轉(zhuǎn)化為向量，在漏洞搜索等應(yīng)用中展現(xiàn)了優(yōu)勢?，F(xiàn)有的方法將函數(shù)表示為匯編指令序列、控制流圖的拓撲結(jié)構(gòu)或若干路徑，都沒有克服不同編譯環(huán)境導(dǎo)致控制流圖結(jié)構(gòu)變化的干擾。為此,設(shè)計了基于基本塊樹（Basic Block Tree, BBT)的代碼表示以及構(gòu)建了對應(yīng)的代碼嵌入模型BBTree。首先，二進制函數(shù)被表示為一系列BBT，每個BBT被處理為指令序列；其次，BBTree利用LSTM和BiGRU將基于BBT的代碼表示轉(zhuǎn)化為向量；最后，通過計算向量間的距離去高效衡量對應(yīng)函數(shù)的相似性。在代碼搜索中，BBTree的平均準確率比主流工具提升了24.8%；在漏洞搜索中，BBTree的平均召回率比主流工具提升了26.1%。
中圖分類號：TP311.5
文獻標識碼：A
DOI:10.19358/j.issn.2097-1788.2023.03.010
引用格式：熊敏，薛吟興，徐云.基于代碼嵌入的二進制代碼相似性分析方法［J］.網(wǎng)絡(luò)安全與數(shù)據(jù)治理，2023,42(3):58-67.
文獻標識碼：A
DOI:10.19358/j.issn.2097-1788.2023.03.010
引用格式：熊敏，薛吟興，徐云.基于代碼嵌入的二進制代碼相似性分析方法［J］.網(wǎng)絡(luò)安全與數(shù)據(jù)治理，2023,42(3):58-67.
A binary code similarity analysis method based on code embedding
Xiong Min1,2，Xue Yinxing1，Xu Yun 1,2
(1. School of Computer Science and Technology, University of Science and Technology of China, Hefei 230026, China; 2. Key Laboratory of High Performance Computing of Anhui Province, Hefei 230026, China)
Abstract： Code embedding utilizes neural network models to convert binary code into a vector, showing advantages in applications such as vulnerability searching. Existing methods represent functions as assembly instruction sequences, topology structures of control flow graphs, or several paths.However, none of them can overcome the interference produced by the structural changes in control flow graphs caused by different compilation environments.To this end, this paper designs a basic block tree (BBT)-based code representation and builds a corresponding code embedding model named BBTree.Firstly, the binary function is represented as a series of BBTs, and each BBT is processed into an instruction sequence Secondly, BBTree utilizes LSTM and Bi.GRU to convert the BBT.based code representation into a numerical vector Last, the distance between vectors is calculated to efficiently measure the similarity of corresponding functions. In code search, BBTree’s average accuracy rate is 24.8% higher than mainstream tools; in vulnerability search, BBTree’s average recall rate is 26.1% higher than mainstream tools.
Key words :
0 引言
由于商業(yè)程序、遺留程序和惡意代碼的源碼不公開， 因此，對這些程序進行二進制代碼相似性分析具有很多安全應(yīng)用，比如抄襲檢測、惡意軟件檢測、漏洞搜索等。相似性分析旨在根據(jù)已有的二進制代碼（如已揭露的漏洞等）在代碼庫中搜索出語義相似的二進制代碼，從而探測出潛在的漏洞，維護程序的安全。二進制代碼嵌入作為一種新興的相似性分析技術(shù)，利用神經(jīng)網(wǎng)絡(luò)模型將二進制函數(shù)的代碼表示轉(zhuǎn)化為數(shù)值向量，不僅學(xué)習(xí)了二進制代碼的語義，還可以通過計算向量間的距離去定量分析對應(yīng)函數(shù)的相似性。
本文詳細內(nèi)容請下載：http://m.xxav2194.com/resource/share/2000005257
作者信息：
熊敏1,2，薛吟興1，徐云1,2
（1.中國科學(xué)技術(shù)大學(xué)計算機科學(xué)與技術(shù)學(xué)院，安徽合肥230026；2.安徽省高性能計算重點實驗室，安徽合肥230026）
此內(nèi)容為AET網(wǎng)站原創(chuàng)，未經(jīng)授權(quán)禁止轉(zhuǎn)載。