Merge pull request #111 from kx499/master

Handful of changes to VirusTotal module

misp_modules/modules/expansion/__init__.py

@@ -2,4 +2,4 @@ from . import _vmray
 
 __all__ = ['vmray_submit', 'asn_history', 'circl_passivedns', 'circl_passivessl',
            'countrycode', 'cve', 'dns', 'domaintools', 'eupi', 'ipasn', 'passivetotal', 'sourcecache',
-           'virustotal', 'whois', 'shodan', 'reversedns', 'geoip_country', 'wiki']
+           'virustotal', 'whois', 'shodan', 'reversedns', 'geoip_country', 'wiki', 'iprep']

misp_modules/modules/expansion/iprep.py

@@ -0,0 +1,88 @@
+# -*- coding: utf-8 -*-
+
+import json
+import requests
+
+
+misperrors = {'error': 'Error'}
+mispattributes = {'input': ['ip-src', 'ip-dst'], 'output': ['text']}
+moduleinfo = {'version': '1.0', 'author': 'Keith Faber',
+              'description': 'Query IPRep Data for IP Address',
+              'module-type': ['expansion']}
+
+moduleconfig = ['apikey']
+
+
+def handler(q=False):
+    if q is False:
+        return False
+    request = json.loads(q)
+    if request.get('ip-src'):
+        toquery = request['ip-src']
+    elif request.get('ip-dst'):
+        toquery = request['ip-dst']
+    else:
+        misperrors['error'] = "Unsupported attributes type"
+        return misperrors
+
+    if not request.get('config') and not request['config'].get('apikey'):
+        misperrors['error'] = 'IPRep api key is missing'
+        return misperrors
+
+    err, rep = parse_iprep(toquery, request['config'].get('apikey'))
+    if len(err) > 0:
+        misperrors['error'] = ','.join(err)
+        return misperrors
+    return {'results': rep}
+
+
+def parse_iprep(ip, api):
+    meta_fields = ['origin', 'Query_Time', 'created_on', 'IP_Lookup_History', 'IPs_in_collection', '_id', 'disclaimer',
+                   'MaxMind_Free_GeoIP', 'Unique_Lookups', 'query_result']
+    rep = []
+    err = []
+    full_text = ''
+    url = 'https://www.packetmail.net/iprep.php/%s' % ip
+    try:
+        data = requests.get(url, params={'apikey': api}).json()
+    except:
+        return ['Error pulling data'], rep
+    # print '%s' % data
+    for name, val in data.items():
+        if name not in meta_fields:
+            try:
+                context = val['context']
+                if type(context) is list:
+                    if context[0].get('alert'):
+                        context = ','.join([hit['alert']['signature'] for hit in context])
+                    elif context[0].get('signature'):
+                        context = ','.join([hit['signature'] for hit in context])
+                    elif context[0].get('target_port') and context[0].get('protocol'):
+                        context = ','.join(
+                            ['Port Attacked: %s %s' % (hit['target_port'], hit['protocol']) for hit in context])
+                    elif context[0].get('phishing_kit') and context[0].get('url'):
+                        context = ','.join(['%s (%s)' % (hit['phishing_kit'], hit['url']) for hit in context])
+                    else:
+                        context = ';'.join(['%s: %s' % (k, v) for k, v in context[0].items()])
+
+                if val.get('special_note'):
+                    context += '; ' + val['special_note']
+
+                misp_val = context
+                full_text += '\n%s' % context
+                misp_comment = 'IPRep Source %s: %s' % (name, val['last_seen'])
+                rep.append({'types': mispattributes['output'], 'categories':['External analysis'], 'values': misp_val, 'comment': misp_comment})
+            except:
+                err.append('Error parsing source: %s' % name)
+
+    rep.append({'types': ['freetext'], 'values': full_text , 'comment': 'Free text import of IPRep'})
+    return err, rep
+
+
+def introspection():
+    return mispattributes
+
+
+def version():
+    moduleinfo['config'] = moduleconfig
+    return moduleinfo

misp_modules/modules/expansion/virustotal.py

@@ -7,7 +7,8 @@ import os
 
 misperrors = {'error': 'Error'}
 mispattributes = {'input': ['hostname', 'domain', "ip-src", "ip-dst", "md5", "sha1", "sha256", "sha512"],
-                  'output':['domain', "ip-src", "ip-dst", "text", "md5", "sha1", "sha256", "sha512", "ssdeep", "authentihash", "filename"]
+                  'output': ['domain', "ip-src", "ip-dst", "text", "md5", "sha1", "sha256", "sha512", "ssdeep",
+                             "authentihash", "filename"]
                   }
 
 # possible module-types: 'expansion', 'hover' or both
@@ -18,6 +19,8 @@ moduleinfo = {'version': '2', 'author': 'Hannah Ward',
 # config fields that your code expects from the site admin
 moduleconfig = ["apikey", "event_limit"]
 limit = 5  # Default
+comment = '%s: Enriched via VT'
+
 
 def handler(q=False):
     global limit
@@ -55,12 +58,17 @@ def handler(q=False):
     r["results"] = uniq
     return r
 
+
 def getHash(hash, key, do_not_recurse=False):
     global limit
     toReturn = []
+    try:
         req = requests.get("https://www.virustotal.com/vtapi/v2/file/report",
                            params={"allinfo": 1, "apikey": key, 'resource': hash}
                            ).json()
+    except:
+        return []
+
     if req["response_code"] == 0:
         # Nothing found
         return []
@@ -68,19 +76,24 @@ def getHash(hash, key, do_not_recurse = False):
     toReturn += getMoreInfo(req, key)
     return toReturn
 
+
 def getIP(ip, key, do_not_recurse=False):
     global limit
     toReturn = []
+    try:
         req = requests.get("https://www.virustotal.com/vtapi/v2/ip-address/report",
                            params={"ip": ip, "apikey": key}
                            ).json()
+    except:
+        return []
+
     if req["response_code"] == 0:
         # Nothing found
         return []
 
     if "resolutions" in req:
         for res in req["resolutions"][:limit]:
-        toReturn.append( {"types":["domain"], "values":[res["hostname"]]})
+            toReturn.append({"types": ["domain"], "values": [res["hostname"]], "comment":comment % ip})
             # Pivot from here to find all domain info
             if not do_not_recurse:
                 toReturn += getDomain(res["hostname"], key, True)
@@ -88,25 +101,34 @@ def getIP(ip, key, do_not_recurse = False):
     toReturn += getMoreInfo(req, key)
     return toReturn
 
+
 def getDomain(domain, key, do_not_recurse=False):
     global limit
     toReturn = []
+    try:
         req = requests.get("https://www.virustotal.com/vtapi/v2/domain/report",
                            params={"domain": domain, "apikey": key}
                            ).json()
+    except:
+        return []
+
     if req["response_code"] == 0:
         # Nothing found
         return []
 
     if "resolutions" in req:
         for res in req["resolutions"][:limit]:
-        toReturn.append( {"types":["ip-dst", "ip-src"], "values":[res["ip_address"]]})
+            toReturn.append({"types": ["ip-dst", "ip-src"], "values": [res["ip_address"]], "comment":comment % domain})
             # Pivot from here to find all info on IPs
             if not do_not_recurse:
                 toReturn += getIP(res["ip_address"], key, True)
+    if "subdomains" in req:
+        for subd in req["subdomains"]:
+            toReturn.append({"types": ["domain"], "values": [subd], "comment": comment % domain})
     toReturn += getMoreInfo(req, key)
     return toReturn
 
+
 def findAll(data, keys):
     a = []
     if isinstance(data, dict):
@@ -122,6 +144,7 @@ def findAll(data, keys):
 
     return a
 
+
 def getMoreInfo(req, key):
     global limit
     r = []
@@ -131,22 +154,25 @@ def getMoreInfo(req, key):
     r.append({"types": ["md5", "sha1", "sha256", "sha512"], "values": hashes})
     for hsh in hashes[:limit]:
         # Search VT for some juicy info
+        try:
             data = requests.get("http://www.virustotal.com/vtapi/v2/file/report",
                                 params={"allinfo": 1, "apikey": key, "resource": hsh}
                                 ).json()
+        except:
+            continue
 
         # Go through each key and check if it exists
         if "submission_names" in data:
-        r.append({'types':["filename"], "values":data["submission_names"]})
+            r.append({'types': ["filename"], "values": data["submission_names"], "comment":comment % hsh})
 
         if "ssdeep" in data:
-        r.append({'types':["ssdeep"], "values":[data["ssdeep"]]})
+            r.append({'types': ["ssdeep"], "values": [data["ssdeep"]], "comment":comment % hsh})
 
         if "authentihash" in data:
-        r.append({"types":["authentihash"], "values":[data["authentihash"]]})
+            r.append({"types": ["authentihash"], "values": [data["authentihash"]], "comment":comment % hsh})
 
         if "ITW_urls" in data:
-        r.append({"types":["url"], "values":data["ITW_urls"]})
+            r.append({"types": ["url"], "values": data["ITW_urls"], "comment":comment % hsh})
 
         # Get the malware sample
         sample = requests.get("https://www.virustotal.com/vtapi/v2/file/download",
@@ -165,10 +191,11 @@ def getMoreInfo(req, key):
 
     return r
 
+
 def introspection():
     return mispattributes
 
+
 def version():
     moduleinfo['config'] = moduleconfig
     return moduleinfo
-