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 eje�G d d
� d
e��Zeje�G d d � d e��ZdS )
� )�absolute_import�division�print_functionN)�utils)�AlreadyFinalized�
InvalidKey�UnsupportedAlgorithm�_Reasons)� _get_backend)� HMACBackend)�
constant_time�hmac)�KeyDerivationFunctionc @ s. e Zd Zd
dd�Zdd� Zdd� Zdd � ZdS ) �HKDFNc C sh t |�}t|t�s 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� algorithm�lengthr �info�backend� 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 �update�finalize)r � key_material�hr 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|�|�st�d 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
dd�Zdd� Zdd� Zdd � ZdS ) r Nc C sz t |�}t|t�s 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 �
ValueError�format�_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 �sixZint2byte�appendr% �join)r r& �outputZcounterr' r r r! �_expand[ s zHKDFExpand._expandc C s& t jd|� | jrt�d| _| j|�S )Nr& T)r r) r7 r r? )r r& r r r! r* i s
zHKDFExpand.derivec C s t j| j|�|�st�d 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_interface�objectr r r r r r! � s $