From 7da6d4102035f3fc0d75e922155ce3ee78f03601 Mon Sep 17 00:00:00 2001 From: chrisr3d Date: Mon, 19 Nov 2018 09:33:49 +0100 Subject: [PATCH] add: Update to include domain & hostname composite attributes --- misp_modules/modules/expansion/eupi.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/misp_modules/modules/expansion/eupi.py b/misp_modules/modules/expansion/eupi.py index e230bcf..561efb2 100755 --- a/misp_modules/modules/expansion/eupi.py +++ b/misp_modules/modules/expansion/eupi.py @@ -4,7 +4,7 @@ import json from pyeupi import PyEUPI misperrors = {'error': 'Error'} -mispattributes = {'input': ['hostname', 'domain', 'url'], 'output': ['freetext']} +mispattributes = {'input': ['hostname', 'hostname|port', 'domain', 'domain|ip', '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']} @@ -18,8 +18,12 @@ def handler(q=False): request = json.loads(q) if request.get('hostname'): toquery = request['hostname'] + elif request.get('hostname|port'): + toquery, _ = request['hostname|port'].split('|') elif request.get('domain'): toquery = request['domain'] + elif request.get('domain|ip'): + toquery, _ = request['domain|ip'].split('|') elif request.get('url'): toquery = request['url'] else: