File: //opt/alt/python37/lib/python3.7/site-packages/pyasn1/codec/ber/__pycache__/decoder.cpython-37.pyc
B
��f�- � 2 @ s0 d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z d dlm
Z
d dlm Z d dlm Z d d l
mZ d d
lmZ d d lmZmZmZmZ d d lmZ d d
lmZ d dlmZ d dlmZ d dlm Z d dlm Z d dlm Z dddgZ ej e!ej"d�Z#ej$Z$ej%Z%G dd� de&�Z'G dd� de'�Z(G dd � d e(�Z)e)� Z*G d d � d e(�Z+G d d � d e+�Z,G d!d"� d"e(�Z-G d#d$� d$e(�Z.G d%d&� d&e(�Z/G d'd(� d(e(�Z0G d)d*� d*e(�Z1G d+d,� d,e'�Z2G d-d.� d.e2�Z3G d/d0� d0e3�Z4G d1d2� d2e4�Z5G d3d4� d4e4�Z6G d5d6� d6e3�Z7G d7d8� d8e7�Z8G d9d:� d:e7�Z9G d;d<� d<e3�Z:G d=d>� d>e(�Z;G d?d@� d@e.�Z<G dAdB� dBe.�Z=G dCdD� dDe.�Z>G dEdF� dFe.�Z?G dGdH� dHe.�Z@G dIdJ� dJe.�ZAG dKdL� dLe.�ZBG dMdN� dNe.�ZCG dOdP� dPe.�ZDG dQdR� dRe.�ZEG dSdT� dTe.�ZFG dUdV� dVe.�ZGG dWdX� dXe.�ZHG dYdZ� dZe.�ZIe jJjKe+� e jLjKe,� e jMjKe-� e jNjKe.� e jOjKe/� e jPjKe0� e jQjKe+� e jRjKe1� e jSjKe4� e jTjKe7� e jUjKe:� ejVjKe<� ejWjKe=� ejXjKe>� ejYjKe?� ejZjKe@� ej[jKeA� ej\jKeB� ej]jKeC� ej^jKeD� ej_jKeE� ej`jKeF� e jajKeG� e jbjKeH� e jcjKeI� iZde jTjee8� e jfjee9� e jSjee5� e jgjee6� e jUjee:� e jhjee;� iZiedZjeiZkxDed�l� D ]8Zmemjndk �r�emjnjojeZeeedk �r�eeeik�r�emeiee< �q�W d[d\� epd]�D �\
ZqZrZsZtZuZvZwZxZyZzed^�Z{G d_d`� d`e&�Z|G dad� de&�Z}G dbd� de&�Z~e~� ZdS )c� N)�debug)�error)�eoo)�asSeekableStream)�
isEndOfStream)�peekIntoStream)�readFromStream)�_MISSING)�
from_bytes)�oct2int� octs2ints� ints2octs�null)� PyAsn1Error)�base)�char)�tag)�tagmap)�univ)�useful�StreamingDecoder�Decoder�decode)�flagsc @ s0 e Zd ZdZddd�Zd dd�Zedd� �ZdS )
�AbstractPayloadDecoderNc K s t �d|f ��dS )ztDecode value with fixed byte length.
The decoder is allowed to consume as many bytes as necessary.
z(SingleItemDecoder not implemented for %sN)r r ) �self� substrate�asn1Spec�tagSet�length�state� decodeFun� substrateFun�options� r$ �I/opt/alt/python37/lib/python3.7/site-packages/pyasn1/codec/ber/decoder.py� valueDecoder* s z#AbstractPayloadDecoder.valueDecoderc K s t �d|f ��dS )zsDecode value with undefined length.
The decoder is allowed to consume as many bytes as necessary.
z5Indefinite length mode decoder not implemented for %sN)r r ) r r r r r r r! r" r# r$ r$ r% �indefLenValueDecoder4 s z+AbstractPayloadDecoder.indefLenValueDecoderc C s d|kr| |d< |S )N�
asn1Objectr$ )r( r# r$ r$ r% �_passAsn1Object> s z&AbstractPayloadDecoder._passAsn1Object)NNNNN)NNNNN)�__name__�
__module__� __qualname__�protoComponentr& r' � staticmethodr) r$ r$ r$ r% r ' s
r c @ s e Zd Zedd� �Zdd� ZdS )� AbstractSimplePayloadDecoderc c s xt |||�D ]
}|V qW d S )N)r )r( r r r# �chunkr$ r$ r% �substrateCollectorG s z/AbstractSimplePayloadDecoder.substrateCollectorc K s@ |� d�r|S |d kr&| jj||d�S |tkr2|S |�|�S d S )NZnative)r )�getr- �clone�noValue)r r r �valuer# r$ r$ r% �_createComponentL s
z-AbstractSimplePayloadDecoder._createComponentN)r* r+ r, r. r1 r6 r$ r$ r$ r% r/ F s r/ c @ s* e Zd Ze�d�Zddd�Zddd�ZdS ) �RawPayloadDecoder� Nc k s^ |r8| j ||df|�} x || |||�D ]
}
|
V q&W d S x |||||f|�D ]
} | V qLW d S )Nr8 )r6 ) r r r r r r r! r" r# r( r0 r5 r$ r$ r% r& Z s
z RawPayloadDecoder.valueDecoderc k sz |r8| j ||df|�} x || |||�D ]
}
|
V q&W d S x<x6|||||fddi|��D ]} | tjkrhd S | V qVW q:W d S )Nr8 �allowEooT)r6 r � endOfOctets) r r r r r r r! r" r# r( r0 r5 r$ r$ r% r' i s
z&RawPayloadDecoder.indefLenValueDecoder)NNNNN)NNNNN)r* r+ r, r �Anyr- r& r' r$ r$ r$ r% r7 W s
r7 c @ s e Zd Ze�d�Zddd�ZdS )�IntegerPayloadDecoderr Nc k sn |d j tjkrt�d��x$t|||�D ]} t| t�r(| V q(W | rRt| dd�}
nd}
| j |||
f|�V d S )Nr zSimple tag format expectedT)�signed)
� tagFormatr �tagFormatSimpler r r �
isinstance�SubstrateUnderrunErrorr
r6 ) r r r r r r r! r" r# r0 r5 r$ r$ r% r& � s
z"IntegerPayloadDecoder.valueDecoder)NNNNN)r* r+ r, r �Integerr- r& r$ r$ r$ r% r< � s
r< c @ s e Zd Ze�d�Zdd� ZdS )�BooleanPayloadDecoderr c K s t j| |||rdpdf|�S )N� r )r< r6 )r r r r5 r# r$ r$ r% r6 � s z&BooleanPayloadDecoder._createComponentN)r* r+ r, r �Booleanr- r6 r$ r$ r$ r% rC � s
rC c @ s. e Zd Ze�d�ZdZddd�Zd dd�ZdS )
�BitStringPayloadDecoderr$ TNc k s |r8| j ||tf|�} x || |||�D ]
}
|
V q&W d S |sFt�d��x t|�D ]}
t|
t�rP|
V qPW |
rvt�d��|d jtj k�r x$t
|d|�D ]} t| t�r�| V q�W t | �} | dkr�t�d| ��x(t
||d |�D ]}
t|
t�r�|
V q�W | j j
|
d| d�} | j ||| f|�V d S | j�s:t�d| jj ��t�rHtd � | j}| j j
tdd
�}
|�� }x�|�� | |k �r�x4||| j fd |i|��D ]}t|t��r�|V �q�W t|d �} | dk�r�t�d| ��| j j
|dd � d|
| d �}
�qhW | j |||
f|�V d S )
NzEmpty BIT STRING substrater rD � zTrailing bits overflow %sT)�internalFormat�paddingz*Constructed encoding form prohibited at %sz$assembling constructed serialization)rH r" )rH �prependrI )r6 r4 r r r r@ rA r> r r? r �ordr- �fromOctetString�supportConstructedForm� __class__r* �LOGr1 r �tellr )r r r r r r r! r" r# r( r0 � trailingBitsr5 � bitString�current_position� componentr$ r$ r% r& � sb
z$BitStringPayloadDecoder.valueDecoderc k s� |r8| j ||tf|�} x || |||�D ]
}
|
V q&W d S | j}| jjtdd�} x�x>||| jf|dd�|��D ] } | tjkr|P t| t �rl| V qlW | tjkr�P t
| d �}
|
dkr�t � d|
��| jj| dd � d| |
d�} qPW | j ||| f|�V d S ) NT)rH )r" r9 r rG zTrailing bits overflow %srD )rH rJ rI )
r6 r4 r1 r- rL r r r: r@ rA r r r )r r r r r r r! r" r# r( r0 rR rT rQ r$ r$ r% r' � s2
z,BitStringPayloadDecoder.indefLenValueDecoder)NNNNN)NNNNN) r* r+ r, r � BitStringr- rM r& r' r$ r$ r$ r% rF � s
O rF c @ s. e Zd Ze�d�ZdZddd�Zd dd�ZdS )
�OctetStringPayloadDecoderr8 TNc k s& |r8| j ||tf|�} x || |||�D ]
}
|
V q&W d S |d jtjkr�x$t|||�D ]}
t|
t�rV|
V qVW | j |||
f|�V d S | js�t �
d| j j ��t
r�t
d� | j}t} |�� } xP|�� | |k �r x0||| jfd|i|��D ]}
t|
t�r�|
V q�W | |
7 } q�W | j ||| f|�V d S )Nr z*Constructed encoding form prohibited at %sz$assembling constructed serializationr" )r6 r4 r> r r? r r@ rA rM r r rN r* rO r1 r rP r- )r r r r r r r! r" r# r( r0 �header�original_positionrT r$ r$ r% r& * s4
z&OctetStringPayloadDecoder.valueDecoderc
k s� |rB|| j k rB| j||tf|�} x || |||�D ]
}
|
V q0W d S | j }t} xXx>||| jf|dd�|��D ] } t| t�r~| V | tjkrjP qjW | tjkr�P | | 7 } qNW | j||| f|�V d S )NT)r" r9 ) r1 r6 r4 r r- r@ rA r r: )
r r r r r r r! r" r# r( r0 rW rT r$ r$ r% r' W s&
z.OctetStringPayloadDecoder.indefLenValueDecoder)NNNNN)NNNNN) r* r+ r, r � OctetStringr- rM r& r' r$ r$ r$ r% rV & s
, rV c @ s e Zd Ze�d�Zddd�ZdS )�NullPayloadDecoderr8 Nc k sn |d j tjkrt�d��x$t|||�D ]} t| t�r(| V q(W | j||df|�}
| rdt�d| ��|
V d S )Nr zSimple tag format expectedr8 z&Unexpected %d-octet substrate for Null) r> r r? r r r r@ rA r6 ) r r r r r r r! r" r# r0 rT r$ r$ r% r& s
z NullPayloadDecoder.valueDecoder)NNNNN)r* r+ r, r �Nullr- r& r$ r$ r$ r% rZ | s
rZ c @ s e Zd Ze�d�Zddd�ZdS )� ObjectIdentifierPayloadDecoderr$ Nc k s� |d j tjkrt�d��x$t|||�D ]} t| t�r(| V q(W | sNt�d��t| �} d}
d} t | �} x�| | k �r| | }
| d7 } |
dk r�|
|
f7 }
qh|
dk�r|
}d}
xD|dkr�|
d> |d@ }
| | kr�t�d |
f ��| | }| d7 } q�W |
|
d> | f7 }
qh|
dkrht�d
��qhW d|
d k�r4d k�rBn n
d |
}
n|d
|
d k�r^dk�r�n n d|
d d
f|
dd � }
n>|
d dk�r�d|
d d f|
dd � }
nt�d| d ��| j
|||
f|�V d S )Nr zSimple tag format expectedzEmpty substrater$ rD � rG � z#Short substrate for sub-OID past %sz"Invalid octet 0x80 in OID encoding�' )r �( �O �P � z Malformed first OID octet: %s) r> r r? r r r r@ rA r �lenr6 )r r r r r r r! r" r# r0 �oid�indexZ substrateLenZsubIdZ nextSubIdr$ r$ r% r& � sJ
z+ObjectIdentifierPayloadDecoder.valueDecoder)NNNNN)r* r+ r, r �ObjectIdentifierr- r& r$ r$ r$ r% r\ � s
r\ c @ s e Zd Ze�� Zddd�ZdS )�RealPayloadDecoderNc k s� |d j tjkrt�d��x$t|||�D ]} t| t�r(| V q(W | s\| j||df|�V d S t | d �}
| dd � } |
d@ �r�| s�t�d��t
r�t
d� |
d@ d } | d kr�t | d �} | dd � } | d | � | | d � } } | r�| s�t�d
��t | d �d@ �rd �p
d}
x0| �r<|
d K }
|
t | d �O }
| dd � } �qW |
d ? d@ }|d
k�r^t�d��|dk�rr|
d9 }
n|d
k�r�|
d 9 }
d}x0| �r�|d K }|t | d �O }| dd � } �q�W |
d@ �r�| }|
d
? d@ }|d
| 9 }|d
|
f}n�|
d@ �r t
�rt
d� |
d@ �rd�pd}n�|
d@ dk�r�| �s:t�d��t
�rHt
d� y`|
d@ dk�rht | �ddf}n>|
d@ d
k�r�t | �}n&|
d@ dk�r�t | �}nt�d|
��W n t
k
�r� t�d��Y nX nt�d|
��| j|||f|�V d S )Nr zSimple tag format expectedg rD r] z Incomplete floating-point valuez decoding binary encoded REAL� � zReal exponent screwed���� rc zIllegal Real base�@ zdecoding infinite REALz-inf�inf� z decoding character encoded REAL�
zUnknown NR (tag %s)zBad character Real syntaxzUnknown encoding (tag %s))r> r r? r r r r@ rA r6 r rO �int�float�
ValueError)r r r r r r r! r" r# r0 �fo�n�eo�e�b�pZsfr5 r$ r$ r% r& � s�
z RealPayloadDecoder.valueDecoder)NNNNN)r* r+ r, r �Realr- r& r$ r$ r$ r% rh � s rh c @ s e Zd ZdZdS )�!AbstractConstructedPayloadDecoderN)r* r+ r, r- r$ r$ r$ r% r{ = s r{ c @ sB e Zd ZdZdZdd� Zdd� Zd dd�Zd
dd �Zdd
d �Z dS )� ConstructedPayloadDecoderBaseNc C s
t � �d S )N)�NotImplementedError)r r( �idxr$ r$ r% �_getComponentTagMapE s z1ConstructedPayloadDecoderBase._getComponentTagMapc C s
t � �d S )N)r} )r r( r r~ r$ r$ r% �_getComponentPositionByTypeH s z9ConstructedPayloadDecoderBase._getComponentPositionByTypec
k s d }g }t � }|�� } x�|dks0|�� | | k r�x$||f|�D ]}
t|
t�r>|
V q>W |dkrj|
tjkrjP |�|
� |�|
j� t |�dkr�| j
} n| j } | j t
j| jjf|j�� d�}qW tr�td| � x(t|�D ] \} }
|j| |
dddd� q�W |V d S )Nrk rD )r z@guessed %r container type (pass `asn1Spec` to guide the decoder)F)�verifyConstraints� matchTags�matchConstraints)�setrP r@ rA r r: �append�addr rd �protoRecordComponent�protoSequenceComponentr3 r �TagSet�baseTagZ superTagsrO � enumerate�setComponentByPosition)
r r r r! r r# r( �
componentsZcomponentTypesrX rT r- r~ r$ r$ r% �_decodeComponentsSchemalessK s4
z9ConstructedPayloadDecoderBase._decodeComponentsSchemalessc k s |d j tjkrt�d��|�� } |r�|d k r8|�� }
n&| jd k rR| jj|d�}
n | j| j f}
x ||
|||�D ]
} | V qnW d S |d k�r x2| j
|f|||d�|��D ]}
t |
t �r�|
V q�W |�� | | k �rt
�rx$t||d�D ]} t | t �r�| V q�W t
dt| �t�| �f � |
V d S |�� }
|
�� | �|
|�}|jtjjtjjfk�r�|j}
|jtjjk}| �ov|
j }t
�r�t
d| �r�d�p�d |�r�d
�p�d |f � t� }d}�x6|�� | |k �r�|
�s�d }nz|�r�|
j}nlyD|�r�|
| j }n.|
| j �s|
| j �r |
� |�}n
|
| j }W n& t k
�rN t�d |f ��Y nX x*|||f|�D ]}t |t ��r`|V �q`W |�s�|
�r�|�r�|
�!|j"�}n&|
| j �s�|
| j �r�|
�#|j"|�}|
j$||d d d d
� |�%|� |d7 }�q�W t
�r t
d| � |
�rx|
j&�'|��s&t�d|
j(j) ��|
j*�r�|�+di �}t
�rpt
d� x&|�,� D ]\}}t
d||f � �qRW |�s�|�+dd ��r��x t-|
j.�D �]�\}}|j/�s��q�|j �r�|
�0|�j1�sq�|
�2|j/j3�}y || }W n� t4k
�r� t
�r@t
d|
j(j)|j3|
j(j)|j/j3f � x(|j/�,� D ]\}}t
d||f � �q"W y|j/| }W n. t4k
�r| t
�rtt
d|f � �w�Y nX Y nX t
�r�t
d||f � |
�0|�}|jtj5jtj6jfk�r x�t-|�D ]T\}} t7|| �8� �} x0|| fd|i|��D ]}t |t ��r�|V �q�W |||< �q�W nPt7|
�0|��8� �} x0|| fd|i|��D ]}t |t ��rH|V �qHW |
�$||� �q�W n|
j9} | �r| �n~|j}t
�r�t
d| � d}x`|�� | |k �rx*|||f|�D ]}t |t ��r�|V �q�W |
j$||d d d d
� |d7 }�q�W |
V d S )Nr z Constructed tag format expected)r )r r! r )�contextz)Unused trailing %d octets encountered: %sz5decoding %sdeterministic %s type %r chosen by type IDznon-r8 �SETz"Excessive components decoded at %rF)r� r� r� rD zseen component indices %sz,ASN.1 object %s has uninitialized components� openTypesz user-specified open types map:z%s -> %r�decodeOpenTypeszJdefault open types map of component "%s.%s" governed by component "%s.%s":z1failed to resolve open type by governing value %rz+resolved open type %r by governing value %rr z+decoding type %r chosen by given `asn1Spec`):r> r �tagFormatConstructedr r rP r3 r- r� r� r� r@ rA rO r rd r �hexdump�clearr) �typeIdr �Sequence�Set�
componentType�hasOptionalOrDefaultr� � tagMapUniquer( �
isOptional� isDefaulted�getTagMapNearPosition�
IndexError�getPositionByType�effectiveTagSet�getPositionNearTyper� r� �requiredComponents�issubsetrN r* � hasOpenTypesr2 �itemsr� �
namedTypes�openType�getComponentByPosition�isValue�getComponentByName�name�KeyError�SetOf�
SequenceOfr �asOctets�isInconsistent) r r r r r r r! r" r# rX r( r0 Ztrailingr� � isSetType�isDeterministic� seenIndicesr~ r� rT r� �k�v� namedType�governingValuer� �containerValue�pos�containerElement�stream�
inconsistencyr$ r$ r% r& ~ s
z*ConstructedPayloadDecoderBase.valueDecoderc k s |d j tjkrt�d��|d k r||d k r4|�� } n&| jd k rN| jj|d�} n | j| jf} x || |||�D ]
}
|
V qjW d S |d kr�x:| j |f|||d�t
|dd���D ]} t | t �r�| V q�W | V d S |�� } | �
� | �| |�}|jtjjtjjfk�rz| j} | jtjjk} | �o | j }
t�rTtd|
�r:d�p<d | �rHd
�pJd |f � t� }d}�xPt| �|k�rvd }nz| �r�| j}nlyD|
�r�| | j}n.| | j�s�| | j�r�| � |�}n
| | j}W n& t k
�r� t�d | f ��Y nX x@|||fd di|��D ]&}t |t ��r |V |t j k�rP �qW |t j k�r@P |
�s�| �r�| �r`| � |j!�}n&| | j�sx| | j�r�| �"|j!|�}| j#||d
d
d
d� |�$|� |d7 }�qbW t�r�td| � | �r
| j%�&|��s�t�d| j'j( ��| j)�rh|�*di �}t�r4td� x&|�+� D ]\}}td||f � �qW |�sH|�*dd
��rx�x,t,| j-�D �]
\}}|j.�sl�qV|j�r�| �/|�j0�s��qV| �1|j.j2�}y || }W n� t3k
�rF t�rtd| j'j(|j2| j'j(|j.j2f � x(|j.�+� D ]\}}td||f � �q�W y|j.| }W n. t3k
�r@ t�r8td|f � �wVY nX Y nX t�r^td||f � | �/|�}|jtj4jtj5jfk�r�x�t,|�D ]j\}}t6|| �7� �}xF||fd|it
|dd���D ]&}t |t ��r�|V |t j k�r�P �q�W |||< �q�W nft6| �/|��7� �}xR||fd|it
|dd���D ]2}t |t ��r@|V |t j k�rNP | �#||� �q*W �qVW n| j8}|�r
|�n�|j} t�r�td| � d}xrx@||| fd di|��D ]&}t |t ��r�|V |t j k�r�P �q�W |t j k�r�P | j#||d
d
d
d� |d7 }�q�W | V d S )Nr z Constructed tag format expected)r )r r! r T)r9 z5decoding %sdeterministic %s type %r chosen by type IDznon-r8 r� z"Excessive components decoded at %rr9 F)r� r� r� rD zseen component indices %sz,ASN.1 object %s has uninitialized componentsr� z user-specified open types map:z%s -> %rr� zJdefault open types map of component "%s.%s" governed by component "%s.%s":z1failed to resolve open type by governing value %rz+resolved open type %r by governing value %rr z+decoding type %r chosen by given `asn1Spec`)9r> r r� r r r3 r- r� r� r� �dictr@ rA r� r) r� r r� r� r� r� rO r� rd r� r( r� r� r� r� r r: r� r� r� r� r� r� r� rN r* r� r2 r� r� r� r� r� r� r� r� r� r� r� r r� r� ) r r r r r r r! r" r# r( r0 r� r� r� r� r~ rT r� r� r� r� r� r� r� r� r� r� r� r� r$ r$ r% r' Z s*
z2ConstructedPayloadDecoderBase.indefLenValueDecoder)NNN)NNNNN)NNNNN)
r* r+ r, r� r� r r� r� r& r' r$ r$ r$ r% r| A s
2
\ r| c @ s e Zd Ze�� Ze�� ZdS )�"SequenceOrSequenceOfPayloadDecoderN)r* r+ r, r r� r� r� r� r$ r$ r$ r% r� E s r� c @ s e Zd Ze�� ZdS )�SequencePayloadDecoderN)r* r+ r, r r� r- r$ r$ r$ r% r� J s r� c @ s e Zd Ze�� ZdS )�SequenceOfPayloadDecoderN)r* r+ r, r r� r- r$ r$ r$ r% r� N s r� c @ s e Zd Ze�� Ze�� ZdS )�SetOrSetOfPayloadDecoderN)r* r+ r, r r� r� r� r� r$ r$ r$ r% r� R s r� c @ s e Zd Ze�� ZdS )�SetPayloadDecoderN)r* r+ r, r r� r- r$ r$ r$ r% r� W s r� c @ s e Zd Ze�� ZdS )�SetOfPayloadDecoderN)r* r+ r, r r� r- r$ r$ r$ r% r� [ s r� c @ s( e Zd Ze�� Zddd�Zddd�ZdS )�ChoicePayloadDecoderNc
k s |d kr| j j|d�} n|�� } |rFx || |||�D ]
}
|
V q4W d S | �| |�}| j|kr�trntd|f � xl||| jf|�D ]} t| t�r�| V q�W nBtr�td|f � x.||| j|||f|�D ]} t| t�r�| V q�W | j} tr�td| | f � | j | | ddddd� | V d S )N)r z'decoding %s as explicitly tagged CHOICEz decoding %s as untagged CHOICEz*decoded component %s, effective tag set %sF)r� r� r� � innerFlag)
r- r3 r) r rO ZcomponentTagMapr@ rA r� �setComponentByType)
r r r r r r r! r" r# r( r0 rT r� r$ r$ r% r& b s>
z!ChoicePayloadDecoder.valueDecoderc k s: |d kr| j j|d�} n|�� } |rFx || |||�D ]
}
|
V q4W d S | �| |�}| j|k} trxtd|| rndppdf � x�| r�||| jjft|dd��} n ||| jj|||ft|dd��} x^| D ]V}
t|
t �r�|
V |
t
j kr�P |
j }tr�td|
|f � | j
||
ddddd � | s�P q�W | �r*|
t
j krzP qzW | V d S )
N)r z decoding %s as %stagged CHOICEz explicitly ZunT)r9 z*decoded component %s, effective tag set %sF)r� r� r� r� )r- r3 r) r rO r� r� r� r@ rA r r: r� r� )r r r r r r r! r" r# r( r0 �isTagged�iteratorrT r� r$ r$ r% r' � sN
z)ChoicePayloadDecoder.indefLenValueDecoder)NNNNN)NNNNN)r* r+ r, r �Choicer- r& r' r$ r$ r$ r% r� _ s
2 r� c @ s( e Zd Ze�� Zddd�Zddd�ZdS )�AnyPayloadDecoderNc
k s |d krd} n"|j tjkr&||jk} n
||jk} | r�|j}
|�� } |�|
tj � || |
7 }t
r�x"t ||�D ]} t | t
�rl| V qlW t
dt�| � � |r�x*|| j||tf|�|||�D ]
} | V q�W d S x$t|||�D ]} t | t
�r�| V q�W | j||| f|�V d S )NTz&decoding as untagged ANY, substrate %s)rN r �TagMap�tagMapr �markedPositionrP �seek�os�SEEK_SETrO r r@ rA r r� r6 r4 r )
r r r r r r r! r" r# Z
isUntagged� fullPosition�currentPositionr0 r$ r$ r% r& � s4
z AnyPayloadDecoder.valueDecoderc k s� |d krd} n"|j tjkr&||jk} n
||jk} | rFt}
tr�td� n\|j} |�� } |� | t
j � x(t || | |�D ]}
t
|
t�rt|
V qtW tr�tdt�|
� � | j}|r�|| jk r�| j||tf|�}
x(||
|
| |t|
� |�D ]
}
|
V q�W d S t�rtd� | j}x`xB|||f|dd�|��D ]&}t
|t��r<|V |tjk�r&P �q&W |tjk�r^P |
|7 }
�q W |�rz|
V n| j|||
f|�V d S )NFzdecoding as tagged ANYz-decoding as untagged ANY, header substrate %sz$assembling constructed serializationT)r" r9 )rN r r� r� r r rO r� rP r� r� r� r r@ rA r r� r- r1 r6 r4 rd r r: )r r r r r r r! r" r# r� r0 r� r� r( rT r$ r$ r% r' sT
z&AnyPayloadDecoder.indefLenValueDecoder)NNNNN)NNNNN)r* r+ r, r r; r- r& r' r$ r$ r$ r% r� � s
( r� c @ s e Zd Ze�� ZdS )�UTF8StringPayloadDecoderN)r* r+ r, r �
UTF8Stringr- r$ r$ r$ r% r� P s r� c @ s e Zd Ze�� ZdS )�NumericStringPayloadDecoderN)r* r+ r, r �
NumericStringr- r$ r$ r$ r% r� T s r� c @ s e Zd Ze�� ZdS )� PrintableStringPayloadDecoderN)r* r+ r, r �PrintableStringr- r$ r$ r$ r% r� X s r� c @ s e Zd Ze�� ZdS )�TeletexStringPayloadDecoderN)r* r+ r, r �
TeletexStringr- r$ r$ r$ r% r� \ s r� c @ s e Zd Ze�� ZdS )� VideotexStringPayloadDecoderN)r* r+ r, r �VideotexStringr- r$ r$ r$ r% r� ` s r� c @ s e Zd Ze�� ZdS )�IA5StringPayloadDecoderN)r* r+ r, r � IA5Stringr- r$ r$ r$ r% r� d s r� c @ s e Zd Ze�� ZdS )�GraphicStringPayloadDecoderN)r* r+ r, r �
GraphicStringr- r$ r$ r$ r% r� h s r� c @ s e Zd Ze�� ZdS )�VisibleStringPayloadDecoderN)r* r+ r, r �
VisibleStringr- r$ r$ r$ r% r� l s r� c @ s e Zd Ze�� ZdS )�GeneralStringPayloadDecoderN)r* r+ r, r �
GeneralStringr- r$ r$ r$ r% r� p s r� c @ s e Zd Ze�� ZdS )� UniversalStringPayloadDecoderN)r* r+ r, r �UniversalStringr- r$ r$ r$ r% r� t s r� c @ s e Zd Ze�� ZdS )�BMPStringPayloadDecoderN)r* r+ r, r � BMPStringr- r$ r$ r$ r% r� x s r� c @ s e Zd Ze�� ZdS )� ObjectDescriptorPayloadDecoderN)r* r+ r, r �ObjectDescriptorr- r$ r$ r$ r% r� } s r� c @ s e Zd Ze�� ZdS )� GeneralizedTimePayloadDecoderN)r* r+ r, r �GeneralizedTimer- r$ r$ r$ r% r� � s r� c @ s e Zd Ze�� ZdS )�UTCTimePayloadDecoderN)r* r+ r, r �UTCTimer- r$ r$ r$ r% r� � s r� c C s g | ]}|�qS r$ r$ )�.0�xr$ r$ r% �
<listcomp>� s r� rp )r r c @ sF e Zd ZeZe� ZdZeZe Z e
e
fdd�Z ddde ddfdd�Z
dS )�SingleItemDecoderTc K s8 |t k r |n| j| _|t k r |n| j| _i | _i | _d S )N)r �TAG_MAP�_tagMap�TYPE_MAP�_typeMap� _tagCache� _tagSetCache)r r� �typeMapZignoredr$ r$ r% �__init__� s zSingleItemDecoder.__init__Nc # k s� |� dd�} tr&tdtj|||f � | r�| jr�x$t|d|�D ]}
t|
t�r>|
V q>W |
tkrvtrjtd� t j
V d S |� dt j
� | j} | j} | j}
| j}t}|�� |_�x�|tk �r�|tk�rPd}x$t|d|�D ]}t|t�r�|V q�W t|�}y |
| }W n� tk
�r� |}|d @ }|d
@ }|d @ }|d k�r�d}d }d }xpx(t|d|�D ]}t|t��rT|V �qTW |�s�t�d
��t|�}|d7 }|dK }||d@ O }|d@ �sFP �qFW tj|||d�}|�r�||
|< Y nX |d k�r2|�r$y || }W n* tk
�r t� d|�}|||< Y nX n t� d|�}n|| }t }t�rPtd| � |t k�rnx(t|d|�D ]}t|t��rh|V �qhW t|�}|dk �r�|}n�|dk�r8|d@ }x(t|||�D ]}t|t��r�|V �q�W t |�}t |�|k�rt�d|t |�|f ��d }x"|D ]} |dK }|t | �O }�qW |d7 }nd}|dk�rX| j�sXt�!d��t"}t�rntd| � |t"k�r�|d k�r�t#}nt$}|t#k�rby | | } W n tk
�r� d } Y nX | �r�t%}n@y| |d d� } W n tk
�r� d } Y nX | �rt%}nt&}t�rbtd| �r$| j'j(�p&d|t%k�r6d�p8d f � tj�)| d k�rVd �p^| j*j'j(� |t$k�r�|j't+j,k�rBy || } W n tk
�r� d } Y nX t�rztd � x,|j-�.� D ] \}} td || j'j(f � �q�W |j/�r td � x,|j/�.� D ] \}} td || j'j(f � �q�W td!| d k�r0d�p6| �0� |f � n8||j1k�sZ||j2k�rv|} t�rztd"|j'j( � nd } | d k �r&y$| | j3 } t�r�td#| j3f � W nf tk
�r t� | j1j4| j1j4�} y | | } t�r�td$| f � W n tk
�r d } Y nX Y nX | �r | }t%}nt&}nd } t&}t�r�td%|t%k�rL| j'j(�pNd|t%k�r^d�p`d f � tj�)| d k�r~d �p�| j'j(� |t%k�r�|�5d&d��s�|�s�d'd(� }|�� }!|dk�r�x�| j6||||t"| |f|�D ]}t|t��r�|V �q�W n�x6| j7||||t"| |f|�D ]}t|t��r|V �qW |�� |! }"|�s`|"|k�r`t!d)|"|f ��n |�r�|"|k�r�t!d*|"|f ��t�r�td+| j'j(|j'j(t|t8j9��r�|�:� �p�|f � t}P |t&k�r<|�r�|d j;tj<k�r�|d j=tj>k�r�t?} t%}n
d } | j@}t�r<td,| �r | j'j(�p"d|t%k�r2d�p4d-f � |tAk�rf| jB} t�rbtd.| j'j( � t%}|tCkr�t�!d/||f ��q�W t�r�tj� � td0tj � |V d S )1Nr9 FzWdecoder called at scope %s with state %d, working with up to %s octets of substrate: %src z end-of-octets sentinel found���TrD ro � � r z'Short octet stream on long tag decodingrG r^ r] )�tagClassr> �tagIdr$ z$tag decoded into %s, decoding lengthz %s<%s at %srl rk z6Indefinite length encoding not supported by this codecz value length decoded into %dz/codec %s chosen by a built-in type, decoding %sz<none>r5 zas explicit tag�?z!candidate ASN.1 spec is a map of:z
%s -> %szbut neither of: z,new candidate ASN.1 spec is %s, chosen by %szcandidate ASN.1 spec is %sz8value decoder chosen for an ambiguous type by type ID %sz value decoder chosen by base %sz*codec %s chosen by ASN.1 spec, decoding %sZ
recursiveFlagc s s
| V dS )ae Legacy hack to keep the recursiveFlag=False option supported.
The decode(..., substrateFun=userCallback) option was introduced in 0.1.4 as a generalization
of the old recursiveFlag=False option. Users should pass their callback instead of using
recursiveFlag.
Nr$ )r( �
_substrate�_length�_optionsr$ r$ r% r" � s z0SingleItemDecoder.__call__.<locals>.substrateFunz%Read %s bytes instead of expected %s.z(Read %s bytes are more than expected %s.z&codec %s yields type %s, value:
%s
...z codec %s chosen, decoding %sz
as failurez codec %s chosen, decoding valuez%s not in asn1Spec: %rz%decoder left scope %s, call completed)D�poprO r Zscope�supportIndefLengthr r@ rA � EOO_SENTINELr r: r� r� �SEEK_CURr� r� r� r� r4 rP r� �stStop� stDecodeTagrK r� r r �Tagr� �stDecodeLength�listrd r r �stGetValueDecoder�stGetValueDecoderByTag�stGetValueDecoderByAsn1Spec�
stDecodeValue�stTryAsExplicitTagrN r* �pushr- r r� Z presentTypesr� Z skipTypesZprettyPrintTyper r� r� r� r2 r' r& r ZAsn1ItemZ prettyPrintr> r� r� ZtagClassUniversal�rawPayloadDecoder�defaultErrorState�stDumpRawValue�defaultRawDecoder�stErrorCondition)#r r r r r r r! r" r# r9 Z
eoo_candidater� r� ZtagCacheZ tagSetCacher5 Z
isShortTagZ firstByteZ
firstOctetZlastTagZ
integerTagr� r> r ZlengthOctetIdxZ integerByte�sizeZ
encodedLengthZ lengthOctetZconcreteDecoderZ
chosenSpecr� Z
baseTagSetrX Z bytesReadr$ r$ r% �__call__� s�
0"
&
4
"
0
zSingleItemDecoder.__call__)r* r+ r, r r r� r r r� r� r r� r
r r$ r$ r$ r% r� � s r� c @ s& e Zd ZdZeZddd�Zdd� ZdS )r a� Create an iterator that turns BER/CER/DER byte stream into ASN.1 objects.
On each iteration, consume whatever BER/CER/DER serialization is
available in the `substrate` stream-like object and turns it into
one or more, possibly nested, ASN.1 objects.
Parameters
----------
substrate: :py:class:`file`, :py:class:`io.BytesIO`
BER/CER/DER serialization in form of a byte stream
Keyword Args
------------
asn1Spec: :py:class:`~pyasn1.type.base.PyAsn1Item`
A pyasn1 type object to act as a template guiding the decoder.
Depending on the ASN.1 structure being decoded, `asn1Spec` may
or may not be required. One of the reasons why `asn1Spec` may
me required is that ASN.1 structure is encoded in the *IMPLICIT*
tagging mode.
Yields
------
: :py:class:`~pyasn1.type.base.PyAsn1Item`, :py:class:`~pyasn1.error.SubstrateUnderrunError`
Decoded ASN.1 object (possibly, nested) or
:py:class:`~pyasn1.error.SubstrateUnderrunError` object indicating
insufficient BER/CER/DER serialization on input to fully recover ASN.1
objects from it.
In the latter case the caller is advised to ensure some more data in
the input stream, then call the iterator again. The decoder will resume
the decoding process using the newly arrived data.
The `context` property of :py:class:`~pyasn1.error.SubstrateUnderrunError`
object might hold a reference to the partially populated ASN.1 object
being reconstructed.
Raises
------
~pyasn1.error.PyAsn1Error, ~pyasn1.error.EndOfStreamError
`PyAsn1Error` on deserialization error, `EndOfStreamError` on
premature stream closure.
Examples
--------
Decode BER serialisation without ASN.1 schema
.. code-block:: pycon
>>> stream = io.BytesIO(
... b'0 ')
>>>
>>> for asn1Object in StreamingDecoder(stream):
... print(asn1Object)
>>>
SequenceOf:
1 2 3
Decode BER serialisation with ASN.1 schema
.. code-block:: pycon
>>> stream = io.BytesIO(
... b'0 ')
>>>
>>> schema = SequenceOf(componentType=Integer())
>>>
>>> decoder = StreamingDecoder(stream, asn1Spec=schema)
>>> for asn1Object in decoder:
... print(asn1Object)
>>>
SequenceOf:
1 2 3
Nc K s( | j f |�| _t|�| _|| _|| _d S )N)�SINGLE_ITEM_DECODER�_singleItemDecoderr r � _asn1Specr )r r r r# r$ r$ r% r� � s
zStreamingDecoder.__init__c c s\ xVx$| j | j| jf| j�D ]
}|V qW x$t| j�D ]}t|t�rHd V P q4W |rP qW d S )N)r r r r r r@ rA )r r( r0 r$ r$ r% �__iter__� s
zStreamingDecoder.__iter__)N)r* r+ r, �__doc__r� r r� r r$ r$ r$ r% r 7 s I
c @ s. e Zd ZdZeZeddd��Zedd� �Z dS )r znCreate a BER decoder object.
Parse BER/CER/DER octet-stream into one, possibly nested, ASN.1 object.
Nc s� t |�}d|kr.|d � d� fdd� }||d< | j||f|�}xR|D ]J}t|t�r\t�d��ytt|��}W n tjk
r� t}Y nX ||fS W dS )a�
Turns BER/CER/DER octet stream into an ASN.1 object.
Takes BER/CER/DER octet-stream in form of :py:class:`bytes` (Python 3)
or :py:class:`str` (Python 2) and decode it into an ASN.1 object
(e.g. :py:class:`~pyasn1.type.base.PyAsn1Item` derivative) which
may be a scalar or an arbitrary nested structure.
Parameters
----------
substrate: :py:class:`bytes` (Python 3) or :py:class:`str` (Python 2)
BER/CER/DER octet-stream to parse
Keyword Args
------------
asn1Spec: :py:class:`~pyasn1.type.base.PyAsn1Item`
A pyasn1 type object (:py:class:`~pyasn1.type.base.PyAsn1Item`
derivative) to act as a template guiding the decoder.
Depending on the ASN.1 structure being decoded, `asn1Spec` may or
may not be required. Most common reason for it to require is that
ASN.1 structure is encoded in *IMPLICIT* tagging mode.
substrateFun: :py:class:`Union[
Callable[[pyasn1.type.base.PyAsn1Item, bytes, int],
Tuple[pyasn1.type.base.PyAsn1Item, bytes]],
Callable[[pyasn1.type.base.PyAsn1Item, io.BytesIO, int, dict],
Generator[Union[pyasn1.type.base.PyAsn1Item,
pyasn1.error.SubstrateUnderrunError],
None, None]]
]`
User callback meant to generalize special use cases like non-recursive or
partial decoding. A 3-arg non-streaming variant is supported for backwards
compatiblilty in addition to the newer 4-arg streaming variant.
The callback will receive the uninitialized object recovered from substrate
as 1st argument, the uninterpreted payload as 2nd argument, and the length
of the uninterpreted payload as 3rd argument. The streaming variant will
additionally receive the decode(..., **options) kwargs as 4th argument.
The non-streaming variant shall return an object that will be propagated
as decode() return value as 1st item, and the remainig payload for further
decode passes as 2nd item.
The streaming variant shall yield an object that will be propagated as
decode() return value, and leave the remaining payload in the stream.
Returns
-------
: :py:class:`tuple`
A tuple of :py:class:`~pyasn1.type.base.PyAsn1Item` object
recovered from BER/CER/DER substrate and the unprocessed trailing
portion of the `substrate` (may be empty)
Raises
------
: :py:class:`~pyasn1.error.PyAsn1Error`
:py:class:`~pyasn1.error.SubstrateUnderrunError` on insufficient
input or :py:class:`~pyasn1.error.PyAsn1Error` on decoding error.
Examples
--------
Decode BER/CER/DER serialisation without ASN.1 schema
.. code-block:: pycon
>>> s, unprocessed = decode(b'0 ')
>>> str(s)
SequenceOf:
1 2 3
Decode BER/CER/DER serialisation with ASN.1 schema
.. code-block:: pycon
>>> seq = SequenceOf(componentType=Integer())
>>> s, unprocessed = decode(
b'0 ', asn1Spec=seq)
>>> str(s)
SequenceOf:
1 2 3
r" Nc 3 sv y� | |||�}W nJ t k
r\ t�� \}}}|jr8� t|tj�sHt�t� � | ||�}Y nX x|D ]
}|V qdW dS )ar Support both 0.4 and 0.5 style APIs.
substrateFun API has changed in 0.5 for use with streaming decoders. To stay backwards compatible,
we first try if we received a streaming user callback. If that fails,we assume we've received a
non-streaming v0.4 user callback and convert it for streaming on the fly
N)
� TypeError�sys�exc_info�tb_nextr@ �io�BytesIO�AssertionErrorr �_callSubstrateFunV4asV5) r( r r r# Z
substrate_gen�_type�_value� tracebackr5 )�origSubstrateFunr$ r% �substrateFunWrapper� s
z-Decoder.__call__.<locals>.substrateFunWrapperzShort substrate on input)N) r �STREAMING_DECODERr@ rA r �nextr ZEndOfStreamErrorr )�clsr r r# r, ZstreamingDecoderr( �tailr$ )r+ r% r � s P
zDecoder.__call__c c sT |� � }|dkrt|�}| |||�\}}|�|�}|�� |�| tj� |V d S )Nrk )�readrd �write�truncater� r� r )ZsubstrateFunV4r( r r Zsubstrate_bytesr5 Z
nextSubstrate�nbytesr$ r$ r% r' s
z Decoder._callSubstrateFunV4asV5)N)
r* r+ r, r r r- � classmethodr r. r' r$ r$ r$ r% r � s
y)�r$ r� r! Zpyasn1r r Zpyasn1.codec.berr Zpyasn1.codec.streamingr r r r Z
pyasn1.compatr Zpyasn1.compat.integerr
Zpyasn1.compat.octetsr r r
r Z pyasn1.errorr Z pyasn1.typer r r r r r �__all__ZregisterLoggeer* Z
DEBUG_DECODERrO r4 rA �objectr r/ r7 r r< rC rF rV rZ r\ rh r{ r| r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rB r rE rU rY r[ rg Z
Enumeratedrz r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r; r� r� r� �valuesZ typeDecoderr- rN �ranger
r r r r r r r r r r r� r r r r$ r$ r$ r% �<module> s�
) V<n xy
& mc D