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 eZG d d
d
eZe jeG dd de Z dS ) )absolute_importdivisionprint_function)Enum)range)utils)AlreadyFinalized
InvalidKeyUnsupportedAlgorithm_Reasons) _get_backend) HMACBackend)
constant_timehasheshmac)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 ddZdd Zdd Zdd Zd
d ZdS )
KBKDFHMACNc C sJ t |
}
t|
ts tdtjt|tjs6tdtj|
j |sLtdtjt|t
s^t dt|t spt d|sx|r| rt
d|d ks| j| rt
d|d kr| d krt
d|d k rt|t rt d |d krd
}|d krd
}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 labelcontextF) r
isinstancer
r
r ZBACKEND_MISSING_INTERFACEr Z
HashAlgorithmZUNSUPPORTED_HASHZhmac_supportedr TypeErrorr
ValueError_valid_byte_lengthintr _check_bytes
_algorithm_mode_length_rlen_llen _location_label_context_backend_used _fixed_data) self algorithmmodeZlengthZrlenZllenlocationr r Zfixedbackendr r r __init__# sV
zKBKDFHMAC.__init__c C sB t |tstdtjd|}dt| ko4dkn s>dS dS )Nzvalue must be of type int FT)r r# r r int_to_byteslen)r0 valueZ value_binr r r r" k s
z KBKDFHMAC._valid_byte_lengthc C s | j r
ttjd| d| _ | j | jj }dg}tjd| j}|t dt
|d d krdt dxt 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 appendfinalizejoin)r0 r; roundsoutputZr_binihZcounterr r r derivet s&
zKBKDFHMAC.derivec C sB | j rt| j tr| j S tj| jd | j}dj| jd| j |gS )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||std 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_interfaceobjectr r r r r s