a
    ϏPf                     @  st   d dl mZ d dlZd dlmZ d dlmZmZ d dlm	Z	m
Z
mZ d dlmZ G dd deZG d	d
 d
eZdS )    )annotationsN)utils)AlreadyFinalized
InvalidKey)constant_timehasheshmac)KeyDerivationFunctionc                   @  sV   e Zd ZdddddddddZd	d	d
ddZd	d	d
ddZd	d	ddddZdS )HKDFNhashes.HashAlgorithmintbytes | None
typing.Any)	algorithmlengthsaltinfobackendc                 C  sB   || _ |d u rd| j j }ntd| || _t| j ||| _d S )N    r   )
_algorithmdigest_sizer   _check_bytes_salt
HKDFExpand_hkdf_expand)selfr   r   r   r   r    r   h/var/www/html/python-backend/venv/lib/python3.9/site-packages/cryptography/hazmat/primitives/kdf/hkdf.py__init__   s    zHKDF.__init__byteskey_materialreturnc                 C  s"   t | j| j}|| | S N)r   HMACr   r   updatefinalize)r   r!   hr   r   r   _extract#   s    
zHKDF._extractc                 C  s   t d| | j| |S )Nr!   )r   _check_bytesliker   deriver(   r   r!   r   r   r   r*   (   s    zHKDF.deriveNoner!   expected_keyr"   c                 C  s   t | ||std S r#   r   Zbytes_eqr*   r   r   r!   r.   r   r   r   verify,   s    zHKDF.verify)N)__name__
__module____qualname__r   r(   r*   r1   r   r   r   r   r
      s
    r
   c                   @  sT   e Zd ZddddddddZd	d	d
ddZd	d	d
ddZd	d	ddddZdS )r   Nr   r   r   r   )r   r   r   r   c                 C  sX   || _ d|j }||kr(td| d|| _|d u r<d}ntd| || _d| _d S )N   zCannot derive keys larger than z octets.    r   F)r   r   
ValueError_lengthr   r   _info_used)r   r   r   r   r   
max_lengthr   r   r   r   2   s    

zHKDFExpand.__init__r   r    c                 C  s   dg}d}| j jt|d  | jk rtt|| j }||d  || j |t|g |	|
  |d7 }q
d|d | j S )Nr6      )r   r   lenr8   r   r$   r%   r9   r   appendr&   join)r   r!   outputcounterr'   r   r   r   _expandM   s    
zHKDFExpand._expandc                 C  s&   t d| | jrtd| _| |S )Nr!   T)r   r)   r:   r   rC   r+   r   r   r   r*   [   s
    zHKDFExpand.deriver,   r-   c                 C  s   t | ||std S r#   r/   r0   r   r   r   r1   c   s    zHKDFExpand.verify)N)r2   r3   r4   r   rC   r*   r1   r   r   r   r   r   1   s
    r   )
__future__r   typingZcryptographyr   Zcryptography.exceptionsr   r   Zcryptography.hazmat.primitivesr   r   r   Z"cryptography.hazmat.primitives.kdfr	   r
   r   r   r   r   r   <module>   s   "