File: //var/opt/nydus/ops/cryptography/hazmat/backends/openssl/__pycache__/aead.cpython-39.pyc
a
�,�h� � @ s d dl Z d dlmZ e jrVd dlmZ d dlmZmZm Z m
Z
m Z e j eee e
e f Z
dZd Zded�dd �Zed
d �d d
�Zd
eeee je eed�dd�Zd
edd�dd�Zd
edd�dd�Zd
eed�dd�Zd
deee je eed�dd �Zd
deee je eed�d d �ZdS ) � N)�
InvalidTag)�Backend��AESCCM�AESGCM�AESOCB3�AESSIV�ChaCha20Poly1305� � _AEAD_TYPES)�cipher�returnc C s� ddl m}m}m}m}m} t| |�r*dS t| |�rPdt| j�d � d�� d�S t| |�rvdt| j�d � d�� d�S t| |�r�dt| j�d d � d
�� d�S t| |�s�J �dt| j�d � d �� d�S d S ) Nr r s chacha20-poly1305zaes-� z-ccm�asciiz-ocb� z-sivz-gcm)
�+cryptography.hazmat.primitives.ciphers.aeadr r r r r �
isinstance�len�_key�encode)r r r r r r � r �\/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/cryptography/hazmat/backends/openssl/aead.py�_aead_cipher_name s
r r )� cipher_name�backendc C sj | � d�rH|j�|jj| |jj�}|�||jjk� |j�||jj�}n |j�| �}|�||jjk� |S )Ns -siv) �endswith�_libZEVP_CIPHER_fetch�_ffi�NULL�openssl_assert�gcZEVP_CIPHER_freeZEVP_get_cipherbyname)r r �
evp_cipherr r r � _evp_cipher2 s
� r" )r r �key�nonce�tag�tag_len� operationc
C sl t || �}| j�� }| j�|| jj�}| j�||| jj| jj| jjt|t k��} | �
| dk� | j� |t |��} | �
| dk� | j�
|| jjt |�| jj�} | �
| dk� |tkr�|d us�J �| j�
|| jjt |�|�} | �
| dk� n4|�d��r | j�
|| jj|| jj�} | �
| dk� | j�|�}
| j�|�} | j�|| jj| jj| |
t|t k��} | �
| dk� |S )Nr s -ccm)r" r ZEVP_CIPHER_CTX_newr r ZEVP_CIPHER_CTX_freeZEVP_CipherInit_exr �int�_ENCRYPTr Z EVP_CIPHER_CTX_set_key_lengthr �EVP_CIPHER_CTX_ctrlZEVP_CTRL_AEAD_SET_IVLEN�_DECRYPTZEVP_CTRL_AEAD_SET_TAGr � from_buffer) r r r# r$ r% r& r' r! �ctx�resZ nonce_ptrZkey_ptrr r r � _aead_setupB sX
�� � �
�r/ )r �data_lenr
c C s: | j �d�}| j�|| j j|| j j|�}| �|dk� d S �N�int *r )r �newr �EVP_CipherUpdater r )r r- r0 Zintptrr. r r r � _set_lengthz s
�r5 )r �associated_datar
c C s: | j �d�}| j�|| j j||t|��}| �|dk� d S r1 )r r3 r r4 r r r )r r- r6 �outlenr. r r r � _process_aad� s
�r8 )r �datar
c C sd | j �d�}| j �dt|��}| j�||||t|��}|dkrJ| �� t�| j �||d �d d � S )Nr2 �unsigned char[]r )r r3 r r r4 �_consume_errorsr �buffer)r r- r9 r7 �bufr. r r r �
_process_data� s r> )r r r$ r9 r6 �
tag_lengthr
c C s8 ddl m}m} t|�}t| ||j|d |t�} t||�rHt| | t |�� |D ]}
t
| | |
� qLt | | |�} | j �
d�} | j �
dd�}
| j�| |
| �}| �|dk� | | j �|
| d �d d � 7 } | j �
d|�}| j�| | jj||�}| �|dk� | j �|�d d � }t||��r,| �t |�dk� || S | | S d S )Nr �r r r2 r: � )r r r r r/ r r) r r5 r r8 r> r r3 r �EVP_CipherFinal_exr r< r* ZEVP_CTRL_AEAD_GET_TAG)r r r$ r9 r6 r? r r r r- �ad�processed_datar7 r= r. Ztag_bufr% r r r �_encrypt� s2 �
� rE c C s� ddl m}m} t|�|k r t�t||�rD|d |� }||d � }n || d � }|d | � }t|�} t| | |j|||t �}
t||�r�t
| |
t|�� |D ]} t | |
| � q�t||��r"| j �
d�} | j �
dt|��}
| j�|
|
| |t|��}|dk�r| �� t�| j �|
| d �d d � }njt| |
|�}| j �
d�} | j �
dd�}
| j�|
|
| �}|| j �|
| d �d d � 7 }|dk�r�| �� t�|S )Nr r@ r2 r: r
rA )r r r r r r r r/ r r+ r5 r8 r r3 r r4 r; r< r> rB )r r r$ r9 r6 r? r r r% r r- rC r7 r= r. rD r r r �_decrypt� sB
�
rF ) �typingZcryptography.exceptionsr �
TYPE_CHECKINGZ,cryptography.hazmat.backends.openssl.backendr r r r r r r �Unionr r) r+ �bytesr r" �Optionalr( r/ r5 r8 r> �ListrE rF r r r r �<module> sL �� 8 � 0�