Edit File: hkdf.cpython-36.pyc
3 l�_ � @ s� d dl mZmZmZ d dlZd dlmZ d dlmZm Z m Z m Z d dl m Z d dlmZ d dlmZmZ d dlmZ eje�G d d � d e��Zeje�G d d � d e��ZdS ) � )�absolute_import�division�print_functionN)�utils)�AlreadyFinalized� InvalidKey�UnsupportedAlgorithm�_Reasons)� _get_backend)� HMACBackend)� constant_time�hmac)�KeyDerivationFunctionc @ s. e Zd Zd dd�Zdd� Zdd� Zdd � ZdS ) �HKDFNc C sh t |�}t|t�s tdtj��|| _|d kr:d| jj }n tj d|� || _ || _ t | j|||�| _ d S )Nz.Backend object does not implement HMACBackend.� �salt)r � isinstancer r r �BACKEND_MISSING_INTERFACE� _algorithm� digest_sizer � _check_bytes�_salt�_backend� HKDFExpand� _hkdf_expand)�self� algorithm�lengthr �info�backend� r � /usr/lib64/python3.6/hkdf.py�__init__ s z HKDF.__init__c C s( t j| j| j| jd�}|j|� |j� S )N)r )r �HMACr r r �update�finalize)r � key_material�hr r r! �_extract- s z HKDF._extractc C s t jd|� | jj| j|��S )Nr& )r �_check_bytesliker �deriver( )r r& r r r! r* 2 s z HKDF.derivec C s t j| j|�|�st�d S )N)r �bytes_eqr* r )r r& � expected_keyr r r! �verify6 s z HKDF.verify)N)�__name__� __module__� __qualname__r" r( r* r- r r r r! r s r c @ s. e Zd Zd dd�Zdd� Zdd� Zdd � ZdS ) r Nc C sz t |�}t|t�s tdtj��|| _|| _d|j }||krJt dj |���|| _ |d kr^d}n t j d|� || _d| _d S )Nz.Backend object does not implement HMACBackend.� z*Can not derive keys larger than {} octets.� r F)r 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" = s" zHKDFExpand.__init__c C s� dg}d}xr| j jt|�d | jk r|tj|| j | jd�}|j|d � |j| j� |jt j |�� |j |j � � |d7 }q W dj |�d | j� S )Nr2 � )r ���)r r �lenr5 r r# r r$ r6 �sixZint2byte�appendr% �join)r r& �outputZcounterr' r r r! �_expand[ s zHKDFExpand._expandc C s& t jd|� | jrt�d| _| j|�S )Nr& T)r r) r7 r r? )r r& r r r! r* i s zHKDFExpand.derivec C s t j| j|�|�st�d S )N)r r+ r* r )r r& r, r r r! r- q s zHKDFExpand.verify)N)r. r/ r0 r" r? r* r- r r r r! r ; s r )Z __future__r r r r; Z cryptographyr Zcryptography.exceptionsr r r r Z cryptography.hazmat.backendsr 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 $