繁体   English   中英

导入模块时禁止scapy警告消息

[英]suppress scapy warning message when importing the module

我正在编写一个小脚本,使用scapy收集一些信息,然后返回一些xml代码,我将传递给metasploit的xmlrpc接口。 我希望我的脚本只返回xml,没有其他警告等。

我可以通过向我的sr1命令添加选项verbose=0来抑制大多数scapy输出。 我在每个输出之前仍然得到的,并且我假设它在我加载模块时返回此警告,是:

警告:找不到IPv6目标的路由::(没有默认路由?)

通过调用我的脚本,我可以轻松地重定向该输出:

 ./myscript 2> /dev/null

但我想将其纳入脚本中。 为此,我找到了一个提示,一个可以有一个NullDevice类,它不会写任何东西,然后将sys.stderr设置为该NullDevice类的实例化。

这只在我已经加载模块后不幸地工作,所以我仍然有警告,它只重定向发送到stderr的任何后续消息。

如何禁止显示在屏幕上的警告消息?

您可以通过添加以下内容来消除scapy的警告:

logging.getLogger("scapy.runtime").setLevel(logging.ERROR)

导入Scapy 之前 这将禁止所有具有较低严重性的消息而不是错误消息。


例如:

import logging
logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
from scapy.all import *
...

我认为这是正确的方法。

>>> import sys
>>> sys.stderr = None            # suppress stderr
>>> from scapy.all import *
>>> sys.stderr = sys.__stderr__  # restore stderr
>>> print("other errors can be shown", file=sys.stderr)
other errors can be shown
>>> 

我想scapy的python3版本可能会打印来自不同记录器或高级别的消息。 这是我用来抑制模块导入输出的一些代码。

from contextlib import contextmanager

# It looks like redirect_stderr will be part of Python 3.5 as follows:
# from contextlib import redirect_stderr
# Perhaps if you're looking at this code and 3.5 is out, this function could be
# removed.
@contextmanager
def redirect_stderr(new_target):
    """
    A context manager to temporarily redirect stderr. Example use:
    with open(os.devnull, 'w') as f:
        with redirect_stderr(f):
            # stderr redirected to os.devnull. No annoying import messages
            # printed on module import
            from scapy.all import *
    # stderr restored
    """
    import sys
    old_target, sys.stderr = sys.stderr, new_target # replace sys.stdout
    try:
        yield new_target # run some code with the replaced stdout
    finally:
        sys.stderr = old_target # restore to the previous value


# Don't print the annoying warning message that occurs on import
with open(os.devnull, 'w') as errf:
    with redirect_stderr(errf):
        from scapy.all import sr, ICMP, IP, traceroute

使用Python3, 将sys.stder重新定义为None会引发异常AttributeError:'NoneType'对象没有属性'write' 相反,将它定义为os.devnull完成这项工作:

import os
import sys
sys.stderr = os.devnull # suppress stderr
from scapy.all import *
sys.stderr = sys.__stderr__ # restore stderr

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM