使用Python 2.6,我在Windows XP中编写了一个脚本.
该脚本执行以下操作:
输入:域名(即:amazon.com)
该脚本通过dnspython模块查询DNS并返回任何A记录IP地址.输出采用特殊应用程序所需的特殊格式,该应用程序利用此数据.
这在Windows中运行良好,但是当我把它放在我的Linux服务器上时,我得到了一些不寻常和不一致的结果.
第一次运行时,它按预期完成.如果我立即再次运行它,脚本将挂起并且什么也不做,没有输出,脚本也不会结束.如果我使用CTRL-C退出进程,它会打印出来!(几乎就像它已被缓冲,但没有写入终端)
我已经尝试了各种技术来解决这个问题,比如在打印后强制sys.stdout.flush()(尽管如此,print应该会自动刷新)并且没有运气.
如果我等待一段时间(几分钟),然后再次运行脚本,它将再次工作(一次),然后后续尝试将继续失败.我不确定发生了什么......还有其他人经历过这样的事吗?
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 ''''''.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(''' {0} {1} '''.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(''' {0} '''.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() {0} Mail Exchange
使用:python myscript.py domain_to_ip amazon.com
这个脚本有2个参数,第一个映射到要运行的函数,第二个指定域.
显然dnspython 在启动时需要16个字节的高质量随机数.获取它们(来自/ dev/random)可以阻止它们.
如果按Ctrl + C,它实际捕获KeyboardInterupt
异常并返回不太安全的随机数(取自当前系统时间).然后你的程序运行完毕.
有问题的代码在这里:http://www.dnspython.com/docs/1.7.1/html/dns.entropy-pysrc.html
我想我会认为这是dnspython中的一个错误.它应该找到一种不阻塞的方法,然后再回到/ dev/urandom.无论如何,它不应该沉默KeyboardInterrupt
.