misp-modules/misp_modules/modules/expansion/ipasn.py

58 lines
2.0 KiB
Python
Raw Normal View History

2016-05-01 12:09:33 +02:00
# -*- coding: utf-8 -*-
import json
from pyipasnhistory import IPASNHistory
from pymisp import MISPAttribute, MISPEvent, MISPObject
2016-05-01 12:09:33 +02:00
misperrors = {'error': 'Error'}
mispattributes = {'input': ['ip-src', 'ip-dst'], 'format': 'misp_standard'}
moduleinfo = {'version': '0.2', 'author': 'Raphaël Vinot',
2016-05-01 12:09:33 +02:00
'description': 'Query an IP ASN history service (https://github.com/CIRCL/IP-ASN-history.git)',
'module-type': ['expansion', 'hover']}
def parse_result(attribute, values):
event = MISPEvent()
initial_attribute = MISPAttribute()
initial_attribute.from_dict(**attribute)
event.add_attribute(**initial_attribute)
mapping = {'asn': ('AS', 'asn'), 'prefix': ('ip-src', 'subnet-announced')}
print(values)
for last_seen, response in values['response'].items():
asn = MISPObject('asn')
asn.add_attribute('last-seen', **{'type': 'datetime', 'value': last_seen})
for feature, attribute_fields in mapping.items():
attribute_type, object_relation = attribute_fields
asn.add_attribute(object_relation, **{'type': attribute_type, 'value': response[feature]})
asn.add_reference(initial_attribute.uuid, 'related-to')
event.add_object(**asn)
event = json.loads(event.to_json())
return {key: event[key] for key in ('Attribute', 'Object')}
2016-05-01 12:09:33 +02:00
def handler(q=False):
if q is False:
return False
request = json.loads(q)
if request.get('attribute') and request['attribute'].get('type') in mispattributes['input']:
toquery = request['attribute']['value']
2016-05-01 12:09:33 +02:00
else:
misperrors['error'] = "Unsupported attributes type"
return misperrors
ipasn = IPASNHistory()
values = ipasn.query(toquery)
2016-05-04 12:52:01 +02:00
2016-05-01 12:09:33 +02:00
if not values:
misperrors['error'] = 'Unable to find the history of this IP'
return misperrors
return {'results': parse_result(request['attribute'], values)}
2016-05-01 12:09:33 +02:00
def introspection():
return mispattributes
def version():
return moduleinfo