관리-도구
편집 파일: kbkdf.cpython-36.opt-1.pyc
3 l�_� � @ s� d dl mZmZmZ d dlmZ d dlmZ d dlm Z d dl m Z m Z m Z mZ d dlmZ d dlmZ d dlmZmZmZ d d lmZ G d d � d e�ZG d d � d e�Ze je�G dd� de ��Z dS )� )�absolute_import�division�print_function)�Enum)�range)�utils)�AlreadyFinalized� InvalidKey�UnsupportedAlgorithm�_Reasons)� _get_backend)� HMACBackend)� constant_time�hashes�hmac)�KeyDerivationFunctionc @ s e Zd ZdZdS )�ModeZctrN)�__name__� __module__� __qualname__Z CounterMode� r r � /usr/lib64/python3.6/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 @ s6 e Zd Zd dd�Zdd� Zdd� Zdd � Zd d � ZdS ) � KBKDFHMACNc C sJ t | �} t| t�s tdtj��t|tj�s6tdtj��| j |�sLtdtj��t|t �s^t d��t|t �spt d��|sx|r�| r�t d��|d ks�| j|� r�t d��|d kr�| d kr�t d��|d k r�t|t� r�t d ��|d kr�d }|d kr�d }tjd |� tjd |� || _|| _|| _|| _|| _|| _|| _|| _| | _d | _| | _ d S )Nz.Backend object does not implement HMACBackend.z5Algorithm supplied is not a supported hash algorithm.z5Algorithm supplied is not a supported hmac algorithm.zmode 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� �label�contextF) r � isinstancer r r ZBACKEND_MISSING_INTERFACEr Z HashAlgorithmZUNSUPPORTED_HASHZhmac_supportedr � TypeErrorr � ValueError�_valid_byte_length�intr � _check_bytes� _algorithm�_mode�_length�_rlen�_llen� _location�_label�_context�_backend�_used� _fixed_data) �self� algorithm�modeZlengthZrlenZllen�locationr r Zfixed�backendr r r �__init__# sV zKBKDFHMAC.__init__c C sB t |t�std��tjd|�}dt|� ko4dkn s>dS dS )Nzvalue must be of type int� � FT)r r# r r � int_to_bytes�len)r0 �valueZ value_binr r r r" k s z KBKDFHMAC._valid_byte_lengthc C s� | j r t�tjd|� d| _ | j | jj }dg}tjd| j�}|t dt |�d �d krdt d��x�t d|d �D ]n}t j|| j| jd�}tj|| j�}| jtjkr�|j|� |j| j� � | jtjkr�|j|� |j|j� � qtW dj|�d | j� S ) N� key_materialTr r6 � � z There are too many iterations.)r4 )r. r r Z_check_bytesliker' r% Z digest_sizer8 r( �powr9 r! r r ZHMACr- r* r r �update�_generate_fixed_inputr �append�finalize�join)r0 r; �rounds�outputZr_bin�i�hZcounterr r r �derivet s& zKBKDFHMAC.derivec C sB | j rt| j t�r| j S tj| jd | j�}dj| jd| j |g�S )Nr= r � ) r/ r �bytesr r8 r' r) rC r+ r, )r0 Zl_valr r r r@ � s z KBKDFHMAC._generate_fixed_inputc C s t j| j|�|�st�d S )N)r Zbytes_eqrH r )r0 r; Z expected_keyr r r �verify� s zKBKDFHMAC.verify)N)r r r r5 r" rH r@ rK r r r r r ! s = $r N) Z __future__r r r �enumr Z six.movesr Z cryptographyr Zcryptography.exceptionsr r r r Z cryptography.hazmat.backendsr Z'cryptography.hazmat.backends.interfacesr Z cryptography.hazmat.primitivesr r r Z"cryptography.hazmat.primitives.kdfr r r Zregister_interface�objectr r r r r �<module> s