관리-도구
편집 파일: aead.cpython-36.opt-1.pyc
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_import�division�print_function)� InvalidTag� c C sd ddl m}m}m} t| |�r"dS t| |�rFdjt| j�d �jd�S djt| j�d �jd�S d S )Nr )�AESCCM�AESGCM�ChaCha20Poly1305s chacha20-poly1305z aes-{}-ccm� �asciiz aes-{}-gcm) �+cryptography.hazmat.primitives.ciphers.aeadr r r � isinstance�format�len�_key�encode)�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|jd��r$| j j || j j|| jj�} | j| dk� | jj|�} | jj|�} | j j|| jj| jj| | t |t k��} | j| dk� |S )Nr s -ccm)�_libZEVP_get_cipherbyname�openssl_assert�_ffi�NULLZEVP_CIPHER_CTX_new�gcZEVP_CIPHER_CTX_freeZEVP_CipherInit_ex�int�_ENCRYPTZ EVP_CIPHER_CTX_set_key_lengthr �EVP_CIPHER_CTX_ctrlZEVP_CTRL_AEAD_SET_IVLEN�_DECRYPTZEVP_CTRL_AEAD_SET_TAG�endswithZ from_buffer) �backend� cipher_name�key�nonce�tagZtag_lenZ operationZ evp_cipher�ctx�resZ 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_data�outlenr& 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| �} | dk�r.| 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 �<module> s 0