diff --git a/misp_modules/modules/expansion/ipasn.py b/misp_modules/modules/expansion/ipasn.py index cfdbaf5..ca10a7a 100755 --- a/misp_modules/modules/expansion/ipasn.py +++ b/misp_modules/modules/expansion/ipasn.py @@ -2,22 +2,40 @@ import json from pyipasnhistory import IPASNHistory +from pymisp import MISPAttribute, MISPEvent, MISPObject misperrors = {'error': 'Error'} -mispattributes = {'input': ['ip-src', 'ip-dst'], 'output': ['freetext']} +mispattributes = {'input': ['ip-src', 'ip-dst'], 'format': 'misp_standard'} moduleinfo = {'version': '0.1', 'author': 'Raphaƫl Vinot', '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')} + + 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'] + if request.get('attribute') and request['attribute'].get('type') in mispattributes['input']: + toquery = request['attribute']['value'] else: misperrors['error'] = "Unsupported attributes type" return misperrors @@ -28,7 +46,7 @@ def handler(q=False): if not values: misperrors['error'] = 'Unable to find the history of this IP' return misperrors - return {'results': [{'types': mispattributes['output'], 'values': [str(values)]}]} + return {'results': parse_result(request['attribute'], values)} def introspection():