a
a @ s d dl Z d dlZd dlmZ d dlmZmZmZ d dlm Z d dl
m Z G dd de j dZ
G d d
d
e j dZG d d d e j dZG d
d de j dZG dd deZee
G dd deZeeee
eeG dd deZeeG dd deZdS ) N)utils)AlreadyFinalizedAlreadyUpdatedNotYetFinalized)CipherAlgorithm)modesc @ sL e Zd ZejeedddZejeedddZejedddZ d S )
CipherContextdatareturnc C s dS )zk
Processes the provided bytes through the cipher and returns the results
as bytes.
N selfr
r r Q/usr/lib64/python3.9/site-packages/cryptography/hazmat/primitives/ciphers/base.pyupdate 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)
__name__
__module__ __qualname__abcabstractmethodbytesr intr r r r r r r s r ) metaclassc @ s" e Zd ZejeddddZdS )AEADCipherContextNr c C s dS )z3
Authenticates the provided bytes.
Nr r
r r r authenticate_additional_data* s z.AEADCipherContext.authenticate_additional_data)r r r r r r r r r r r r ) s r c @ s" e Zd ZejeedddZdS )AEADDecryptionContexttagr 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_tag2 s z'AEADDecryptionContext.finalize_with_tagN)r r r r r r r$ r r r r r! 1 s r! c @ s e Zd ZejedddZdS )AEADEncryptionContextr c 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 abstractpropertyr r# r r r r r% : s r% c @ sB e Zd Zd eejej ejdddZ dd Z
dd Z d d
Z dS ) CipherN) algorithmmodebackendc C s4 t |tstd|d ur$|| || _|| _d S )Nz&Expected interface of CipherAlgorithm.)
isinstancer TypeErrorZvalidate_for_algorithmr( r) )r r( r) r* r r r __init__D s
zCipher.__init__c C sL t | jtjr"| jjd ur"tdddlm} || j | j}| j
|ddS )Nz0Authentication tag must be None when encrypting.r r* Tencrypt) r+ r) r ModeWithAuthenticationTagr#
ValueError,cryptography.hazmat.backends.openssl.backendr* Z create_symmetric_encryption_ctxr( _wrap_ctxr r* ctxr r r encryptorT s zCipher.encryptorc C s* ddl m} || j| j}| j|ddS )Nr r. Fr/ )r3 r* Z create_symmetric_decryption_ctxr( r) r4 r5 r r r decryptora s
zCipher.decryptorc C s0 t | jtjr$|rt|S t|S nt|S d S N)r+ r) r r1 _AEADEncryptionContext_AEADCipherContext_CipherContext)r r6 r0 r r r r4 i s
zCipher._wrap_ctx)N)
r r r r typingOptionalr ZModeAnyr- r7 r8 r4 r r r r r' C s
r' c @ sB e Zd Zdd ZeedddZeedddZedd d
Zd S ) r< c C s
|| _ d S r9 )_ctxr r6 r r r r- u s z_CipherContext.__init__r c C s | j d u rtd| j |S N Context was already finalized.)r@ r r r
r r r r x s
z_CipherContext.updatec C s | j d u rtd| j ||S rB )r@ r r r r r r r } s
z_CipherContext.update_intor c C s& | j d u rtd| j }d | _ |S rB )r@ r r r
r r r r s
z_CipherContext.finalizeN) r r r r- r r r r r r r r r r< s s r< c @ sr e Zd Zdd ZeddddZeeddd Zeedd
d Zed d
dZ eedddZ
eddddZ dS )r; c C s" || _ d| _d| _d | _d| _d S )Nr F)r@ _bytes_processed_aad_bytes_processed_tag_updatedrA r r r r- s
z_AEADCipherContext.__init__N) data_sizer c C sV | j d u rtdd| _| j|7 _| j| j jjkrRtd| j jj| j jjd S )NrC Tz+{} has a maximum encrypted byte limit of {}) r@ r rG rD _modeZ_MAX_ENCRYPTED_BYTESr2 formatname)r rH r r r _check_limit s
z _AEADCipherContext._check_limitr c C s | t| | j|S r9 )rL lenr@ r r
r r r r s z_AEADCipherContext.updatec C s | t| | j||S r9 )rL rM r@ r r r r r r s z _AEADCipherContext.update_intor c C s0 | j d u rtd| j }| j j| _d | _ |S rB )r@ r r r# rF r
r r r r s
z_AEADCipherContext.finalizer" c C s2 | j d u rtd| j |}| j j| _d | _ |S rB )r@ r r$ r# rF )r r# r
r r r r$ s
z$_AEADCipherContext.finalize_with_tagc C sn | j d u rtd| jr td| jt|7 _| j| j jjkr^td | j jj
| j jj| j | d S )NrC z'Update has been called on this context.z%{} has a maximum AAD byte limit of {}) r@ r rG r rE rM rI Z_MAX_AAD_BYTESr2 rJ rK r r
r r r r s
z/_AEADCipherContext.authenticate_additional_data) r r r r- r rL r r r r r$ r r r r r r; s r; c @ s e Zd ZeedddZdS )r: r c C s& | j d urtd| jd us J | jS )Nz4You must finalize encryption before getting the tag.)r@ r rF r r r r r# s
z_AEADEncryptionContext.tagN)r r r propertyr r# r r r r r: s r: )r r= Z cryptographyr Zcryptography.exceptionsr r r Z/cryptography.hazmat.primitives._cipheralgorithmr Z&cryptography.hazmat.primitives.ciphersr ABCMetar r r! r% objectr' Zregister_interfacer< r; r: r r r r s$ 0=