3
l_ @ sd d dl mZmZmZ d dlmZ dZd Zdd Zdd Z dd Z
d
d Z d d
Z dd Z
dd ZdS ) )absolute_importdivisionprint_function)
InvalidTag c C sd ddl m}m}m} t| |r"dS t| |rFdjt| jd jdS djt| jd jdS d S )Nr )AESCCMAESGCMChaCha20Poly1305s chacha20-poly1305z
aes-{}-ccm asciiz
aes-{}-gcm) +cryptography.hazmat.primitives.ciphers.aeadr r r
isinstanceformatlen_keyencode)cipherr r r r /usr/lib64/python3.6/aead.py_aead_cipher_name s
r c C st | j j|}| j|| jjk | j j }| jj|| j j}| j j||| jj| jj| jjt |t
k} | j| dk | j j |t |} | j| dk | j j
|| j jt || jj} | j| dk |tkr| j j
|| j jt ||} | j| dk n4|jdr$| j j
|| j j|| jj} | j| dk | jj|}
| jj|} | j j|| jj| jj| |
t |t
k} | j| dk |S )Nr s -ccm)_libZEVP_get_cipherbynameopenssl_assert_ffiNULLZEVP_CIPHER_CTX_newgcZEVP_CIPHER_CTX_freeZEVP_CipherInit_exint_ENCRYPTZ EVP_CIPHER_CTX_set_key_lengthr EVP_CIPHER_CTX_ctrlZEVP_CTRL_AEAD_SET_IVLEN_DECRYPTZEVP_CTRL_AEAD_SET_TAGendswithZ from_buffer) backend cipher_namekeynoncetagZtag_lenZ operationZ
evp_cipherctxresZ nonce_ptrZkey_ptrr r r _aead_setup sN
r' c C s: | j jd}| jj|| j j|| j j|}| j|dk d S )Nzint *r )r newr EVP_CipherUpdater r )r r% Zdata_lenZintptrr& r r r _set_lengthN s r* c C s: | j jd}| jj|| j j||t|}| j|dk d S )Nzint *r )r r( r r) r r r )r r% associated_dataoutlenr& r r r _process_aadV s r- c C s^ | j jd}| j jdt|}| jj||||t|}| j|dk | j j||d d d S )Nzint *zunsigned char[]r )r r( r r r) r buffer)r r% datar, bufr& r r r
_process_data^ s
r1 c C s ddl m} t|}t| ||j|d |t}t||rDt| |t| t | || t
| ||} | j j d}
| j
j|| j j|
} | j| dk | j|
d dk | j j d|} | j
j|| j
j|| } | j| dk | j j| d d }
| |
S )Nr )r zint *zunsigned char[])r r r r' r r r
r* r r- r1 r r( r EVP_CipherFinal_exr r r ZEVP_CTRL_AEAD_GET_TAGr. )r r r# r/ r+
tag_lengthr r! r% processed_datar, r& Ztag_bufr$ r r r _encryptf s$
r5 c C s2 ddl m} t||k r t|| d }|d | }t|}t| ||j|||t} t||rpt | | t| t
| | | t||r| j j d}
| j j dt|} | j
j| | |
|t|} | dkr| j t| j j| |
d d d }
nBt| | |}
| j j d}
| j
j| | j j|
} | dkr.| j t|
S )Nr )r zint *zunsigned char[]r )r r r r r r' r r r
r* r- r r( r r) Z_consume_errorsr. r1 r2 r )r r r# r/ r+ r3 r r$ r! r% r, r0 r& r4 r r r _decrypt s4
r6 N)Z
__future__r r r Zcryptography.exceptionsr r r r r' r* r- r1 r5 r6 r r r r s 0