File: //lib/python3/dist-packages/cryptography/hazmat/primitives/ciphers/__pycache__/aead.cpython-310.pyc
o
�)%a� � @ sl d dl Z d dlZd dlmZmZ d dlmZ d dlmZ G dd� de �Z
G dd� de �Z G d d
� d
e �Z dS ) � N)�
exceptions�utils)�aead)�backendc @ s� e Zd ZdZdefdd�Zedefdd��Zded ed
ej e defd d �Z
ded ed
ej e defd
d�Z ded ed
eddfdd�Z dS )�ChaCha20Poly1305� �keyc C sD t �| �s
t�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 �M/usr/lib/python3/dist-packages/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_datac C �R |d u rd}t |�| jkst |�| jkrtd��| �|||� t�t| |||d�S �N� �1Data or associated data too long. Max 2**32 bytes� �r � _MAX_SIZE�
OverflowError�
_check_paramsr �_encryptr �r r r r r r r �encrypt" � �zChaCha20Poly1305.encryptc C �. |d u rd}| � |||� t�t| |||d�S �Nr! r# �r' r �_decryptr r) r r r �decrypt4 � zChaCha20Poly1305.decryptNc C s< t �d|� t �d|� t �d|� t|�dkr td��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* r0 r' r r r r r s@
���
����
� ����r c @ s� e Zd ZdZddedefdd�Zededefd d
��Zd ed ed
e j
e defdd�Z d ed ed
e j
e defdd�Z d edefdd�Z
d ed ed
efdd�ZdS )�AESCCMr r# r �
tag_lengthc C sR t �d|� t|�dvrtd��|| _t|t�s td��|dvr$td��|| _d S )Nr �r# � r z)AESCCM key must be 128, 192, or 256 bits.z tag_length must be an integer)� � � �
r2 � r# zInvalid tag_length) r r r r r �
isinstance�int� TypeError� _tag_length)r r r= r r r r P s
zAESCCM.__init__�
bit_lengthr c C �0 t |t�s td��|dvrtd��t�|d �S �Nz bit_length must be an integer)� � � z#bit_length must be 128, 192, or 256rB �rE rF rG r r r �r rI r r r r ^ �
zAESCCM.generate_keyr r r c C sd |d u rd}t |�| jkst |�| jkrtd��| �|||� | �|t |�� t�t| |||| j�S )Nr! r" ) r r% r& r' �_validate_lengthsr r( r rH r) r r r r* h s ��zAESCCM.encryptc C s0 |d u rd}| � |||� t�t| |||| j�S )Nr! )r' r r/ r rH r) r r r r0 } s �zAESCCM.decrypt�data_lenc C s( dt |� }dd| |k rtd��d S )N� � rB zData too long for nonce)r r )r r rS �l_valr r r rR � s �zAESCCM._validate_lengthsc C sP t �d|� t �d|� t �d|� dt|� kr!dks&td�� td��d S )Nr r r � �
z$Nonce must be between 7 and 13 bytesr3 r) r r r r' � s ��zAESCCM._check_paramsN)r# )r5 r6 r7 r% r8 rF r r9 r r: r; r* r0 rR r' r r r r r< M s2 ���
����
�r< c @ s� e Zd ZdZdefdd�Zededefdd��Zd ed
ed e j
e defd d
�Z d ed
ed e j
e defdd�Z d ed
ed eddfdd�Z
dS )�AESGCMr r c C s* t �d|� t|�dvrtd��|| _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__rI r c C rJ rK rO rP r r r r � rQ zAESGCM.generate_keyr r r c C r r r$ r) r r r r* � r+ zAESGCM.encryptc C r, r- r. r) r r r r0 � r1 zAESGCM.decryptNc C sH t �d|� t �d|� t �d|� t|�dk s t|�dkr"td��d S )Nr r r rB rL z%Nonce must be between 8 and 128 bytesr3 r) r r r r' � s �zAESGCM._check_params)r5 r6 r7 r% r8 r r9 rF r r: r; r* r0 r' r r r r rY � s@ ���
����
� ����rY )
r r: � cryptographyr r �$cryptography.hazmat.backends.opensslr �,cryptography.hazmat.backends.openssl.backendr �objectr r< rY r r r r �<module> s ?M