관리-도구
편집 파일: aead.cpython-39.pyc
a �aB$ � @ s| d dl Z d dlZd dlmZmZ d dlmZ d dlmZ G dd� de �Z G dd� de �Z G d d � d e �Z G d d � d e �Z dS ) � 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 �Q/usr/lib64/python3.9/site-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_datar c C sR |d u r 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 r r r r r r �encrypt" s �zChaCha20Poly1305.encryptc C s. |d u r d}| � |||� t�t| |||d�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 �Nz 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 sd |d u r d}t |�| jks(t |�| jkr0td��| �|||� | �|t |�� t�t| |||| j�S )Nr" r# ) r r& r' r( �_validate_lengthsr r) r rF r* r r r r+ n s ��zAESCCM.encryptc C s0 |d u r d}| � |||� t�t| |||| 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 rP Zl_valr r r rO � 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/ rO 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 rM rN r r r r � s zAESGCM.generate_keyr c C sR |d u r d}t |�| jks(t |�| jkr0td��| �|||� t�t| |||d�S r! r% r* r r r r+ � s �zAESGCM.encryptc C s. |d u r d}| � |||� t�t| |||d�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@ rJ 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 rU � s& � � �rU 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 rM rN r r r r � s zAESOCB3.generate_keyr c C sR |d u r d}t |�| jks(t |�| jkr0td��| �|||� t�t| |||d�S r! r% r* r r r r+ � s �zAESOCB3.encryptc C s. |d u r d}| � |||� t�t| |||d�S r, r- r* r r r r/ s zAESOCB3.decryptNc C s< t �d|� t �d|� t �d|� t|�dkr8td��d S )Nr r r r0 z Nonce must be 96-bits (12 bytes)r1 r* r r r r( s zAESOCB3._check_paramsrV r r r r rW � s& � � �rW )r r8 Z cryptographyr r Z$cryptography.hazmat.backends.opensslr Z,cryptography.hazmat.backends.openssl.backendr �objectr r: rU rW r r r r �<module> s ?U?