File: //var/opt/nydus/ops/cryptography/hazmat/primitives/ciphers/__pycache__/aead.cpython-39.pyc
a
�,�h�, � @ s� d dl Z d dlZd dlmZmZ d dlmZ d dlmZ G dd� d�Z G dd� d�Z
G d d
� d
�Z G d d � d �Z G d
d� de
�ZdS )� N)�
exceptions�utils)�aead)�backendc @ sx e Zd ZdZed�dd�Zeed�dd��Zeeej e ed�d d
�Z
eeej e ed�d d �Z eeed
d�dd�Z d
S )�ChaCha20Poly1305����keyc C sD t �| �st�dtjj��t�d|� t|�dkr:t d��|| _
d S )Nz<ChaCha20Poly1305 is not supported by this version of OpenSSLr � z&ChaCha20Poly1305 key must be 32 bytes.) r �aead_cipher_supportedr �UnsupportedAlgorithm�_Reasons�UNSUPPORTED_CIPHERr �_check_byteslike�len�
ValueError�_key��selfr � r �^/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/cryptography/hazmat/primitives/ciphers/aead.py�__init__ s
� zChaCha20Poly1305.__init__)�returnc C s
t �d�S )Nr
)�os�urandom)�clsr r r � generate_key s z ChaCha20Poly1305.generate_key��nonce�data�associated_datar c C sT |d u r d}t |�| jks(t |�| jkr0td��| �|||� t�t| |||gd�S �N� �5Data or associated data too long. Max 2**31 - 1 bytes� �r � _MAX_SIZE�
OverflowError�
_check_paramsr �_encryptr �r r r r r r r �encrypt" s �zChaCha20Poly1305.encryptc C s0 |d u r d}| � |||� t�t| |||gd�S �Nr" r$ �r( r �_decryptr r* r r r �decrypt4 s zChaCha20Poly1305.decryptNc C s< t �d|� t �d|� t �d|� t|�dkr8td��d S )Nr r r � zNonce must be 12 bytes�r r � _check_bytesr r r* r r r r( @ s
z ChaCha20Poly1305._check_params)
�__name__�
__module__� __qualname__r& �bytesr � classmethodr �typing�Optionalr+ r/ r( r r r r r s&
� � �r c @ s� e Zd ZdZdeed�dd�Zeeed�dd��Zeee j
e ed �d
d �Z eee j
e ed �d d
�Z eedd�dd�Z
eeedd �dd�ZdS )�AESCCMr r$ )r �
tag_lengthc C sl t �d|� t|�dvr td��|| _t|t�s8td��|dvrHtd��|| _t �
| �sht � dt j
j��d S )Nr �r$ � r
z)AESCCM key must be 128, 192, or 256 bits.z tag_length must be an integer)� � � �
r0 � r$ zInvalid tag_lengthz2AESCCM is not supported by this version of OpenSSL)r r r r r �
isinstance�int� TypeError� _tag_lengthr r r r r
r )r r r; r r r r P s
�zAESCCM.__init__��
bit_lengthr c C s0 t |t�std��|dvr"td��t�|d �S �N� bit_length must be an integer)� � � z#bit_length must be 128, 192, or 256r@ �rC rD rE r r r �r rH r r r r d s
zAESCCM.generate_keyr c C sf |d u r d}t |�| jks(t |�| jkr0td��| �|||� | �|t |�� t�t| |||g| j�S )Nr" r# ) r r& r' r( �_validate_lengthsr r) r rF r* r r r r+ n s ��zAESCCM.encryptc C s2 |d u r d}| � |||� t�t| |||g| j�S )Nr" )r( r r. r rF r* r r r r/ � s �zAESCCM.decryptN)r �data_lenr c C s( dt |� }dd| |k r$td��d S )N� � r@ zData too long for nonce)r r )r r rQ Zl_valr r r rP � s zAESCCM._validate_lengthsc C sJ t �d|� t �d|� t �d|� dt|� kr<dksFn td��d S )Nr r r � �
z$Nonce must be between 7 and 13 bytesr1 r* r r r r( � s
zAESCCM._check_params)r$ )r3 r4 r5 r& r6 rD r r7 r r8 r9 r+ r/ rP r( r r r r r: M s$ � � �r: c @ sz e Zd ZdZed�dd�Zeeed�dd��Zeee j
e ed�d d
�Z eee j
e ed�d d �Z eeed
d�dd�Z
d
S )�AESGCMr r c C s* t �d|� t|�dvr td��|| _d S )Nr r< z)AESGCM key must be 128, 192, or 256 bits.)r r r r r r r r r r � s zAESGCM.__init__rG c C s0 t |t�std��|dvr"td��t�|d �S rI rN rO r r r r � s
zAESGCM.generate_keyr c C sT |d u r d}t |�| jks(t |�| jkr0td��| �|||� t�t| |||gd�S r! r% r* r r r r+ � s �zAESGCM.encryptc C s0 |d u r d}| � |||� t�t| |||gd�S r, r- r* r r r r/ � s zAESGCM.decryptNc C sH t �d|� t �d|� t �d|� t|�dk s<t|�dkrDtd��d S )Nr r r r@ rK z%Nonce must be between 8 and 128 bytesr1 r* r r r r( � s
zAESGCM._check_params�r3 r4 r5 r& r6 r r7 rD r r8 r9 r+ r/ r( r r r r rV � s& � � �rV c @ sz e Zd ZdZed�dd�Zeeed�dd��Zeee j
e ed�d d
�Z eee j
e ed�d d �Z eeed
d�dd�Z
d
S )�AESOCB3r r c C sD t �d|� t|�dvr td��|| _t�| �s@t�dtj j
��d S )Nr r< z*AESOCB3 key must be 128, 192, or 256 bits.z0OCB3 is not supported by this version of OpenSSL� r r r r r r r r r r
r r r r r r � s
�zAESOCB3.__init__rG c C s0 t |t�std��|dvr"td��t�|d �S rI rN rO r r r r � s
zAESOCB3.generate_keyr c C sT |d u r d}t |�| jks(t |�| jkr0td��| �|||� t�t| |||gd�S r! r% r* r r r r+ � s �zAESOCB3.encryptc C s0 |d u r d}| � |||� t�t| |||gd�S r, r- r* r r r r/
s zAESOCB3.decryptNc C sH t �d|� t �d|� t �d|� t|�dk s<t|�dkrDtd��d S )Nr r r r0 rR z%Nonce must be between 12 and 15 bytesr1 r* r r r r( s
zAESOCB3._check_paramsrW r r r r rX � s&
� � �rX c @ s� e Zd ZdZed�dd�Zeeed�dd��Zee j
e j e ed�d d
�Z ee j
e j e ed�d d �Z
ee j d
d�dd�Zd
S )�AESSIVr r c C sD t �d|� t|�dvr td��|| _t�| �s@t�dtj j
��d S )Nr )r
�0 �@ z)AESSIV key must be 256, 384, or 512 bits.z3AES-SIV is not supported by this version of OpenSSLrY r r r r r ) s
�zAESSIV.__init__rG c C s0 t |t�std��|dvr"td��t�|d �S )NrJ )rM i� i z#bit_length must be 256, 384, or 512r@ rN rO r r r r 6 s
zAESSIV.generate_key)r r r c sX |d u r g }� � ||� t|�� jks<t� fdd�|D ��rDtd��t�t� d||d�S )Nc 3 s | ]}t |�� jkV qd S �N)r r& )�.0�ad�r r r � <genexpr>J s z!AESSIV.encrypt.<locals>.<genexpr>r# r" r$ )r( r r& �anyr' r r) r �r r r r r` r r+ @ s ��zAESSIV.encryptc C s, |d u r g }| � ||� t�t| d||d�S r, r- rc r r r r/ T s zAESSIV.decryptNc C s4 t �d|� t|t�r(tdd� |D ��s0td��d S )Nr c s s | ]}t |t�V qd S r] )rC r6 )r^ �xr r r ra f s z'AESSIV._check_params.<locals>.<genexpr>z/associated_data must be a list of bytes or None)r r2 rC �list�allrE rc r r r r( ` s
�zAESSIV._check_params)r3 r4 r5 r& r6 r r7 rD r r8 r9 �Listr+ r/ r( r r r r rZ & s
� � �rZ )r r8 Z cryptographyr r Z$cryptography.hazmat.backends.opensslr Z,cryptography.hazmat.backends.openssl.backendr r r: rV rX �objectrZ r r r r �<module> s ?U?E