�
�3Tf� � � � d dl mZ d dlZd dlmZ d dlmZmZ d dlm Z m
Z
m Z d dl m
Z
G d� de
� Z G d � d
e
� Zy) � )� annotationsN)�utils)�AlreadyFinalized�
InvalidKey)�
constant_time�hashes�hmac)�KeyDerivationFunctionc �B � e Zd Z d dd�Zdd�Zdd�Zd d�Zy)
�HKDFNc � � || _ |�d| j j z }nt j d|� || _ t | j ||� | _ y )N� �salt)�
_algorithm� digest_sizer � _check_bytes�_salt�
HKDFExpand� _hkdf_expand)�self� algorithm�lengthr �info�backends �I/usr/lib/python3/dist-packages/cryptography/hazmat/primitives/kdf/hkdf.py�__init__z
HKDF.__init__ sP � � $��� �<��T�_�_� 8� 8�8�D� � � �v�t� ,���
� &�t����� E��� c � � t j | j | j � }|j |� |j � S �N)r �HMACr r �update�finalize)r � key_material�hs r �_extractz
HKDF._extract# s2 � � �I�I�d�j�j�$�/�/� 2�� ��� � ��z�z�|�r c � � t j d|� | j j | j |� � S )Nr# )r �_check_bytesliker �deriver% �r r# s r r( z HKDF.derive( s2 � �
� � �~�|�<�� � �'�'��
�
�l�(C�D�Dr c �Z � t j | j |� |� st �y r �r �bytes_eqr( r �r r# � expected_keys r �verifyz HKDF.verify, �' � � �%�%�d�k�k�,�&?� �N� � � Or r )
r �hashes.HashAlgorithmr �intr �typing.Optional[bytes]r r3 r �
typing.Any�r# �bytes�returnr6 �r# r6 r. r6 r7 �None)�__name__�
__module__� __qualname__r r% r( r/ � r r r r sR � � #�
F�'�F� �F� %� F�
%� F� �
F�& �
E� r r c �> � e Zd Z d dd�Zdd�Zdd�Zd d�Zy)
r Nc � � || _ d|j z }||kD rt d|� d�� �|| _ |�d}nt j
d|� || _ d| _ y )N� z Cannot derive keys larger than z octets.r r F)r r �
ValueError�_lengthr r �_info�_used)r r r r r �
max_lengths r r zHKDFExpand.__init__2 sm � � $����9�0�0�0�
� �J� � �1�*� �X�F��
� �� � �<��D� � � �v�t� ,���
���
r c �. � dg}d}| j j t |� dz
z | j k r�t j
|| j � }|j
|d � |j
| j � |j
t |g� � |j |j � � |dz
}| j j t |� dz
z | j k r��dj |� d | j S )Nr � ���) r r �lenrB r r r! rC r6 �appendr" �join)r r# �output�counterr$ s r �_expandzHKDFExpand._expandM s� � �������o�o�)�)�S��[�1�_�=�� � �L�� � �,����8�A�
�H�H�V�B�Z� �
�H�H�T�Z�Z� �
�H�H�U�G�9�%� &� �M�M�!�*�*�,� '� �q�L�G�
�o�o�)�)�S��[�1�_�=�� � �L� �x�x�� ��$�,�,�/�/r c � � t j d|� | j rt �d| _ | j |� S )Nr# T)r r' rD r rN r) s r r( zHKDFExpand.derive[ s5 � �
� � �~�|�<� �:�:�"� "���
��|�|�L�)�)r c �Z � t j | j |� |� st �y r r+ r- s r r/ zHKDFExpand.verifyc r0 r r )r r1 r r2 r r3 r r4 r5 r8 )r: r; r< r rN r( r/ r= r r r r 1 sA � � #� �'�� �� %� �
� �6 0� *� r r )�
__future__r �typing� cryptographyr �cryptography.exceptionsr r � cryptography.hazmat.primitivesr r r �"cryptography.hazmat.primitives.kdfr
r r r= r r �<module>rW s7 ��
#�
� � @� F� F� D� � � �D4 �&� 4 r |