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.
76 lines
2.3 KiB
76 lines
2.3 KiB
# -*- coding: utf-8 -*- |
|
|
|
import json |
|
from pyeupi import PyEUPI |
|
|
|
misperrors = {'error': 'Error'} |
|
mispattributes = {'input': ['hostname', 'domain', 'url'], 'output': ['freetext']} |
|
moduleinfo = {'version': '0.1', 'author': 'Raphaël Vinot', |
|
'description': 'Query the Phishing Initiative service (https://phishing-initiative.lu)', |
|
'module-type': ['expansion', 'hover']} |
|
|
|
moduleconfig = ['apikey', 'url'] |
|
|
|
|
|
def handler(q=False): |
|
if q is False: |
|
return False |
|
request = json.loads(q) |
|
if request.get('hostname'): |
|
toquery = request['hostname'] |
|
elif request.get('domain'): |
|
toquery = request['domain'] |
|
elif request.get('url'): |
|
toquery = request['url'] |
|
else: |
|
misperrors['error'] = "Unsupported attributes type" |
|
return misperrors |
|
|
|
if not request.get('config') and not (request['config'].get('apikey') and request['config'].get('url')): |
|
misperrors['error'] = 'EUPI authentication is missing' |
|
return misperrors |
|
|
|
pyeupi = PyEUPI(request['config']['apikey'], request['config']['url']) |
|
|
|
if 'event_id' in request: |
|
return handle_expansion(pyeupi, toquery) |
|
else: |
|
return handle_hover(pyeupi, toquery) |
|
|
|
|
|
def handle_expansion(pyeupi, url): |
|
results = pyeupi.search_url(url=url) |
|
|
|
if results.get('results'): |
|
to_return = '' |
|
for r in results['results']: |
|
if r['tag_label'] != 'phishing': |
|
continue |
|
to_return += ' {} {} {} '.format(r['url'], r['domain'], r['ip_address']) |
|
if to_return: |
|
return {'results': [{'types': mispattributes['output'], 'values': to_return}]} |
|
else: |
|
misperrors['error'] = 'Unknown in the EUPI service' |
|
return misperrors |
|
else: |
|
return {'results': [{'types': mispattributes['output'], 'values': ''}]} |
|
|
|
|
|
def handle_hover(pyeupi, url): |
|
try: |
|
result = pyeupi.lookup(url=url)['results'][0] |
|
except (KeyError, IndexError): |
|
misperrors['error'] = 'Error in EUPI lookup' |
|
return misperrors |
|
|
|
return {'results': [{'types': mispattributes['output'], |
|
'values': result['tag_label'].title()}]} |
|
|
|
|
|
def introspection(): |
|
return mispattributes |
|
|
|
|
|
def version(): |
|
moduleinfo['config'] = moduleconfig |
|
return moduleinfo
|
|
|