3
l�_b � @ s� d dl mZmZmZ d dlZd dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ dd � Zdd
d �Zd d
� Zddd�Zddd�Zddd�ZG dd� de�ZdS )� )�absolute_import�division�print_functionN)� _get_backend)�Cipher)�AES)�ECB)�bytes_eqc C s� t t| �t� |�j� }t|�}xttd�D ]h}xbt|�D ]V}|j||| �}tjdtj d|d d� �d || | d A �}|dd � ||< q6W q(W |dj
|� S )N� z>Q� r � � i����) r r r � encryptor�len�range�update�struct�pack�unpack�join) � wrapping_key�a�r�backendr �n�j�i�b� r � /usr/lib64/python3.6/keywrap.py�
_wrap_core s ,r c sx t |�}t| �d kr td��t� �dk r0td��t� �d dkrHtd��d }� fd
d �tdt� �d�D �}t| |||�S )
N� � � z/The wrapping key must be a valid AES key lengthz)The key to wrap must be at least 16 bytesr r z-The key to wrap must be a multiple of 8 bytess ��������c s g | ]}� ||d � �qS )r r )�.0r )� key_to_wrapr r �
1 s z aes_key_wrap..)r! r" r# )r r �
ValueErrorr r )r r% r r r r )r% r � aes_key_wrap% s r( c
C s� t t| �t� |�j� }t|�}x�ttd��D ]p}xjtt|��D ]Z}tjdtj d|�d || | d A �|| }|j
|�} | d d� }| dd � ||< q>W q,W ||fS )Nr
z>Qr r r i����) r r r � decryptorr �reversedr r r r r )
r r r r r) r r r Zatrr r r r � _unwrap_core5 s "
r+ c s� t |�}t| �d kr td��dtjdt� �� }dt� �d d }� d| � t� �dkr�tt| �t� |�j� }|j |� �}|S � fd d
�t
d t� �d�D �}t | |||�S d S )
Nr! r" r# z/The wrapping key must be a valid AES key lengths �YY�z>ir � c s g | ]}� ||d � �qS )r r )r$ r )r% r r r& \ s z-aes_key_wrap_with_padding..r )r! r" r# ) r r r' r r r r r r r r r )r r% r �aivZpadr r r r )r% r �aes_key_wrap_with_paddingL s r. c st t |�}t� �dk r td��t| �dkr0td��t� �dkrztt| �t� |�j� }|j� �}|d d� }|dd � }d}nL� fdd �t d
t� �d�D �}|j
d
�} t|�}t | | ||�\}}d j |�}t
jd |d
d � �\}
d| |
}t|d d
� d� �sNd|d |
k �o d| kn �sN|d
k�rTt|| d � d| � �rTt� �|d
k�rb|S |d | � S d S )Nr! zMust be at least 16 bytesr" r# z/The wrapping key must be a valid AES key lengthr r c s g | ]}� ||d � �qS )r r )r$ r )� wrapped_keyr r r& q s z/aes_key_unwrap_with_padding..r r
z>I� s �YY�r, )r! r" r# )r r �
InvalidUnwrapr' r r r r) r r �popr+ r r r r ) r r/ r r) r r �datar r Z
encrypted_aivZmlir )r/ r �aes_key_unwrap_with_padding` s2
*&
r4 c s� t |�}t� �dk r td��t� �d dkr4td��t| �d
krHtd��d }� fd
d �tdt� �d�D �}|jd�}t| |||�\}}t||�s�t� �d j|�S )Nr" zMust be at least 24 bytesr r z-The wrapped key must be a multiple of 8 bytesr! r# z/The wrapping key must be a valid AES key lengths ��������c s g | ]}� ||d � �qS )r r )r$ r )r/ r r r& � s z"aes_key_unwrap..r
)r! r" r# ) r r r1 r' r r2 r+ r r )r r/ r r- r r r )r/ r �aes_key_unwrap� s
r5 c @ s e Zd ZdS )r1 N)�__name__�
__module__� __qualname__r r r r r1 � s r1 )N)N)N)N)Z
__future__r r r r Z cryptography.hazmat.backendsr Z&cryptography.hazmat.primitives.ciphersr Z1cryptography.hazmat.primitives.ciphers.algorithmsr Z,cryptography.hazmat.primitives.ciphers.modesr Z,cryptography.hazmat.primitives.constant_timer r r( r+ r. r4 r5 � Exceptionr1 r r r r � s
+