a
au @ s d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z e
e
ej e
e
dddZ de
e
ej
e
d d
d Ze
e
ej e
eje
ej e
f dd d
Zde
e
ej
e
d ddZde
e
ej
e
dddZde
e
ej
e
dddZG dd deZdS ) N)Cipher)AES)ECB)bytes_eq) wrapping_keyarreturnc C s t t| t }t|}tdD ]d}t|D ]V}|||| }tdt d|d d d || | d A }|dd ||< q0q$|
dksJ |d | S )N >Q r ) r r r encryptorlenrangeupdatestructpackunpackfinalizejoin)r r r r njib r L/usr/lib64/python3.9/site-packages/cryptography/hazmat/primitives/keywrap.py
_wrap_core s (r )r key_to_wrapbackendr c sn t | dvrtdt dk r(tdt d dkr@tdd} fd d
tdt dD }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
7 r z aes_key_wrap..)r
ValueErrorr r )r r r! r r r r+ r aes_key_wrap( s r. c C s t t| t }t|}ttdD ]l}tt|D ]Z}tdt d|d || | d A || }|
|}|d d }|dd ||< q8q(| dksJ ||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 | dvrtddtdt }dt d d } d| t dkrtt| t }|| }| dksJ |S fdd t
d
t dD }t | ||S d S ) Nr" r& YYz>ir r c s g | ]} ||d qS r( r r) r+ r r r, i r z-aes_key_wrap_with_padding..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 rtdt | dvr(tdt dkrtt| t }| }| dksbJ |d d }|dd }d}nJ fdd t d
t dD }|
d
} t |}t | | |\}}d |}t
d |d d \}
d| |
}t|d d d
rLd|d |
k r$d| krLn n$|d
krRt|| d d| sRt |d
kr`|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..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 rtdt d dkr,tdt | dvr@tdd} fd d
tdt dD }|d}t| ||\}}t||st 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..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 bytesListr Anyr. Tupler1 r5 r< r= Exceptionr9 r r r r sX 1