mirror of https://github.com/MISP/misp-modules
148 lines
6.8 KiB
Python
148 lines
6.8 KiB
Python
import json
|
|
import logging
|
|
import sys
|
|
import os
|
|
from apiosintDS import apiosintDS
|
|
|
|
log = logging.getLogger('apiosintDS')
|
|
log.setLevel(logging.DEBUG)
|
|
apiodbg = logging.StreamHandler(sys.stdout)
|
|
apiodbg.setLevel(logging.DEBUG)
|
|
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
apiodbg.setFormatter(formatter)
|
|
log.addHandler(apiodbg)
|
|
|
|
misperrors = {'error': 'Error'}
|
|
|
|
mispattributes = {'input': ["domain", "domain|ip", "hostname", "ip-dst", "ip-src", "ip-dst|port", "ip-src|port", "url",
|
|
"md5", "sha1", "sha256", "filename|md5", "filename|sha1", "filename|sha256"],
|
|
'output': ["domain", "ip-dst", "url", "comment", "md5", "sha1", "sha256"]
|
|
}
|
|
|
|
moduleinfo = {'version': '0.1', 'author': 'Davide Baglieri aka davidonzo',
|
|
'description': 'On demand query API for OSINT.digitalside.it project.',
|
|
'module-type': ['expansion', 'hover']}
|
|
|
|
moduleconfig = ['import_related_hashes', 'cache', 'cache_directory']
|
|
|
|
|
|
def handler(q=False):
|
|
if q is False:
|
|
return False
|
|
request = json.loads(q)
|
|
tosubmit = []
|
|
if request.get('domain'):
|
|
tosubmit.append(request['domain'])
|
|
elif request.get('domain|ip'):
|
|
tosubmit.append(request['domain|ip'].split('|')[0])
|
|
tosubmit.append(request['domain|ip'].split('|')[1])
|
|
elif request.get('hostname'):
|
|
tosubmit.append(request['hostname'])
|
|
elif request.get('ip-dst'):
|
|
tosubmit.append(request['ip-dst'])
|
|
elif request.get('ip-src'):
|
|
tosubmit.append(request['ip-src'])
|
|
elif request.get('ip-dst|port'):
|
|
tosubmit.append(request['ip-dst|port'].split('|')[0])
|
|
elif request.get('ip-src|port'):
|
|
tosubmit.append(request['ip-src|port'].split('|')[0])
|
|
elif request.get('url'):
|
|
tosubmit.append(request['url'])
|
|
elif request.get('md5'):
|
|
tosubmit.append(request['md5'])
|
|
elif request.get('sha1'):
|
|
tosubmit.append(request['sha1'])
|
|
elif request.get('sha256'):
|
|
tosubmit.append(request['sha256'])
|
|
elif request.get('filename|md5'):
|
|
tosubmit.append(request['filename|md5'].split('|')[1])
|
|
elif request.get('filename|sha1'):
|
|
tosubmit.append(request['filename|sha1'].split('|')[1])
|
|
elif request.get('filename|sha256'):
|
|
tosubmit.append(request['filename|sha256'].split('|')[1])
|
|
else:
|
|
return False
|
|
|
|
submitcache = False
|
|
submitcache_directory = False
|
|
import_related_hashes = False
|
|
|
|
r = {"results": []}
|
|
|
|
if request.get('config'):
|
|
if request['config'].get('cache') and request['config']['cache'].lower() == "yes":
|
|
submitcache = True
|
|
if request['config'].get('import_related_hashes') and request['config']['import_related_hashes'].lower() == "yes":
|
|
import_related_hashes = True
|
|
if submitcache:
|
|
cache_directory = request['config'].get('cache_directory')
|
|
if cache_directory and len(cache_directory) > 0:
|
|
if os.access(cache_directory, os.W_OK):
|
|
submitcache_directory = cache_directory
|
|
else:
|
|
ErrorMSG = "Cache directory is not writable. Please fix it before."
|
|
log.debug(str(ErrorMSG))
|
|
misperrors['error'] = ErrorMSG
|
|
return misperrors
|
|
else:
|
|
ErrorMSG = "Value for Plugin.Enrichment_apiosintds_cache_directory is empty but cache option is enabled as recommended. Please set a writable cache directory in plugin settings."
|
|
log.debug(str(ErrorMSG))
|
|
misperrors['error'] = ErrorMSG
|
|
return misperrors
|
|
else:
|
|
log.debug("Cache option is set to " + str(submitcache) + ". You are not using the internal cache system and this is NOT recommended!")
|
|
log.debug("Please, consider to turn on the cache setting it to 'Yes' and specifing a writable directory for the cache directory option.")
|
|
try:
|
|
response = apiosintDS.request(entities=tosubmit, cache=submitcache, cachedirectory=submitcache_directory, verbose=True)
|
|
r["results"] += reversed(apiosintParser(response, import_related_hashes))
|
|
except Exception as e:
|
|
log.debug(str(e))
|
|
misperrors['error'] = str(e)
|
|
return r
|
|
|
|
|
|
def apiosintParser(response, import_related_hashes):
|
|
ret = []
|
|
if isinstance(response, dict):
|
|
for key in response:
|
|
for item in response[key]["items"]:
|
|
if item["response"]:
|
|
comment = item["item"] + " IS listed by OSINT.digitalside.it. Date list: " + response[key]["list"]["date"]
|
|
if key == "url":
|
|
if "hashes" in item.keys():
|
|
if "sha256" in item["hashes"].keys():
|
|
ret.append({"types": ["sha256"], "values": [item["hashes"]["sha256"]]})
|
|
if "sha1" in item["hashes"].keys():
|
|
ret.append({"types": ["sha1"], "values": [item["hashes"]["sha1"]]})
|
|
if "md5" in item["hashes"].keys():
|
|
ret.append({"types": ["md5"], "values": [item["hashes"]["md5"]]})
|
|
|
|
if len(item["related_urls"]) > 0:
|
|
for urls in item["related_urls"]:
|
|
if isinstance(urls, dict):
|
|
itemToInclude = urls["url"]
|
|
if import_related_hashes:
|
|
if "hashes" in urls.keys():
|
|
if "sha256" in urls["hashes"].keys():
|
|
ret.append({"types": ["sha256"], "values": [urls["hashes"]["sha256"]], "comment": "Related to: " + itemToInclude})
|
|
if "sha1" in urls["hashes"].keys():
|
|
ret.append({"types": ["sha1"], "values": [urls["hashes"]["sha1"]], "comment": "Related to: " + itemToInclude})
|
|
if "md5" in urls["hashes"].keys():
|
|
ret.append({"types": ["md5"], "values": [urls["hashes"]["md5"]], "comment": "Related to: " + itemToInclude})
|
|
ret.append({"types": ["url"], "values": [itemToInclude], "comment": "Related to: " + item["item"]})
|
|
else:
|
|
ret.append({"types": ["url"], "values": [urls], "comment": "Related URL to: " + item["item"]})
|
|
else:
|
|
comment = item["item"] + " IS NOT listed by OSINT.digitalside.it. Date list: " + response[key]["list"]["date"]
|
|
ret.append({"types": ["text"], "values": [comment]})
|
|
return ret
|
|
|
|
|
|
def introspection():
|
|
return mispattributes
|
|
|
|
|
|
def version():
|
|
moduleinfo['config'] = moduleconfig
|
|
return moduleinfo
|