MISP
/
misp-modules
mirror of https://github.com/MISP/misp-modules
2
0
Fork 0
Code Issues Releases Wiki Activity

Handling of error added

pull/2/head
Alexandre Dulaunoy 2016-02-29 21:44:50 +01:00
parent 6c0e905737
commit dd3195eff8
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
modules/expansion/dns.py
View File

@ -1,6 +1,7 @@
import json import json
import dns.resolver import dns.resolver
misperrors = {'error' : 'Error'}
mispattributes = {'input': ['hostname', 'domain'], 'output': ['ip-src', 'ip-dst']} mispattributes = {'input': ['hostname', 'domain'], 'output': ['ip-src', 'ip-dst']}
moduleinfo = "0.1" moduleinfo = "0.1"
@ -16,13 +17,20 @@ def handler(q=False):
else: else:
return False return False
r = dns.resolver.Resolver() r = dns.resolver.Resolver()
r.nameservers = ['8.8.8.8'] r.timeout = 2
r.lifetime = 2
r.nameservers = ['8.1.8.7']
try: try:
answer = r.query(toquery, 'A') answer = r.query(toquery, 'A')
except dns.resolver.NXDOMAIN: except dns.resolver.NXDOMAIN:
return False misperrors['error'] = "NXDOMAIN"
return misperrors
except dns.exception.Timeout: 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])]}]} r = {'results': [{'types': mispattributes['output'], 'values':[str(answer[0])]}]}
return r return r