mirror of https://github.com/MISP/misp-modules
Modules for expansion services, import and export in MISP
http://misp.github.io/misp-modules
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
2.0 KiB
57 lines
2.0 KiB
# -*- coding: utf-8 -*- |
|
|
|
import json |
|
from pyipasnhistory import IPASNHistory |
|
from pymisp import MISPAttribute, MISPEvent, MISPObject |
|
|
|
misperrors = {'error': 'Error'} |
|
mispattributes = {'input': ['ip-src', 'ip-dst'], 'format': 'misp_standard'} |
|
moduleinfo = {'version': '0.2', '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('attribute') and request['attribute'].get('type') in mispattributes['input']: |
|
toquery = request['attribute']['value'] |
|
else: |
|
misperrors['error'] = "Unsupported attributes type" |
|
return misperrors |
|
|
|
ipasn = IPASNHistory() |
|
values = ipasn.query(toquery) |
|
|
|
if not values: |
|
misperrors['error'] = 'Unable to find the history of this IP' |
|
return misperrors |
|
return {'results': parse_result(request['attribute'], values)} |
|
|
|
|
|
def introspection(): |
|
return mispattributes |
|
|
|
|
|
def version(): |
|
return moduleinfo
|
|
|