File: //lib/python3/dist-packages/cryptography/hazmat/primitives/kdf/__pycache__/concatkdf.cpython-38.pyc
U
�e�]� � @ s� d dl mZmZmZ d dlZd dlmZ d dlmZm Z m
Z
m Z d dl m
Z
d dl mZ d dlmZmZmZ d dlmZ d d
� Zd d � Zd
d� Ze�e�G dd� de��Ze�e�G dd� de��ZdS )� )�absolute_import�division�print_functionN)�utils)�AlreadyFinalized�
InvalidKey�UnsupportedAlgorithm�_Reasons)� HMACBackend)� HashBackend)�
constant_time�hashes�hmac)�KeyDerivationFunctionc C s t �d| �S )Nz>I)�structZpack)�n� r �N/usr/lib/python3/dist-packages/cryptography/hazmat/primitives/kdf/concatkdf.py�
_int_to_u32be s r c C s8 | j d }||kr td�|���|d k r4t�d|� d S )Nl �� z(Can not derive keys larger than {} bits.� otherinfo)Z digest_size�
ValueError�formatr � _check_bytes)� algorithm�lengthr Z
max_lengthr r r �_common_args_checks s
��r c C s� t �d| � dg}d}d}||krr|� }|�t|�� |�| � |�|� |�|�� � |t|d �7 }|d7 }qd�|�d |� S )N� key_material� r � ���)r Z_check_byteslike�updater �append�finalize�len�join)r r Zauxfnr �outputZoutlenZcounter�hr r r �_concatkdf_derive" s
r' c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
ConcatKDFHashc C sT t |||� || _|| _|| _| jd kr.d| _t|t�sDtdtj��|| _ d| _
d S )Nr z.Backend object does not implement HashBackend.F) r �
_algorithm�_length�
_otherinfo�
isinstancer r r �BACKEND_MISSING_INTERFACE�_backend�_used)�selfr r r �backendr r r �__init__6 s
�zConcatKDFHash.__init__c C s t �| j| j�S �N)r
ZHashr) r. �r0 r r r �_hashG s zConcatKDFHash._hashc C s$ | j r
t�d| _ t|| j| j| j�S �NT)r/ r r' r* r5 r+ �r0 r r r r �deriveJ s �zConcatKDFHash.derivec C s t �| �|�|�st�d S r3 �r Zbytes_eqr8 r �r0 r Z expected_keyr r r �verifyQ s zConcatKDFHash.verifyN)�__name__�
__module__� __qualname__r2 r5 r8 r; r r r r r( 4 s r( c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
ConcatKDFHMACc C sz t |||� || _|| _|| _| jd kr.d| _|d krBd|j }n t�d|� || _t|t �sjt
dt j ��|| _
d| _d S )Nr � �saltz.Backend object does not implement HMACBackend.F)r r) r* r+ Z
block_sizer r �_saltr, r
r r r- r. r/ )r0 r r rA r r1 r r r r2 X s"
�zConcatKDFHMAC.__init__c C s t �| j| j| j�S r3 )r ZHMACrB r) r. r4 r r r �_hmacp s zConcatKDFHMAC._hmacc C s$ | j r
t�d| _ t|| j| j| j�S r6 )r/ r r' r* rC r+ r7 r r r r8 s s �zConcatKDFHMAC.derivec C s t �| �|�|�st�d S r3 r9 r: r r r r; z s zConcatKDFHMAC.verifyN)r< r= r> r2 rC r8 r; r r r r r? V s r? ) Z
__future__r r r r Z cryptographyr Zcryptography.exceptionsr r r r Z'cryptography.hazmat.backends.interfacesr
r Z cryptography.hazmat.primitivesr r
r Z"cryptography.hazmat.primitives.kdfr r r r' Zregister_interface�objectr( r? r r r r �<module> s !