�
�3Tfb � �\ � d dl mZ d dlZd dlZd dlmZ d dlmZ d dlm Z m
Z
G d� dej �� Z dd �Z
dd
�Z d d �Z dd �Z d d
�Z G d� d� Z G d� de � Z G d� de � Z G d� d� Z G d� de � Z G d� de � Zy) � )� annotationsN)�utils)�AlreadyFinalized)�check_ansix923_padding�check_pkcs7_paddingc �X � e Zd Zej dd�� Zej dd�� Zy)�PaddingContextc � � y)zR
Pads the provided bytes and returns any available data as bytes.
N� )�self�datas �H/usr/lib/python3/dist-packages/cryptography/hazmat/primitives/padding.py�updatezPaddingContext.update � � � c � � y)z6
Finalize the padding, returns bytes.
Nr �r s r �finalizezPaddingContext.finalize r r N�r
�bytes�returnr �r r )�__name__�
__module__� __qualname__�abc�abstractmethodr r r r r r r s4 � ���� � � �
��� � � r r )� metaclassc �n � d| cxk rdk st d� � t d� �| dz dk7 r t d� �y )Nr i� z%block_size must be in range(0, 2041).� z#block_size must be a multiple of 8.)�
ValueError)�
block_sizes r �_byte_padding_checkr# sI � �
�� #�t� #��@�A�A�
$��@�A�A��A�~����>�?�?� r c � � | � t d� �t j d|� | t |� z
} t | � |dz z }| d ||dz z }| ||dz z d } | |fS )N� Context was already finalized.r
r )r r �_check_bytesliker �len��buffer_r
r" �finished_blocks�results r �_byte_padding_updater, ( s{ � � �� � ?�@�@� ���6�4�(� �u�T�{��G��'�l�z�Q��7�O�
�:��*��/�:�
;�F��o��q� �9�;�<�G� �F�?�r c �T � | � t d� �|dz t | � z
}| ||� z S )Nr% r )r r' )r) r" � paddingfn�pad_sizes r �_byte_padding_padr0 : s7 � �
�� � ?�@�@��Q���W� �-�H� �Y�x�(� (�(r c �� � | � t d� �t j d|� | t |� z
} t t | � |dz z dz
d� }| d ||dz z }| ||dz z d } | |fS )Nr% r
r � r )r r r& r �maxr' r( s r �_byte_unpadding_updater4 F s� � � �� � ?�@�@� ���6�4�(� �u�T�{��G��#�g�,�:��?�;�a�?��C�O�
�:��*��/�:�
;�F��o��q� �9�;�<�G� �F�?�r c � � | � t d� �t | � |dz k7 r t d� � || � }|s t d� �| d }| d | S )Nr% r zInvalid padding bytes.���)r r' r! )r) r" �checkfn�validr/ s r �_byte_unpadding_checkr9 X sb � �
�� � ?�@�@�
�7�|�z�Q��&��1�2�2� �G� �E� ��1�2�2��r�{�H� �:�X�I� � r c �$ � e Zd Zdd�Zdd�Zdd�Zy)�PKCS7c �( � t |� || _ y �N�r# r" �r r" s r �__init__zPKCS7.__init__m � � ��J�'�$��r c �, � t | j � S r= )�_PKCS7PaddingContextr" r s r �padderz PKCS7.padderq s � �#�D�O�O�4�4r c �, � t | j � S r= )�_PKCS7UnpaddingContextr" r s r �unpadderzPKCS7.unpaddert s � �%�d�o�o�6�6r N�r" �int�r r �r r r r@ rD rG r r r r; r; l s � �%�5�7r r; c �8 � e Zd ZU ded< dd�Zd d�Zd
d�Zd d�Zy) rC �typing.Optional[bytes]�_bufferc � � || _ d| _ y �Nr �r" rN r? s r r@ z _PKCS7PaddingContext.__init__{ � � �$����� r c �X � t | j || j � \ | _ }|S r= �r, rN r" �r r
r+ s r r z_PKCS7PaddingContext.update� �) � � 3� �L�L�$����
� �� �f� �
r c � � t |g� |z S r= �r �r �sizes r �_paddingz _PKCS7PaddingContext._padding� s � ��d�V�}�t�#�#r c �j � t | j | j | j � }d | _ |S r= �r0 rN r" r[ �r r+ s r r z _PKCS7PaddingContext.finalize� �- � �"� �L�L�$�/�/�4�=�=�
�� �� ��
r NrH r �rZ rI r r r �r r r �__annotations__r@ r r[ r r r r rC rC x s � �
#�#��
� $�r rC c �0 � e Zd ZU ded< dd�Zdd�Zd d�Zy)
rF rM rN c � � || _ d| _ y rP rQ r? s r r@ z _PKCS7UnpaddingContext.__init__� rR r c �X � t | j || j � \ | _ }|S r= �r4 rN r" rU s r r z _PKCS7UnpaddingContext.update� �) � � 5� �L�L�$����
� �� �f� �
r c �^ � t | j | j t � }d | _ |S r= )r9 rN r" r r^ s r r z _PKCS7UnpaddingContext.finalize� s* � �&� �L�L�$�/�/�+>�
�� �� ��
r NrH r r �r r r rb r@ r r r r r rF rF � s � �
#�#��
� r rF c �$ � e Zd Zdd�Zdd�Zdd�Zy)�ANSIX923c �( � t |� || _ y r= r> r? s r r@ zANSIX923.__init__� rA r c �, � t | j � S r= )�_ANSIX923PaddingContextr" r s r rD zANSIX923.padder� s � �&�t���7�7r c �, � t | j � S r= )�_ANSIX923UnpaddingContextr" r s r rG zANSIX923.unpadder� s � �(�� � �9�9r NrH rJ rK r r r rk rk � s � �%�8�:r rk c �8 � e Zd ZU ded< dd�Zd d�Zd
d�Zd d�Zy) rn rM rN c � � || _ d| _ y rP rQ r? s r r@ z _ANSIX923PaddingContext.__init__� rR r c �X � t | j || j � \ | _ }|S r= rT rU s r r z _ANSIX923PaddingContext.update� rV r c �@ � t dg� |dz
z t |g� z S )Nr r2 rX rY s r r[ z _ANSIX923PaddingContext._padding� s" � ��a�S�z�T�A�X�&���v� �6�6r c �j � t | j | j | j � }d | _ |S r= r] r^ s r r z _ANSIX923PaddingContext.finalize� r_ r NrH r r` r ra r r r rn rn � s � �
#�#��
� 7�r rn c �0 � e Zd ZU ded< dd�Zdd�Zd d�Zy)
rp rM rN c � � || _ d| _ y rP rQ r? s r r@ z"_ANSIX923UnpaddingContext.__init__� rR r c �X � t | j || j � \ | _ }|S r= rf rU s r r z _ANSIX923UnpaddingContext.update� rg r c �^ � t | j | j t � }d | _ |S r= )r9 rN r" r r^ s r r z"_ANSIX923UnpaddingContext.finalize� s+ � �&� �L�L� �O�O� "�
��
�� ��
r NrH r r ri r r r rp rp � s � �
#�#��
� r rp )r" rI r �None)r) rM r
r r" rI r ztyping.Tuple[bytes, bytes])r) rM r" rI r. z typing.Callable[[int], bytes]r r )r) rM r" rI r7 z typing.Callable[[bytes], int]r r )�
__future__r r �typing� cryptographyr �cryptography.exceptionsr �"cryptography.hazmat.bindings._rustr r �ABCMetar r# r, r0 r4 r9 r; rC rF rk rn rp r r r �<module>r� s ��
#�
�
� � 4�� �s�{�{� � @��
#��+0��>A�� ��$ )�
#� )�� )� -� )� � )��
#��+0��>A�� ��$ �
#� �� � +� � � �( 7� 7��>� �2�^� �, :� :��n� �2�� r |