�
�3Tf. � � � d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dl m
Z
dd�Z d dd�Z dd �Z
d dd
�Z d dd �Z d dd �Z G d
� de� Zy)� )� annotationsN)�Cipher)�AES)�ECB)�bytes_eqc � � t t | � t � � j � }t |� }t d� D ]h }t |� D ]X }|j
||| z � }t j |d d d�� ||z |z dz z j dd�� }|dd ||< �Z �j |j � dk( sJ �|dj |� z S ) N� � �big�� byteorder� ��lengthr
���� ) r r r � encryptor�len�range�update�int�
from_bytes�to_bytes�finalize�join)� wrapping_key�a�rr �n�j�i�bs �H/usr/lib/python3/dist-packages/cryptography/hazmat/primitives/keywrap.py�
_wrap_corer$ s� � � �s�<�(�#�%�0�:�:�<�I� �A��A�
�1�X� ���q�� �A� � � ��Q�q�T��*�A����q��!�u���6�1�q�5�A�+��/�J��h�a�5�h�1�
� �R�S�6�A�a�D� � � � � � �3� &�&� &� �s�x�x��{�?�r c � � t | � dvr t d� �t |� dk r t d� �t |� dz dk7 r t d� �d}t dt |� d� D �cg c]
}|||dz �� }}t | ||� S c c}w ) 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� ��������)r �
ValueErrorr r$ )r � key_to_wrap�backendr r! r s r# � aes_key_wrapr/ ' s� � �
�<�� �,��J�K�K�
�;��"� ��D�E�E�
�;��!��q� ��H�I�I�+�A�).�q�#�k�2B�A�)F�G�A��Q��Q�� �G�A�G� �l�A�q� )�)�� Hs �(Bc � � t t | � t � � j � }t |� }t t
d� � D ]s }t t
|� � D ]Z }t j |d�� ||z |z dz z j dd�� || z }|j |� }|d d }|dd ||< �\ �u |j � dk( sJ �||fS ) Nr r r r r
r r r ) r r r � decryptorr �reversedr r r r r r ) r r r r1 r r r! �atrr" s r# � _unwrap_corer4 : s� � � �s�<�(�#�%�0�:�:�<�I� �A��A�
�e�A�h�
� ���%��(�#� �A����q�E��2�q�1�u��k�A�o�F��h�a�5�h�1�A�a�D�9�C�
� � ��%�A��"�1��A��R�S�6�A�a�D� � � � � � �3� &�&� &� �a�4�Kr c �� � t | � dvr t d� �dt |� j dd�� z }dt |� dz z
dz }|d|z z }t |� dk( rVt t | � t � � j
� }|j ||z � }|j � d k( sJ �|S t d
t |� d� D �cg c]
}|||dz �� }}t | ||� S c c}w ) Nr& r* � �YY�� r r r
� r r ) r r, r r r r r r r r r$ ) r r- r. �aiv�padr r" r! r s r# �aes_key_wrap_with_paddingr; Q s � �
�<�� �,��J�K�K�
��K� 0� 9� 9��E� !:� !� �C�
��K� �1�$� %��
*�C� ��#�
�-�K�
�;��1� ��3�|�,�c�e�4�>�>�@� � � � �S�;� .� /���!�!�#�s�*�*�*���-2�1�c�+�6F��-J� K��[��Q��U�
#� K�� K��,��Q�/�/��
Ls �C.c � � t |� dk r t d� �t | � dvr t d� �t |� dk( r^t t | � t � � j
� }|j |� }|j � dk( sJ �|d d }|dd }d}nft dt |� d� D �cg c]
}|||dz �� } }| j d� }
t | � }t | |
| � \ }} dj | � }t j |d d d
� � } d|z | z
} t |d d d � r8d|dz
z | cxk rd|z k r$n t � �| dk7 r t || d d
| z � s
t � �| dk( r|S |d | S c c}w )Nr' zMust be at least 16 bytesr& r* r r
r r r7 r r r6 r8 )r �
InvalidUnwrapr, r r r r1 r r r �popr4 r r r r )
r � wrapped_keyr. r1 �outr �datar r! r �
encrypted_aiv�mlir" s
r# �aes_key_unwrap_with_paddingrD j s� � �
�;��"� �� 7�8�8�
�<�� �,��J�K�K�
�;��2� ��3�|�,�c�e�4�>�>�@� �� � �{�+���!�!�#�s�*�*�*� ���G���1�2�w��
��-2�1�c�+�6F��-J� K��[��Q��U�
#� K�� K����a��
� ��F���L�-��;���1��x�x��{�� �.�.��1�2��%�.�
0�C�
�Q��#�
�A� �Q�r��U� /� 0��A��E�{�S�)�A��E�)� �o� �
��F�8�D�!���I�w��{�;��o� ��A�v�� ��C�a�R�y���/
Ls �6Fc � � t |� dk r t d� �t |� dz dk7 r t d� �t | � dvr t d� �d}t dt |� d� D �cg c]
}|||dz �� }}|j d� }t | ||� \ }}t
||� s
t � �d j |� S c c}w )
Nr( zMust be at least 24 bytesr
r z-The wrapped key must be a multiple of 8 bytesr& r* r+ r )r r= r, r r> r4 r r )r r? r. r9 r! r r s r# �aes_key_unwraprF � s� � �
�;��"� �� 7�8�8�
�;��!��q� �� K�L�L�
�<�� �,��J�K�K�
-�C�).�q�#�k�2B�A�)F�G�A��Q��Q�� �G�A�G� ���a��A� � �a�� +�D�A�q� �A�s� ��o� � �8�8�A�;���
Hs �(Cc � � e Zd Zy)r= N)�__name__�
__module__� __qualname__� r r# r= r= � s � �r r= )r �bytesr rL r �typing.List[bytes]�returnrL )N)r rL r- rL r. �
typing.AnyrN rL )r rL r rL r rM rN z'typing.Tuple[bytes, typing.List[bytes]])r rL r? rL r. rO rN rL )�
__future__r �typing�&cryptography.hazmat.primitives.ciphersr �1cryptography.hazmat.primitives.ciphers.algorithmsr �,cryptography.hazmat.primitives.ciphers.modesr �,cryptography.hazmat.primitives.constant_timer r$ r/ r4 r; rD rF � Exceptionr= rK r r# �<module>rW s0 ��
#�
� 9� A� <� A���� �� �� � �6 �*��*��*� �*� � *�&��� �� �� -� �4 �0��0��0� �0� � 0�8 �+��+��+� �+� � +�b ������ �� � �0 �I� r |