File: //lib/python3/dist-packages/cryptography/hazmat/primitives/kdf/__pycache__/kbkdf.cpython-38.pyc
U
�e�]) � @ s� d dl mZmZmZ d dlmZ d dlmZ d dlm Z d dl
m Z m Z m
Z
mZ d dlmZ d dlmZmZmZ d dlmZ G d d
� d
e�ZG d d � d e�Ze �e�G d
d� de��ZdS )� )�absolute_import�division�print_function)�Enum)�range)�utils)�AlreadyFinalized�
InvalidKey�UnsupportedAlgorithm�_Reasons)� HMACBackend)�
constant_time�hashes�hmac)�KeyDerivationFunctionc @ s e Zd ZdZdS )�ModeZctrN)�__name__�
__module__� __qualname__Z CounterMode� r r �J/usr/lib/python3/dist-packages/cryptography/hazmat/primitives/kdf/kbkdf.pyr s r c @ s e Zd ZdZdZdS )�CounterLocationZ before_fixedZ after_fixedN)r r r � BeforeFixed�
AfterFixedr r r r r s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) � KBKDFHMACc C s> t |
t�stdtj��t |tj�s.tdtj��|
�|�sDtdtj��t |t �sVt
d��t |t �sht
d��|sp|r|| r|t d��|d ks�| �
|�s�t d��|d kr�| d kr�t d��|d k r�t |t�s�t
d ��|d kr�d
}|d kr�d
}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� �label�contextF) �
isinstancer r
r ZBACKEND_MISSING_INTERFACEr Z
HashAlgorithmZUNSUPPORTED_HASHZhmac_supportedr � TypeErrorr �
ValueError�_valid_byte_length�intr � _check_bytes�
_algorithm�_mode�_length�_rlen�_llen� _location�_label�_context�_backend�_used� _fixed_data) �self� algorithm�modeZlengthZrlenZllen�locationr r Zfixed�backendr r r �__init__ sX
� �
�
zKBKDFHMAC.__init__c C s@ t |t�std��t�d|�}dt|� kr6dks<n dS dS )Nzvalue must be of type int� � FT)r r" r r � int_to_bytes�len)r/ �valueZ value_binr r r r! Z s
z KBKDFHMAC._valid_byte_lengthc C s� | j r
t�t�d|� d| _ | j | jj }dg}t�d| j�}|t dt
|�d �d krdt d��t d|d �D ]n}t
j|| j| jd�}t�|| j�}| jtjkr�|�|� |�| �� � | jtjkr�|�|� |�|�� � qrd�|�d | j� S ) N� key_materialTr r5 � � z There are too many iterations.)r3 )r- r r Z_check_bytesliker& r$ Z digest_sizer7 r' �powr8 r r r ZHMACr, r) r r �update�_generate_fixed_inputr �append�finalize�join)r/ r: �rounds�outputZr_bin�i�hZcounterr r r �derivec s&
zKBKDFHMAC.derivec C sB | j rt| j t�r| j S t�| jd | j�}d�| jd| j |g�S )Nr< r � )
r. r �bytesr r7 r& r( rB r* r+ )r/ Zl_valr r r r? � s z KBKDFHMAC._generate_fixed_inputc C s t �| �|�|�st�d S )N)r
Zbytes_eqrG r )r/ r: Z expected_keyr r r �verify� s zKBKDFHMAC.verifyN)r r r r4 r! rG r? rJ r r r r r s
; $r N) Z
__future__r r r �enumr Z six.movesr Z cryptographyr Zcryptography.exceptionsr r r
r Z'cryptography.hazmat.backends.interfacesr Z cryptography.hazmat.primitivesr
r r Z"cryptography.hazmat.primitives.kdfr r r Zregister_interface�objectr r r r r �<module> s