o
    hD                     @  s   d dl mZ d dlZd dlZd dlmZ d dlmZ d dlm	Z	m
Z
mZ G dd dejdZd(ddZd)ddZd*ddZd)ddZd+ddZG d d! d!Zee	 ee
 G d"d# d#ZG d$d% d%eZG d&d' d'eZdS ),    )annotationsN)utils)AlreadyFinalized)PKCS7PaddingContextPKCS7UnpaddingContextcheck_ansix923_paddingc                   @  s,   e Zd Zejd	ddZejd
ddZdS )PaddingContextdatabytesreturnc                 C     dS )zR
        Pads the provided bytes and returns any available data as bytes.
        N )selfr	   r   r   Y/var/www/html/venv/lib/python3.10/site-packages/cryptography/hazmat/primitives/padding.pyupdate       zPaddingContext.updatec                 C  r   )z6
        Finalize the padding, returns bytes.
        Nr   r   r   r   r   finalize   r   zPaddingContext.finalizeNr	   r
   r   r
   r   r
   )__name__
__module____qualname__abcabstractmethodr   r   r   r   r   r   r      s
    r   )	metaclass
block_sizeintr   Nonec                 C  s<   d|   krdkst d t d| d dkrt dd S )Nr   i  z%block_size must be in range(0, 2041).   z#block_size must be a multiple of 8.)
ValueError)r   r   r   r   _byte_padding_check!   s   r!   buffer_bytes | Noner	   r
   tuple[bytes, bytes]c                 C  sh   | d u rt dtd| | t|7 } t| |d  }| d ||d   }| ||d  d  } | |fS )NContext was already finalized.r	   r   )r   r   _check_bytesliker
   lenr"   r	   r   Zfinished_blocksresultr   r   r   _byte_padding_update)   s   r*   	paddingfntyping.Callable[[int], bytes]c                 C  s,   | d u rt d|d t|  }| || S )Nr%   r   )r   r'   )r"   r   r+   pad_sizer   r   r   _byte_padding_pad;   s   r.   c                 C  sr   | d u rt dtd| | t|7 } tt| |d  d d}| d ||d   }| ||d  d  } | |fS )Nr%   r	   r      r   )r   r   r&   r
   maxr'   r(   r   r   r   _byte_unpadding_updateG   s   r1   checkfntyping.Callable[[bytes], int]c                 C  sR   | d u rt dt| |d krtd|| }|std| d }| d |  S )Nr%   r   zInvalid padding bytes.)r   r'   r    )r"   r   r2   validr-   r   r   r   _byte_unpadding_checkY   s   r6   c                   @  *   e Zd ZdddZdddZdd	d
ZdS )PKCS7r   r   c                 C     t | || _d S Nr!   r   r   r   r   r   r   __init__n      
zPKCS7.__init__r   r   c                 C  
   t | jS r:   )r   r   r   r   r   r   padderr      
zPKCS7.padderc                 C  r?   r:   )r   r   r   r   r   r   unpadderu   rA   zPKCS7.unpadderNr   r   r   r   r   r   r   r=   r@   rB   r   r   r   r   r8   m       

r8   c                   @  r7   )ANSIX923r   r   c                 C  r9   r:   r;   r<   r   r   r   r=   ~   r>   zANSIX923.__init__r   r   c                 C  r?   r:   )_ANSIX923PaddingContextr   r   r   r   r   r@      rA   zANSIX923.padderc                 C  r?   r:   )_ANSIX923UnpaddingContextr   r   r   r   r   rB      rA   zANSIX923.unpadderNrC   rD   rE   r   r   r   r   rG   }   rF   rG   c                   @  s>   e Zd ZU ded< dddZdd
dZdddZdddZdS )rH   r#   _bufferr   r   c                 C     || _ d| _d S N    r   rJ   r<   r   r   r   r=         
z _ANSIX923PaddingContext.__init__r	   r
   r   c                 C     t | j|| j\| _}|S r:   )r*   rJ   r   r   r	   r)   r   r   r   r         

z_ANSIX923PaddingContext.updatesizec                 C  s   t dg|d  t |g S )Nr   r/   )r
   )r   rS   r   r   r   _padding   s   z _ANSIX923PaddingContext._paddingc                 C  s   t | j| j| j}d | _|S r:   )r.   rJ   r   rT   r   r)   r   r   r   r      s
   z _ANSIX923PaddingContext.finalizeNrC   r   )rS   r   r   r
   r   )r   r   r   __annotations__r=   r   rT   r   r   r   r   r   rH      s   
 


rH   c                   @  s4   e Zd ZU ded< dddZdd
dZdddZdS )rI   r#   rJ   r   r   c                 C  rK   rL   rN   r<   r   r   r   r=      rO   z"_ANSIX923UnpaddingContext.__init__r	   r
   r   c                 C  rP   r:   )r1   rJ   r   rQ   r   r   r   r      rR   z _ANSIX923UnpaddingContext.updatec                 C  s   t | j| jt}d | _|S r:   )r6   rJ   r   r   rU   r   r   r   r      s   z"_ANSIX923UnpaddingContext.finalizeNrC   r   r   )r   r   r   rV   r=   r   r   r   r   r   r   rI      s
   
 

rI   )r   r   r   r   )r"   r#   r	   r
   r   r   r   r$   )r"   r#   r   r   r+   r,   r   r
   )r"   r#   r   r   r2   r3   r   r
   )
__future__r   r   typingZcryptographyr   Zcryptography.exceptionsr   Z"cryptography.hazmat.bindings._rustr   r   r   ABCMetar   r!   r*   r.   r1   r6   r8   registerrG   rH   rI   r   r   r   r   <module>   s$   






