관리-도구
편집 파일: 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