File: //lib/python3/dist-packages/cryptography/hazmat/primitives/kdf/__pycache__/hkdf.cpython-38.pyc
U
�e�]�
� @ s� d dl mZmZmZ d dlZd dlmZ d dlmZm Z m
Z
m Z d dl m
Z
d dlmZmZ d dlmZ e�e�G dd � d e��Ze�e�G d
d � d e��ZdS ) � )�absolute_import�division�print_functionN)�utils)�AlreadyFinalized�
InvalidKey�UnsupportedAlgorithm�_Reasons)� HMACBackend)�
constant_time�hmac)�KeyDerivationFunctionc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�HKDFc C s` t |t�stdtj��|| _|d kr2d| jj }n t�d|� || _ || _
t | j|||�| _ d S )N�.Backend object does not implement HMACBackend.� �salt)
�
isinstancer
r r �BACKEND_MISSING_INTERFACE�
_algorithm� digest_sizer � _check_bytes�_salt�_backend�
HKDFExpand� _hkdf_expand)�self� algorithm�lengthr �info�backend� r �I/usr/lib/python3/dist-packages/cryptography/hazmat/primitives/kdf/hkdf.py�__init__ s
� z
HKDF.__init__c C s( t j| j| j| jd�}|�|� |�� S )N�r )r �HMACr r r �update�finalize)r � key_material�hr r r! �_extract( s
z
HKDF._extractc C s t �d|� | j�| �|��S )Nr' )r �_check_bytesliker �deriver) �r r' r r r! r+ - s z HKDF.derivec C s t �| �|�|�st�d S �N�r Zbytes_eqr+ r �r r' Z expected_keyr r r! �verify1 s z HKDF.verifyN)�__name__�
__module__� __qualname__r" r) r+ r0 r r r r! r s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r c C sr t |t�stdtj��|| _|| _d|j }||krBtd� |���|| _
|d krVd}n t � d|� || _
d| _d S )Nr � z*Can not derive keys larger than {} octets.� r F)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" 8 s(
�
�� zHKDFExpand.__init__c C s� dg}d}| j jt|�d | jk rztj|| j | jd�}|�|d � |�| j� |�t �
|�� |� |� � � |d7 }q
d�
|�d | j� S )Nr5 � r# ���)r r �lenr8 r r$ r r% r9 �sixZint2byte�appendr& �join)r r' �outputZcounterr( r r r! �_expandV s
zHKDFExpand._expandc C s& t �d|� | jrt�d| _| �|�S )Nr' T)r r* r: r rB r, r r r! r+ d s
zHKDFExpand.derivec C s t �| �|�|�st�d S r- r. r/ r r r! r0 l s zHKDFExpand.verifyN)r1 r2 r3 r" rB r+ r0 r r r r! r 6 s r )Z
__future__r r r r> Z cryptographyr Zcryptography.exceptionsr r r r Z'cryptography.hazmat.backends.interfacesr
Z cryptography.hazmat.primitivesr r Z"cryptography.hazmat.primitives.kdfr
Zregister_interface�objectr r r r r r! �<module> s #