2009-11-25 127 views
1

使用Python 2.6,我在Windows XP中编写了一个脚本。Python:打印不起作用,脚本无休止地挂起

的脚本执行以下操作:

输入:域名(例如:amazon.com)

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

这在Windows中正常工作,但是当我把它放在我的Linux服务器上时,我得到了一些不寻常和不一致的结果。

第一次运行时,它按预期完成。如果我立即再次运行该脚本,该脚本将挂起,不执行任何操作,不输出,脚本也不会结束。如果我使用CTRL-C退出该过程,它会打印! (几乎像它已被缓冲,但没有写入终端)

我已经尝试过各种技术来解决这个问题,就像打印后强制sys.stdout.flush()(尽管如此,打印应该会自动刷新)并没有运气。

如果我等了一段时间(几分钟),再次运行脚本,它将再次运行(一次),然后后续尝试将继续失败。我不确定发生了什么......有没有其他人经历过这样的事情?

在Windows和Linux(Ubuntu)上的Python 2.6。

这里是我的脚本:

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() 

用途:蟒蛇myscript.py domain_to_ip amazon.com

2个参数,这个剧本,第一个映射到函数运行,第二个指定该域名。

+0

Jason的回答是正确的,当阅读/ dev/random时脚本挂起。 我评论了/ dev/random读出,以便它总是通过time.time()生成它的编号,并且它工作正常。 – brildum 2009-11-30 14:38:31

回答

5

显然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无声。

+0

哇,这是可能的,但我从来没有像这样枯竭我的熵池。对于OP,如果您想验证这一点,请尝试在Python会话中执行'strace'以查看它是否停留在等待从/ dev/random读取的系统调用中。 – 2009-11-26 07:42:23

+0

在我的印象中,耗尽熵池是Linux上相当普遍的问题。 – 2009-11-28 14:27:07

1

您是否尝试过做

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()