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 |j
� dks�t �|dj |� S )N� z>Q� r � � i����)
r r r � encryptor�len�range�update�struct�pack�unpack�finalize�AssertionError�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 |j � dks�t �||fS )Nr
z>Qr r r r
i����)
r r r � decryptorr �reversedr r r 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 |� �}|j
� d ks�t �|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 � r
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 r' r �aivZpadr r r r )r' r! �aes_key_wrap_with_paddingL s r0 c s� t |�}t� �dk r td��t| �dkr0td��t� �dkr�tt| �t� |�j� }|j� �}|j � dkslt
�|d d� }|dd � }d}nL� fd d
�t d t� �d�D �}|j d �} t|�}t
| | ||�\}}dj|�}tjd |d
d � �\}
d| |
}t|d d
� d� �s^d|d |
k �o.d| kn �s^|d k�rdt|| d � d| � �rdt� �|d k�rr|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 r c s g | ]}� ||d � �qS )r r )r& r )� wrapped_keyr r! r( q s z/aes_key_unwrap_with_padding..r z>I� s �YY�r. )r# r$ r% )r r �
InvalidUnwrapr) r r r r+ r r r r �popr- r r r r ) r r1 r r+ r r �datar r Z
encrypted_aivZmlir )r1 r! �aes_key_unwrap_with_padding` s4
*&
r6 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 )r1 r r! r( � s z"aes_key_unwrap..r
)r# r$ r% ) r r r3 r) r r4 r- r r )r r1 r r/ r r r )r1 r! �aes_key_unwrap� s
r7 c @ s e Zd ZdS )r3 N)�__name__�
__module__� __qualname__r r r r! r3 � s r3 )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- r0 r6 r7 � Exceptionr3 r r r r! � s
+