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

beautify code

Browse Source
This commit is contained in:
Babak Farrokhi 2016-04-08 12:32:50 +04:30
parent 11badcd9c3
commit 0b068b849b
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
dnstraceroute.py
View File

@ -27,8 +27,8 @@
import getopt import getopt
import signal import signal
import sys
import socket import socket
import sys
import time import time
import dns.rdatatype import dns.rdatatype
@ -127,9 +127,9 @@ def main():
icmp_socket.bind(("", dnsport)) icmp_socket.bind(("", dnsport))
icmp_socket.settimeout(timeout) icmp_socket.settimeout(timeout)
try: # send DNS request try: # send DNS request
stime = time.time() stime = time.time()
answers = resolver.query(hostname, dnsrecord, ipttl = ttl) answers = resolver.query(hostname, dnsrecord, ipttl=ttl)
except dns.resolver.NoNameservers as e: except dns.resolver.NoNameservers as e:
if not quiet: if not quiet:
print("no or bad response:", e) print("no or bad response:", e)
@ -153,7 +153,7 @@ def main():
curr_addr = None curr_addr = None
curr_host = None curr_host = None
try: # expect ICMP response try: # expect ICMP response
_, curr_addr = icmp_socket.recvfrom(512) _, curr_addr = icmp_socket.recvfrom(512)
curr_addr = curr_addr[0] curr_addr = curr_addr[0]
@ -182,12 +182,13 @@ def main():
if curr_addr: if curr_addr:
print("%d\t%s (%s) %d ms" % (ttl, curr_name, curr_addr, elapsed)) print("%d\t%s (%s) %d ms" % (ttl, curr_name, curr_addr, elapsed))
else: else:
print("%d\t *" % ttl) print("%d\t *" % ttl)
ttl += 1 ttl += 1
hops += 1 hops += 1
if (hops >= count) or (curr_addr == dnsserver) or reached: if (hops >= count) or (curr_addr == dnsserver) or reached:
break break
if __name__ == '__main__': if __name__ == '__main__':
main() main()