fix udp ipv6

This commit is contained in:
breakwa11 2015-06-16 21:55:36 +08:00
parent 4b52f83376
commit 17122c299c

View file

@ -351,14 +351,18 @@ class TCPRelayHandler(object):
# TODO use logging when debug completed # TODO use logging when debug completed
self.destroy() self.destroy()
def _is_support_ipv6(self):
local = socket.gethostbyaddr(socket.gethostname())
for ip in local:
if ':' in ip:
return True
return False
def _create_remote_socket(self, ip, port): def _create_remote_socket(self, ip, port):
addrs = None
if self._remote_udp: if self._remote_udp:
try: if self._is_support_ipv6():
addrs = socket.getaddrinfo("::", 0, 0, socket.SOCK_DGRAM, socket.SOL_UDP) addrs = socket.getaddrinfo("::", 0, 0, socket.SOCK_DGRAM, socket.SOL_UDP)
except Exception as e: else:
pass
if addrs is None:
addrs = socket.getaddrinfo("0.0.0.0", 0, 0, socket.SOCK_DGRAM, socket.SOL_UDP) addrs = socket.getaddrinfo("0.0.0.0", 0, 0, socket.SOCK_DGRAM, socket.SOL_UDP)
else: else:
addrs = socket.getaddrinfo(ip, port, 0, socket.SOCK_STREAM, socket.SOL_TCP) addrs = socket.getaddrinfo(ip, port, 0, socket.SOCK_STREAM, socket.SOL_TCP)