File: //lib/python3/dist-packages/pyasn1/codec/ber/__pycache__/encoder.cpython-38.pyc
U
�Z�\ � 8 @ s� d dl mZ d dl mZ d dlmZ d dlmZ d dlmZm Z m
Z
m Z m Z m
Z
d dlmZ d dlmZ d dlmZ d d lmZ d
gZG d d � d e�ZG d
d� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Z G dd � 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!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� 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 � ejAj>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� ej<j>e� i ZBG d%d&� d&e�ZCeCe=eB�ZDd'S )(� )�debug)�error)�eoo)�to_bytes)�int2oct�oct2int� ints2octs�null�str2octs� isOctetsType)�char)�tag)�univ)�useful�encodec @ s> e Zd ZdZdZee�Zdd� Zdd� Zdd� Z d d
d �Z
d S )
�AbstractItemEncoderT)r r c C sv |\}}}||B }|r |t jO }|dk r2||B fS |d@ f}|dL }|rdd|d@ B f| }|dL }qD|dB f| S d S )N� � � � )r
ZtagFormatConstructed)�self� singleTag�
isConstructedZtagClassZ tagFormatZtagIdZ
encodedTag� substrate� r �:/usr/lib/python3/dist-packages/pyasn1/codec/ber/encoder.py� encodeTag s
z AbstractItemEncoder.encodeTagc C sl |s| j rdS |dk r |fS d}|r<|d@ f| }|dL }q t|�}|dkrZt�d| ��d|B f| S d S )N)r r r � � �~ zLength octets overflow (%d))�supportIndefLenMode�lenr � PyAsn1Error)r Zlength�defModer Z substrateLenr r r � encodeLength, s
z AbstractItemEncoder.encodeLengthc K s t �d��d S )NzNot implemented)r r" �r �value�asn1Spec� encodeFun�optionsr r r � encodeValue; s z AbstractItemEncoder.encodeValueNc K s |d kr|j }n|j }|s6| j|||f|�\}}}|S |�dd�} t|j�D ]�\}
} | } |
s�| j|||f|�\}}}|s�|r�|�dd�r�| S |s�d} | �| |�}
|
| �t|�| �7 }
|r�t|
�| }| s�|| j 7 }qL|
| }| sL|| j
7 }qL|s�t|�}|S )Nr# T�
ifNotEmptyF) �tagSetr* �get� enumerateZ superTagsr r$ r! r �eooOctetsSubstrate�eooIntegerSubstrate)r r& r' r( r) r, r r ZisOctetsr# �idxr ZdefModeOverride�headerr r r r > sN ��
��
zAbstractItemEncoder.encode)NN) �__name__�
__module__� __qualname__r r0 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* u s z EndOfOctetsEncoder.encodeValueN�r3 r4 r5 r* r r r r r6 t s r6 c @ s e Zd ZdZdd� ZdS )�BooleanEncoderFc K s |rdp
dddfS )N)� �r Fr r% r r r r* | s zBooleanEncoder.encodeValueN�r3 r4 r5 r r* r r r r r: y s r: c @ s e Zd ZdZdZdd� ZdS )�IntegerEncoderFc K s, |dkr| j rdS dS tt|�dd�ddfS )Nr )r FF)r< FFT)ZsignedF)�supportCompactZeror �intr% r r r r* � s
zIntegerEncoder.encodeValueN)r3 r4 r5 r r? r* r r r r r> � s r> c @ s e Zd Zdd� ZdS )�BitStringEncoderc
K s� |d k r|� |�}t|�}|d r4|d|d > }n|}|�dd�}|rXt|�|d kr~|�� }tt|�d | �| ddfS |jj} | r�t�| | �}
nt�� }
|j |
d�}d} t }| |k r�| } t
| |d |�} |||| | � |f|�7 }q�|ddfS )Nr � maxChunkSizer FT�r, ) �cloner! r- �asOctetsr r, �baseTagr
�TagSetr �min)
r r& r' r( r) Z valueLengthZ alignedValuerB r rF r, �stop�startr r r r* � s,
z BitStringEncoder.encodeValueNr9 r r r r rA � s rA c @ s e Zd Zdd� ZdS )�OctetStringEncoderc K s
|d kr|� � }n t|�s*|�|�� � }n|}|�dd�}|rJt|�|krT|ddfS |d kr�|jj}|rvt�||�}nt�� }|j|d�}n6t|�s�|jj}|r�t�||�}nt�� }|j|d�}d} t }|| | | � }
|
s�q�|||
|f|�7 }| |7 } q�|ddfS d S )NrB r FTrC )
rE r rD r- r! r, rF r
rG r ) r r& r' r( r) r rB rF r, �pos�chunkr r r r* � s8
z OctetStringEncoder.encodeValueNr9 r r r r rK � s rK c @ s e Zd ZdZdd� ZdS )� NullEncoderFc K s
t ddfS r7 r8 r% r r r r* � s zNullEncoder.encodeValueNr= r r r r rN � s rN c @ s e Zd ZdZdd� ZdS )�ObjectIdentifierEncoderFc K s� |d k r|� |�}|�� }z|d }|d }W n$ tk
rR t�d|f ��Y nX d| krhdkr�n nn|dkr�|d f|dd � }nL|dkr�|f|dd � }n0|dkr�|d f|dd � }nt�d|f ��n0|dkr�|d f|dd � }nt�d|f ��d }|D ]�} d| k�r.d
k�r>n n || f7 }n\| d
k�r�| d
@ f}
| d L } | �r~d | d
@ B f|
}
| d L } �qZ||
7 }nt�d
| |f ���q|ddfS )Nr r; z Short OID %s�' �( � �P z"Impossible first/second arcs at %sr r r r zNegative OID arc %s at %sF)rD ZasTuple�
IndexErrorr r" ) r r& r' r( r) Zoid�first�secondZoctetsZsubOid�resr r r r* � s@
z#ObjectIdentifierEncoder.encodeValueNr= r r r r rO � s rO c @ s0 e Zd ZdZdZedd� �Zdd� Zdd� Zd S )
� RealEncoderr rR c C s� d\}}| dk rd}|dk r d}| |9 } |dkrZ| dt |�d | 9 } t |�d | }n0|dkr�| dt |�d | 9 } t |�d | }t| �| kr�| |9 } |d 8 }q�q�q�|t| �||fS )
N)r; r; r ���r rR � � � r; )�absr@ )�m�encbase�e�msZesr r r �_dropFloatingPoint. s$ z RealEncoder._dropFloatingPointc C s |\}}}dddg}|j |kr.| �||j |�S | j |krH| �|| j |�S |||g}|||g}d}d} td�}td�D ]�}
| �||
||
||
�\}||
< ||
< ||
< t||
�t|�k s�t||
�t|�krt||
|k rt||
}t||
�}||
} qt||| |fS )NrR r r[ r; �infrZ )�
binEncBaserb �float�ranger] r@ ) r r& r^ �br` ZencBaseZmantissaZ exponentaZsignr_ �ir r r �_chooseEncBaseE s,
�4
zRealEncoder._chooseEncBasec K s� |d k r|� |�}|jr dS |jr&dS |\}}}|s>tddfS |dkrjtd||dkrXdpZd |f �ddfS |d
k�r�d }| �|�\} }}
}| dk r�|d O }|
d
kr�|d
@ dkr�|d
L }|d
7 }q�nX|
dkr�|d@ dkr�|dL }|d
7 }q�|dO }n(|d@ dk�r|dL }|d
7 }q�|dO }d} |d
@ dk�r@|d
L }| d
7 } �q | dk�rTt�d��|| d
> O }t} |dk�sx|dk�r�t|d@ �} n�|dk�r�t|d@ �| } |dL }�q�|dk�r�| �r�t | d �d @ �r�td�| } |dk�r| �rt | d �d @ �std�| } t
| �}
|
dk�r$t�d��|
d
k�r0n@|
d
k�rD|d
O }n,|
dk�rX|d
O }n|dO }t|
d@ �| } t}|�r�t|d@ �| }|dL }�qtt|�| | }|ddfS t�d| ��d S )N))�@ FF))�A FFFT�
z%dE%s%dr �+� rR r rj r; r r rZ r[ � r\ � zScale factor overflowrY r )r rY zReal exponent overflowzProhibited Real base %s) rD Z isPlusInfZ
isMinusInfr r
ri r r" r r r! )r r& r' r( r) r^ rg r` Zfora r_ ZsfZeo�nZpor r r r r* ] s�
$
" "
zRealEncoder.encodeValueN) r3 r4 r5 r rd � staticmethodrb ri r* r r r r rX * s
rX c @ s e Zd ZdZdd� ZdS )�SequenceEncoderFc K s� t }|d kr�|�� |j}t|�� �D ]�\}}|rn|| } | jrH|jsHq&| jrZ|| jkrZq&| j rn|j
| jd� |||f|�}
|r�| j r�| j} | j r�| �
|�s�||
| f|�}
||
7 }q&n�t|jj�D ]�\}} z|| j }W n* tk
�r t�d| j|f ��Y nX | j�r| j|k�rq�| j�r0|| jk�r0q�| j �rF|j
| jd� |||| f|�}
| j �r�| j} | j �r�| �
|��s�||
| f|�}
||
7 }q�|ddfS )N)r+ z#Component name "%s" not found in %rT)r �verifySizeSpec�
componentTyper. �valuesZ
isOptionalZisValueZ isDefaultedZ
asn1Object�omitEmptyOptionals�updateZopenTyper, ZisSameTypeWith�
namedTypes�name�KeyErrorr r" ) r r& r' r( r) r ry r1 � component� namedTyperM ZwrapTyper r r r* � sJ
zSequenceEncoder.encodeValueN)r3 r4 r5 rw r* r r r r rs � s rs c @ s e Zd Zdd� ZdS )�SequenceOfEncoderc K sN |d kr|� � n|j}t}t|�D ] \}}|||| |f|�7 }q$|ddfS )NT)rt ru r r. )r r& r' r( r) r r1 r| r r r r* � s
z SequenceOfEncoder.encodeValueNr9 r r r r r~ � s r~ c @ s e Zd Zdd� ZdS )�
ChoiceEncoderc s~ |d kr� � � }nX� fdd�|jjD �}t|�dkrRt�dt|�rFdpHd� f ��|d }� | }|| }|||f|�ddfS ) Nc s g | ]}|j � kr|j �qS r )rz )�.0r} �r& r r �
<listcomp> s
�z-ChoiceEncoder.encodeValue.<locals>.<listcomp>r; z %s components for Choice at %rz Multiple zNone r T)Z getComponentru ry r! r r" )r r& r' r( r) r| �namesrz r r� r r* � s
zChoiceEncoder.encodeValueNr9 r r r r r � s r c @ s e Zd Zdd� ZdS )�
AnyEncoderc K s< |d kr|� � }nt|�s(|�|�� � }||�dd� dfS )Nr# T)rE r rD r- r% r r r r* s
zAnyEncoder.encodeValueNr9 r r r r r�
s r� c @ s* e Zd ZdZdZi fdd�Zddd�ZdS )�EncoderNc C s || _ || _d S )N)�_Encoder__tagMap�_Encoder__typeMap)r �tagMap�typeMapr r r �__init__` s zEncoder.__init__c
K s� z|d kr|j }n|j }W n$ tk
r> t�d|f ��Y nX tjtj@ rTtj}nd }|r�|d|�dd� rrdptd|�dd�|d kr�|�� p�|�� |f � | j d k r�|j
| j d � | j d k r�|j
| j d
� z&| j | }|r�|d |j
j|f � W n� tk
�r� |d k�r|j}n|j}t�|j|j�}z| j| }W n( tk
�rd t�d ||f ��Y nX |�r�|d
|j
j|f � Y nX |j||| f|�} |�r�|d|t| �t�| �f � | S )Nz<Value %r is not ASN.1 type instance and "asn1Spec" not givenzBencoder called in %sdef mode, chunk size %s for type %s, value:
%sr# T�inrn rB r )r# )rB z)using value codec %s chosen by type ID %szNo encoder for %r (%s)z(using value codec %s chosen by tagSet %sz;codec %s built %s octets of substrate: %s
encoder completed)�typeId�AttributeErrorr r" r �loggerZ flagEncoderr- ZprettyPrintType�fixedDefLengthModerx �fixedChunkSizer� � __class__r3 r{ r, r
rG rF r� r r! Zhexdump)
r r& r' r) r� r� ZconcreteEncoderr, Z
baseTagSetr r r r �__call__d sL
� 8�
zEncoder.__call__)N)r3 r4 r5 r� r� r� r� r r r r r� [ s r� N)EZpyasn1r r Zpyasn1.codec.berr Zpyasn1.compat.integerr Zpyasn1.compat.octetsr r r r r
r Z pyasn1.typer r
r r �__all__�objectr r6 r: r> rA rK rN rO rX rs r~ r r� Z endOfOctetsr, ZBooleanZIntegerZ BitStringZ OctetStringZNullZObjectIdentifierZ
EnumeratedZRealZ
SequenceOfZSetOfZChoiceZ
UTF8StringZ
NumericStringZPrintableStringZ
TeletexStringZVideotexStringZ IA5StringZ
GraphicStringZ
VisibleStringZ
GeneralStringZUniversalStringZ BMPStringZObjectDescriptorZGeneralizedTimeZUTCTimer� r� �Set�SequenceZAnyr� r� r r r r r �<module> s _%:5|F �" �#v