python - 如何识别一个字节字符串是否是一个 pickled 对象？

Question

如何识别一个字节字符串对象是否是一个 pickled 对象。

例如，在下面的字典中，我如何识别b是一个pickled对象而其他对象不是？

example = {'a':1, 'b':b'\x80\x03K\x02.', 'c':b'3'}

Answer 1

尽管有一些方法可以判断流是否可能是 pickle 流(例如，第一个字节始终是 \x80，后跟协议(protocol)号，这里 \x03，最后一个字节也总是 b'.')，一般来说很难确定(第一个字节可能是巧合)字节是随机字符串中的 \x80)。所以你最好简单地尝试解开它，看看是否可行:

import pickle

def is_pickle_stream(stream):
    try:
        pickle.loads(stream)
        return True
    except UnpicklingError:
        return False

如果流是 pickle 流，上述方法将返回 True，否则返回 False，如果 loads 则抛出异常抛出异常(内存不足等)。