File: //lib/python3/dist-packages/cryptography/hazmat/primitives/ciphers/__pycache__/aead.cpython-38.pyc
U
�e�]% � @ sx d dl mZmZmZ d dlZd dlmZmZ d dlm Z d dl
m Z G dd� de �Z
G dd � d e �ZG d
d � d e �ZdS ) � )�absolute_import�division�print_functionN)�
exceptions�utils)�aead)�backendc @ s< e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd
d � Z d S )
�ChaCha20Poly1305� c C sD t �| �st�dtjj��t�d|� t|�dkr:t d��|| _
d S )Nz<ChaCha20Poly1305 is not supported by this version of OpenSSL�key� 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__c C s
t �d�S )Nr )�os�urandom)�clsr r r � generate_key s z ChaCha20Poly1305.generate_keyc C sR |d kr d}t |�| jks(t |�| jkr0td��| �|||� t�t| |||d�S �N� �1Data or associated data too long. Max 2**32 bytes� �r � _MAX_SIZE�
OverflowError�
_check_paramsr �_encryptr �r �nonce�data�associated_datar r r �encrypt" s � �zChaCha20Poly1305.encryptc C s. |d kr d}| � |||� t�t| |||d�S �Nr r! �r% r �_decryptr r' r r r �decrypt1 s �zChaCha20Poly1305.decryptc 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_paramsN�
�__name__�
__module__� __qualname__r# r � classmethodr r+ r/ r% r r r r r s
r c @ sF e Zd ZdZddd�Zedd� �Zdd� Zd d
� Zd d � Z d
d� Z
dS )�AESCCMr
r! c C sl t �d|� t|�dkr td��|| _t|t�s8td��|dkrHtd��|| _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 Z
tag_lengthr r r r E s
�zAESCCM.__init__c C s0 t |t�std��|dkr"td��t�|d �S �Nz bit_length must be an integer)� � � z#bit_length must be 128, 192, or 256r= �r@ rA rB r r r �r �
bit_lengthr r r r Y s
zAESCCM.generate_keyc C sd |d kr d}t |�| jks(t |�| jkr0td��| �|||� | �|t |�� t�t| |||| j�S )Nr r ) r r# r$ r% �_validate_lengthsr r& r rC r' r r r r+ c s � �zAESCCM.encryptc C s0 |d kr d}| � |||� t�t| |||| j�S )Nr )r% r r. r rC r' r r r r/ s s �zAESCCM.decryptc C s( dt |� }dd| |k r$td��d S )N� � r= zNonce too long for data)r r )r r( Zdata_lenZl_valr r r rK | 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_paramsN)r! ) r4 r5 r6 r# r r7 r r+ r/ rK r% r r r r r8 B s
r8 c @ s< e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd
d � Z d S )
�AESGCMr
c C s* t �d|� t|�dkr td��|| _d S )Nr r9 z)AESGCM key must be 128, 192, or 256 bits.)r r r r r r r r r r � s zAESGCM.__init__c C s0 t |t�std��|dkr"td��t�|d �S rD rH rI r r r r � s
zAESGCM.generate_keyc C sR |d kr d}t |�| jks(t |�| jkr0td��| �|||� t�t| |||d�S r r" r' r r r r+ � s � �zAESGCM.encryptc C s. |d kr d}| � |||� t�t| |||d�S r, r- r' r r r r/ � s �zAESGCM.decryptc C s< t �d|� t �d|� t �d|� t|�dkr8td��d S )Nr( r) r* r z Nonce must be at least 1 byter1 r' r r r r% � s
zAESGCM._check_paramsNr3 r r r r rP � s
rP )Z
__future__r r r r Z cryptographyr r Z$cryptography.hazmat.backends.opensslr Z,cryptography.hazmat.backends.openssl.backendr �objectr r8 rP r r r r �<module> s 4I