25 lines
1.2 KiB
Python
25 lines
1.2 KiB
Python
|
import networkx as nx
|
||
|
|
import os
|
||
|
|
from tqdm import tqdm
|
||
|
|
import csv
|
||
|
|
#用于直接读取gexf信息保存的代码
|
||
|
|
def write_csv(CFG_dir = "../CFG_data/malware",csv_save_path="../CFG_data/malware_msg.csv",header = ['malware_name','nodes_num','edgs_num']):
|
||
|
|
file_list=os.listdir(CFG_dir)[:10000]
|
||
|
|
# benign_list=os.listdir(bengin_CFG_dir)
|
||
|
|
csv_data = []
|
||
|
|
|
||
|
|
for item_name in tqdm(file_list):
|
||
|
|
file=os.path.join(CFG_dir,item_name)
|
||
|
|
G=nx.read_gexf(file)
|
||
|
|
nodes_num = len(G.nodes())
|
||
|
|
edgs_num = len(G.edges())
|
||
|
|
csv_data.append([item_name, str(nodes_num), str(edgs_num)])
|
||
|
|
|
||
|
|
with open(csv_save_path,'w',encoding='utf-8',newline='') as fp:
|
||
|
|
writer =csv.writer(fp)
|
||
|
|
writer.writerow(header)
|
||
|
|
writer.writerows(csv_data)
|
||
|
|
print("成功结束")
|
||
|
|
if __name__ == '__main__':
|
||
|
|
write_csv(CFG_dir="../cfg_data_with_feature/benign1", csv_save_path="../cfg_data_with_feature/benign_msg1.csv",header=['benign_name', 'nodes_num', 'edgs_num','insert_point_count'])
|
||
|
|
write_csv(CFG_dir="../cfg_data_with_feature/malware1", csv_save_path="../cfg_data_with_feature/malware_msg1.csv",header=['malware_name', 'nodes_num', 'edgs_num', 'insert_point_count'])
|