| Current File : //lib64/python3.9/site-packages/cryptography/hazmat/primitives/__pycache__/keywrap.cpython-39.pyc |
a
�au � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z e
e
ej e
e
d�dd�Z de
e
ej
e
d �d
d �Ze
e
ej e
eje
ej e
f d�d d
�Zde
e
ej
e
d �dd�Zde
e
ej
e
d�dd�Zde
e
ej
e
d�dd�ZG dd� de�ZdS )� N)�Cipher)�AES)�ECB)�bytes_eq)� wrapping_key�a�r�returnc C s� t t| �t� ��� }t|�}td�D ]d}t|�D ]V}|�||| �}t�dt� d|d d� �d || | d A �}|dd � ||< q0q$|�
� dks�J �|d� |� S )N� �>Q� r � ���� ) r r r � encryptor�len�range�update�struct�pack�unpack�finalize�join)r r r r �n�j�i�b� r �L/usr/lib64/python3.9/site-packages/cryptography/hazmat/primitives/keywrap.py�
_wrap_core s (�r )r � key_to_wrap�backendr c sn t | �dvrtd��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 �r r r �
<listcomp>7 r z aes_key_wrap.<locals>.<listcomp>)r �
ValueErrorr r )r r r! r r r r+ r � aes_key_wrap( s r. c C s� t t| �t� ��� }t|�}ttd��D ]l}tt|��D ]Z}t�dt� d|�d || | d A �|| }|�
|�}|d d� }|dd � ||< q8q(|� � dks�J �||fS )Nr
r r r
r r r ) r r r � decryptorr �reversedr r r r r r ) r r r r/ r r r Zatrr r r r � _unwrap_core; s ���
r1 c s� t | �dvrtd��dt�dt � �� }dt � �d d }� d| � t � �dkr�tt| �t� ��� }|�|� �}|� � dks�J �|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, i r z-aes_key_wrap_with_padding.<locals>.<listcomp>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_paddingV s r5 )r � wrapped_keyr! r c sr t � �dk rtd��t | �dvr(td��t � �dkr�tt| �t� ��� }|�� �}|�� dksbJ �|d d� }|dd � }d}nJ� fdd �t d
t � �d�D �}|�
d
�} t |�}t | | |�\}}d� |�}t
�d |d d � �\}
d| |
}t|d d � d
��rLd|d |
k �r$d| k�rLn n$|d
k�rRt|| d � d| ��sRt� �|d
k�r`|S |d | � S d S )Nr# zMust be at least 16 bytesr" r& r r r
c s g | ]}� ||d � �qS r( r r) �r6 r r r, � r z/aes_key_unwrap_with_padding.<locals>.<listcomp>r z>I� r2 r3 )r �
InvalidUnwrapr- r r r r/ r r r �popr1 r r r r ) r r6 r! r/ r r �datar r Z
encrypted_aivZmlir r7 r �aes_key_unwrap_with_paddingm s@
� �
� ��
r< c s� t � �dk rtd��t � �d dkr,td��t | �dvr@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) r7 r r r, � r z"aes_key_unwrap.<locals>.<listcomp>r )r r9 r- r r: r1 r r )r r6 r! r4 r r r r7 r �aes_key_unwrap� s
r= c @ s e Zd ZdS )r9 N)�__name__�
__module__� __qualname__r r r r r9 � s r9 )N)N)N)N)r �typingZ&cryptography.hazmat.primitives.ciphersr Z1cryptography.hazmat.primitives.ciphers.algorithmsr Z,cryptography.hazmat.primitives.ciphers.modesr Z,cryptography.hazmat.primitives.constant_timer �bytes�Listr �Anyr. �Tupler1 r5 r< r= � Exceptionr9 r r r r �<module> sX � �� � �� �� 1 ��