| Current File : //lib64/python3.9/site-packages/cryptography/hazmat/primitives/kdf/__pycache__/hkdf.cpython-39.pyc |
a
�a� � @ sh d dl Z d dlmZ d dlmZmZ d dlmZmZm Z d dl
m Z G dd� de �Z G dd � d e �Z
dS )
� N)�utils)�AlreadyFinalized�
InvalidKey)�
constant_time�hashes�hmac)�KeyDerivationFunctionc @ sf e Zd Zd
ejeeje eje ej d�dd�Z
eed�dd�Z eed�dd �Z eedd
�d d �Z
dS )�HKDFN)� 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 �M/usr/lib64/python3.9/site-packages/cryptography/hazmat/primitives/kdf/hkdf.py�__init__ s z
HKDF.__init__�� 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�r � expected_keyr c C s t �| �|�|�st�d S r �r Zbytes_eqr$ r �r r r' r r r �verify. s z HKDF.verify)N)�__name__�
__module__� __qualname__r �
HashAlgorithm�int�typing�Optional�bytes�Anyr r" r$ r* r r r r r s �� r c @ s^ e Zd Zd
ejeeje ej d�dd�Z
eed�dd�Z eed�dd �Z eedd
�d d �Z
dS )r N)r
r r
r c C sV || _ d|j }||kr&td�|���|| _|d u r:d}n t�d|� || _d| _d S )N� z)Cannot derive keys larger than {} octets.� r
F) r r �
ValueError�format�_lengthr r �_info�_used)r r
r r
r Z
max_lengthr r r r 4 s
� zHKDFExpand.__init__r c C s� dg}d}| j jt|�d | jk rtt�|| j �}|�|d � |�| j� |�t|g�� |� |�
� � |d7 }q
d� |�d | j� S )Nr5 � ���) r r �lenr8 r r r r9 r2 �appendr �join)r r �outputZcounterr! r r r �_expandO s
zHKDFExpand._expandc C s& t �d|� | jrt�d| _| �|�S )Nr T)r r# r: r rA r% r r r r$ ] s
zHKDFExpand.deriver& c C s t �| �|�|�st�d S r r( r) r r r r* e s zHKDFExpand.verify)N)r+ r, r- r r. r/ r0 r1 r2 r3 r rA r$ r* r r r r r 3 s �� r )r0 Z cryptographyr Zcryptography.exceptionsr r Z cryptography.hazmat.primitivesr r r Z"cryptography.hazmat.primitives.kdfr r r r r r r �<module> s "