3
l�_U � @ s� d dl mZmZmZ d dlZd dlZd dlmZ d dlm Z m
Z
m Z m Z m
Z
d dlmZ d dlmZ d dlmZ ejej�G dd � d e��Zejej�G d
d � d e��Zejej�G d d
� d
e��Zejej�G dd� de��Zejej�G dd� de��Zejej�G dd� de��Z G dd� de�Z ej e�G dd� de��Z ej e�ej e�ej e�G dd� de����Z ej e �G dd� de ��Z!dS ) � )�absolute_import�division�print_functionN)�utils)�AlreadyFinalized�AlreadyUpdated�NotYetFinalized�UnsupportedAlgorithm�_Reasons)� _get_backend)�
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 � /usr/lib64/python3.6/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 �update2 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_into9 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 0 s r c @ s e Zd Zejdd� �ZdS )�AEADCipherContextc C s dS )z3
Authenticates the provided bytes.
Nr )r r r r r � authenticate_additional_dataI s z.AEADCipherContext.authenticate_additional_dataN)r r r r r! r# r r r r r" G 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_tagR s z'AEADDecryptionContext.finalize_with_tagN)r r r r r! r& r r r r r$ P 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% \ s zAEADEncryptionContext.tagN)r r r r r r% r r r r r' Z s r' c @ s. e Zd Zd
dd�Zdd� Zdd� Zdd � ZdS ) �CipherNc C sX t |�}t|t�s tdtj��t|t�s0td��|d k rB|j|� || _ || _
|| _ d S )Nz0Backend object does not implement CipherBackend.z&Expected interface of CipherAlgorithm.) r �
isinstancer r r
ZBACKEND_MISSING_INTERFACEr � TypeErrorZvalidate_for_algorithm� algorithm�mode�_backend)r r+ r, Zbackendr r r �__init__e s
zCipher.__init__c C sB t | jtj�r"| jjd k r"td��| jj| 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 � encryptorw s zCipher.encryptorc C s | j j| j| j�}| j|dd�S )NF)r/ )r- Z create_symmetric_decryption_ctxr+ r, r2 )r r3 r r r � decryptor� s zCipher.decryptorc C s0 t | jtj�r$|rt|�S t|�S nt|�S d S )N)r) r, r
r0 �_AEADEncryptionContext�_AEADCipherContext�_CipherContext)r r3 r/ r r r r2 � s
zCipher._wrap_ctx)N)r r r r. r4 r5 r2 r r r r r( d s
r( c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r8 c C s
|| _ d S )N)�_ctx)r r3 r r r r. � s z_CipherContext.__init__c C s | j d krtd��| j j|�S )Nz Context was already finalized.)r9 r r )r r r r r r � s
z_CipherContext.updatec C s | j d krtd��| j j||�S )Nz Context was already finalized.)r9 r r )r r r r r r r � s
z_CipherContext.update_intoc C s& | j d krtd��| j j� }d | _ |S )Nz Context was already finalized.)r9 r r )r r r r r r � s
z_CipherContext.finalizeN)r r r r. r r r r r r r r8 � s r8 c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )r7 c C s" || _ d| _d| _d | _d| _d S )Nr F)r9 �_bytes_processed�_aad_bytes_processed�_tag�_updated)r r3 r r r r. � s
z_AEADCipherContext.__init__c C sV | j d krtd��d| _| j|7 _| j| j jjkrRtdj| j jj| j jj���d S )Nz Context was already finalized.Tz+{} has a maximum encrypted byte limit of {}) r9 r r= r: �_modeZ_MAX_ENCRYPTED_BYTESr1 �formatr )r Z data_sizer r r � _check_limit� s
z _AEADCipherContext._check_limitc C s | j t|�� | jj|�S )N)r@ �lenr9 r )r r r r r r � s z_AEADCipherContext.updatec C s | j t|�� | jj||�S )N)r@ rA r9 r )r r r r r r r � s z _AEADCipherContext.update_intoc C s0 | j d krtd��| j j� }| j j| _d | _ |S )Nz Context was already finalized.)r9 r r r% r< )r r r r r r � s
z_AEADCipherContext.finalizec C s2 | j d krtd��| j j|�}| j j| _d | _ |S )Nz Context was already finalized.)r9 r r& r% r< )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^tdj | j jj
| j jj���| j j |� d S )Nz Context was already finalized.z'Update has been called on this context.z%{} has a maximum AAD byte limit of {}) r9 r r= r r; rA r> Z_MAX_AAD_BYTESr1 r? r r# )r r r r r r# � s
z/_AEADCipherContext.authenticate_additional_dataN)
r r r r. r@ r r r r& r# r r r r r7 � s r7 c @ s e Zd Zedd� �ZdS )r6 c C s | j d k rtd��| jS )Nz4You must finalize encryption before getting the tag.)r9 r r< )r r r r r% � s
z_AEADEncryptionContext.tagN)r r r �propertyr% r r r r r6 � s r6 )"Z
__future__r r r r ZsixZ cryptographyr Zcryptography.exceptionsr r r r r
Z cryptography.hazmat.backendsr Z'cryptography.hazmat.backends.interfacesr Z&cryptography.hazmat.primitives.ciphersr
Z
add_metaclass�ABCMeta�objectr r r r" r$ r' r( Zregister_interfacer8 r7 r6 r r r r � s8 . =