Current File : //snap/core/current/usr/lib/python3.5/__pycache__/pickletools.cpython-35.pyc
|�gqf � O @ s d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d d g Z e j Z d Z d Z d Z
d Z d
Z Gd d � d e
� Z d d
l m Z d d � Z e d d d d d e d d � Z d d � Z e d d d d d e d d � Z d d � Z e d d d d d e d d � Z d d � Z e d d d d d e d d! � Z d" d# � Z e d d$ d d% d e d d& � Z d' d' d( d) � Z e d d* d e d e d d+ � Z d, d- � Z e d d. d e d e d d/ � Z d0 d1 � Z e d d2 d e d e d d3 � Z! d4 d5 � Z" e d d6 d e d e" d d7 � Z# d8 d9 � Z$ e d d: d e
d e$ d d; � Z% d< d= � Z& e d d> d e d e& d d? � Z' d@ d= � Z& e d d> d e d e& d dA � Z' dB dC � Z( e d dD d e d e( d dE � Z) dF dG � Z* e d dH d e d e* d dI � Z+ dJ dK � Z, e d dL d e d e, d dM � Z- dN dO � Z. e d dP d e d e. d dQ � Z/ dR dS � Z0 e d dT d e d e0 d dU � Z1 dV dW � Z2 e d dX d e d e2 d dY � Z3 dZ d[ � Z4 d\ d] � Z5 e d d^ d e d e4 d d_ � Z6 e d d` d e d e5 d da � Z7 db dc � Z8 e d dd d e d e8 d de � Z9 df dg � Z: e d dh d d% d e: d di � Z; d dj l m< Z< dk dl � Z= e d dm d e d e= d dn � Z> do dp � Z? e d dq d e
d e? d dr � Z@ Gds dt � dt e
� ZA eA d du dv eB d dw � ZC ZD eA d dx dv eB eE f d dy � ZF eA d dz dv eE d d{ � ZG eA d d| dv eH d d} � ZI eA d d~ dv eJ eK f d d � ZL ZM eA d d� dv eJ d d� � ZN eA d d� dv eK d d� � ZO eA d d� dv eP d � d d� � ZQ eA d d� dv eR d d� � ZS eA d d� dv eT d d� � ZU eA d d� dv eV d d� � ZW eA d d� dv eX d d� � ZY eA d d� dv eX d d� � ZZ eA d d� dv e
d d� � Z[ eA d d� dv eA d d� � Z\ eA d d� dv eA d d� � Z] Gd� d� � d� e
� Z^ e^ Z_ e_ d d� d� d� d� e6 d� g d� eF g d� d d d� � e_ d d� d� d� d� e d� g d� eC g d� d d d� � e_ d d� d� d� d� e d� g d� eC g d� d d d� � e_ d d� d� d� d� e d� g d� eC g d� d d d� � e_ d d� d� d� d� e7 d� g d� eC g d� d d d� � e_ d d� d� d� d� e> d� g d� eC g d� d d d� � e_ d d� d� d� d� e@ d� g d� eC g d� d d d� � e_ d d� d� d� d� e d� g d� eL g d� d d d� � e_ d d� d� d� d� e% d� g d� eL g d� d d d� � e_ d d� d� d� d� e# d� g d� eL g d� d d d� � e_ d d� d� d� d� e) d� g d� eN g d� d d d� � e_ d d� d� d� d� e' d� g d� eN g d� d d d� � e_ d d� d� d� d� e+ d� g d� eN g d� d d d� � e_ d d� d� d� d� d d� g d� eQ g d� d d d� � e_ d d� d� d� d� d d� g d� eG g d� d d d� � e_ d d� d� d� d� d d� g d� eG g d� d d d� � e_ d d� d� d� d� e- d� g d� eO g d� d d d� � e_ d d� d� d� d� e/ d� g d� eO g d� d d d� � e_ d d� d� d� d� e1 d� g d� eO g d� d d d� � e_ d d� d� d� d� e3 d� g d� eO g d� d d d� � e_ d d� d� d� d� e9 d� g d� eI g d� d d d� � e_ d d� d� d� d� e; d� g d� eI g d� d d d� � e_ d d� d� d� d� d d� g d� eU g d� d d d� � e_ d d� d� d� d� d d� eU e[ g d� eU g d� d d d� � e_ d d� d� d� d� d d� eU e\ e] g d� eU g d� d d d� � e_ d d� d� d� d� d d� e\ e] g d� eU g d� d d d� � e_ d d� d� d� d� d d� g d� eS g d� d d d� � e_ d d� d� d� d� d d� e\ e] g d� eS g d� d d d� � e_ d d� d� d� d� d d� e[ g d� eS g d� d d d� � e_ d d� d� d� d� d d� e[ e[ g d� eS g d� d d d� � e_ d d� d� d� d� d d� e[ e[ e[ g d� eS g d� d d d� � e_ d d� d� d� d� d d� g d� eW g d� d d d� � e_ d d� d� d� d� d d� e\ e] g d� eW g d� d d d� � e_ d d d� dd� d d� eW e[ e[ g d� eW g d� d d d� e_ d dd� dd� d d� eW e\ e] g d� eW g d� d d d� e_ d dd� dd� d d� g d� eY g d� d d d� e_ d d d� d
d� d d� eY e\ e] g d� eY g d� d d d � e_ d d d� d
d� d d� e\ e] g d� eZ g d� d d d� e_ d dd� dd� d d� e[ g d� g d� d d d� e_ d dd� dd� d d� e[ g d� e[ e[ g d� d d d� e_ d dd� dd� d d� g d� e\ g d� d d d� e_ d dd� dd� d d� e\ e] g d� g d� d d d� e_ d dd� d d� e6 d� g d� e[ g d� d d d � e_ d d d� d d� e d� g d� e[ g d� d d d � e_ d d!d� d"d� e d� g d� e[ g d� d d d#� e_ d d$d� d%d� e6 d� g d� g d� d d d&� e_ d d'd� d(d� e d� g d� g d� d d d)� e_ d d*d� d+d� e d� g d� g d� d d d,� e_ d d-d� d.d� d d� e[ g d� e[ g d� d d d/� e_ d d0d� d1d� e d� g d� e[ g d� d d d2� e_ d d3d� d4d� e d� g d� e[ g d� d d d5� e_ d d6d� d7d� e d� g d� e[ g d� d d d8� e_ d d9d� d:d� e! d� g d� e[ g d� d d d;� e_ d d<d� d=d� d d� eO eO g d� e[ g d� d d d>� e_ d d?d� d@d� d d� e[ e[ g d� e[ g d� d d dA� e_ d dBd� dCd� d d� e[ e[ g d� e[ g d� d d dD� e_ d dEd� dFd� e! d� e\ e] g d� e[ g d� d d dG� e_ d dHd� dId� d d� e\ e[ e] g d� e[ g d� d d dJ� e_ d dKd� dLd� d d� e[ e[ g d� e[ g d� d d dM� e_ d dNd� dOd� d d� e[ e[ e[ g d� e[ g d� d d dP� e_ d dQd� dRd� e d� g d� g d� d d dS� e_ d dTd� dUd� d d� e[ g d� g d� d d dV� e_ d dWd� dXd� e d� g d� g d� d d dY� e_ d dZd� d[d� e d� g d� e[ g d� d d d\� e_ d d]d� d^d� d d� e[ g d� e[ g d� d d d_� gA Z` [_ i Za i Zb x� ec e` � D]� \ Zd Ze ee jf ea k r�eg d`ee jf ea ee jf ed f � � ee jh eb k r�eg daee jh eb ee jh ed f � � ed ea ee jf <ed eb ee jh <q�W[a [b [d [e i Zi x e` D] Ze ee ei ee jh <q-W[e dbdcdd� Zj ej � [j dbdedf� Zk dgd � Zl dhd � Zm d d d d did � Zn Gdjdk� dk� Zo dlZp dmZq dnep doeq i Zr dpdq� Zs et drk r�d d lu Zu eu jv dsdt� Zw ew jx dudveu jy dw� dxdydzd{�ew jx d|d}d~e jz dveu jy d� dzd��ew jx d�d�d�d�dzd��ew jx d�d�d~d dveB dzd��ew jx d�d�d�d�dzd��ew jx d�d�d~d�dzd��ew jx d�d�d�d�dzd��ew jx d�d�d�dzd��ew j{ � Z| e| j} r&es � n� e| j~ r5d�n d Z~ e| j rRew j� � n� e� e| j � d k r�en e| j d e| j� d e| j� e~ � nq e| j� r�i n d Z� xY e| j D]N Z� e| j� j� d e� jf � Z� e| j� j� e� d�� en e� e| j� e� e| j� e~ � q�Wd S(� ar "Executable documentation" for the pickle module.
Extensive comments about the pickle protocols and pickle-machine opcodes
can be found here. Some functions meant for external use:
genops(pickle)
Generate all the opcodes in a pickle, as (opcode, arg, position) triples.
dis(pickle, out=None, memo=None, indentlevel=4)
Print a symbolic disassembly of a pickle.
� N�dis�genops�optimize� � � � � c @ s" e Z d Z d Z d d � Z d S) �ArgumentDescriptor�name�n�reader�docc C s� t | t � s t � | | _ t | t � rT | d k sZ | t t t t t f k sZ t � | | _
| | _ t | t � s� t � | | _ d S)Nr )
�
isinstance�str�AssertionErrorr �int�
UP_TO_NEWLINE�TAKEN_FROM_ARGUMENT1�TAKEN_FROM_ARGUMENT4�TAKEN_FROM_ARGUMENT4U�TAKEN_FROM_ARGUMENT8Ur r
r )�selfr r r
r � r �!/usr/lib/python3.5/pickletools.py�__init__� s zArgumentDescriptor.__init__N)znamer zreaderzdoc)�__name__�
__module__� __qualname__� __slots__r r r r r r
� s
r
)�unpackc C s- | j d � } | r | d St d � � d S)zG
>>> import io
>>> read_uint1(io.BytesIO(b'\xff'))
255
r r z'not enough data in stream to read uint1N)�read�
ValueError)�f�datar r r �
read_uint1� s r% r �uint1r r
r zOne-byte unsigned integer.c C sB | j d � } t | � d k r2 t d | � d St d � � d S)z�
>>> import io
>>> read_uint2(io.BytesIO(b'\xff\x00'))
255
>>> read_uint2(io.BytesIO(b'\xff\xff'))
65535
r z<Hr z'not enough data in stream to read uint2N)r! �len�_unpackr" )r# r$ r r r �
read_uint2� s r) �uint2z)Two-byte unsigned integer, little-endian.c C sB | j d � } t | � d k r2 t d | � d St d � � d S)z�
>>> import io
>>> read_int4(io.BytesIO(b'\xff\x00\x00\x00'))
255
>>> read_int4(io.BytesIO(b'\x00\x00\x00\x80')) == -(2**31)
True
r z<ir z¬ enough data in stream to read int4N)r! r' r( r" )r# r$ r r r � read_int4� s r+ �int4z8Four-byte signed integer, little-endian, 2's complement.c C sB | j d � } t | � d k r2 t d | � d St d � � d S)z�
>>> import io
>>> read_uint4(io.BytesIO(b'\xff\x00\x00\x00'))
255
>>> read_uint4(io.BytesIO(b'\x00\x00\x00\x80')) == 2**31
True
r z<Ir z'not enough data in stream to read uint4N)r! r' r( r" )r# r$ r r r �
read_uint4 s r- �uint4z*Four-byte unsigned integer, little-endian.c C sB | j d � } t | � d k r2 t d | � d St d � � d S)z�
>>> import io
>>> read_uint8(io.BytesIO(b'\xff\x00\x00\x00\x00\x00\x00\x00'))
255
>>> read_uint8(io.BytesIO(b'\xff' * 8)) == 2**64-1
True
� z<Qr z'not enough data in stream to read uint8N)r! r' r( r" )r# r$ r r r �
read_uint8& s r0 �uint8r/ z+Eight-byte unsigned integer, little-endian.Tc C s� | j � } | j d � s' t d � � | d d � } | r� xc d D]K } | j | � rD | j | � s~ t d | | f � � | d d
� } PqD Wt d | � � | r� t j | � d j d
� } | S)au
>>> import io
>>> read_stringnl(io.BytesIO(b"'abcd'\nefg\n"))
'abcd'
>>> read_stringnl(io.BytesIO(b"\n"))
Traceback (most recent call last):
...
ValueError: no string quotes around b''
>>> read_stringnl(io.BytesIO(b"\n"), stripquotes=False)
''
>>> read_stringnl(io.BytesIO(b"''\n"))
''
>>> read_stringnl(io.BytesIO(b'"abcd"'))
Traceback (most recent call last):
...
ValueError: no newline found when trying to read stringnl
Embedded escapes are undone in the result.
>>> read_stringnl(io.BytesIO(br"'a\n\\b\x00c\td'" + b"\n'e'"))
'a\n\\b\x00c\td'
s
z-no newline found when trying to read stringnlNr � "� 'z,strinq quote %r not found at both ends of %rzno string quotes around %rr �ascii���)r2 r3 r5 )�readline�endswithr" �
startswith�codecs�
escape_decode�decode)r# r; � stripquotesr$ �qr r r �
read_stringnl; s
r> �stringnlz�A newline-terminated string.
This is a repr-style string, with embedded escapes, and
bracketing quotes.
c C s t | d d �S)Nr< F)r> )r# r r r �read_stringnl_noescapet s r@ �stringnl_noescapea A newline-terminated string.
This is a str-style string, without embedded escapes,
or bracketing quotes. It should consist solely of
printable ASCII characters.
c C s d t | � t | � f S)zp
>>> import io
>>> read_stringnl_noescape_pair(io.BytesIO(b"Queue\nEmpty\njunk"))
'Queue Empty'
z%s %s)r@ )r# r r r �read_stringnl_noescape_pair� s rB �stringnl_noescape_paira� A pair of newline-terminated strings.
These are str-style strings, without embedded
escapes, or bracketing quotes. They should
consist solely of printable ASCII characters.
The pair is returned as a single string, with
a single blank separating the two strings.
c C sl t | � } | d k s t � | j | � } t | � | k rL | j d � St d | t | � f � � d S)z�
>>> import io
>>> read_string1(io.BytesIO(b"\x00"))
''
>>> read_string1(io.BytesIO(b"\x03abcdef"))
'abc'
r zlatin-1z2expected %d bytes in a string1, but only %d remainN)r% r r! r' r; r" )r# r r$ r r r � read_string1� s
rD �string1z�A counted string.
The first argument is a 1-byte unsigned int giving the number
of bytes in the string, and the second argument is that many
bytes.
c C sv t | � } | d k r( t d | � � | j | � } t | � | k rV | j d � St d | t | � f � � d S)aP
>>> import io
>>> read_string4(io.BytesIO(b"\x00\x00\x00\x00abc"))
''
>>> read_string4(io.BytesIO(b"\x03\x00\x00\x00abcdef"))
'abc'
>>> read_string4(io.BytesIO(b"\x00\x00\x00\x03abcdef"))
Traceback (most recent call last):
...
ValueError: expected 50331648 bytes in a string4, but only 6 remain
r zstring4 byte count < 0: %dzlatin-1z2expected %d bytes in a string4, but only %d remainN)r+ r" r! r' r; )r# r r$ r r r � read_string4� s
rF �string4z�A counted string.
The first argument is a 4-byte little-endian signed int giving
the number of bytes in the string, and the second argument is
that many bytes.
c C sc t | � } | d k s t � | j | � } t | � | k rC | St d | t | � f � � d S)z�
>>> import io
>>> read_bytes1(io.BytesIO(b"\x00"))
b''
>>> read_bytes1(io.BytesIO(b"\x03abcdef"))
b'abc'
r z1expected %d bytes in a bytes1, but only %d remainN)r% r r! r' r" )r# r r$ r r r � read_bytes1� s rH �bytes1z�A counted bytes string.
The first argument is a 1-byte unsigned int giving the number
of bytes in the string, and the second argument is that many
bytes.
c C sc t | � } | d k s t � | j | � } t | � | k rC | St d | t | � f � � d S)z�
>>> import io
>>> read_bytes1(io.BytesIO(b"\x00"))
b''
>>> read_bytes1(io.BytesIO(b"\x03abcdef"))
b'abc'
r z1expected %d bytes in a bytes1, but only %d remainN)r% r r! r' r" )r# r r$ r r r rH � s z�A counted bytes string.
The first argument is a 1-byte unsigned int giving the number
of bytes, and the second argument is that many bytes.
c C s� t | � } | d k s t � | t j k r= t d | � � | j | � } t | � | k rb | St d | t | � f � � d S)aN
>>> import io
>>> read_bytes4(io.BytesIO(b"\x00\x00\x00\x00abc"))
b''
>>> read_bytes4(io.BytesIO(b"\x03\x00\x00\x00abcdef"))
b'abc'
>>> read_bytes4(io.BytesIO(b"\x00\x00\x00\x03abcdef"))
Traceback (most recent call last):
...
ValueError: expected 50331648 bytes in a bytes4, but only 6 remain
r z#bytes4 byte count > sys.maxsize: %dz1expected %d bytes in a bytes4, but only %d remainN)r- r �sys�maxsizer" r! r' )r# r r$ r r r � read_bytes4 s
rL �bytes4z�A counted bytes string.
The first argument is a 4-byte little-endian unsigned int giving
the number of bytes, and the second argument is that many bytes.
c C s� t | � } | d k s t � | t j k r= t d | � � | j | � } t | � | k rb | St d | t | � f � � d S)a�
>>> import io, struct, sys
>>> read_bytes8(io.BytesIO(b"\x00\x00\x00\x00\x00\x00\x00\x00abc"))
b''
>>> read_bytes8(io.BytesIO(b"\x03\x00\x00\x00\x00\x00\x00\x00abcdef"))
b'abc'
>>> bigsize8 = struct.pack("<Q", sys.maxsize//3)
>>> read_bytes8(io.BytesIO(bigsize8 + b"abcdef")) #doctest: +ELLIPSIS
Traceback (most recent call last):
...
ValueError: expected ... bytes in a bytes8, but only 6 remain
r z#bytes8 byte count > sys.maxsize: %dz1expected %d bytes in a bytes8, but only %d remainN)r0 r rJ rK r" r! r' )r# r r$ r r r � read_bytes83 s rN �bytes8z�A counted bytes string.
The first argument is an 8-byte little-endian unsigned int giving
the number of bytes, and the second argument is that many bytes.
c C sD | j � } | j d � s' t d � � | d d � } t | d � S)zm
>>> import io
>>> read_unicodestringnl(io.BytesIO(b"abc\\uabcd\njunk")) == 'abc\uabcd'
True
s
z4no newline found when trying to read unicodestringnlNr zraw-unicode-escaper5 )r6 r7 r" r )r# r$ r r r �read_unicodestringnlU s
rP �unicodestringnlz�A newline-terminated Unicode string.
This is raw-unicode-escape encoded, so consists of
printable ASCII characters, and may contain embedded
escape sequences.
c C so t | � } | d k s t � | j | � } t | � | k rO t | d d � St d | t | � f � � d S)a�
>>> import io
>>> s = 'abcd\uabcd'
>>> enc = s.encode('utf-8')
>>> enc
b'abcd\xea\xaf\x8d'
>>> n = bytes([len(enc)]) # little-endian 1-byte length
>>> t = read_unicodestring1(io.BytesIO(n + enc + b'junk'))
>>> s == t
True
>>> read_unicodestring1(io.BytesIO(n + enc[:-1]))
Traceback (most recent call last):
...
ValueError: expected 7 bytes in a unicodestring1, but only 6 remain
r zutf-8�
surrogatepassz9expected %d bytes in a unicodestring1, but only %d remainN)r% r r! r' r r" )r# r r$ r r r �read_unicodestring1o s rS �unicodestring1aA A counted Unicode string.
The first argument is a 1-byte little-endian signed int
giving the number of bytes in the string, and the second
argument-- the UTF-8 encoding of the Unicode string --
contains that many bytes.
c C s� t | � } | d k s t � | t j k r= t d | � � | j | � } t | � | k rn t | d d � St d | t | � f � � d S)a�
>>> import io
>>> s = 'abcd\uabcd'
>>> enc = s.encode('utf-8')
>>> enc
b'abcd\xea\xaf\x8d'
>>> n = bytes([len(enc), 0, 0, 0]) # little-endian 4-byte length
>>> t = read_unicodestring4(io.BytesIO(n + enc + b'junk'))
>>> s == t
True
>>> read_unicodestring4(io.BytesIO(n + enc[:-1]))
Traceback (most recent call last):
...
ValueError: expected 7 bytes in a unicodestring4, but only 6 remain
r z+unicodestring4 byte count > sys.maxsize: %dzutf-8rR z9expected %d bytes in a unicodestring4, but only %d remainN)r- r rJ rK r" r! r' r )r# r r$ r r r �read_unicodestring4� s rU �unicodestring4aA A counted Unicode string.
The first argument is a 4-byte little-endian signed int
giving the number of bytes in the string, and the second
argument-- the UTF-8 encoding of the Unicode string --
contains that many bytes.
c C s� t | � } | d k s t � | t j k r= t d | � � | j | � } t | � | k rn t | d d � St d | t | � f � � d S)a�
>>> import io
>>> s = 'abcd\uabcd'
>>> enc = s.encode('utf-8')
>>> enc
b'abcd\xea\xaf\x8d'
>>> n = bytes([len(enc)]) + bytes(7) # little-endian 8-byte length
>>> t = read_unicodestring8(io.BytesIO(n + enc + b'junk'))
>>> s == t
True
>>> read_unicodestring8(io.BytesIO(n + enc[:-1]))
Traceback (most recent call last):
...
ValueError: expected 7 bytes in a unicodestring8, but only 6 remain
r z+unicodestring8 byte count > sys.maxsize: %dzutf-8rR z9expected %d bytes in a unicodestring8, but only %d remainN)r0 r rJ rK r" r! r' r )r# r r$ r r r �read_unicodestring8� s rW �unicodestring8aB A counted Unicode string.
The first argument is an 8-byte little-endian signed int
giving the number of bytes in the string, and the second
argument-- the UTF-8 encoding of the Unicode string --
contains that many bytes.
c C sB t | d d d d �} | d k r( d S| d k r8 d St | � S)z�
>>> import io
>>> read_decimalnl_short(io.BytesIO(b"1234\n56"))
1234
>>> read_decimalnl_short(io.BytesIO(b"1234L\n56"))
Traceback (most recent call last):
...
ValueError: invalid literal for int() with base 10: b'1234L'
r; Fr< s 00s 01T)r> r )r# �sr r r �read_decimalnl_short� s rZ c C sH t | d d d d �} | d d � d k r> | d d � } t | � S) z�
>>> import io
>>> read_decimalnl_long(io.BytesIO(b"1234L\n56"))
1234
>>> read_decimalnl_long(io.BytesIO(b"123456789012345678901234L\n6"))
123456789012345678901234
r; Fr< r N� Lr5 r5 )r> r )r# rY r r r �read_decimalnl_long� s r\ �decimalnl_shorta� A newline-terminated decimal integer literal.
This never has a trailing 'L', and the integer fit
in a short Python int on the box where the pickle
was written -- but there's no guarantee it will fit
in a short Python int on the box where the pickle
is read.
�decimalnl_longz�A newline-terminated decimal integer literal.
This has a trailing 'L', and can represent integers
of any size.
c C s" t | d d d d �} t | � S)zO
>>> import io
>>> read_floatnl(io.BytesIO(b"-1.25\n6"))
-1.25
r; Fr< )r> �float)r# rY r r r � read_floatnl' s r` �floatnla� A newline-terminated decimal floating literal.
In general this requires 17 significant digits for roundtrip
identity, and pickling then unpickling infinities, NaNs, and
minus zero doesn't work across boxes, or on some boxes even
on itself (e.g., Windows can't read the strings it produces
for infinities or NaNs).
c C sB | j d � } t | � d k r2 t d | � d St d � � d S)z�
>>> import io, struct
>>> raw = struct.pack(">d", -1.25)
>>> raw
b'\xbf\xf4\x00\x00\x00\x00\x00\x00'
>>> read_float8(io.BytesIO(raw + b"\n"))
-1.25
r/ z>dr z(not enough data in stream to read float8N)r! r' r( r" )r# r$ r r r � read_float8= s
rb �float8a An 8-byte binary representation of a float, big-endian.
The format is unique to Python, and shared with the struct
module (format string '>d') "in theory" (the struct and pickle
implementations don't share the code -- they should). It's
strongly related to the IEEE-754 double format, and, in normal
cases, is in fact identical to the big-endian 754 double format.
On other boxes the dynamic range is limited to that of a 754
double, and "add a half and chop" rounding is used to reduce
the precision to 53 bits. However, even on a 754 box,
infinities, NaNs, and minus zero may not be handled correctly
(may not survive roundtrip pickling intact).
)� decode_longc C sC t | � } | j | � } t | � | k r9 t d � � t | � S)a+
>>> import io
>>> read_long1(io.BytesIO(b"\x00"))
0
>>> read_long1(io.BytesIO(b"\x02\xff\x00"))
255
>>> read_long1(io.BytesIO(b"\x02\xff\x7f"))
32767
>>> read_long1(io.BytesIO(b"\x02\x00\xff"))
-256
>>> read_long1(io.BytesIO(b"\x02\x00\x80"))
-32768
z'not enough data in stream to read long1)r% r! r' r" rd )r# r r$ r r r �
read_long1c s
re �long1a A binary long, little-endian, using 1-byte size.
This first reads one byte as an unsigned size, then reads that
many bytes and interprets them as a little-endian 2's-complement long.
If the size is 0, that's taken as a shortcut for the long 0L.
c C s_ t | � } | d k r( t d | � � | j | � } t | � | k rU t d � � t | � S)ag
>>> import io
>>> read_long4(io.BytesIO(b"\x02\x00\x00\x00\xff\x00"))
255
>>> read_long4(io.BytesIO(b"\x02\x00\x00\x00\xff\x7f"))
32767
>>> read_long4(io.BytesIO(b"\x02\x00\x00\x00\x00\xff"))
-256
>>> read_long4(io.BytesIO(b"\x02\x00\x00\x00\x00\x80"))
-32768
>>> read_long1(io.BytesIO(b"\x00\x00\x00\x00"))
0
r zlong4 byte count < 0: %dz'not enough data in stream to read long4)r+ r" r! r' rd )r# r r$ r r r �
read_long4� s rg �long4a� A binary representation of a long, little-endian.
This first reads four bytes as a signed size (but requires the
size to be >= 0), then reads that many bytes and interprets them
as a little-endian 2's-complement long. If the size is 0, that's taken
as a shortcut for the int 0, although LONG1 should really be used
then instead (and in any case where # of bytes < 256).
c @ s. e Z d Z d Z d d � Z d d � Z d S)
� StackObjectr �obtyper c C s� t | t � s t � | | _ t | t � sB t | t � sB t � t | t � rw x# | D] } t | t � sX t � qX W| | _ t | t � s� t � | | _ d S)N)r r r r �type�tuplerj r )r r rj r Z containedr r r r � s $
zStackObject.__init__c C s | j S)N)r )r r r r �__repr__� s zStackObject.__repr__N)znamezobtypezdoc)r r r r r rm r r r r ri � s
ri r rj zA Python integer object.Z int_or_boolz#A Python integer or boolean object.�boolzA Python boolean object.r_ zA Python float object.Z bytes_or_strz*A Python bytes or (Unicode) string object.�byteszA Python bytes object.r z!A Python (Unicode) string object.�NonezThe Python None object.rl zA Python tuple object.�listzA Python list object.�dictzA Python dict object.�setzA Python set object.� frozensetzA Python frozenset object.�anyz Any kind of object whatsoever.Zmarkaz 'The mark' is a unique object.
Opcodes that operate on a variable number of objects
generally don't embed the count of objects in the opcode,
or pull it off the stack. Instead the MARK opcode is used
to push a special marker object on the stack, and then
some other opcodes grab all the objects from the top of
the stack down to (but not including) the topmost marker
object.
�
stackslicea� An object representing a contiguous slice of the stack.
This is used in conjunction with markobject, to represent all
of the stack following the topmost markobject. For example,
the POP_MARK opcode changes the stack from
[..., markobject, stackslice]
to
[...]
No matter how many object are on the stack after the topmost
markobject, POP_MARK gets rid of all of them (including the
topmost markobject too).
c @ s" e Z d Z d Z d d � Z d
S) �
OpcodeInfor �code�arg� stack_before� stack_after�protor c C se t | t � s t � | | _ t | t � s3 t � t | � d k sK t � | | _ | d k su t | t � su t � | | _ t | t � s� t � x# | D] } t | t � s� t � q� W| | _
t | t � s� t � x# | D] } t | t � s� t � q� W| | _ t | t � r4d | k o/t
j k n s:t � | | _ t | t � sXt � | | _ d S)Nr r )r r r r r' rx r
ry rq ri rz r{ r �pickleZHIGHEST_PROTOCOLr| r ) r r rx ry rz r{ r| r �xr r r r T s& !
4 zOpcodeInfo.__init__N)znamezcodezargz stack_beforez stack_afterzprotozdoc)r r r r r r r r r rw 5 s rw ZINTrx �Iry rz r{ r| a� Push an integer or bool.
The argument is a newline-terminated decimal literal string.
The intent may have been that this always fit in a short Python int,
but INT can be generated in pickles written on a 64-bit box that
require a Python long on a 32-bit box. The difference between this
and LONG then is that INT skips a trailing 'L', and produces a short
int whenever possible.
Another difference is due to that, when bool was introduced as a
distinct type in 2.3, builtin names True and False were also added to
2.2.2, mapping to ints 1 and 0. For compatibility in both directions,
True gets pickled as INT + "I01\n", and False as INT + "I00\n".
Leading zeroes are never produced for a genuine integer. The 2.3
(and later) unpicklers special-case these and return bool instead;
earlier unpicklers ignore the leading "0" and return the int.
ZBININT�Ja1 Push a four-byte signed integer.
This handles the full range of Python (short) integers on a 32-bit
box, directly as binary bytes (1 for the opcode and 4 for the integer).
If the integer is non-negative and fits in 1 or 2 bytes, pickling via
BININT1 or BININT2 saves space.
ZBININT1�Kz�Push a one-byte unsigned integer.
This is a space optimization for pickling very small non-negative ints,
in range(256).
ZBININT2�Mz�Push a two-byte unsigned integer.
This is a space optimization for pickling small positive ints, in
range(256, 2**16). Integers in range(256) can also be pickled via
BININT2, but BININT1 instead saves a byte.
ZLONG�La� Push a long integer.
The same as INT, except that the literal ends with 'L', and always
unpickles to a Python long. There doesn't seem a real purpose to the
trailing 'L'.
Note that LONG takes time quadratic in the number of digits when
unpickling (this is simply due to the nature of decimal->binary
conversion). Proto 2 added linear-time (in C; still quadratic-time
in Python) LONG1 and LONG4 opcodes.
ZLONG1� z|Long integer using one-byte length.
A more efficient encoding of a Python long; the long1 encoding
says it all.ZLONG4� z~Long integer using found-byte length.
A more efficient encoding of a Python long; the long4 encoding
says it all.�STRING�Sa� Push a Python string object.
The argument is a repr-style string, with bracketing quote characters,
and perhaps embedded escapes. The argument extends until the next
newline character. These are usually decoded into a str instance
using the encoding given to the Unpickler constructor. or the default,
'ASCII'. If the encoding given was 'bytes' however, they will be
decoded as bytes object instead.
Z BINSTRING�Ta� Push a Python string object.
There are two arguments: the first is a 4-byte little-endian
signed int giving the number of bytes in the string, and the
second is that many bytes, which are taken literally as the string
content. These are usually decoded into a str instance using the
encoding given to the Unpickler constructor. or the default,
'ASCII'. If the encoding given was 'bytes' however, they will be
decoded as bytes object instead.
ZSHORT_BINSTRING�Ua� Push a Python string object.
There are two arguments: the first is a 1-byte unsigned int giving
the number of bytes in the string, and the second is that many
bytes, which are taken literally as the string content. These are
usually decoded into a str instance using the encoding given to
the Unpickler constructor. or the default, 'ASCII'. If the
encoding given was 'bytes' however, they will be decoded as bytes
object instead.
ZBINBYTES�Bz�Push a Python bytes object.
There are two arguments: the first is a 4-byte little-endian unsigned int
giving the number of bytes, and the second is that many bytes, which are
taken literally as the bytes content.
ZSHORT_BINBYTES�Cz�Push a Python bytes object.
There are two arguments: the first is a 1-byte unsigned int giving
the number of bytes, and the second is that many bytes, which are taken
literally as the string content.
Z BINBYTES8� z�Push a Python bytes object.
There are two arguments: the first is an 8-byte unsigned int giving
the number of bytes in the string, and the second is that many bytes,
which are taken literally as the string content.
ZNONE�NzPush None on the stack.ZNEWTRUE� z&True.
Push True onto the stack.ZNEWFALSE� z'True.
Push False onto the stack.�UNICODE�Vz�Push a Python Unicode string object.
The argument is a raw-unicode-escape encoding of a Unicode string,
and so may contain embedded escape sequences. The argument extends
until the next newline character.
ZSHORT_BINUNICODE� a Push a Python Unicode string object.
There are two arguments: the first is a 1-byte little-endian signed int
giving the number of bytes in the string. The second is that many
bytes, and is the UTF-8 encoding of the Unicode string.
Z
BINUNICODE�Xa Push a Python Unicode string object.
There are two arguments: the first is a 4-byte little-endian unsigned int
giving the number of bytes in the string. The second is that many
bytes, and is the UTF-8 encoding of the Unicode string.
Z BINUNICODE8� a Push a Python Unicode string object.
There are two arguments: the first is an 8-byte little-endian signed int
giving the number of bytes in the string. The second is that many
bytes, and is the UTF-8 encoding of the Unicode string.
ZFLOAT�Fa� Newline-terminated decimal float literal.
The argument is repr(a_float), and in general requires 17 significant
digits for roundtrip conversion to be an identity (this is so for
IEEE-754 double precision values, which is what Python float maps to
on most boxes).
In general, FLOAT cannot be used to transport infinities, NaNs, or
minus zero across boxes (or even on a single box, if the platform C
library can't read the strings it produces for such things -- Windows
is like that), but may do less damage than BINFLOAT on boxes with
greater precision or dynamic range than IEEE-754 double.
ZBINFLOAT�Ga� Float stored in binary form, with 8 bytes of data.
This generally requires less than half the space of FLOAT encoding.
In general, BINFLOAT cannot be used to transport infinities, NaNs, or
minus zero, raises an exception if the exponent exceeds the range of
an IEEE-754 double, and retains no more than 53 bits of precision (if
there are more than that, "add a half and chop" rounding is used to
cut it back to 53 significant bits).
Z
EMPTY_LIST�]zPush an empty list.ZAPPEND�az�Append an object to a list.
Stack before: ... pylist anyobject
Stack after: ... pylist+[anyobject]
although pylist is really extended in-place.
ZAPPENDS�ez�Extend a list by a slice of stack objects.
Stack before: ... pylist markobject stackslice
Stack after: ... pylist+stackslice
although pylist is really extended in-place.
ZLIST�las Build a list out of the topmost stack slice, after markobject.
All the stack entries following the topmost markobject are placed into
a single Python list, which single list object replaces all of the
stack from the topmost markobject onward. For example,
Stack before: ... markobject 1 2 3 'abc'
Stack after: ... [1, 2, 3, 'abc']
Z EMPTY_TUPLE�)zPush an empty tuple.ZTUPLE�tav Build a tuple out of the topmost stack slice, after markobject.
All the stack entries following the topmost markobject are placed into
a single Python tuple, which single tuple object replaces all of the
stack from the topmost markobject onward. For example,
Stack before: ... markobject 1 2 3 'abc'
Stack after: ... (1, 2, 3, 'abc')
ZTUPLE1� z�Build a one-tuple out of the topmost item on the stack.
This code pops one value off the stack and pushes a tuple of
length 1 whose one item is that value back onto it. In other
words:
stack[-1] = tuple(stack[-1:])
ZTUPLE2� a Build a two-tuple out of the top two items on the stack.
This code pops two values off the stack and pushes a tuple of
length 2 whose items are those values back onto it. In other
words:
stack[-2:] = [tuple(stack[-2:])]
ZTUPLE3� a Build a three-tuple out of the top three items on the stack.
This code pops three values off the stack and pushes a tuple of
length 3 whose items are those values back onto it. In other
words:
stack[-3:] = [tuple(stack[-3:])]
Z
EMPTY_DICT�}zPush an empty dict.ZDICT�da� Build a dict out of the topmost stack slice, after markobject.
All the stack entries following the topmost markobject are placed into
a single Python dict, which single dict object replaces all of the
stack from the topmost markobject onward. The stack slice alternates
key, value, key, value, .... For example,
Stack before: ... markobject 1 2 3 'abc'
Stack after: ... {1: 2, 3: 'abc'}
ZSETITEMrY z�Add a key+value pair to an existing dict.
Stack before: ... pydict key value
Stack after: ... pydict
where pydict has been modified via pydict[key] = value.
ZSETITEMS�ua\ Add an arbitrary number of key+value pairs to an existing dict.
The slice of the stack following the topmost markobject is taken as
an alternating sequence of keys and values, added to the dict
immediately under the topmost markobject. Everything at and after the
topmost markobject is popped, leaving the mutated dict at the top
of the stack.
Stack before: ... pydict markobject key_1 value_1 ... key_n value_n
Stack after: ... pydict
where pydict has been modified via pydict[key_i] = value_i for i in
1, 2, ..., n, and in that order.
Z EMPTY_SET� zPush an empty set.ZADDITEMS� a$ Add an arbitrary number of items to an existing set.
The slice of the stack following the topmost markobject is taken as
a sequence of items, added to the set immediately under the topmost
markobject. Everything at and after the topmost markobject is popped,
leaving the mutated set at the top of the stack.
Stack before: ... pyset markobject item_1 ... item_n
Stack after: ... pyset
where pyset has been modified via pyset.add(item_i) = item_i for i in
1, 2, ..., n, and in that order.
Z FROZENSET� az Build a frozenset out of the topmost slice, after markobject.
All the stack entries following the topmost markobject are placed into
a single Python frozenset, which single frozenset object replaces all
of the stack from the topmost markobject onward. For example,
Stack before: ... markobject 1 2 3
Stack after: ... frozenset({1, 2, 3})
�POP�0z<Discard the top stack item, shrinking the stack by one item.ZDUP�2z=Push the top stack item onto the stack again, duplicating it.�MARK�(z�Push markobject onto the stack.
markobject is a unique object, used by other opcodes to identify a
region of the stack containing a variable number of objects for them
to work on. See markobject.doc for more detail.
ZPOP_MARK�1a Pop all the stack objects at and above the topmost markobject.
When an opcode using a variable number of stack objects is done,
POP_MARK is used to remove those objects, and to remove the markobject
that delimited their starting position on the stack.
�GET�gz�Read an object from the memo and push it on the stack.
The index of the memo object to push is given by the newline-terminated
decimal string following. BINGET and LONG_BINGET are space-optimized
versions.
�BINGET�hz�Read an object from the memo and push it on the stack.
The index of the memo object to push is given by the 1-byte unsigned
integer following.
� LONG_BINGET�jz�Read an object from the memo and push it on the stack.
The index of the memo object to push is given by the 4-byte unsigned
little-endian integer following.
�PUT�pz�Store the stack top into the memo. The stack is not popped.
The index of the memo location to write into is given by the newline-
terminated decimal string following. BINPUT and LONG_BINPUT are
space-optimized versions.
�BINPUTr= z�Store the stack top into the memo. The stack is not popped.
The index of the memo location to write into is given by the 1-byte
unsigned integer following.
� LONG_BINPUT�rz�Store the stack top into the memo. The stack is not popped.
The index of the memo location to write into is given by the 4-byte
unsigned little-endian integer following.
�MEMOIZE� z�Store the stack top into the memo. The stack is not popped.
The index of the memo location to write is the number of
elements currently present in the memo.
ZEXT1� a� Extension code.
This code and the similar EXT2 and EXT4 allow using a registry
of popular objects that are pickled by name, typically classes.
It is envisioned that through a global negotiation and
registration process, third parties can set up a mapping between
ints and object names.
In order to guarantee pickle interchangeability, the extension
code registry ought to be global, although a range of codes may
be reserved for private use.
EXT1 has a 1-byte integer argument. This is used to index into the
extension registry, and the object at that index is pushed on the stack.
ZEXT2� zNExtension code.
See EXT1. EXT2 has a two-byte integer argument.
ZEXT4� zOExtension code.
See EXT1. EXT4 has a four-byte integer argument.
ZGLOBAL�ca� Push a global object (module.attr) on the stack.
Two newline-terminated strings follow the GLOBAL opcode. The first is
taken as a module name, and the second as a class name. The class
object module.class is pushed on the stack. More accurately, the
object returned by self.find_class(module, class) is pushed on the
stack, so unpickling subclasses can override this form of lookup.
Z STACK_GLOBAL� z7Push a global object (module.attr) on the stack.
ZREDUCE�RaL Push an object built from a callable and an argument tuple.
The opcode is named to remind of the __reduce__() method.
Stack before: ... callable pytuple
Stack after: ... callable(*pytuple)
The callable and the argument tuple are the first two items returned
by a __reduce__ method. Applying the callable to the argtuple is
supposed to reproduce the original object, or at least get it started.
If the __reduce__ method returns a 3-tuple, the last component is an
argument to be passed to the object's __setstate__, and then the REDUCE
opcode is followed by code to create setstate's argument, and then a
BUILD opcode to apply __setstate__ to that argument.
If not isinstance(callable, type), REDUCE complains unless the
callable has been registered with the copyreg module's
safe_constructors dict, or the callable has a magic
'__safe_for_unpickling__' attribute with a true value. I'm not sure
why it does this, but I've sure seen this complaint often enough when
I didn't want to <wink>.
ZBUILD�ba� Finish building an object, via __setstate__ or dict update.
Stack before: ... anyobject argument
Stack after: ... anyobject
where anyobject may have been mutated, as follows:
If the object has a __setstate__ method,
anyobject.__setstate__(argument)
is called.
Else the argument must be a dict, the object must have a __dict__, and
the object is updated via
anyobject.__dict__.update(argument)
ZINST�iaq Build a class instance.
This is the protocol 0 version of protocol 1's OBJ opcode.
INST is followed by two newline-terminated strings, giving a
module and class name, just as for the GLOBAL opcode (and see
GLOBAL for more details about that). self.find_class(module, name)
is used to get a class object.
In addition, all the objects on the stack following the topmost
markobject are gathered into a tuple and popped (along with the
topmost markobject), just as for the TUPLE opcode.
Now it gets complicated. If all of these are true:
+ The argtuple is empty (markobject was at the top of the stack
at the start).
+ The class object does not have a __getinitargs__ attribute.
then we want to create an old-style class instance without invoking
its __init__() method (pickle has waffled on this over the years; not
calling __init__() is current wisdom). In this case, an instance of
an old-style dummy class is created, and then we try to rebind its
__class__ attribute to the desired class object. If this succeeds,
the new instance object is pushed on the stack, and we're done.
Else (the argtuple is not empty, it's not an old-style class object,
or the class object does have a __getinitargs__ attribute), the code
first insists that the class object have a __safe_for_unpickling__
attribute. Unlike as for the __safe_for_unpickling__ check in REDUCE,
it doesn't matter whether this attribute has a true or false value, it
only matters whether it exists (XXX this is a bug). If
__safe_for_unpickling__ doesn't exist, UnpicklingError is raised.
Else (the class object does have a __safe_for_unpickling__ attr),
the class object obtained from INST's arguments is applied to the
argtuple obtained from the stack, and the resulting instance object
is pushed on the stack.
NOTE: checks for __safe_for_unpickling__ went away in Python 2.3.
NOTE: the distinction between old-style and new-style classes does
not make sense in Python 3.
ZOBJ�oa� Build a class instance.
This is the protocol 1 version of protocol 0's INST opcode, and is
very much like it. The major difference is that the class object
is taken off the stack, allowing it to be retrieved from the memo
repeatedly if several instances of the same class are created. This
can be much more efficient (in both time and space) than repeatedly
embedding the module and class names in INST opcodes.
Unlike INST, OBJ takes no arguments from the opcode stream. Instead
the class object is taken off the stack, immediately above the
topmost markobject:
Stack before: ... markobject classobject stackslice
Stack after: ... new_instance_object
As for INST, the remainder of the stack above the markobject is
gathered into an argument tuple, and then the logic seems identical,
except that no __safe_for_unpickling__ check is done (XXX this is
a bug). See INST for the gory details.
NOTE: In Python 2.3, INST and OBJ are identical except for how they
get the class object. That was always the intent; the implementations
had diverged for accidental reasons.
ZNEWOBJ� aL Build an object instance.
The stack before should be thought of as containing a class
object followed by an argument tuple (the tuple being the stack
top). Call these cls and args. They are popped off the stack,
and the value returned by cls.__new__(cls, *args) is pushed back
onto the stack.
Z NEWOBJ_EX� au Build an object instance.
The stack before should be thought of as containing a class
object followed by an argument tuple and by a keyword argument dict
(the dict being the stack top). Call these cls and args. They are
popped off the stack, and the value returned by
cls.__new__(cls, *args, *kwargs) is pushed back onto the stack.
�PROTO� z�Protocol version indicator.
For protocol 2 and above, a pickle must start with this opcode.
The argument is the protocol version, an int in range(2, 256).
�STOP�.z�Stop the unpickling machine.
Every pickle ends with this opcode. The object at the top of the stack
is popped, and that's the result of unpickling. The stack should be
empty then.
�FRAME� z�Indicate the beginning of a new frame.
The unpickler may use this opcode to safely prefetch data from its
underlying stream.
ZPERSID�Pa Push an object identified by a persistent ID.
The pickle module doesn't define what a persistent ID means. PERSID's
argument is a newline-terminated str-style (no embedded escapes, no
bracketing quote characters) string, which *is* "the persistent ID".
The unpickler passes this string to self.persistent_load(). Whatever
object that returns is pushed on the stack. There is no implementation
of persistent_load() in Python's unpickler: it must be supplied by an
unpickler subclass.
Z BINPERSID�QaX Push an object identified by a persistent ID.
Like PERSID, except the persistent ID is popped off the stack (instead
of being a string embedded in the opcode bytestream). The persistent
ID is passed to self.persistent_load(), and whatever object that
returns is pushed on the stack. See PERSID for more detail.
z%repeated name %r at indices %d and %dz%repeated code %r at indices %d and %dFc C s� t j � } xt j D]} t j d | � sE | r t d | � q t t | � } t | t � sv t
| � d k r� | r t d | | f � q | j d � } | | k r| r� t d | | f � | | } | j | k r� t
d | | | j f � � | | =q t
d | | f � � q W| r|d g } x4 | j � D]& \ } } | j d
| j | f � q=Wt
d j | � � � d S) Nz[A-Z][A-Z0-9_]+$z0skipping %r: it doesn't look like an opcode namer z5skipping %r: value %r doesn't look like a pickle codezlatin-1z+checking name %r w/ code %r for consistencyzBfor pickle code %r, pickle.py uses name %r but we're using name %rzPpickle.py appears to have a pickle opcode with name %r and code %r, but we don'tz=we appear to have pickle opcodes that pickle.py doesn't have:z name %r with code %r�
)�code2op�copyr} �__all__�re�match�print�getattrr ro r' r; r r" �items�append�join)�verboser� r Z
picklecoder� �msgrx r r r �assure_pickle_consistency� s> "
r� c c sF t | t � r t j | � } t | d � r9 | j } n d d � } x� | � } | j d � } t j | j d � � } | d k r� | d k r� t
d � � n( t
d | d k r� d n | | f � � | j d k r� d } n | j j | � } | r| | | | � f Vn | | | f V| d
k rH | j
d k s=t � PqH Wd S) N�tellc S s d S)Nr r r r r �<lambda>� s z_genops.<locals>.<lambda>r zlatin-1� z#pickle exhausted before seeing STOPz!at position %s, opcode %r unknownz <unknown>� .r� )r � bytes_types�io�BytesIO�hasattrr� r! r� �getr; r" ry r
r r )r$ �
yield_end_posZgetpos�posrx �opcodery r r r �_genops� s0
r� c C s
t | � S)ax Generate all the opcodes in a pickle.
'pickle' is a file-like object, or string, containing the pickle.
Each opcode in the pickle is generated, from the current pickle position,
stopping after a STOP opcode is delivered. A triple is generated for
each opcode:
opcode, arg, pos
opcode is an OpcodeInfo record, describing the current opcode.
If the opcode has an argument embedded in the pickle, arg is its decoded
value, as a Python object. If the opcode doesn't have an argument, arg
is None.
If the pickle has a tell() method, pos was the value of pickle.tell()
before reading the current opcode. If the pickle is a bytes object,
it's wrapped in a BytesIO object, and the latter's tell() result is
used. Else (the pickle doesn't have a tell(), and it's not obvious how
to query its current position) pos is None.
)r� )r} r r r r � s c C s� d } d } t � } i } g } d } d } xXt | d d �D]D\ } } }
} d | j k r� | j | � | j | | f � q@ | j d k r� t | � } | j | � | j | | f � q@ d | j k r� q@ d | j k r| j | k r� | j } d | | <| j | | f � q@ | j d
k rq| | k r<| } |
d k r[| |
| � } q�| j |
| f � q@ | j |
| f � q@ W~ t j � }
|
j | � t
j |
| � } | d k r�| j j
� d } x� | D]� \ } } | | k r)| | k rq�| j | � } | | | <| d 7} n2 | | k rK| j | | � } n | | | � } | j j � | j | � q�W| j j � |
j � S)
z7Optimize a pickle string by removing unused PUT opcodesr� r� r r� r� Tr� r� Nr� r r )rs r� r �addr� r' r| r� r� �writer} Z_PicklerZframerZ
start_framing�putr� Z commit_frameZ end_framing�getvalue)r� r� r� ZoldidsZnewids�opcodesr| Z protoheaderr� ry r� Zend_pos�idx�outZpickler�opr$ r r r r � sd %
c C se g } | d k r i } d } g } d | } d } | }
x�t | � D]�\ } } }
|
d k r| t d |
d d d | �d t | j � d d � | t | � | j f } t | | j � } | j } | j } t | � } d } t
| k s| j d k r�| r�| d t
k r�t
| k s(t � t
| k rJ| d! t k sJt � | r�| j
� } | d k rqd } n
d
| } x | d" t
k r�| j
� q~W| j
� y | j t
� } Wq�t k
r�| j d k s�t � d } Yq�Xn
d } } | j d# k r�| j d k r#t | � } n | d k s5t � | } | | k rTd | } q�| scd } q�| d$ t
k r|d } q�| d% | | <nM | j d& k r�| | k r�t | � d k s�t � | | g } n
d | } | d k s�| r;| d d t | j � 7} | d k r'| d t | � 7} | r;| d | 7} | r�| d |
t | � 7} t | � }
|
d k rw| }
| d | j j d d � d 7} t | d | �| r�t | � � t | � | k r�t d | t | � f � � | r�| | d � =t
| k r't
| k st � | j |
� | j | � qG Wt d | d | �| rat d | � � d S)'aK Produce a symbolic disassembly of a pickle.
'pickle' is a file-like object, or string, containing a (at least one)
pickle. The pickle is disassembled from the current position, through
the first STOP opcode encountered.
Optional arg 'out' is a file-like object to which the disassembly is
printed. It defaults to sys.stdout.
Optional arg 'memo' is a Python dict, used as the pickle's memo. It
may be mutated by dis(), if the pickle contains PUT or BINPUT opcodes.
Passing the same memo object to another dis() call then allows disassembly
to proceed across multiple pickles that were all created by the same
pickler with the same memo. Ordinarily you don't need to worry about this.
Optional arg 'indentlevel' is the number of blanks by which to indent
a new MARK level. It defaults to 4.
Optional arg 'annotate' if nonzero instructs dis() to add short
description of the opcode on each line of disassembled output.
The value given to 'annotate' must be an integer and is used as a
hint for the column where annotation should start. The default
value is 0, meaning no annotations.
In addition to printing the disassembly, some sanity checks are made:
+ All embedded opcode arguments "make sense".
+ Explicit and implicit pop operations have enough items on the stack.
+ When an opcode implicitly refers to a markobject, a markobject is
actually on the stack.
+ A memo entry isn't referenced before it's defined.
+ The markobject isn't stored in the memo.
+ A memo entry isn't redefined.
Nr � z%5d:�end�filez %-4s %s%sr� z (MARK at unknown opcode offset)z (MARK at %d)r zno MARK exists on stackr� r� r� r� zmemo key %r already definedz'stack is empty -- can't store into memoz"can't store markobject in the memor� r� r� z&memo key %r has never been stored into�
�2 r� z3tries to pop %d items from stack with only %d itemsz highest protocol among opcodes =z stack not empty after STOP: %rr5 r5 r5 r5 r5 )zPUTr� r� zMEMOIZEr5 r5 )zGETr� r� )r r� �reprrx r' r �maxr| rz r{ �
markobjectr rv �pop�indexr" r �splitr� �extend)r} r� �memo� indentlevel�annotate�stackZmaxprotoZ markstackZ indentchunkZerrormsgZannocolr� ry r� �lineZbeforeZafterZnumtopopZmarkmsgZmarkposZmemo_idxr r r r ) s� -
!
c @ s e Z d Z d d � Z d S)�_Examplec C s
| | _ d S)N)�value)r r r r r r � s z_Example.__init__N)r r r r r r r r r � s r a�
>>> import pickle
>>> x = [1, 2, (3, 4), {b'abc': "def"}]
>>> pkl0 = pickle.dumps(x, 0)
>>> dis(pkl0)
0: ( MARK
1: l LIST (MARK at 0)
2: p PUT 0
5: L LONG 1
9: a APPEND
10: L LONG 2
14: a APPEND
15: ( MARK
16: L LONG 3
20: L LONG 4
24: t TUPLE (MARK at 15)
25: p PUT 1
28: a APPEND
29: ( MARK
30: d DICT (MARK at 29)
31: p PUT 2
34: c GLOBAL '_codecs encode'
50: p PUT 3
53: ( MARK
54: V UNICODE 'abc'
59: p PUT 4
62: V UNICODE 'latin1'
70: p PUT 5
73: t TUPLE (MARK at 53)
74: p PUT 6
77: R REDUCE
78: p PUT 7
81: V UNICODE 'def'
86: p PUT 8
89: s SETITEM
90: a APPEND
91: . STOP
highest protocol among opcodes = 0
Try again with a "binary" pickle.
>>> pkl1 = pickle.dumps(x, 1)
>>> dis(pkl1)
0: ] EMPTY_LIST
1: q BINPUT 0
3: ( MARK
4: K BININT1 1
6: K BININT1 2
8: ( MARK
9: K BININT1 3
11: K BININT1 4
13: t TUPLE (MARK at 8)
14: q BINPUT 1
16: } EMPTY_DICT
17: q BINPUT 2
19: c GLOBAL '_codecs encode'
35: q BINPUT 3
37: ( MARK
38: X BINUNICODE 'abc'
46: q BINPUT 4
48: X BINUNICODE 'latin1'
59: q BINPUT 5
61: t TUPLE (MARK at 37)
62: q BINPUT 6
64: R REDUCE
65: q BINPUT 7
67: X BINUNICODE 'def'
75: q BINPUT 8
77: s SETITEM
78: e APPENDS (MARK at 3)
79: . STOP
highest protocol among opcodes = 1
Exercise the INST/OBJ/BUILD family.
>>> import pickletools
>>> dis(pickle.dumps(pickletools.dis, 0))
0: c GLOBAL 'pickletools dis'
17: p PUT 0
20: . STOP
highest protocol among opcodes = 0
>>> from pickletools import _Example
>>> x = [_Example(42)] * 2
>>> dis(pickle.dumps(x, 0))
0: ( MARK
1: l LIST (MARK at 0)
2: p PUT 0
5: c GLOBAL 'copy_reg _reconstructor'
30: p PUT 1
33: ( MARK
34: c GLOBAL 'pickletools _Example'
56: p PUT 2
59: c GLOBAL '__builtin__ object'
79: p PUT 3
82: N NONE
83: t TUPLE (MARK at 33)
84: p PUT 4
87: R REDUCE
88: p PUT 5
91: ( MARK
92: d DICT (MARK at 91)
93: p PUT 6
96: V UNICODE 'value'
103: p PUT 7
106: L LONG 42
111: s SETITEM
112: b BUILD
113: a APPEND
114: g GET 5
117: a APPEND
118: . STOP
highest protocol among opcodes = 0
>>> dis(pickle.dumps(x, 1))
0: ] EMPTY_LIST
1: q BINPUT 0
3: ( MARK
4: c GLOBAL 'copy_reg _reconstructor'
29: q BINPUT 1
31: ( MARK
32: c GLOBAL 'pickletools _Example'
54: q BINPUT 2
56: c GLOBAL '__builtin__ object'
76: q BINPUT 3
78: N NONE
79: t TUPLE (MARK at 31)
80: q BINPUT 4
82: R REDUCE
83: q BINPUT 5
85: } EMPTY_DICT
86: q BINPUT 6
88: X BINUNICODE 'value'
98: q BINPUT 7
100: K BININT1 42
102: s SETITEM
103: b BUILD
104: h BINGET 5
106: e APPENDS (MARK at 3)
107: . STOP
highest protocol among opcodes = 1
Try "the canonical" recursive-object test.
>>> L = []
>>> T = L,
>>> L.append(T)
>>> L[0] is T
True
>>> T[0] is L
True
>>> L[0][0] is L
True
>>> T[0][0] is T
True
>>> dis(pickle.dumps(L, 0))
0: ( MARK
1: l LIST (MARK at 0)
2: p PUT 0
5: ( MARK
6: g GET 0
9: t TUPLE (MARK at 5)
10: p PUT 1
13: a APPEND
14: . STOP
highest protocol among opcodes = 0
>>> dis(pickle.dumps(L, 1))
0: ] EMPTY_LIST
1: q BINPUT 0
3: ( MARK
4: h BINGET 0
6: t TUPLE (MARK at 3)
7: q BINPUT 1
9: a APPEND
10: . STOP
highest protocol among opcodes = 1
Note that, in the protocol 0 pickle of the recursive tuple, the disassembler
has to emulate the stack in order to realize that the POP opcode at 16 gets
rid of the MARK at 0.
>>> dis(pickle.dumps(T, 0))
0: ( MARK
1: ( MARK
2: l LIST (MARK at 1)
3: p PUT 0
6: ( MARK
7: g GET 0
10: t TUPLE (MARK at 6)
11: p PUT 1
14: a APPEND
15: 0 POP
16: 0 POP (MARK at 0)
17: g GET 1
20: . STOP
highest protocol among opcodes = 0
>>> dis(pickle.dumps(T, 1))
0: ( MARK
1: ] EMPTY_LIST
2: q BINPUT 0
4: ( MARK
5: h BINGET 0
7: t TUPLE (MARK at 4)
8: q BINPUT 1
10: a APPEND
11: 1 POP_MARK (MARK at 0)
12: h BINGET 1
14: . STOP
highest protocol among opcodes = 1
Try protocol 2.
>>> dis(pickle.dumps(L, 2))
0: \x80 PROTO 2
2: ] EMPTY_LIST
3: q BINPUT 0
5: h BINGET 0
7: \x85 TUPLE1
8: q BINPUT 1
10: a APPEND
11: . STOP
highest protocol among opcodes = 2
>>> dis(pickle.dumps(T, 2))
0: \x80 PROTO 2
2: ] EMPTY_LIST
3: q BINPUT 0
5: h BINGET 0
7: \x85 TUPLE1
8: q BINPUT 1
10: a APPEND
11: 0 POP
12: h BINGET 1
14: . STOP
highest protocol among opcodes = 2
Try protocol 3 with annotations:
>>> dis(pickle.dumps(T, 3), annotate=1)
0: \x80 PROTO 3 Protocol version indicator.
2: ] EMPTY_LIST Push an empty list.
3: q BINPUT 0 Store the stack top into the memo. The stack is not popped.
5: h BINGET 0 Read an object from the memo and push it on the stack.
7: \x85 TUPLE1 Build a one-tuple out of the topmost item on the stack.
8: q BINPUT 1 Store the stack top into the memo. The stack is not popped.
10: a APPEND Append an object to a list.
11: 0 POP Discard the top stack item, shrinking the stack by one item.
12: h BINGET 1 Read an object from the memo and push it on the stack.
14: . STOP Stop the unpickling machine.
highest protocol among opcodes = 2
a=
>>> import pickle
>>> import io
>>> f = io.BytesIO()
>>> p = pickle.Pickler(f, 2)
>>> x = [1, 2, 3]
>>> p.dump(x)
>>> p.dump(x)
>>> f.seek(0)
0
>>> memo = {}
>>> dis(f, memo=memo)
0: \x80 PROTO 2
2: ] EMPTY_LIST
3: q BINPUT 0
5: ( MARK
6: K BININT1 1
8: K BININT1 2
10: K BININT1 3
12: e APPENDS (MARK at 5)
13: . STOP
highest protocol among opcodes = 2
>>> dis(f, memo=memo)
14: \x80 PROTO 2
16: h BINGET 0
18: . STOP
highest protocol among opcodes = 2
Zdisassembler_testZdisassembler_memo_testc C s d d l } | j � S)Nr )�doctestZtestmod)r r r r �_test�
s r �__main__Z descriptionz$disassemble one or more pickle files� pickle_filerk Zbr�nargs�*�helpzthe pickle filez-oz--output�default�wz+the file where the output should be writtenz-mz--memo�action�
store_truez#preserve memo between disassembliesz-lz
--indentlevelz8the number of blanks by which to indent a new MARK levelz-az
--annotatez2annotate each line with a short opcode descriptionz-pz
--preamblez==> {name} <==zMif more than one pickle file is specified, print this before each disassemblyz-tz--testzrun self-test suitez-vz)run verbosely; only affects self-test run� r� )��__doc__r9 r� r} r� rJ r� r� r r r r r �objectr
Zstructr r( r% r& r) r* r+ r, r- r. r0 r1 r> r? r@ rA rB rC rD rE rF rG rH rI rL rM rN rO rP rQ rS rT rU rV rW rX rZ r\ r] r^ r` ra rb rc rd re rf rg rh ri r ZpyintZpylongrn Zpyinteger_or_boolZpyboolr_ Zpyfloatro r Zpybytes_or_strZpystringZpybytesZ pyunicoderk Zpynonerl Zpytuplerq Zpylistrr Zpydictrs ZpysetZ pyfrozensetZ anyobjectr� rv rw r r� Zname2iZcode2i� enumerater� r� r r" rx r� r� r� r r r r Z _dis_testZ
_memo_testZ__test__r r �argparse�ArgumentParser�parser� add_argumentZFileType�stdout�
parse_args�argsZtestr� r Z
print_helpr' �outputr� r� r# Zpreamble�formatr� r r r r �<module> s� �$ /
;
+
& >��
!
Mini Shell