File: //usr/lib/python3.6/site-packages/pyasn1/codec/ber/__pycache__/encoder.cpython-36.pyc
3
u?�Y�G � 8 @ st d dl mZmZmZmZ d dlmZ d dlmZm Z m
Z
m Z m Z d dl
mZ d dlmZmZ dgZG dd� de�ZG d d
� d
e�ZG d d � d e�ZG d
d� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Z G dd� de�Z G dd � d e�Z G d d � d e�Z G d d � d e�Z ej!j"e� ej#j"e� ej$j"e� ej%j"e� ej&j"e� ej'j"e� ej(j"e� ej)j"e� ej*j"e � ej+j"e � ej,j"e � ej-j"e � ej.j"e� ej/j"e� ej0j"e� ej1j"e� ej2j"e� ej3j"e� ej4j"e� ej5j"e� ej6j"e� ej7j"e� ej8j"e� ej9j"e� ej:j"e� ej;j"e� iZ<ej#j=e� ej$j=e� ej%j=e� ej&j=e� ej'j=e� ej(j=e� ej)j=e� ej*j=e � ej>j=e � ej,j=e � ej?j=e � ej+j=e � ej-j=e � ej@j=e � ej.j=e� ej/j=e� ej0j=e� ej1j=e� ej2j=e� ej3j=e� ej4j=e� ej5j=e� ej6j=e� ej7j=e� ej8j=e� ej9j=e� ej:j=e� ej;j=e� i ZAG d!d"� d"e�ZBeBe<eA�ZCd#S )$� )�tag�univ�char�useful)�eoo)�int2oct�oct2int� ints2octs�null�str2octs)�to_bytes)�debug�error�encodec @ s< e Zd ZdZd Zee�Zdd� Zdd� Zdd� Z d d
� Z
d S )
�AbstractItemEncoder� r c C sz |\}}}||B }|r |t jO }|dk r2||B fS |d@ f}|dL }x"|rfd|d@ B f| }|dL }qFW |dB f| S d S )N� � � � )r ZtagFormatConstructed)�self� singleTag�
isConstructedZtagClassZ tagFormatZtagIdZ
encodedTag� substrate� r � /usr/lib/python3.6/encoder.py� encodeTag s
z AbstractItemEncoder.encodeTagc C sr | r| j rdS |dk r |fS f }x |r@|d@ f| }|dL }q$W t|�}|dkr`tjd| ��d|B f| S d S )Nr � � �~ zLength octets overflow (%d))r )�supportIndefLenMode�lenr � PyAsn1Error)r Zlength�defModer Z substrateLenr r r � encodeLength' s z AbstractItemEncoder.encodeLengthc K s t jd��d S )NzNot implemented)r r" )r �value� encodeFun�optionsr r r � encodeValue6 s z AbstractItemEncoder.encodeValuec
K s� |j }|s$| j||f|�\}}}|S |jdd�}x�t|j�D ]�\} }
|} | s�| j||f|�\}}}|jdd�rx| rx|S |s�d} | j|
|�} | | jt|�| �7 } |r�t| �| }| s�|| j 7 }q<| | }| s<|| j
7 }q<W |s�t|�}|S )Nr# TZ
ifNotEmptyF) �tagSetr( �get� enumerateZ superTagsr r$ r! r �eooOctetsSubstrate�eooIntegerSubstrate)
r r% r& r' r) r r ZisOctetsr# �idxr ZdefModeOverride�headerr r r r 9 s6 zAbstractItemEncoder.encodeN)r r ) �__name__�
__module__� __qualname__r r- r r, r r$ r( r r r r r r s r c @ s e Zd Zdd� ZdS )�EndOfOctetsEncoderc K s
t ddfS )NFT)r
)r r% r& r' r r r r( m s z EndOfOctetsEncoder.encodeValueN)r0 r1 r2 r( r r r r r3 l s r3 c @ s e Zd ZdZdd� ZdS )�BooleanEncoderFc K s |rdp
dddfS )Nr r F)r )r r )r r% r& r' r r r r( t s zBooleanEncoder.encodeValueN)r0 r1 r2 r r( r r r r r4 q s r4 c @ s e Zd ZdZdZdd� ZdS )�IntegerEncoderFc K s2 |dkr | j rf ddfS dS tt|�dd�ddfS )Nr FT)Zsigned�r )r6 FF)�supportCompactZeror �int)r r% r& r' r r r r( | s
zIntegerEncoder.encodeValueN)r0 r1 r2 r r7 r( r r r r r5 x s r5 c @ s e Zd Zdd� ZdS )�BitStringEncoderc
K s� t |�}|d r"|d|d > }n|}|jdd�}| sHt |�|d krn|j� }tt |�d | �| ddfS |jtj|jj|jj�d�}d}t }x:||k r�|} t
| |d |�}|||| |� f|�7 }q�W |ddfS )Nr � maxChunkSizer FT)r) ) r! r* �asOctetsr �cloner �TagSetr) �baseTagr
�min)
r r% r& r' Z valueLengthZ alignedValuer: r �stop�startr r r r( � s"
z BitStringEncoder.encodeValueN)r0 r1 r2 r( r r r r r9 � s r9 c @ s e Zd Zdd� ZdS )�OctetStringEncoderc K s� |j dd�}| s t|�|kr,|j� ddfS tj|jj|jj�}d}t}x<|j|||| � |d�}|sjP |||f|�7 }||7 }qJW |ddfS d S )Nr: r FT)r) ) r* r! r; r r= r) r> r
r< ) r r% r& r' r: �
baseTagSet�posr �chunkr r r r( � s z OctetStringEncoder.encodeValueN)r0 r1 r2 r( r r r r rB � s rB c @ s e Zd ZdZdd� ZdS )� NullEncoderFc K s
t ddfS )NFT)r
)r r% r& r' r r r r( � s zNullEncoder.encodeValueN)r0 r1 r2 r r( r r r r rF � s rF c @ s e Zd ZdZdd� ZdS )�ObjectIdentifierEncoderFc
K s� |j � }y|d }|d }W n$ tk
r@ tjd|f ��Y nX d| koTdkn r�|dkrz|d f|dd � }q�|dkr�|f|dd � }q�|dkr�|d f|dd � }q�tjd|f ��n0|dkr�|d f|dd � }ntjd|f ��f }x�|D ]�}d| k�od kn �r.||f7 }n`|d k�r||d @ f} |d
L }x&|�rpd |d @ B f| } |d
L }�qLW || 7 }ntjd ||f ���qW |d
d
fS )Nr r z Short OID %s�' �( � �P z"Impossible first/second arcs at %sr r r zNegative OID arc %s at %sF)ZasTuple�
IndexErrorr r" )
r r% r& r' Zoid�first�secondZoctetsZsubOid�resr r r r( � s<
z#ObjectIdentifierEncoder.encodeValueN)r0 r1 r2 r r( r r r r rG � s rG c @ s0 e Zd ZdZdZedd� �Zdd� Zdd� Zd S )
� RealEncoderr rJ c C s� d\}}| dk rd }|dk r d
}| |9 } |dkrZ| dt |�d | 9 } t |�d | }n0|dkr�| dt |�d | 9 } t |�d | }x$t| �| kr�| |9 } |d8 }q�P q�W |t| �||fS ) Nr r r rJ � � � )r r ���rT )�absr8 )�m�encbase�e�msZesr r r �_dropFloatingPoint� s& z RealEncoder._dropFloatingPointc C s |\}}}dddg}|j |kr.| j||j |�S | j |krH| j|| j |�S |||g}|||g}d}d} td�}x�td�D ]�}
| j||
||
||
�\}||
< ||
< ||
< t||
�t|�k s�t||
�t|�krv||
|k rv||
}t||
�}||
} qvW ||| |fS )NrJ r rR r �infrQ )�
binEncBaserZ �float�rangerU r8 ) r r% rV �brX ZencBaseZmantissaZ exponentaZsignrW �ir r r �_chooseEncBase s$
.4 zRealEncoder._chooseEncBasec K s� |j r
dS |jrdS |\}}}|s,tddfS |dkrXtd||dkrFdpHd |f �ddfS |d
k�r�d }| j|�\}}} }|dk r�|dO }| d
kr�x�|d @ dkr�|d L }|d 7 }q�W nb| d
kr�x |d@ dkr�|dL }|d 7 }q�W |dO }n.x$|d@ dk�r |dL }|d 7 }� q�W |dO }d}
x$|d @ dk�r>|d L }|
d 7 }
�q W |
dk�rTtjd��||
d
> O }t} |dk�sx|d k�r�t|d@ �} n�x(|d k�r�t|d@ �| } |d
L }�q�W |dk�r�| �r�t| d �d @ �r�td�| } |d k�r| �rt| d �d @ �rtd�| } t | �} | dk�r*tjd��| d k�r6n@| d
k�rJ|d O }n,| dk�r^|d
O }n|dO }t| d@ �| } t}
x$|�r�t|d@ �|
}
|d
L }�q|W t|�| |
}|ddfS tjd| ��d S ) N�@ F�A T�
z%dE%s%dr �+� rJ r r r r rQ rR � rS � zScale factor overflowr zReal exponent overflowzProhibited Real base %s�rb )ri FF�rc )rj FFrT rT )r rT rT )
Z isPlusInfZ
isMinusInfr
r ra r r" r r r! )r r% r& r' rV r_ rX ZforY rW ZsfZeo�nZpor r r r r( & s|
$
" $
zRealEncoder.encodeValueN) r0 r1 r2 r r\ � staticmethodrZ ra r( r r r r rP � s
rP c @ s e Zd Zdd� ZdS )�SequenceEncoderc K s� |j � |j}t}t|�}xb|dkr||d8 }|rf|| jrH|| j rHq || jrf|| || jkrfq ||| f|�| }q W |ddfS )Nr r T)�verifySizeSpecZ
componentTyper
r! Z
isOptionalZisValueZ isDefaultedZ
asn1Object)r r% r& r' Z
namedTypesr r. r r r r( m s
zSequenceEncoder.encodeValueN)r0 r1 r2 r( r r r r rm l s rm c @ s e Zd Zdd� ZdS )�SequenceOfEncoderc K sH |j � t}t|�}x(|dkr<|d8 }||| f|�| }qW |ddfS )Nr r T)rn r
r! )r r% r& r' r r. r r r r( � s
z SequenceOfEncoder.encodeValueN)r0 r1 r2 r( r r r r ro � s ro c @ s e Zd Zdd� ZdS )�
ChoiceEncoderc K s ||j � f|�ddfS )NT)Z getComponent)r r% r& r' r r r r( � s zChoiceEncoder.encodeValueN)r0 r1 r2 r( r r r r rp � s rp c @ s e Zd Zdd� ZdS )�
AnyEncoderc K s |j � |jdd� dfS )Nr# T)r; r* )r r% r& r' r r r r( � s zAnyEncoder.encodeValueN)r0 r1 r2 r( r r r r rq � s rq c @ s( e Zd ZdZdZi fdd�Zdd� ZdS )�EncoderNc C s || _ || _d S )N)�_Encoder__tagMap�_Encoder__typeMap)r �tagMap�typeMapr r r �__init__� s zEncoder.__init__c K sF t jt j@ rt j}nd }|rR|d|jdd� r2dp4d|jdd�|j� |j� f � | jd k rj|j| jd� | jd k r�|j| jd � |j }y| j
|j }W n\ t k
r� t
j|j j|j j�}y| j| }W n$ t k
r� tjd
|f ��Y nX Y nX |�r|d |jj|f � |j|| f|�}|�rB|d |t|�t j|�f � |S )
NzBencoder called in %sdef mode, chunk size %s for type %s, value:
%sr# T�inrf r: r )r# )r: zNo encoder for %sz!using value codec %s chosen by %sz;codec %s built %s octets of substrate: %s
encoder completed)r
�loggerZ flagEncoderr* ZprettyPrintTypeZ prettyPrint�fixedDefLengthMode�update�fixedChunkSizer) rt �typeId�KeyErrorr r= r> rs r r" � __class__r0 r r! Zhexdump)r r% r' ry r) ZconcreteEncoderrC r r r r �__call__� s0 6
zEncoder.__call__)r0 r1 r2 rz r| rw r� r r r r rr � s rr N)DZ pyasn1.typer r r r Zpyasn1.codec.berr Zpyasn1.compat.octetsr r r r
r Zpyasn1.compat.integerr Zpyasn1r
r �__all__�objectr r3 r4 r5 r9 rB rF rG rP rm ro rp rq Z endOfOctetsr) ZBooleanZIntegerZ BitStringZ OctetStringZNullZObjectIdentifierZ
EnumeratedZRealZ
SequenceOfZSetOfZChoiceZ
UTF8StringZ
NumericStringZPrintableStringZ
TeletexStringZVideotexStringZ IA5StringZ
GraphicStringZ
VisibleStringZ
GeneralStringZUniversalStringZ BMPStringZObjectDescriptorZGeneralizedTimeZUTCTimeru r} �Set�SequenceZAnyrv rr r r r r r �<module> s� \ 2y
K