在 Python 中捕获 Scapy 函数输出

Question

我正在尝试将 scapy 函数（traceroute）的输出捕获到 python 脚本中的字符串。我知道我需要将此函数通过管道传输到 stdout （就像使用 subproces.call() 所做的那样，但不确定如何使用 scapy 执行此操作，有人能够提供任何帮助吗？我是 Python 新手。

下面是相关代码。

#!/usr/bin/env python
from scapy.all import traceroute

traceroute('www.google.com')

Answer 1

您还可以像这样调用traceroute：

trace, _ = traceroute("www.example.org", verbose=0)
# trace.get_trace() returns a rather impractical format, so we need
# to convert it. First, we only want the first trace available
hosts = trace.get_trace().values()[0]

# hosts will be in the format { 1: ("1.2.3.4",     False), 
#                               2: ("10.20.30.40", False) ... }
# We convert it to ["1.2.3.4", "10.20.30.40", ...] here:
ips = [hosts[i][0] for i in range(1, len(hosts) + 1)]

之后ips 变量将包含作为跟踪一部分的主机列表。

Answer 2

您可以通过使用类似文件的对象（例如 StringIO）修补 sys.stdout 来做到这一点：

#!/usr/bin/env python                              
from scapy.all import traceroute
from StringIO import StringIO
import sys

stdout = StringIO()
sys.stdout = stdout
result, unanswered = traceroute('www.google.com')
sys.stdout = sys.__stdout__

print 'Captured stdout:', stdout.getvalue()

无论如何，请注意，您需要的信息可能已经在traceroute 方法返回的对象：

print result.summary()                                      
print unanswered.summary()

注意：您可以在这个 问题。