From e7cb15a0c43dcfbd0132c8a41f69a16114ad07a3 Mon Sep 17 00:00:00 2001 From: Sebdraven Date: Fri, 5 Mar 2021 11:22:53 +0100 Subject: [PATCH] Update yeti.py add ip-dst to enrich --- misp_modules/modules/expansion/yeti.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) diff --git a/misp_modules/modules/expansion/yeti.py b/misp_modules/modules/expansion/yeti.py index 78b7b4b..8c6f68a 100644 --- a/misp_modules/modules/expansion/yeti.py +++ b/misp_modules/modules/expansion/yeti.py @@ -63,9 +63,21 @@ def handler(q=False): if q is False: return False request = json.loads(q) - print(request) - attribute = request['attribute'] - print(attribute) + + if 'url' in request: + yeti_url = request['url'] + if 'apikey' in request: + apikey = request['apikey'] + if apikey and yeti_url: + yeti_client = Yeti(yeti_url,apikey) + if request.get('ip-dst'): + obs_value = request['ip-dst'] + + if yeti_client: + obs=yeti_client.search(obs_value) + print(obs) + else: + misperrors['error'] = 'Yeti Config Error' def version():