File: //usr/lib/python3/dist-packages/cryptography/hazmat/primitives/__pycache__/keywrap.cpython-310.pyc
o
�)%a' � @ s d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z de de d ej
e d
e fd d �Z dde d
e d
e fdd�Zde de d ej
e d
eje ej
e f fdd�Z dde d
e d
e fdd�Z dde de d
e fdd�Z dde de d
e fdd�ZG dd� de�ZdS ) � N)� _get_backend)�Cipher)�AES)�ECB)�bytes_eq� wrapping_key�a�r�returnc C s� t t| �t� |��� }t|�}td�D ]2}t|�D ]+}|�||| �}t�dt� d|d d� �d || | d A �}|dd � ||< qq|�
� dksNJ �|d� |� S )N� �>Q� r � ���� ) r r r � encryptor�len�range�update�struct�pack�unpack�finalize�join) r r r �backendr �n�j�i�b� r �H/usr/lib/python3/dist-packages/cryptography/hazmat/primitives/keywrap.py�
_wrap_core s (�� r! � key_to_wrapc sx t |�}t| �dvrtd��t� �dk rtd��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 � g | ]
}� ||d � �qS �r
r ��.0r �r" r r �
<listcomp>5 � z aes_key_wrap.<locals>.<listcomp>)r 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 ]6}tt|��D ]-}t�dt� d|�d || | d A �|| }|�
|�} | d d� }| dd � ||< q q|� � dksTJ �||fS )Nr r r r r
r r ) r r r � decryptorr �reversedr r r r r r )
r r r r r2 r r r �atrr r r r � _unwrap_core9 s ���
�r5 c s� t |�}t| �dvrtd��dt�dt� �� }dt� �d d }� d| � t� �dkrJtt| �t� |��� }|� |� �}|�
� dksHJ �|S � fdd �t d
t� �d�D �}t | |||�S ) Nr# r' � �YY�z>ir
� r c r) r* r r+ r- r r r. d r/ z-aes_key_wrap_with_padding.<locals>.<listcomp>r )
r r r0 r r r r r r r r r r! )r r" r �aiv�padr r r r r- r �aes_key_wrap_with_paddingR s r: � wrapped_keyc sr t |�}t� �dk rtd��t| �dvrtd��t� �dkrEtt| �t� |��� }|�� �}|� � dks6J �|d d� }|dd � }d}n&� fdd �t
d
t� �d�D �}|� d
�} t|�}t | | ||�\}}d�
|�}t�d |d d � �\}
d| |
}t|d d � d
�r�d|d |
k r�d| kr�n t� �|d
kr�t|| d � d| �s�t� �|d
kr�|S |d | � S )Nr$ zMust be at least 16 bytesr# r' r r
r c r) r* r r+ �r; r r r. { r/ z/aes_key_unwrap_with_padding.<locals>.<listcomp>r z>I� r6 r7 )r r �
InvalidUnwrapr0 r r r r2 r r r �popr5 r r r r ) r r; r r2 r r �datar r �
encrypted_aiv�mlir r< r �aes_key_unwrap_with_paddingh s8
�" �rC c s� t |�}t� �dk rtd��t� �d dkrtd��t| �dvr$td��d}� fd d
�tdt� �d�D �}|�d�}t| |||�\}}t||�sKt� �d �|�S ) Nr% zMust be at least 24 bytesr
r z-The wrapped key must be a multiple of 8 bytesr# r' r( c r) r* r r+ r< r r r. � r/ z"aes_key_unwrap.<locals>.<listcomp>r ) r r r> r0 r r? r5 r r )r r; r r8 r r r r< r �aes_key_unwrap� s
rD c @ s e Zd ZdS )r> N)�__name__�
__module__� __qualname__r r r r r> � s r> )N)r �typing� cryptography.hazmat.backendsr �&cryptography.hazmat.primitives.ciphersr �1cryptography.hazmat.primitives.ciphers.algorithmsr �,cryptography.hazmat.primitives.ciphers.modesr �,cryptography.hazmat.primitives.constant_timer �bytes�Listr! r1 �Tupler5 r: rC rD � Exceptionr> r r r r �<module> st ���
����
����
����
����
�.���
�