Edit File: aead.cpython-36.pyc
3 l�_� � @ 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� Zd d � Zd d � Z d S )�ChaCha20Poly1305� � c C sD t j| �stjdtjj��tjd|� t|�dkr:t d��|| _ d S )Nz<ChaCha20Poly1305 is not supported by this version of OpenSSL�keyr z&ChaCha20Poly1305 key must be 32 bytes.) r Zaead_cipher_supportedr ZUnsupportedAlgorithmZ_ReasonsZUNSUPPORTED_CIPHERr �_check_byteslike�len� ValueError�_key)�selfr � r � /usr/lib64/python3.6/aead.py�__init__ s zChaCha20Poly1305.__init__c C s t jd�S )Nr )�os�urandom)�clsr r r � generate_key s z ChaCha20Poly1305.generate_keyc C sR |d kr d}t |�| jks(t |�| jkr0td��| j|||� tjt| |||d�S )N� z1Data 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}| j |||� tjt| |||d�S )Nr r )r r �_decryptr )r r r r! r r r �decrypt/ s zChaCha20Poly1305.decryptc C s< t jd|� t jd|� t jd|� t|�dkr8td��d S )Nr r r! � zNonce must be 12 bytes)r r � _check_bytesr r )r r r r! r r r r 6 s z ChaCha20Poly1305._check_paramsNl ) �__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 dd� Z dS )�AESCCMr r r c C sR t jd|� t|�dkr td��|| _t|t�s8td��|dkrHtd ��|| _d S )Nr r � r z)AESCCM key must be 128, 192, or 256 bits.z tag_length must be an integer� � � � r% � zInvalid tag_length)r r, r )r- r. r/ r0 r% r1 r ) r r r r r � isinstance�int� TypeError� _tag_length)r r Z tag_lengthr r r r A s zAESCCM.__init__c C s0 t |t�std��|dkr"td��tj|d �S )Nz bit_length must be an integer� � � z#bit_length must be 128, 192, or 256r/ )r6 r7 r8 )r2 r3 r4 r r r )r � bit_lengthr r r r O s zAESCCM.generate_keyc C sd |d kr d}t |�| jks(t |�| jkr0td��| j|||� | j|t |�� tjt| |||| j�S )Nr z1Data or associated data too long. Max 2**32 bytes) r r r r �_validate_lengthsr r r r5 )r r r r! r r r r" Y s zAESCCM.encryptc C s0 |d kr d}| j |||� tjt| |||| j�S )Nr )r r r# r r5 )r r r r! r r r r$ i s zAESCCM.decryptc C s( dt |� }dd| |k r$td��d S )N� r r/ zData too long for nonce)r r )r r Zdata_lenZl_valr r r r: r s zAESCCM._validate_lengthsc C sL t jd|� t jd|� t jd|� dt|� ko:dkn sHtd��d S )Nr r r! � � z$Nonce must be between 7 and 13 bytes)r r r& r r )r r r r! r r r r y s zAESCCM._check_paramsNl )r ) r' r( r) r r r* r r" r$ r: r r r r r r+ > s r+ c @ s< e Zd ZdZdd� Zedd� �Zdd� Zd d � Zd d � Z d S )�AESGCMr r c C s* t jd|� t|�dkr td��|| _d S )Nr r r, r z)AESGCM key must be 128, 192, or 256 bits.)r r, r )r r r r r )r r r r r r � s zAESGCM.__init__c C s0 t |t�std��|dkr"td��tj|d �S )Nz bit_length must be an integerr6 r7 r8 z#bit_length must be 128, 192, or 256r/ )r6 r7 r8 )r2 r3 r4 r r r )r r9 r r r r � s zAESGCM.generate_keyc C sR |d kr d}t |�| jks(t |�| jkr0td��| j|||� tjt| |||d�S )Nr z1Data or associated data too long. Max 2**32 bytesr )r r r r r r r )r r r r! r r r r" � s zAESGCM.encryptc C s. |d kr d}| j |||� tjt| |||d�S )Nr r )r r r# r )r r r r! r r r r$ � s zAESGCM.decryptc C s< t jd|� t jd|� t jd|� t|�dkr8td��d S )Nr r r! r z Nonce must be at least 1 byte)r r r& r r )r r r r! r r r r � s zAESGCM._check_paramsNl ) r' r( r) r r r* r r" r$ r r r r r r> � s r> )Z __future__r r r r Z cryptographyr r Z$cryptography.hazmat.backends.opensslr Z,cryptography.hazmat.backends.openssl.backendr �objectr r+ r> r r r r �<module> s 0C
Back to File Manager