o
    c]$                     @   s  d dl mZ d dlmZ d dlmZmZ d dlmZ d dlm	Z	 dgZ
G dd dejZG d	d
 d
ejZG dd deZG dd deejZG dd deejZG dd dejZG dd dejZG dd dejZG dd dejZej Zeejje ejje e	jje e	jje ejje ej j!e i ej" Z"e"ejj!e ejj!e e	jj!e e	jj!e ej#j!e ejj!e ej j!e ej$j!e i G dd dej%Z%e%ee"Z&dS )    )error)encoder)str2octsnull)univ)usefulencodec                   @      e Zd Zdd ZdS )BooleanEncoderc                 K   s   |dkrd}nd}|ddfS )Nr   )r   )   F )selfvalueasn1Spec	encodeFunoptions	substrater   r   :/usr/lib/python3/dist-packages/pyasn1/codec/cer/encoder.pyencodeValue   s   
zBooleanEncoder.encodeValueN__name__
__module____qualname__r   r   r   r   r   r
          r
   c                   @   r	   )RealEncoderc                 C   s   |\}}}|  |||S )N)Z_dropFloatingPoint)r   r   mber   r   r   _chooseEncBase   s   
zRealEncoder._chooseEncBaseN)r   r   r   r   r   r   r   r   r      r   r   c                   @   sL   e Zd ZedZedZedZedZedZedZ	dZ
dZd	d
 ZdS )TimeEncoderMixInZ+-,.0      c                 K   sp  |d ur	| |}| }| j|v s| j|v rtd| |d | jkr,td| | j|v r8td| | j|v rd}t	|}t
|| jd t|d }|| | jkrp|| | jkre||= d}|d8 }|| | jksY|d7 }|t|k r|| | jkr||d = d}|r| |}| jt|  k r| jk sn td	| |jd
d tjj| |||fi |S )NzMust be UTC time: %rz#Missing "Z" time zone specifier: %rz!Comma in fractions disallowed: %rF      TzLength constraint violated: %r  )ZmaxChunkSize)ZcloneZ	asNumbers	PLUS_CHAR
MINUS_CHARr   PyAsn1ErrorZ_CHAR
COMMA_CHARDOT_CHARlistminindexlen	ZERO_CHAR
MIN_LENGTH
MAX_LENGTHupdater   OctetStringEncoderr   )r   r   r   r   r   ZnumbersZ
isModifiedZsearchIndexr   r   r   r   ,   sD   




zTimeEncoderMixIn.encodeValueN)r   r   r   ordr/   r,   r-   r0   r1   r6   r7   r8   r   r   r   r   r   r   !   s    r   c                   @      e Zd ZdZdZdS )GeneralizedTimeEncoderr&      Nr   r   r   r7   r8   r   r   r   r   r=   f       r=   c                   @   r<   )UTCTimeEncoder
      Nr?   r   r   r   r   rA   k   r@   rA   c                   @   r	   )SetOfEncoderc                    sz   | j |||fi |}t|dkr5tdttt|  fdd|D }|jdd d dd |D }t|d	d	fS )
Nr*    c                    s   g | ]
}|  |fqS r   )ljust.0xZmaxLenZzeror   r   
<listcomp>y   s    z,SetOfEncoder.encodeValue.<locals>.<listcomp>c                 S   s   | d S )Nr   r   )rI   r   r   r   <lambda>|   s    z*SetOfEncoder.encodeValue.<locals>.<lambda>keyc                 S   s   g | ]}|d  qS )r*   r   rG   r   r   r   rK   ~   s    T)_encodeComponentsr5   r   maxmapsortr   join)r   r   r   r   r   chunksZpaddedChunksr   rJ   r   r   q   s   zSetOfEncoder.encodeValueNr   r   r   r   r   rD   p   r   rD   c                   @   r	   )SequenceOfEncoderc                 K   sD   | ddrt|stddfS | j|||fi |}t|ddfS )N
ifNotEmptyFT)getr5   r   rO   rS   )r   r   r   r   r   rT   r   r   r   r      s   
zSequenceOfEncoder.encodeValueNr   r   r   r   r   rU      r   rU   c                   @   s    e Zd Zedd Zdd ZdS )
SetEncoderc                 C   sB   | \}}|du r
|}|j tjj kr|js|jr|jS |jjS |jS )z^Sort SET components by tag

        Sort regardless of the Choice value (static sort)
        N)typeIdr   ZChoicetagSetcomponentTypeZ	minTagSet)ZcomponentAndType	componentr   r   r   r   _componentSortKey   s   zSetEncoder._componentSortKeyc              	   K   s  t }g }i }|d u rJ|j}|r||j}	t| D ].\}
}|	r;|	|
 }|jr+|js+q|jr4||jkr4q||t	|< nd |t	|< |
||f qnEt|jjD ]>\}
}z||j }W n tyl   td|j|f w |jrv|j|vrvqP|jr||jkrqP||t	|< |
|||
 f qPt|| jdD ]9\}}|t	| }|r|j|jd |||fi |}|r|jr|j}|jr||s|||fi |}||7 }q|ddfS )Nz#Component name "%s" not found in %rrM   )rV   T)r   ZisInconsistentr[   	enumeratevaluesZ
isOptionalZisValueZisDefaultedZ
asn1Objectidappend
namedTypesnameKeyErrorr   r.   sortedr]   r9   ZopenTyperZ   ZisSameTypeWith)r   r   r   r   r   r   compsZcompsMapZinconsistencyrb   idxr\   Z	namedTypecompZcompTypechunkZwrapTyper   r   r   r      sT   


zSetEncoder.encodeValueN)r   r   r   staticmethodr]   r   r   r   r   r   rX      s    
rX   c                   @   s   e Zd ZdZdS )SequenceEncoderTN)r   r   r   ZomitEmptyOptionalsr   r   r   r   rk      s    rk   c                   @   r<   )EncoderFr+   N)r   r   r   ZfixedDefLengthModeZfixedChunkSizer   r   r   r   rl     r@   rl   N)'Zpyasn1r   Zpyasn1.codec.berr   Zpyasn1.compat.octetsr   r   Zpyasn1.typer   r   __all__ZIntegerEncoderr
   r   objectr   r:   r=   rA   rU   rD   rk   rX   ZtagMapcopyr9   ZBooleanrZ   ZRealZGeneralizedTimeZUTCTimeZSetOfSequencerY   ZtypeMapSetZ
SequenceOfrl   r   r   r   r   r   <module>   sJ   	EZ
















1