File: //lib/python3/dist-packages/cryptography/hazmat/primitives/ciphers/__pycache__/base.cpython-38.pyc
U
�e�]� � @ s� d dl mZmZmZ d dlZd dlZd dlmZ d dlm Z m
Z
m Z m Z m
Z
d dlmZ d dlmZ e�ej�G dd� de��Ze�ej�G d d
� d
e��Ze�ej�G d d � d e��Ze�ej�G d
d� de��Ze�ej�G dd� de��Ze�ej�G dd� de��ZG dd� de�Ze� e�G dd� de��Z e� e�e� e�e� e�G dd� de����Z e� e�G dd� de ��Z dS )� )�absolute_import�division�print_functionN)�utils)�AlreadyFinalized�AlreadyUpdated�NotYetFinalized�UnsupportedAlgorithm�_Reasons)�
CipherBackend)�modesc @ s( e Zd Zejdd� �Zejdd� �ZdS )�CipherAlgorithmc C s dS )zE
A string naming this mode (e.g. "AES", "Camellia").
N� ��selfr r �M/usr/lib/python3/dist-packages/cryptography/hazmat/primitives/ciphers/base.py�name s zCipherAlgorithm.namec C s dS )zW
The size of the key being used as an integer in bits (e.g. 128, 256).
Nr r r r r �key_size s zCipherAlgorithm.key_sizeN)�__name__�
__module__� __qualname__�abc�abstractpropertyr r r r r r r
s
r
c @ s e Zd Zejdd� �ZdS )�BlockCipherAlgorithmc C s dS )zK
The size of a block as an integer in bits (e.g. 64, 128).
Nr r r r r �
block_size% s z BlockCipherAlgorithm.block_sizeN)r r r r r r r r r r r # s r c @ s6 e Zd Zejdd� �Zejdd� �Zejdd� �ZdS )�
CipherContextc C s dS )zk
Processes the provided bytes through the cipher and returns the results
as bytes.
Nr �r �datar 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 Zbufr r r � update_into5 s zCipherContext.update_intoc 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 �abstractmethodr r r! r r r r r , s
r c @ s e Zd Zejdd� �ZdS )�AEADCipherContextc C s dS )z3
Authenticates the provided bytes.
Nr r r r r � authenticate_additional_dataE s z.AEADCipherContext.authenticate_additional_dataN)r r r r r" r$ r r r r r# C s r# c @ s e Zd Zejdd� �ZdS )�AEADDecryptionContextc C s dS )z�
Returns the results of processing the final block as bytes and allows
delayed passing of the authentication tag.
Nr )r �tagr r r �finalize_with_tagN s z'AEADDecryptionContext.finalize_with_tagN)r r r r r" r' r r r r r% L s r% c @ s e Zd Zejdd� �ZdS )�AEADEncryptionContextc C s dS )zb
Returns tag bytes. This is only available after encryption is
finalized.
Nr r r r r r& X s zAEADEncryptionContext.tagN)r r r r r r& r r r r r( V s r( c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�Cipherc C sP t |t�stdtj��t |t�s(td��|d k r:|�|� || _|| _ || _
d S )Nz0Backend object does not implement CipherBackend.z&Expected interface of CipherAlgorithm.) �
isinstancer r r
ZBACKEND_MISSING_INTERFACEr
� TypeErrorZvalidate_for_algorithm� algorithm�mode�_backend)r r, r- Zbackendr r r �__init__a 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)
r* r- r �ModeWithAuthenticationTagr& �
ValueErrorr. Z create_symmetric_encryption_ctxr, � _wrap_ctx�r �ctxr r r � encryptorr s � �zCipher.encryptorc C s | j �| j| j�}| j|dd�S )NFr0 )r. Z create_symmetric_decryption_ctxr, r- r4 r5 r r r � decryptor} s
�zCipher.decryptorc C s0 t | jtj�r$|rt|�S t|�S nt|�S d S �N)r* r- r r2 �_AEADEncryptionContext�_AEADCipherContext�_CipherContext)r r6 r1 r r r r4 � s
zCipher._wrap_ctxN)r r r r/ r7 r8 r4 r r r r r) ` s r) c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r< c C s
|| _ d S r9 )�_ctxr5 r r r r/ � s z_CipherContext.__init__c C s | j d krtd��| j �|�S �N� Context was already finalized.)r= r r r r r r r � s
z_CipherContext.updatec C s | j d krtd��| j �||�S r> )r= r r r r r r r � s
z_CipherContext.update_intoc C s& | j d krtd��| j �� }d | _ |S r> )r= r r! r r r r r! � s
z_CipherContext.finalizeN)r r r r/ r r r! r r r r r< � s r< c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )r; c C s" || _ d| _d| _d | _d| _d S )Nr F)r= �_bytes_processed�_aad_bytes_processed�_tag�_updatedr5 r r r r/ � s
z_AEADCipherContext.__init__c C sV | j d krtd��d| _| j|7 _| j| j jjkrRtd�| j jj| j jj���d S )Nr? Tz+{} has a maximum encrypted byte limit of {}) r= r rC r@ �_modeZ_MAX_ENCRYPTED_BYTESr3 �formatr )r � data_sizer r r � _check_limit� s
��z _AEADCipherContext._check_limitc C s | � t|�� | j�|�S r9 )rG �lenr= r r r r r r � s z_AEADCipherContext.updatec C s | � t|�� | j�||�S r9 )rG rH r= r r r r r r � s z _AEADCipherContext.update_intoc C s0 | j d krtd��| j �� }| j j| _d | _ |S r> )r= r r! r& rB r r r r r! � s
z_AEADCipherContext.finalizec C s2 | j d krtd��| j �|�}| j j| _d | _ |S r> )r= r r' r& rB )r r& r r r r r' � s
z$_AEADCipherContext.finalize_with_tagc C sn | j d krtd��| jr td��| jt|�7 _| j| j jjkr^td� | j jj
| j jj���| j � |� d S )Nr? z'Update has been called on this context.z%{} has a maximum AAD byte limit of {}) r= r rC r rA rH rD Z_MAX_AAD_BYTESr3 rE r r$ r r r r r$ � s
��z/_AEADCipherContext.authenticate_additional_dataN)
r r r r/ rG r r r! r' r$ r r r r r; � s r; c @ s e Zd Zedd� �ZdS )r: c C s | j d k rtd��| jS )Nz4You must finalize encryption before getting the tag.)r= r rB r r r r r& � s
z_AEADEncryptionContext.tagN)r r r �propertyr& r r r r r: � s r: ) Z
__future__r r r r ZsixZ cryptographyr Zcryptography.exceptionsr r r r r
Z'cryptography.hazmat.backends.interfacesr Z&cryptography.hazmat.primitives.ciphersr Z
add_metaclass�ABCMeta�objectr
r r r# r% r( r) Zregister_interfacer< r; r: r r r r �<module> s6
-=