correctly resolve dns server hostname when layer 3 protocol specified (close #9)

This commit is contained in:
Babak Farrokhi 2016-05-09 13:56:35 +04:30
parent 0a5b5f2380
commit 988c439599
Signed by: farrokhi
GPG Key ID: 6B267AD85D632E9A

View File

@ -145,7 +145,7 @@ def main():
ipaddress.ip_address(dnsserver)
except ValueError: # so it is not a valid IPv4 or IPv6 address, so try to resolve host name
try:
dnsserver = socket.getaddrinfo(dnsserver, port=None)[1][4][0]
dnsserver = socket.getaddrinfo(dnsserver, port=None, family=af)[1][4][0]
except OSError:
print('Error: cannot resolve hostname:', dnsserver)
sys.exit(1)