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 ejejG dd d eZejejG d
d d eZejejG d d
d
eZejejG dd deZejejG dd deZejejG dd deZ G dd deZ ej eG dd deZ ej eej eej eG dd deZ ej e G dd de Z!dS ) )absolute_importdivisionprint_functionN)utils)AlreadyFinalizedAlreadyUpdatedNotYetFinalizedUnsupportedAlgorithm_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.pyname 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__abcabstractpropertyr 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
ddZdd Zdd Zdd ZdS ) CipherNc C sX t |}t|ts tdtjt|ts0td|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 algorithmmode_backend)r r+ r, Zbackendr r r __init__e s
zCipher.__init__c C sB t | jtjr"| jjd k r"td| jj| j| j}| j |ddS )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|ddS )NF)r/ )r- Z create_symmetric_decryption_ctxr+ r, r2 )r r3 r r r decryptor s zCipher.decryptorc C s0 t | jtjr$|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 krtdd| _| j|7 _| j| j jjkrRtdj| j jj| j jjd 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_metaclassABCMetaobjectr r r r" r$ r' r( Zregister_interfacer8 r7 r6 r r r r s8 . =