twisted.names.client可以强制返回ipv4地址吗？ （不修改resolv.conf或ETH设备）

Question

我遇到的问题是因为在我的以太网接口上启用了IPv6 - 所以当我不想要它们时，主机名解析有时会产生IPv6结果。

是否可以将twisted.names.client变为仅返回ipv4地址而不对操作系统/名称解析/以太网配置进行任何更改？

示例代码：

#!/usr/bin/python

from twisted.internet import defer, reactor
from twisted.names import client

def got_arg(*args):
    print 'got_arg',args

def get_arg(arg):
    d = client.getHostByName('www.google.com').addCallback(got_arg)

if __name__ == '__main__':
    get_arg('www.google.com')
    reactor.run()

此代码始终在我的Linux机器上为我提供了一个ipv6地址：

got_arg ('2001:4860:4001:800::1010',)

我希望能够强制执行ipv4查找，而无需修改下面的任何内容。有任何想法吗？由于缺乏强制，至少有一个dict或元组与ipv4 / ipv6分解的结果会很好。

Answer 1

我想出来了..你必须使用client.lookupAllRecords然后测试返回的记录类型..

#!/usr/bin/python

from twisted.internet import defer, reactor
from twisted.names import client, dns

def got_arg(*args):
    for a in args[0][0]:
        if a.payload.TYPE == dns.A:
            print 'A    - ipv4',a.payload
        elif a.payload.TYPE == dns.AAAA:
            print 'AAAA - ipv6',a.payload


def get_arg(arg):
    d = client.lookupAllRecords('www.google.com').addCallback(got_arg)

if __name__ == '__main__':
    get_arg('www.google.com')
    reactor.run()

修改

我意识到这个解决方案是不够的 - 特别是在第一批记录只包含指向另一个地址的单个CNAME的情况下..

原因是twisted.names.common.extractRecord（）首先查找ipv6地址并立即返回它们 - 没有明显的机制来覆盖此行为。

所以，我为此设置了一个hacky解决方案，以便我们甚至不会尝试通过制作一个丑陋的猴子补丁来解决ipv6地址链。

#!/usr/bin/python

import socket
from twisted.names import dns
from twisted.names import common

def myExtractRecord(resolver, name, answers, level=10):
    if not level:
        return None
    for r in answers:
        if r.name == name and r.type == dns.A:
            return socket.inet_ntop(socket.AF_INET, r.payload.address)
    for r in answers:
        if r.name == name and r.type == dns.CNAME:
            result = myExtractRecord(
                resolver, r.payload.name, answers, level - 1)
            if not result:
                return resolver.getHostByName(
                    str(r.payload.name), effort=level - 1)
            return result
    # No answers, but maybe there's a hint at who we should be asking about
    # this
    for r in answers:
        if r.type == dns.NS:
            from twisted.names import client
            r = client.Resolver(servers=[(str(r.payload.name), dns.PORT)])
            return r.lookupAddress(str(name)
                ).addCallback(
                    lambda (ans, auth, add):
                        myExtractRecord(r, name, ans + auth + add, level - 1))

common.extractRecord = myExtractRecord

我把它放在dnsclient.py中并在我的主编程中使用它：

from twisted.names import client
import dnsclient

如果有一个更优雅的解决方案，我会全力以赴。我认为如果getHostByName接受了dns.A，dns.AAAA，dns.A＆amp;＆amp;的掩码，这会更友好。 dns.AAAA用于确定哪些记录走在树上。