From dd3195eff87caf284a8695898bf101a03ed8f26d Mon Sep 17 00:00:00 2001 From: Alexandre Dulaunoy Date: Mon, 29 Feb 2016 21:44:50 +0100 Subject: [PATCH] Handling of error added --- modules/expansion/dns.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/modules/expansion/dns.py b/modules/expansion/dns.py index 92befc8..a07a9bf 100755 --- a/modules/expansion/dns.py +++ b/modules/expansion/dns.py @@ -1,6 +1,7 @@ import json import dns.resolver +misperrors = {'error' : 'Error'} mispattributes = {'input': ['hostname', 'domain'], 'output': ['ip-src', 'ip-dst']} moduleinfo = "0.1" @@ -16,13 +17,20 @@ def handler(q=False): else: return False r = dns.resolver.Resolver() - r.nameservers = ['8.8.8.8'] + r.timeout = 2 + r.lifetime = 2 + r.nameservers = ['8.1.8.7'] try: answer = r.query(toquery, 'A') except dns.resolver.NXDOMAIN: - return False + misperrors['error'] = "NXDOMAIN" + return misperrors except dns.exception.Timeout: - return False + misperrors['error'] = "Timeout" + return misperrors + except: + misperrors['error'] = "DNS resolving error" + return misperrors r = {'results': [{'types': mispattributes['output'], 'values':[str(answer[0])]}]} return r