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_importdivisionprint_functionN)
exceptionsutils)aead)backendc @ s< e Zd ZdZdd Zedd Zdd Zd d
Zd d Z d
S )ChaCha20Poly1305 c C sD t j| stjdtjjtjd| 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 |tstd|dkr"tdtj|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| |||dS )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| |||dS )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|dkr8tdd 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