Edit File: keywrap.cpython-36.pyc
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! � <listcomp>1 s z aes_key_wrap.<locals>.<listcomp>)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.<locals>.<listcomp>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.<locals>.<listcomp>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.<locals>.<listcomp>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! �<module> s +