File: //lib/python3/dist-packages/cryptography/hazmat/backends/openssl/__pycache__/aead.cpython-38.pyc
U
�e�]v � @ sd d dl mZmZmZ d dlmZ dZd Zdd� Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� ZdS )� )�absolute_import�division�print_function)�
InvalidTag� c C sr ddl m}m}m} t| |�r"dS t| |�rFd�t| j�d ��d�S t| |�sTt �d�t| j�d ��d�S d S )Nr )�AESCCM�AESGCM�ChaCha20Poly1305s chacha20-poly1305z
aes-{}-ccm� �asciiz
aes-{}-gcm)
�+cryptography.hazmat.primitives.ciphers.aeadr r r �
isinstance�format�len�_key�encode�AssertionError)�cipherr r r � r �K/usr/lib/python3/dist-packages/cryptography/hazmat/backends/openssl/aead.py�_aead_cipher_name s
r c
C st | j �|�}| �|| jjk� | j �� }| j�|| j j�}| j �||| jj| jj| jjt |t
k��} | �| dk� | j � |t |��} | �| dk� | j �
|| j jt |�| jj�} | �| dk� |tkr�| j �
|| j jt |�|�} | �| dk� n4|�d��r$| j �
|| j j|| jj�} | �| dk� | j�|�}
| j�|�} | j �|| jj| jj| |
t |t
k��} | �| dk� |S )Nr s -ccm)�_libZEVP_get_cipherbyname�openssl_assert�_ffi�NULLZEVP_CIPHER_CTX_new�gcZEVP_CIPHER_CTX_freeZEVP_CipherInit_ex�int�_ENCRYPTZ EVP_CIPHER_CTX_set_key_lengthr �EVP_CIPHER_CTX_ctrlZEVP_CTRL_AEAD_SET_IVLEN�_DECRYPTZEVP_CTRL_AEAD_SET_TAG�endswithZ from_buffer) �backend� cipher_name�key�nonce�tagZtag_lenZ operationZ
evp_cipher�ctx�resZ nonce_ptrZkey_ptrr r r � _aead_setup sd
� � � �
�r( c C s: | j �d�}| j�|| j j|| j j|�}| �|dk� d S �N�int *r )r �newr �EVP_CipherUpdater r )r! r& Zdata_lenZintptrr' r r r � _set_lengthH s �r- c C s: | j �d�}| j�|| j j||t|��}| �|dk� d S r) )r r+ r r, r r r )r! r& �associated_data�outlenr' r r r � _process_aadT s �r0 c C s^ | j �d�}| j �dt|��}| j�||||t|��}| �|dk� | j �||d �d d � S )Nr* �unsigned char[]r )r r+ r r r, r �buffer)r! r& �datar/ �bufr' r r r �
_process_data\ s
r5 c C s� ddl m} t|�}t| ||j|d |t�}t||�rDt| |t|�� t | ||� t
| ||�} | j � d�}
| j
�|| j j|
�} | �| dk� | �|
d dk� | j � d|�} | j
�|| j
j|| �} | �| dk� | j �| �d d � }
| |
S )Nr �r r* r1 )r r r r( r r r
r- r r0 r5 r r+ r �EVP_CipherFinal_exr r r ZEVP_CTRL_AEAD_GET_TAGr2 )r! r r$ r3 r. �
tag_lengthr r" r& �processed_datar/ r' Ztag_bufr% r r r �_encryptd s: �
�r: c C s2 ddl m} t|�|k r t�|| d � }|d | � }t|�}t| ||j|||t�} t||�rpt | | t|�� t
| | |� t||�r�| j � d�}
| j � dt|��} | j
�| | |
|t|��} | dkr�| �� t�| j �| |
d �d d � }
nBt| | |�}
| j � d�}
| j
�| | j j|
�} | dk�r.| �� t�|
S )Nr r6 r* r1 r )r r r r r r( r r r
r- r0 r r+ r r, Z_consume_errorsr2 r5 r7 r )r! r r$ r3 r. r8 r r% r" r& r/ r4 r' r9 r r r �_decrypt sB �
r; N)Z
__future__r r r Zcryptography.exceptionsr r r r r( r- r0 r5 r: r; r r r r �<module> s
-