farrokhi/dnsdiag
1
0
Fork 0
Code Issues Pull Requests Releases Activity

add "-n" option to disable reverse lookups

Browse Source
This commit is contained in:
Babak Farrokhi 2016-04-13 20:50:23 +04:30
parent d90cc4a8d5
commit 09a67904b1
Signed by: farrokhi
GPG Key ID: 6B267AD85D632E9A
1 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
dnstraceroute.py
View File

@ -120,10 +120,11 @@ def main():
dest_port = 53 dest_port = 53
hops = 0 hops = 0
as_lookup = False as_lookup = False
should_resolve = True
try: try:
opts, args = getopt.getopt(sys.argv[1:], "aqhc:s:t:w:p:", opts, args = getopt.getopt(sys.argv[1:], "aqhc:s:t:w:p:n",
["help", "output=", "count=", "server=", "quiet", "type=", "wait=", "asn", "port"]) ["help", "count=", "server=", "quiet", "type=", "wait=", "asn", "port"])
except getopt.GetoptError as err: except getopt.GetoptError as err:
# print help information and exit: # print help information and exit:
print(err) # will print something like "option -a not recognized" print(err) # will print something like "option -a not recognized"
@ -149,6 +150,8 @@ def main():
dnsrecord = a dnsrecord = a
elif o in ("-p", "--port"): elif o in ("-p", "--port"):
dest_port = int(a) dest_port = int(a)
elif o in ("-n"):
should_resolve = False
elif o in ("-a", "--asn"): elif o in ("-a", "--asn"):
if has_whois: if has_whois:
as_lookup = True as_lookup = True
@ -168,7 +171,8 @@ def main():
ttl = 1 ttl = 1
reached = False reached = False
print("%s DNS: %s:%d, hostname: %s, rdatatype: %s" % (__PROGNAME__, dnsserver, dest_port, hostname, dnsrecord)) if not quiet:
print("%s DNS: %s:%d, hostname: %s, rdatatype: %s" % (__PROGNAME__, dnsserver, dest_port, hostname, dnsrecord))
while True: while True:
if should_stop: if should_stop:
@ -234,15 +238,18 @@ def main():
elapsed -= timeout * 1000 elapsed -= timeout * 1000
try: if should_resolve:
if curr_addr: try:
curr_name = socket.gethostbyaddr(curr_addr)[0] if curr_addr:
except socket.error: curr_name = socket.gethostbyaddr(curr_addr)[0]
except socket.error:
curr_name = curr_addr
except SystemExit:
pass
except:
print("unxpected error: ", sys.exc_info()[0])
else:
curr_name = curr_addr curr_name = curr_addr
except SystemExit:
pass
except:
print("unxpected error: ", sys.exc_info()[0])
if curr_addr: if curr_addr:
as_name = "" as_name = ""