o
    .]$                     @   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.10/site-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)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   numbersZ
isModifiedZsearchIndexr   r   r   r   ,   sD   




zTimeEncoderMixIn.encodeValueN)r   r   r   ordr0   r-   r.   r1   r2   r7   r8   r9   r   r   r   r   r   r   !   s    r   c                   @      e Zd ZdZdZdS )GeneralizedTimeEncoderr&      Nr   r   r   r8   r9   r   r   r   r   r?   f       r?   c                   @   r>   )UTCTimeEncoder
      NrA   r   r   r   r   rC   k   rB   rC   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maxLenZzeror   r   
<listcomp>y   s    z,SetOfEncoder.encodeValue.<locals>.<listcomp>c                 S   s   | d S )Nr   r   )rK   r   r   r   <lambda>|   s    z*SetOfEncoder.encodeValue.<locals>.<lambda>keyc                 S   s   g | ]}|d  qS )r*   r   rI   r   r   r   rN   ~   s    T)_encodeComponentsr6   r   maxmapsortr   join)r   r   r   r   r   chunksZpaddedChunksr   rL   r   r   q   s   zSetOfEncoder.encodeValueNr   r   r   r   r   rF   p   r   rF   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)getr6   r   rR   rV   )r   r   r   r   r   rW   r   r   r   r      s   
zSequenceOfEncoder.encodeValueNr   r   r   r   r   rX      r   rX   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 %rrP   )rY   T)r   ZisInconsistentr^   	enumeratevaluesZ
isOptionalZisValueZisDefaultedZ
asn1Objectidappend
namedTypesnameKeyErrorr   r/   sortedr`   r:   ZopenTyper]   ZisSameTypeWith)r   r   r   r   r   r   compsZcompsMapZinconsistencyre   idxr_   Z	namedTypecompZcompTypechunkZwrapTyper   r   r   r      sT   


zSetEncoder.encodeValueN)r   r   r   staticmethodr`   r   r   r   r   r   r[      s    
r[   c                   @   s   e Zd ZdZdS )SequenceEncoderTN)r   r   r   ZomitEmptyOptionalsr   r   r   r   rn      s    rn   c                   @   r>   )EncoderFr+   N)r   r   r   ZfixedDefLengthModeZfixedChunkSizer   r   r   r   ro     rB   ro   N)'pyasn1r   Zpyasn1.codec.berr   Zpyasn1.compat.octetsr   r   Zpyasn1.typer   r   __all__ZIntegerEncoderr
   r   objectr   r;   r?   rC   rX   rF   rn   r[   ZtagMapcopyr:   ZBooleanr]   RealZGeneralizedTimeZUTCTimeZSetOfSequencer\   ZtypeMapSetZ
SequenceOfro   r   r   r   r   r   <module>   sJ   	EZ








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