File: //lib/python3/dist-packages/cryptography/hazmat/primitives/__pycache__/keywrap.cpython-38.pyc
U
�e�]V � @ s� d dl mZmZmZ d dlZd dlmZ d dlmZ d dl m
Z
d dl m Z dd� Z
d d
� Zd d � Zd
d� Zdd� Zdd� ZG dd� de�ZdS )� )�absolute_import�division�print_functionN)�Cipher)�AES)�ECB)�bytes_eqc C s� t t| �t� |��� }t|�}td�D ]d}t|�D ]V}|�||| �}t�dt� d|d d� �d || | d A �}|dd � ||< q2q&|�
� dks�t �|d� |� S )N� �>Q� r � ���� )
r r r � encryptor�len�range�update�struct�pack�unpack�finalize�AssertionError�join) � wrapping_key�a�r�backendr �n�j�i�b� r! �H/usr/lib/python3/dist-packages/cryptography/hazmat/primitives/keywrap.py�
_wrap_core s &�r# c sp t | �dkrtd��t � �dk r(td��t � �d dkr@td��d}� fd d
�tdt � �d�D �}t| |||�S ) N�� � � �/The wrapping key must be a valid AES key lengthr% z)The key to wrap must be at least 16 bytesr r z-The key to wrap must be a multiple of 8 bytes� ��������c s g | ]}� ||d � �qS �r r! ��.0r �� key_to_wrapr! r" �
<listcomp>/ s z aes_key_wrap.<locals>.<listcomp>)r �
ValueErrorr r# )r r. r r r r! r- r" � aes_key_wrap$ s r1 c
C s� t t| �t� |��� }t|�}ttd��D ]l}tt|��D ]Z}t�dt� d|�d || | d A �|| }|�
|�} | d d� }| dd � ||< q:q*|� � dks�t �||fS )Nr r
r r r r
r )
r r r � decryptorr �reversedr r r r r r r )
r r r r r2 r r r Zatrr r! r! r" � _unwrap_core3 s ��
r4 c s� t | �dkrtd��dt�dt � �� }dt � �d d }� d| � t � �dkr�tt| �t� |��� }|�|� �}|� � dks�t
�|S � fdd �t d
t � �d�D �}t | |||�S d S ) Nr$ r( � �YY�z>ir � r c s g | ]}� ||d � �qS r* r! r+ r- r! r" r/ V s z-aes_key_wrap_with_padding.<locals>.<listcomp>r )
r r0 r r r r r r r r r r r# )r r. r �aivZpadr r r r! r- r" �aes_key_wrap_with_paddingG s r8 c sv t � �dk rtd��t | �dkr(td��t � �dkr�tt| �t� |��� }|�� �}|�� dksdt �|d d� }|dd � }d}nL� fdd �t
d
t � �d�D �}|� d
�} t |�}t | | ||�\}}d�
|�}t�d |d d � �\}
d| |
}t|d d � d
��rPd|d |
k �r(d| k�rPn n$|d
k�rVt|| d � d| ��sVt� �|d
k�rd|S |d | � S d S )Nr% zMust be at least 16 bytesr$ r( r r r c s g | ]}� ||d � �qS r* r! r+ �� wrapped_keyr! r" r/ j s z/aes_key_unwrap_with_padding.<locals>.<listcomp>r z>I� r5 r6 )r �
InvalidUnwrapr0 r r r r2 r r r r �popr4 r r r r ) r r: r r2 r r �datar r Z
encrypted_aivZmlir! r9 r" �aes_key_unwrap_with_paddingZ sB
�
�
� ��
r? c s� t � �dk rtd��t � �d dkr,td��t | �dkr@td��d}� fd d
�tdt � �d�D �}|�d�}t| |||�\}}t||�s�t� �d �|�S ) Nr& zMust be at least 24 bytesr r z-The wrapped key must be a multiple of 8 bytesr$ r( r) c s g | ]}� ||d � �qS r* r! r+ r9 r! r" r/ � s z"aes_key_unwrap.<locals>.<listcomp>r )r r< r0 r r= r4 r r )r r: r r7 r r r! r9 r" �aes_key_unwrap� s
r@ c @ s e Zd ZdS )r< N)�__name__�
__module__� __qualname__r! r! r! r" r< � s r< )Z
__future__r r r r Z&cryptography.hazmat.primitives.ciphersr Z1cryptography.hazmat.primitives.ciphers.algorithmsr Z,cryptography.hazmat.primitives.ciphers.modesr Z,cryptography.hazmat.primitives.constant_timer r# r1 r4 r8 r? r@ � Exceptionr< r! r! r! r" �<module> s +