可靠地在Python中获取IPV6地址

时间:2013-04-29 10:56:38

标签: python sockets ipv6

我现在这样做:

def get_inet_ip():
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    s.connect(('mysite.com', 80))
    return s.getsockname()[0]

这是基于: Finding local IP addresses using Python's stdlib

然而,这看起来有点可疑。据我所知,它打开一个到mysite.com:80的套接字,然后返回该套接字的第一个地址,假设它是一个IPv4地址。这看起来有点狡猾...我不认为我们可以保证这是事实。

这是我的第一个问题,是否安全?在启用IPv6的服务器上,是否可能意外返回IPv6地址?

我的第二个问题是,我如何以类似的方式获取IPv6地址。我将修改函数以采用可选的ipv6参数。

2 个答案:

答案 0 :(得分:7)

问题是,你只是想连接,还是你真的想要地址?

如果您只想连接,可以

s = socket.create_connection(('mysite.com', 80))

并建立连接。

但是,如果您对该地址感兴趣,可以采取以下方式之一:

def get_ip_6(host, port=0):
    import socket
    # search only for the wanted v6 addresses
    result = socket.getaddrinfo(host, port, socket.AF_INET6)
    return result # or:
    return result[0][4][0] # just returns the first answer and only the address

或者,更接近another, already presented solution

def get_ip_6(host, port=0):
     # search for all addresses, but take only the v6 ones
     alladdr = socket.getaddrinfo(host,port)
     ip6 = filter(
         lambda x: x[0] == socket.AF_INET6, # means its ip6
         alladdr
     )
     # if you want just the sockaddr
     # return map(lambda x:x[4],ip6)
     return list(ip6)[0][4][0]

答案 1 :(得分:2)

您应该使用函数socket.getaddrinfo()

获取IPv6的示例代码

def get_ip_6(host,port=80):
    # discard the (family, socktype, proto, canonname) part of the tuple
    # and make sure the ips are unique
    alladdr = list(
        set(
            map(
                lambda x: x[4],
                socket.getaddrinfo(host,port)
            )
        )
    )
    ip6 = filter(
        lambda x: ':' in x[0], # means its ip6
        alladdr
    )
    return ip6