File: //lib/python3/dist-packages/cryptography/hazmat/primitives/kdf/__pycache__/hkdf.cpython-310.pyc
o
�)%a� � @ s� d dl Z d dlmZ d dlmZmZmZmZ d dlm Z d dl
m Z d dl m
Z
mZmZ d dlmZ G dd � d e�ZG d
d � d e�ZdS ) � N)�utils)�AlreadyFinalized�
InvalidKey�UnsupportedAlgorithm�_Reasons)� _get_backend)� HMACBackend)�
constant_time�hashes�hmac)�KeyDerivationFunctionc
@ sr e Zd Z ddejdedeje deje fdd�Z ded efd
d �Z
ded efd d
�Z deded dfdd�Z dS )�HKDFN� algorithm�length�salt�infoc C sh t |�}t|t�stdtj��|| _|d u r d| jj }nt� d|� || _
|| _ t | j|||�| _
d S )N�.Backend object does not implement HMACBackend.� r )r �
isinstancer r r �BACKEND_MISSING_INTERFACE�
_algorithm� digest_sizer � _check_bytes�_salt�_backend�
HKDFExpand� _hkdf_expand)�selfr r r r �backend� r �I/usr/lib/python3/dist-packages/cryptography/hazmat/primitives/kdf/hkdf.py�__init__ s
� z
HKDF.__init__� key_material�returnc C s( t j| j| j| jd�}|�|� |�� S )N�r )r �HMACr r r �update�finalize)r r" �hr r r �_extract2 s
z
HKDF._extractc C s t �d|� | j�| �|��S )Nr" )r �_check_bytesliker �deriver) �r r" r r r r+ 7 s z HKDF.derive� expected_keyc C � t �| �|�|�s t�d S �N�r �bytes_eqr+ r �r r" r- r r r �verify; � �z HKDF.verifyr/ )
�__name__�
__module__� __qualname__r
�
HashAlgorithm�int�typing�Optional�bytesr! r) r+ r3 r r r r r
s ����
� r
c @ sh e Zd Z ddejdedeje fdd�Z dedefd d
�Z
dedefd d �Z ded
eddfdd�Z dS )r Nr r r c C sz t |�}t|t�stdtj��|| _|| _d|j }||kr%t d�
|���|| _ |d u r/d}nt �
d|� || _d| _d S )Nr � z*Can not derive keys larger than {} octets.� r F)r r r r r r r r r �
ValueError�format�_lengthr r �_info�_used)r r r r r �
max_lengthr r r r! A s&
�
�
zHKDFExpand.__init__r" r# c C s� dg}d}| j jt|�d | jk rItj|| j | jd�}|�|d � |�| j� |�t |g�� |�
|� � � |d7 }| j jt|�d | jk sd� |�d | j� S )Nr> � r$ ���)
r r �lenrA r r% r r& rB r< �appendr' �join)r r" �output�counterr( r r r �_expande s �zHKDFExpand._expandc C s& t �d|� | jr t�d| _| �|�S )Nr" T)r r* rC r rL r, r r r r+ s s
zHKDFExpand.deriver- c C r. r/ r0 r2 r r r r3 { r4 zHKDFExpand.verifyr/ )
r5 r6 r7 r
r8 r9 r: r; r< r! rL r+ r3 r r r r r @ s ���
�$r )r: � cryptographyr �cryptography.exceptionsr r r r � cryptography.hazmat.backendsr �'cryptography.hazmat.backends.interfacesr � cryptography.hazmat.primitivesr r
r �"cryptography.hazmat.primitives.kdfr r
r r r r r �<module> s +