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
ZdddZdddZdddZG dd deZdS ) )absolute_importdivisionprint_functionN) _get_backend)Cipher)AES)ECB)bytes_eqc C s t t| t |j }t|}xttdD ]h}xbt|D ]V}|j||| }tjdtj d|d d d || | d A }|dd ||< q6W q(W |j
dkst |dj | S )N z>Q r i)
r r r encryptorlenrangeupdatestructpackunpackfinalizeAssertionErrorjoin) wrapping_keyarbackendr njib r /usr/lib64/python3.6/keywrap.py
_wrap_core s ,r" c sx t |}t| d kr tdt dk r0tdt d dkrHtdd } fd
d tdt dD }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|}xttdD ]p}xjtt|D ]Z}tjdtj d|d || | d A || }|j
|} | d d }| dd ||< q>W q,W |j dkst ||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 tddtjdt }dt d d } d| t dkrtt| t |j }|j | }|j
d kst |S fd
d t d t dD }t
| |||S d S )Nr# r$ r% z/The wrapping key must be a valid AES key lengths YYz>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 tdt| dkr0tdt dkrtt| t |j }|j }|j dkslt
|d d }|dd }d}nL fd d
t d t dD }|j d } t|}t
| | ||\}}dj|}tjd |d
d \}
d| |
}t|d d
d s^d|d |
k o.d| kn s^|d krdt|| d d| rdt |d krr|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 YYr. )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 tdt d dkr4tdt| d
krHtdd } fd
d tdt dD }|jd}t| |||\}}t||st 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
+