o
    ÖèFhÇ  ã                   @  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                   @  s8   e Zd Z	dddd„Zddd„Zddd„Zddd„ZdS )ÚHKDFNÚ	algorithmúhashes.HashAlgorithmÚlengthÚintÚsaltúbytes | NoneÚinfoÚbackendú
typing.Anyc                 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   ú`/var/www/html/loop/nvenv/lib/python3.10/site-packages/cryptography/hazmat/primitives/kdf/hkdf.pyÚ__init__   s   zHKDF.__init__Úkey_materialÚbytesÚ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Úexpected_keyÚNonec                 C  ó   t  |  |¡|¡st‚d S r"   ©r   Úbytes_eqr)   r   ©r   r   r+   r   r   r   Úverify,   ó   ÿzHKDF.verifyr"   )
r   r   r   r   r   r   r   r   r   r   ©r   r    r!   r    ©r   r    r+   r    r!   r,   )Ú__name__Ú
__module__Ú__qualname__r   r'   r)   r1   r   r   r   r   r
      s    ú

r
   c                   @  s8   e Zd Z	ddd
d„Zddd„Zddd„Z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    r!   c                 C  s    dg}d}| j jt|ƒd  | jk rFt || j ¡}| |d ¡ | | j¡ | t|gƒ¡ | 	| 
¡ ¡ |d7 }| j jt|ƒd  | jk sd |¡d | j… S )Nr9   é   éÿÿÿÿ)r   r   Úlenr;   r   r#   r$   r<   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   rF   r*   r   r   r   r)   [   s
   
zHKDFExpand.deriver+   r,   c                 C  r-   r"   r.   r0   r   r   r   r1   c   r2   zHKDFExpand.verifyr"   )r   r   r   r   r   r   r   r   r3   r4   )r5   r6   r7   r   rF   r)   r1   r   r   r   r   r   1   s    û

r   )Ú
__future__r   ÚtypingÚcryptographyr   Úcryptography.exceptionsr   r   Úcryptography.hazmat.primitivesr   r   r   Ú"cryptography.hazmat.primitives.kdfr	   r
   r   r   r   r   r   Ú<module>   s   "