Question

使用Python 2.6，我在Windows XP中编写了一个脚本。

该脚本执行以下操作：

输入：域名（即：amazon.com）

该脚本通过dnspython模块查询DNS并返回任何A记录IP地址。输出是使用此数据的特定应用程序所需的特殊格式。

这在Windows中运行良好，但是当我将它放在我的Linux服务器上时，我会得到一些不寻常且不一致的结果。

第一次运行时，按预期完成。如果我立即再次运行它，脚本将挂起并且什么也不做，没有输出，脚本也不会结束。如果我使用CTRL-C退出进程，它会打印出来！（几乎就像它已被缓冲，但没有写入终端）

我已经尝试了各种技术来解决这个问题，例如在打印后强制执行sys.stdout.flush（）（尽管如此，print应该会自动刷新）并且没有运气。

如果我等待一段时间（几分钟），然后再次运行脚本，它将再次运行（一次），然后后续尝试将继续失败。我不确定发生了什么......还有其他人经历过这样的事吗？

Windows和Linux上的Python 2.6（Ubuntu）。

这是我的剧本：

from dns.resolver import Resolver
from dns.exception import DNSException
from cStringIO import StringIO
import sys

def maltego_transform(entities, messages = ''):    
    print '''<MaltegoMessage>
<MaltegoTransformResponseMessage>
<Entities>
{0}
</Entities>
<UIMessages>
{1}
</UIMessages>
</MaltegoTransformResponseMessage>
</MaltegoMessage>'''.format(entities, messages)

def domain_to_ip(domain):
    resolver = Resolver()
    results = []
    for type in ['A', 'AAAA']:
        try:
            query = resolver.query(domain, type)
        except DNSException:
            query = []
        results += query

    entities = StringIO()
    for answer in results:
        entities.write('''<Entity Type="IPAddress"><Value>{0}</Value></Entity>'''.format(answer))
    maltego_transform(entities.getvalue())

def domain_to_mxdomain(domain):
    resolver = Resolver()
    try:
        query = resolver.query(domain, 'MX')
    except DNSException:
        query = []

    entities = StringIO()
    for answer in query:
        entities.write('''<Entity Type="Domain"><Value>{0}</Value>
<AdditionalFields><Field Name="DomainType" DisplayName="Domain Type">Mail Exchange</Field></AdditionalFields>
</Entity>'''.format(answer.exchange))
    maltego_transform(entities.getvalue())

def main():
    options = {'domain_to_ip' : domain_to_ip,
               'domain_to_mxdomain' : domain_to_mxdomain}
    if len(sys.argv) > 2:
        func = options.get(sys.argv[1], None)
        if func:
            func(sys.argv[2])

if __name__ == '__main__':
    main()

使用：python myscript.py domain_to_ip amazon.com

此脚本有2个参数，第一个映射到要运行的函数，第二个指定域。

Answer 1

显然dnspython在启动时需要16个字节的high-quality random numbers。获取它们（来自/ dev / random）可以阻止。

如果按Ctrl + C，它实际上会捕获KeyboardInterupt异常并返回不太安全的随机数（取自当前系统时间）。然后你的程序就完成了。

有问题的代码在这里：http://www.dnspython.com/docs/1.7.1/html/dns.entropy-pysrc.html

我想我会认为这是dnspython中的一个错误。它应该找到一种不阻塞的方法，然后再回到/ dev / urandom。无论如何，它不应该使KeyboardInterrupt沉默。

Answer 2

你试过吗

entities = StringIO()
for answer in results:
    entities.write('''<Entity Type="IPAddress"><Value>{0}</Value></Entity>'''.format(answer))
entities.flush()
maltego_transform(entities.getvalue())
entities.close()

Python：打印不起作用，脚本无休止地挂起

2 个答案: