File: //lib/python3/dist-packages/cryptography/hazmat/primitives/ciphers/__pycache__/base.cpython-38.pyc
U
�)%a� � @ sB d dl Z d dlZd dlmZ d dlmZmZmZmZm Z d dl
m Z d dl m
Z
d dlmZ d dlmZ G dd � d e jd
�ZG d d � d e jd
�ZG d
d� de jd
�ZG dd� de jd
�ZG dd� de jd
�ZG dd� de�Ze�e�G dd� de��Ze�e�e�e�e�e�G dd� de����Z e�e�G dd� de ��Z dS )� N)�utils)�AlreadyFinalized�AlreadyUpdated�NotYetFinalized�UnsupportedAlgorithm�_Reasons)� _get_backend)�
CipherBackend)�CipherAlgorithm)�modesc @ s e Zd Zejed�dd��ZdS )�BlockCipherAlgorithm��returnc C s dS )zK
The size of a block as an integer in bits (e.g. 64, 128).
N� ��selfr r �M/usr/lib/python3/dist-packages/cryptography/hazmat/primitives/ciphers/base.py�
block_size s z BlockCipherAlgorithm.block_sizeN)�__name__�
__module__� __qualname__�abc�abstractproperty�intr r r r r r s r )� metaclassc @ sL e Zd Zejeed�dd��Zejeed�dd��Zejed�dd��Z d S )
�
CipherContext��datar c C s dS )zk
Processes the provided bytes through the cipher and returns the results
as bytes.
Nr �r r r r r �update s zCipherContext.updatec C s dS )z�
Processes the provided bytes and writes the resulting data into the
provided buffer. Returns the number of bytes written.
Nr �r r �bufr r r � update_into' s zCipherContext.update_intor
c C s dS )zM
Returns the results of processing the final block as bytes.
Nr r r r r �finalize. s zCipherContext.finalizeN)
r r r r �abstractmethod�bytesr r r" r# r r r r r s r c @ s" e Zd Zejedd�dd��ZdS )�AEADCipherContextNr c C s dS )z3
Authenticates the provided bytes.
Nr r r r r � authenticate_additional_data6 s z.AEADCipherContext.authenticate_additional_data)r r r r r$ r% r' r r r r r& 5 s r& c @ s" e Zd Zejeed�dd��ZdS )�AEADDecryptionContext��tagr c C s dS )z�
Returns the results of processing the final block as bytes and allows
delayed passing of the authentication tag.
Nr )r r* r r r �finalize_with_tag> s z'AEADDecryptionContext.finalize_with_tagN)r r r r r$ r% r+ r r r r r( = s r( c @ s e Zd Zejed�dd��ZdS )�AEADEncryptionContextr
c C s dS )zb
Returns tag bytes. This is only available after encryption is
finalized.
Nr r r r r r* G s zAEADEncryptionContext.tagN)r r r r r r% r* r r r r r, F s r, c @ s> e Zd Zd eejej d�dd�Zdd� Z dd� Z
d d
� Z dS ) �CipherN)� algorithm�modec C sX t |�}t|t�s tdtj��t|t�s0td��|d k rB|�|� || _ || _
|| _ d S )Nz0Backend object does not implement CipherBackend.z&Expected interface of CipherAlgorithm.) r �
isinstancer r r �BACKEND_MISSING_INTERFACEr
� TypeErrorZvalidate_for_algorithmr. r/ �_backend)r r. r/ �backendr r r �__init__P s
�
zCipher.__init__c C sB t | jtj�r"| jjd k r"td��| j�| j| j�}| j |dd�S )Nz0Authentication tag must be None when encrypting.T��encrypt)
r0 r/ r �ModeWithAuthenticationTagr* �
ValueErrorr3 � create_symmetric_encryption_ctxr. � _wrap_ctx�r �ctxr r r � encryptorg s � �zCipher.encryptorc C s | j �| j| j�}| j|dd�S )NFr6 )r3 � create_symmetric_decryption_ctxr. r/ r; r< r r r � decryptorr s
�zCipher.decryptorc C s0 t | jtj�r$|rt|�S t|�S nt|�S d S �N)r0 r/ r r8 �_AEADEncryptionContext�_AEADCipherContext�_CipherContext)r r= r7 r r r r; x s
zCipher._wrap_ctx)N) r r r r
�typing�Optionalr ZModer5 r> r@ r; r r r r r- O s �
� r- c @ sB e Zd Zdd� Zeed�dd�Zeed�dd�Zed�d d
�Zd S ) rD c C s
|| _ d S rA )�_ctxr< r r r r5 � s z_CipherContext.__init__r c C s | j d krtd��| j �|�S �N� Context was already finalized.)rG r r r r r r r � s
z_CipherContext.updatec C s | j d krtd��| j �||�S rH )rG r r" r r r r r" � s
z_CipherContext.update_intor
c C s& | j d krtd��| j �� }d | _ |S rH )rG r r# r r r r r# � s
z_CipherContext.finalizeN) r r r r5 r% r r r" r# r r r r rD � s rD c @ sp e Zd Zdd� Zed�dd�Zeed�dd�Zeed�d d
�Zed �d d
�Z eed�dd�Z
edd�dd�Z dS )rC c C s" || _ d| _d| _d | _d| _d S )Nr F)rG �_bytes_processed�_aad_bytes_processed�_tag�_updatedr< r r r r5 � s
z_AEADCipherContext.__init__)� data_sizec C sV | j d krtd��d| _| j|7 _| j| j jjkrRtd�| j jj| j jj���d S )NrI Tz+{} has a maximum encrypted byte limit of {}) rG r rM rJ �_modeZ_MAX_ENCRYPTED_BYTESr9 �format�name)r rN r r r � _check_limit� s
��z _AEADCipherContext._check_limitr c C s | � t|�� | j�|�S rA )rR �lenrG r r r r r r � s z_AEADCipherContext.updatec C s | � t|�� | j�||�S rA )rR rS rG r" r r r r r" � s z _AEADCipherContext.update_intor
c C s0 | j d krtd��| j �� }| j j| _d | _ |S rH )rG r r# r* rL r r r r r# � s
z_AEADCipherContext.finalizer) c C s2 | j d krtd��| j �|�}| j j| _d | _ |S rH )rG r r+ r* rL )r r* r r r r r+ � s
z$_AEADCipherContext.finalize_with_tagNc C sn | j d krtd��| jr td��| jt|�7 _| j| j jjkr^td� | j jj
| j jj���| j � |� d S )NrI z'Update has been called on this context.z%{} has a maximum AAD byte limit of {}) rG r rM r rK rS rO Z_MAX_AAD_BYTESr9 rP rQ r' r r r r r' � s
��z/_AEADCipherContext.authenticate_additional_data) r r r r5 r rR r% r r" r# r+ r' r r r r rC � s rC c @ s e Zd Zeed�dd��ZdS )rB r
c C s& | j d k rtd��| jd k s t�| jS )Nz4You must finalize encryption before getting the tag.)rG r rL �AssertionErrorr r r r r* � s
�z_AEADEncryptionContext.tagN)r r r �propertyr% r* r r r r rB � s rB ) r rE � cryptographyr �cryptography.exceptionsr r r r r � cryptography.hazmat.backendsr �'cryptography.hazmat.backends.interfacesr Z/cryptography.hazmat.primitives._cipheralgorithmr
�&cryptography.hazmat.primitives.ciphersr �ABCMetar r r& r( r, �objectr- �register_interfacerD rC rB r r r r �<module> s* 3=