3
l�_� � @ s� d dl mZmZmZ d dlmZ d dlmZ d dlm Z d dl
m Z m Z m
Z
mZ d dlmZ d dlmZ d dlmZmZmZ d d lmZ G d
d � d e�ZG d d
� d
e�Ze je�G dd� de ��Z dS )� )�absolute_import�division�print_function)�Enum)�range)�utils)�AlreadyFinalized�
InvalidKey�UnsupportedAlgorithm�_Reasons)� _get_backend)� HMACBackend)�
constant_time�hashes�hmac)�KeyDerivationFunctionc @ s e Zd ZdZdS )�ModeZctrN)�__name__�
__module__� __qualname__Z CounterMode� r r � /usr/lib64/python3.6/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 @ s6 e Zd Zd dd�Zdd� Zdd� Zdd � Zd
d � ZdS )
� KBKDFHMACNc C sJ t |
�}
t|
t�s tdtj��t|tj�s6tdtj��|
j |�sLtdtj��t|t
�s^t d��t|t �spt d��|sx|r�| r�t
d��|d ks�| j|� r�t
d��|d kr�| d kr�t
d��|d k r�t|t� r�t d ��|d kr�d
}|d kr�d
}tjd |� tjd |� || _|| _|| _|| _|| _|| _|| _|| _|
| _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) r �
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__# sV
zKBKDFHMAC.__init__c C sB t |t�std��tjd|�}dt|� ko4dkn s>dS dS )Nzvalue must be of type int� � FT)r r# r r � int_to_bytes�len)r0 �valueZ value_binr r r r" k s
z KBKDFHMAC._valid_byte_lengthc C s� | j r
t�tjd|� d| _ | j | jj }dg}tjd| j�}|t dt
|�d �d krdt d��x�t d|d �D ]n}t
j|| j| jd�}tj|| j�}| jtjkr�|j|� |j| j� � | jtjkr�|j|� |j|j� � qtW dj|�d | j� S ) N� key_materialTr r6 � � z There are too many iterations.)r4 )r. r r Z_check_bytesliker' r% Z digest_sizer8 r( �powr9 r! r r ZHMACr- r* r r �update�_generate_fixed_inputr �append�finalize�join)r0 r; �rounds�outputZr_bin�i�hZcounterr r r �derivet s&
zKBKDFHMAC.derivec C sB | j rt| j t�r| j S tj| jd | j�}dj| jd| j |g�S )Nr= r � )
r/ r �bytesr r8 r' r) rC r+ r, )r0 Zl_valr r r r@ � s z KBKDFHMAC._generate_fixed_inputc C s t j| j|�|�st�d S )N)r Zbytes_eqrH r )r0 r; Z expected_keyr r r �verify� s zKBKDFHMAC.verify)N)r r r r5 r" rH r@ rK 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.backendsr 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 � s