관리-도구
편집 파일: base.cpython-39.opt-1.pyc
a �a � @ s d dl Z d dlZd dlmZ d dlmZmZmZ d dlm Z d dl m Z G dd� de j d�Z G d d � d e j d�ZG d d � d e j d�ZG d d� de j d�ZG dd� de�Ze�e �G dd� de��Ze�e�e�e �e�e�G dd� de����Ze�e�G dd� de��ZdS )� N)�utils)�AlreadyFinalized�AlreadyUpdated�NotYetFinalized)�CipherAlgorithm)�modesc @ sL e Zd Zejeed�dd��Zejeed�dd��Zejed�dd��Z d S ) � CipherContext��data�returnc 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.py�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_into�r 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__�abc�abstractmethod�bytesr �intr r r r r r r s r )� metaclassc @ s" e Zd Zejedd�dd��ZdS )�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 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_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 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# ; s zAEADEncryptionContext.tagN)r r r r �abstractpropertyr r# r r r r r% : s r% c @ sB e Zd Zd eejej ejd�dd�Z dd� Z dd� Z d d � Z dS ) �CipherN)� algorithm�mode�backendc C s4 t |t�std��|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 | jtj�r"| jjd ur"td��ddlm} |�| j | j�}| j |dd�S )Nz0Authentication tag must be None when encrypting.r �r* T��encrypt) r+ r) r �ModeWithAuthenticationTagr# � ValueError�,cryptography.hazmat.backends.openssl.backendr* Z create_symmetric_encryption_ctxr( � _wrap_ctx�r r* �ctxr r r � encryptorT s � �zCipher.encryptorc C s* ddl m} |�| j| j�}| j|dd�S )Nr r. Fr/ )r3 r* Z create_symmetric_decryption_ctxr( r) r4 r5 r r r � decryptora s �zCipher.decryptorc C s0 t | jtj�r$|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 �typing�Optionalr ZMode�Anyr- r7 r8 r4 r r r r r' C s � � r' c @ sB e Zd Zdd� Zeed�dd�Zeed�dd�Zed�d d �Zd S ) r< c C s || _ d S r9 )�_ctx�r 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� Zedd�dd�Zeed�dd �Zeed�d d �Zed �d d�Z eed�dd�Z edd�dd�Z 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 rtd��d| _| j|7 _| j| j jjkrRtd�| j jj| j jj���d S )NrC Tz+{} has a maximum encrypted byte limit of {}) r@ r rG rD �_modeZ_MAX_ENCRYPTED_BYTESr2 �format�name)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 Zeed�dd��ZdS )r: r c C s | j d urtd��| 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 �<module> s$ 0=