mirror of https://github.com/MISP/misp-modules
58 lines
2.5 KiB
Python
58 lines
2.5 KiB
Python
import json
|
|
import requests
|
|
from . import check_input_attribute, checking_error, standard_error_message
|
|
from pymisp import MISPEvent, MISPObject
|
|
|
|
mispattributes = {'input': ['md5', 'sha1', 'sha256'],
|
|
'format': 'misp_standard'}
|
|
moduleinfo = {'version': '0.1', 'author': 'Christian Studer',
|
|
'description': 'Query Malware Bazaar to get additional information about the input hash.',
|
|
'module-type': ['expansion', 'hover']}
|
|
moduleconfig = []
|
|
|
|
|
|
def parse_response(response):
|
|
mapping = {'file_name': {'type': 'filename', 'object_relation': 'filename'},
|
|
'file_size': {'type': 'size-in-bytes', 'object_relation': 'size-in-bytes'},
|
|
'file_type_mime': {'type': 'mime-type', 'object_relation': 'mimetype'},
|
|
'md5_hash': {'type': 'md5', 'object_relation': 'md5'},
|
|
'sha1_hash': {'type': 'sha1', 'object_relation': 'sha1'},
|
|
'sha256_hash': {'type': 'sha256', 'object_relation': 'sha256'},
|
|
'ssdeep': {'type': 'ssdeep', 'object_relation': 'ssdeep'}}
|
|
misp_event = MISPEvent()
|
|
for data in response:
|
|
misp_object = MISPObject('file')
|
|
for feature, attribute in mapping.items():
|
|
if feature in data:
|
|
misp_attribute = {'value': data[feature]}
|
|
misp_attribute.update(attribute)
|
|
misp_object.add_attribute(**misp_attribute)
|
|
misp_event.add_object(**misp_object)
|
|
return {'results': {'Object': [json.loads(misp_object.to_json()) for misp_object in misp_event.objects]}}
|
|
|
|
|
|
def handler(q=False):
|
|
if q is False:
|
|
return False
|
|
request = json.loads(q)
|
|
if not request.get('attribute') or not check_input_attribute(request['attribute'], requirements=('type', 'value')):
|
|
return {'error': f'{standard_error_message}, {checking_error} that is the hash to submit to Malware Bazaar.'}
|
|
attribute = request['attribute']
|
|
if attribute['type'] not in mispattributes['input']:
|
|
return {'error': 'Unsupported attribute type.'}
|
|
url = 'https://mb-api.abuse.ch/api/v1/'
|
|
response = requests.post(url, data={'query': 'get_info', 'hash': attribute['value']}).json()
|
|
query_status = response['query_status']
|
|
if query_status == 'ok':
|
|
return parse_response(response['data'])
|
|
return {'error': 'Hash not found on MALWAREbazzar' if query_status == 'hash_not_found' else f'Problem encountered during the query: {query_status}'}
|
|
|
|
|
|
def introspection():
|
|
return mispattributes
|
|
|
|
|
|
def version():
|
|
moduleinfo['config'] = moduleconfig
|
|
return moduleinfo
|