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 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 � s 0C