3
l_ @ s d dl mZmZmZ d dlZd dlmZ d dlmZm Z m
Z
m Z d dl m
Z
d dlmZ d dlmZmZ d dlmZ ejeG d d
d
eZejeG d d d eZdS )
)absolute_importdivisionprint_functionN)utils)AlreadyFinalized
InvalidKeyUnsupportedAlgorithm_Reasons) _get_backend) HMACBackend)
constant_timehmac)KeyDerivationFunctionc @ s. e Zd Zd
ddZdd Zdd Zdd ZdS ) HKDFNc C sh t |}t|ts tdtj|| _|d kr:d| jj }n tj d| || _
|| _ t | j|||| _
d S )Nz.Backend object does not implement HMACBackend. salt)r
isinstancer r r BACKEND_MISSING_INTERFACE
_algorithm digest_sizer _check_bytes_salt_backend
HKDFExpand _hkdf_expand)self algorithmlengthr infobackend r /usr/lib64/python3.6/hkdf.py__init__ s
z
HKDF.__init__c C s( t j| j| j| jd}|j| |j S )N)r )r
HMACr r r updatefinalize)r key_materialhr r r! _extract- s
z
HKDF._extractc C s t jd| | jj| j|S )Nr& )r _check_bytesliker deriver( )r r& r r r! r* 2 s z HKDF.derivec C s t j| j||std S )N)r bytes_eqr* r )r r& expected_keyr r r! verify6 s z HKDF.verify)N)__name__
__module__ __qualname__r" r( r* r- r r r r! r s
r c @ s. e Zd Zd
ddZdd Zdd Zdd ZdS ) r Nc C sz t |}t|ts tdtj|| _|| _d|j }||krJt dj
||| _ |d kr^d}n t j
d| || _d| _d S )Nz.Backend object does not implement HMACBackend. z*Can not derive keys larger than {} octets. r F)r
r r r r r r r r
ValueErrorformat_lengthr r _info_used)r r r r r Z
max_lengthr r r! r" = s"
zHKDFExpand.__init__c C s dg}d}xr| j jt|d | jk r|tj|| j | jd}|j|d |j| j |jt j
| |j |j |d7 }q W dj
|d | j S )Nr2 )r )r r lenr5 r
r# r r$ r6 sixZint2byteappendr% join)r r& outputZcounterr' r r r! _expand[ s zHKDFExpand._expandc C s& t jd| | jrtd| _| j|S )Nr& T)r r) r7 r r? )r r& r r r! r* i s
zHKDFExpand.derivec C s t j| j||std S )N)r r+ r* r )r r& r, r r r! r- q s zHKDFExpand.verify)N)r. r/ r0 r" r? r* r- r r r r! r ; s
r )Z
__future__r r r r; Z cryptographyr Zcryptography.exceptionsr r r r Z cryptography.hazmat.backendsr
Z'cryptography.hazmat.backends.interfacesr Z cryptography.hazmat.primitivesr r
Z"cryptography.hazmat.primitives.kdfr Zregister_interfaceobjectr r r r r r! s $