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

Add support for EDNS0 flag

Browse Source
This commit is contained in:
Babak Farrokhi 2016-06-26 16:54:22 +04:30
parent 40052f008d
commit 26fdd00647
Signed by: farrokhi
GPG Key ID: 6B267AD85D632E9A
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
dnsping.py
View File

@ -91,13 +91,14 @@ def main():
src_port = 0 src_port = 0
src_ip = None src_ip = None
use_tcp = False use_tcp = False
use_edns = False
af = socket.AF_INET af = socket.AF_INET
hostname = 'wikipedia.org' hostname = 'wikipedia.org'
try: try:
opts, args = getopt.getopt(sys.argv[1:], "qhc:s:t:w:vp:P:S:T46", opts, args = getopt.getopt(sys.argv[1:], "qhc:s:t:w:vp:P:S:T46e",
["help", "count=", "server=", "quiet", "type=", "wait=", "verbose", ["help", "count=", "server=", "quiet", "type=", "wait=", "verbose",
"port=", "srcip=", "tcp", "ipv4", "ipv6", "srcport="]) "port=", "srcip=", "tcp", "ipv4", "ipv6", "srcport=", "edns"])
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"
@ -132,6 +133,8 @@ def main():
af = socket.AF_INET af = socket.AF_INET
elif o in ("-6", "--ipv6"): elif o in ("-6", "--ipv6"):
af = socket.AF_INET6 af = socket.AF_INET6
elif o in ("-e", "--edns"):
use_edns = True
elif o in ("-P", "--srcport"): elif o in ("-P", "--srcport"):
src_port = int(a) src_port = int(a)
if src_port < 1024: if src_port < 1024:
@ -158,6 +161,9 @@ def main():
resolver.port = dst_port resolver.port = dst_port
resolver.retry_servfail = 0 resolver.retry_servfail = 0
if use_edns:
resolver.use_edns(edns=0, payload=8192, ednsflags=dns.flags.edns_from_text('DO'))
response_time = [] response_time = []
i = 0 i = 0