python实现上传样本到virustotal并查询扫描信息的方法
admin
2023-07-31 02:12:18
0

本文实例讲述了python实现上传样本到virustotal并查询扫描信息的方法。分享给大家供大家参考。具体方法如下:

import simplejson 
import urllib 
import urllib2 
import os  
 
MD5 = \"5248f774d2ee0a10936d0b1dc89107f1\" 
MD5 = \"12fa5fb74201d9b6a14f63fbf9a81ff6\" #do not have report on virustotal.com 
       
######################################################################## 
APIKEY = \"e0a50a50e77fxxxxxxxxxxxxxx4f17e31 这里用你自己在virustotal上申请的账号的KEY\" 
 
 
class VirusTotal: 
  \"\"\"\"\"\" 
 
  def __init__(self, md5): 
    \"\"\"Constructor\"\"\" 
    self._virus_dict = {} 
    self._md5 = md5 
     
     
  def repr(self): 
    return str(self._virus_dict) 
   
  def submit_md5(self, file_path): 
    import postfile                                      
    #submit the file 
    FILE_NAME = os.path.basename(file_path)  
               
                                                  
    host = \"www.virustotal.com\"                                
    selector = \"https://www.virustotal.com/vtapi/v2/file/scan\"                 
    fields = [(\"apikey\", APIKEY)] 
    file_to_send = open(file_path, \"rb\").read()                        
    files = [(\"file\", FILE_NAME, file_to_send)]                        
    json = postfile.post_multipart(host, selector, fields, files)               
    print json 
    pass 
   
  def get_report_dict(self): 
    result_dict = {} 
     
    url = \"https://www.virustotal.com/vtapi/v2/file/report\" 
    parameters = {\"resource\": self._md5, 
            \"apikey\": APIKEY} 
    data = urllib.urlencode(parameters) 
    req = urllib2.Request(url, data) 
    response = urllib2.urlopen(req) 
    json = response.read() 
     
    response_dict = simplejson.loads(json) 
    if response_dict[\"response_code\"]: #has result  
      scans_dict = response_dict.get(\"scans\", {}) 
      for anti_virus_comany, virus_name in scans_dict.iteritems(): 
        if virus_name[\"detected\"]: 
          self._virus_dict.setdefault(anti_virus_comany, virus_name[\"result\"]) 
    return self._virus_dict 

返回的结果为:{u\’Sophos\’: u\’Sus/Behav-1010\’},如果有扫描出的结果的话..

调用的方法如下:

MD5 = \"12fa5fb74201d9b6a14f63fbf9a81ff6\" #do not have report on virustotal.com 
MD5 = \"5248f774d2ee0a10936d0b1dc89107f1\" 
FILE_PATH = r\"D:\\backSample\\10\\9af41bc012d66c98ca2f9c68ba38e98f_ICQLiteShell.dll\" 
 
from getVirusTotalInfo import VirusTotal 
#得到扫描结果并打印出来 
virus_total = VirusTotal(MD5) 
print virus_total.get_report_dict() 
 
#提交文件到扫描,以后就可以根据这个MD5取扫描结果了 
virus_total.submit_md5(FILE_PATH) 

希望本文所述对大家的Python程序设计有所帮助。

相关内容

热门资讯

500 行 Python 代码... 语法分析器描述了一个句子的语法结构,用来帮助其他的应用进行推理。自然语言引入了很多意外的歧义,以我们...
定时清理删除C:\Progra... C:\Program Files (x86)下面很多scoped_dir开头的文件夹 写个批处理 定...
65536是2的几次方 计算2... 65536是2的16次方:65536=2⁶ 65536是256的2次方:65536=256 6553...
Mobi、epub格式电子书如... 在wps里全局设置里有一个文件关联,打开,勾选电子书文件选项就可以了。
scoped_dir32_70... 一台虚拟机C盘总是莫名奇妙的空间用完,导致很多软件没法再运行。经过仔细检查发现是C:\Program...
pycparser 是一个用... `pycparser` 是一个用 Python 编写的 C 语言解析器。它可以用来解析 C 代码并构...
小程序支付时提示:appid和... [Q]小程序支付时提示:appid和mch_id不匹配 [A]小程序和微信支付没有进行关联,访问“小...
微信小程序使用slider实现... 众所周知哈,微信小程序里面的音频播放是没有进度条的,但最近有个项目呢,客户要求音频要有进度条控制,所...
Prometheus+Graf... 一,Prometheus概述 1,什么是Prometheus?Prometheus是最初在Sound...
python绘图库Matplo... 本文简单介绍了Python绘图库Matplotlib的安装,简介如下: matplotlib是pyt...