File: //lib/python3/dist-packages/cryptography/hazmat/primitives/kdf/__pycache__/kbkdf.cpython-310.pyc
o
� >bA � @ 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 ej�ZG d
d � d ej�ZG d d
� d
e�ZdS )� N)�utils)�AlreadyFinalized�
InvalidKey�UnsupportedAlgorithm�_Reasons)� _get_backend)� HMACBackend)�
constant_time�hashes�hmac)�KeyDerivationFunctionc @ s e Zd ZdZdS )�Mode�ctrN)�__name__�
__module__� __qualname__� CounterMode� r r �J/usr/lib/python3/dist-packages/cryptography/hazmat/primitives/kdf/kbkdf.pyr
s r
c @ s e Zd ZdZdZdS )�CounterLocation� before_fixed� after_fixedN)r r r � BeforeFixed�
AfterFixedr r r r r s r c @ s� e Zd Z ddejdedededeje de deje
d eje
d
eje
fd d �Z d
ede fdd�Z
de
de
fdd�Zde
fdd�Zde
de
ddfdd�ZdS )� KBKDFHMACN� algorithm�mode�length�rlen�llen�location�label�context�fixedc C sF t |
�}
t|
t�stdtj��t|tj�stdtj��|
� |�s&tdtj��t|t
�s/t d��t|t �s8t d��|s<|rB| rBt
d��|d u sK| �|�sOt
d��|d u r[| d u r[t
d��|d urht|t�sht d ��|d u rnd
}|d u rtd
}t�d |� t�d |� || _|| _|| _|| _|| _|| _|| _|| _|
| _d
| _| | _ d S )Nz.Backend object does not implement HMACBackend.z5Algorithm supplied is not a supported hash algorithm.z5Algorithm supplied is not a supported hmac algorithm.zmode must be of type Modez(location must be of type CounterLocationz9When supplying fixed data, label and context are ignored.z rlen must be between 1 and 4zPlease specify an llenzllen must be an integer� r! r" F) r �
isinstancer r r �BACKEND_MISSING_INTERFACEr
�
HashAlgorithm�UNSUPPORTED_HASH�hmac_supportedr
� TypeErrorr �
ValueError�_valid_byte_length�intr � _check_bytes�
_algorithm�_mode�_length�_rlen�_llen� _location�_label�_context�_backend�_used� _fixed_data) �selfr r r r r r r! r" r# �backendr r r �__init__ s^
� �
�
�
zKBKDFHMAC.__init__�value�returnc C sB t |t�s td��t�d|�}dt|� kr dks dS dS dS )Nzvalue must be of type int� � FT)r% r- r* r � int_to_bytes�len)r: r= � value_binr r r r, g s
�z KBKDFHMAC._valid_byte_length� key_materialc C s� | j rt�t�d|� d| _ | j | jj }dg}t�d| j�}|t dt
|�d �d kr2t d��t d|d �D ]7}t
j|| j| jd�}t�|| j�}| jtjkrW|�|� |�| �� � | jtjkri|�|� |�|�� � q9d�|�d | j� S ) NrD Tr$ r? � � z There are too many iterations.)r; )r8 r r �_check_bytesliker1 r/ � digest_sizerA r2 �powrB r+ �ranger �HMACr7 r4 r r �update�_generate_fixed_inputr �append�finalize�join)r: rD �rounds�output�r_bin�i�h�counterr r r �derivep s&
zKBKDFHMAC.derivec C sB | j r t| j t�r | j S t�| jd | j�}d�| jd| j |g�S )NrF r$ � )
r9 r% �bytesr rA r1 r3 rP r5 r6 )r: �l_valr r r rM � s z KBKDFHMAC._generate_fixed_input� expected_keyc C s t �| �|�|�s t�d S �N)r �bytes_eqrW r )r: rD r[ r r r �verify� s �zKBKDFHMAC.verifyr\ )r r r r
r' r
r- �typing�Optionalr rY r< �boolr, rW rM r^ r r r r r s2 �������� �
�H $r )r_ � cryptographyr �cryptography.exceptionsr r r r � cryptography.hazmat.backendsr �'cryptography.hazmat.backends.interfacesr � cryptography.hazmat.primitivesr r
r �"cryptography.hazmat.primitives.kdfr �Enumr
r r r r r r �<module> s