From 8bcb630340055a34e2b4355cc8b2d796a8e76718 Mon Sep 17 00:00:00 2001 From: chrisr3d Date: Tue, 8 Oct 2019 15:48:26 +0200 Subject: [PATCH] fix: Catching results exceptions properly --- misp_modules/modules/expansion/dbl_spamhaus.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/misp_modules/modules/expansion/dbl_spamhaus.py b/misp_modules/modules/expansion/dbl_spamhaus.py index 529815b..0cccfaf 100644 --- a/misp_modules/modules/expansion/dbl_spamhaus.py +++ b/misp_modules/modules/expansion/dbl_spamhaus.py @@ -49,8 +49,10 @@ def handler(q=False): try: query_result = resolver.query(query, 'A')[0] result = "{} - {}".format(requested_value, dbl_mapping[str(query_result)]) - except Exception as e: - result = str(e) + except dns.resolver.NXDOMAIN as e: + result = e.msg + except Exception: + return {'error': 'Not able to reach dbl.spamhaus.org or something went wrong'} return {'results': [{'types': mispattributes.get('output'), 'values': result}]}