o
    eb                     @  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
 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G d"d# d#eZG d$d% d%eZG d&d' d'ZG d(d) d)eZG d*d+ d+eZdS )0    )annotationsN)utils)AlreadyFinalized)check_ansix923_paddingcheck_pkcs7_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_typing.Optional[bytes]r   r	   typing.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   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_updateF   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_checkX   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__m      
zPKCS7.__init__r
   r   c                 C  
   t | jS r:   )_PKCS7PaddingContextr   r   r   r   r   padderq      
zPKCS7.padderc                 C  r?   r:   )_PKCS7UnpaddingContextr   r   r   r   r   unpaddert   rB   zPKCS7.unpadderNr   r   r
   r   r   r   r   r=   rA   rD   r   r   r   r   r8   l       

r8   c                   @  >   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 )r@   r"   _bufferr   r   c                 C     || _ d| _d S N    r   rJ   r<   r   r   r   r=   {      
z_PKCS7PaddingContext.__init__r   r	   r
   c                 C     t | j|| j\| _}|S r:   r*   rJ   r   r   r   r)   r   r   r   r         

z_PKCS7PaddingContext.updatesizec                 C  s   t |g| S r:   r	   r   rT   r   r   r   _padding   s   z_PKCS7PaddingContext._paddingc                 C     t | j| j| j}d | _|S r:   r.   rJ   r   rW   r   r)   r   r   r   r      
   z_PKCS7PaddingContext.finalizeNrE   r   rT   r   r
   r	   r   r   r   r   __annotations__r=   r   rW   r   r   r   r   r   r@   x      
 


r@   c                   @  4   e Zd ZU ded< dddZdd
dZdddZdS )rC   r"   rJ   r   r   c                 C  rK   rL   rN   r<   r   r   r   r=      rO   z_PKCS7UnpaddingContext.__init__r   r	   r
   c                 C  rP   r:   r1   rJ   r   rR   r   r   r   r      rS   z_PKCS7UnpaddingContext.updatec                 C     t | j| jt}d | _|S r:   )r6   rJ   r   r   rZ   r   r   r   r      s
   
z_PKCS7UnpaddingContext.finalizeNrE   r   r   r   r   r   r^   r=   r   r   r   r   r   r   rC      
   
 

rC   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   rA      rB   zANSIX923.padderc                 C  r?   r:   )_ANSIX923UnpaddingContextr   r   r   r   r   rD      rB   zANSIX923.unpadderNrE   rF   rG   r   r   r   r   re      rH   re   c                   @  rI   )rf   r"   rJ   r   r   c                 C  rK   rL   rN   r<   r   r   r   r=      rO   z _ANSIX923PaddingContext.__init__r   r	   r
   c                 C  rP   r:   rQ   rR   r   r   r   r      rS   z_ANSIX923PaddingContext.updaterT   c                 C  s   t dg|d  t |g S )Nr   r/   rU   rV   r   r   r   rW      s   z _ANSIX923PaddingContext._paddingc                 C  rX   r:   rY   rZ   r   r   r   r      r[   z _ANSIX923PaddingContext.finalizeNrE   r   r\   r   r]   r   r   r   r   rf      r_   rf   c                   @  r`   )rg   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:   ra   rR   r   r   r   r      rS   z _ANSIX923UnpaddingContext.updatec                 C  rb   r:   )r6   rJ   r   r   rZ   r   r   r   r      s   z"_ANSIX923UnpaddingContext.finalizeNrE   r   r   rc   r   r   r   r   rg      rd   rg   )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   typingcryptographyr   cryptography.exceptionsr   "cryptography.hazmat.bindings._rustr   r   ABCMetar   r    r*   r.   r1   r6   r8   r@   rC   re   rf   rg   r   r   r   r   <module>   s$   




