| Current File : //lib64/python3.9/site-packages/cryptography/hazmat/primitives/kdf/__pycache__/kbkdf.cpython-39.pyc |
a
�a� � @ s� d dl Z d dlmZ d dlmZmZmZmZ d dlm Z m
Z
m Z m Z m
Z
d dlmZ G dd� dej�ZG dd � d ej�ZG d
d � d �ZG d d
� d
e�ZG dd� de�ZdS )� N)�utils)�AlreadyFinalized�
InvalidKey�UnsupportedAlgorithm�_Reasons)�ciphers�cmac�
constant_time�hashes�hmac)�KeyDerivationFunctionc @ s e Zd ZdZdS )�ModeZctrN)�__name__�
__module__� __qualname__Z CounterMode� r r �N/usr/lib64/python3.9/site-packages/cryptography/hazmat/primitives/kdf/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
@ sx e Zd Zejeeeeje eeje eje eje d� dd�Z
e ee d�dd��Z
e ee d�dd �Ze d
�d d �Zd
S )�
_KBKDFDeriver) �prf�mode�length�rlen�llen�location�label�context�fixedc
C s t |�s J �t|t�s td��t|t�s0td��|s8|rD| rDtd��|d u sV| �|�s^td��|d u rv| d u rvtd��|d ur�t|t�s�td��|d u r�d}|d u r�d}t� d|� t� d |� || _
|| _ || _ || _
|| _|| _|| _|| _d
| _| | _d S ) Nzmode 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� r r F)�callable�
isinstancer
� TypeErrorr �
ValueError�_valid_byte_length�intr � _check_bytes�_prf�_mode�_length�_rlen�_llen� _location�_label�_context�_used� _fixed_data)
�selfr r r r r r r r r r r r �__init__" s>
� z_KBKDFDeriver.__init__)�value�returnc C s@ t | t�std��t�d| �}dt|� kr6dks<n dS dS )Nzvalue must be of type int� � FT)r" r& r# r � int_to_bytes�len)r4 Z value_binr r r r% W s
z _KBKDFDeriver._valid_byte_length)� key_material�prf_output_sizer5 c C s� | j r
t�t�d|� d| _ | j | }dg}t�d| j�}|tdt|�d �d kr`t d��t
d|d �D ]d}| � |�}t�|| j�}| j t
jkr�|�|� |�| �� � | j t
jkr�|�|� |�|�� � qnd�|�d | j� S )Nr: Tr r6 � � z There are too many iterations.)r0 r r Z_check_bytesliker* r8 r+ �powr9 r$ �ranger( r- r r �update�_generate_fixed_inputr �append�finalize�join) r2 r: r; �rounds�outputZr_bin�i�hZcounterr r r �derivea s&
z_KBKDFDeriver.derive�r5 c C sB | j rt| j t�r| j S t�| jd | j�}d�| jd| j |g�S )Nr= r � )
r1 r" �bytesr r8 r* r, rD r. r/ )r2 Zl_valr r r rA � s z#_KBKDFDeriver._generate_fixed_inputN)r r r �typing�Callabler
r& �Optionalr rL r3 � staticmethod�boolr% rI rA r r r r r ! s � 5 $r c @ sx e Zd Zdejeeeeje e eje
eje
eje
ej d�
dd�Z e
d�dd�Z
e
d�d d
�Ze
e
dd �d d
�ZdS )� KBKDFHMACN)
� algorithmr r r r r r r r �backendc
C s` t |tj�stdtj��ddlm} | �|�s:tdtj��|| _ t
| j |||||||| � | _ d S )Nz5Algorithm supplied is not a supported hash algorithm.r �rT z5Algorithm supplied is not a supported hmac algorithm.)
r" r
�
HashAlgorithmr r ZUNSUPPORTED_HASH�,cryptography.hazmat.backends.openssl.backendrT Zhmac_supported�
_algorithmr r( �_deriver) r2 rS r r r r r r r r rT �osslr r r r3 � s.
�
��zKBKDFHMAC.__init__)r: c C s t �|| j�S �N)r �HMACrX �r2 r: r r r r( � s zKBKDFHMAC._prfrJ c C s | j �|| jj�S r[ )rY rI rX � digest_sizer] r r r rI � s zKBKDFHMAC.derive�r: � expected_keyr5 c C s t �| �|�|�st�d S r[ �r Zbytes_eqrI r �r2 r: r` r r r �verify� s zKBKDFHMAC.verify)N)r r r r
rV r
r& rM rO r rL �Anyr3 r( rI rc r r r r rR � s �� +rR c @ sv e Zd Zdeeeeje eeje eje eje ej d� dd�Z
ed�dd�Z eed�d d
�Z eedd �d d
�Z
dS )� KBKDFCMACN) r r r r r r r r rT c
C sP t |tj�rt |tj�s$tdtj��|| _d | _t | j
|||||||| � | _ d S )N�7Algorithm supplied is not a supported cipher algorithm.) �
issubclassr ZBlockCipherAlgorithmZCipherAlgorithmr r �UNSUPPORTED_CIPHERrX �_cipherr r( rY ) r2 rS r r r r r r r r rT r r r r3 � s,
�
���zKBKDFCMAC.__init__)�_c C s | j d usJ �t�| j �S r[ )ri r ZCMAC)r2 rj r r r r( � s zKBKDFCMAC._prf)r: r5 c C sT | � |�| _| jd usJ �ddlm} |�| j�s>tdtj��| j� || jj
d �S )Nr rU rf r= ) rX ri rW rT Zcmac_algorithm_supportedr r rh rY rI �
block_size)r2 r: rZ r r r rI � s �zKBKDFCMAC.deriver_ c C s t �| �|�|�st�d S r[ ra rb r r r rc s zKBKDFCMAC.verify)N)r r r r
r& rM rO r rL rd r3 r( rI rc r r r r re � s �� $re )rM Z cryptographyr Zcryptography.exceptionsr r r r Z cryptography.hazmat.primitivesr r r r
r Z"cryptography.hazmat.primitives.kdfr �Enumr
r r rR re r r r r �<module> s m7