a
a @ s d dl Z d dlZd dlmZ d dlmZmZ d dlmZm Z m
Z
d dl m Z e
edddZe je
eje dd d
d Zee
ejg e jf eed d
dZG dd de ZG dd de ZdS ) N)utils)AlreadyFinalized
InvalidKey)
constant_timehasheshmac)KeyDerivationFunction)nreturnc C s t d| S )Nz>I)structpack)r r
R/usr/lib64/python3.9/site-packages/cryptography/hazmat/primitives/kdf/concatkdf.py
_int_to_u32be s r ) algorithmlength otherinfor
c C s8 | j d }||kr td||d ur4td| d S )Nl z'Cannot derive keys larger than {} bits.r )Z digest_size
ValueErrorformatr _check_bytes)r r r Z
max_lengthr
r
r _common_args_checks s
r ) key_materialr auxfnr r
c C s t d| dg}d}d}||krr| }|t| || || || |t|d 7 }|d7 }qd|d | S )Nr r )r Z_check_byteslikeupdater appendfinalizelenjoin)r r r r outputZoutlenZcounterhr
r
r _concatkdf_derive$ s
r# c @ s^ e Zd Zdejeeje ej dddZ
ej dddZ eedd d
Z
eedd d d
ZdS )
ConcatKDFHashN)r r r backendc C s4 t ||| || _|| _|d ur$|nd| _d| _d S )Nr F)r
_algorithm_length
_otherinfo_used)selfr r r r% r
r
r __init__< s
zConcatKDFHash.__init__r
c C s t | jS N)r Hashr& r* r
r
r _hashJ s zConcatKDFHash._hashr r
c C s$ | j r
td| _ t|| j| j| jS NT)r) r r# r' r0 r( r* r r
r
r deriveM s zConcatKDFHash.deriver expected_keyr
c C s t | ||std S r- r Zbytes_eqr4 r r* r r6 r
r
r verifyU s zConcatKDFHash.verify)N)__name__
__module__ __qualname__r
HashAlgorithminttypingOptionalbytesAnyr+ r. r0 r4 r9 r
r
r
r r$ ; s r$ c @ sf e Zd Zdejeeje eje ej dddZ
e j dddZ
eedd d
Zeedd d d
ZdS )
ConcatKDFHMACN)r r saltr r% c C st t ||| || _|| _|d ur$|nd| _|jd u rDtd|j|d u rXd|j }n t d| || _
d| _ d S )Nr z {} is unsupported for ConcatKDF rD F) r r& r' r( Z
block_size TypeErrorr namer r _saltr) )r* r r rD r r% r
r
r r+ [ s
zConcatKDFHMAC.__init__r, c C s t | j| jS r- )r HMACrH r& r/ r
r
r _hmacv s zConcatKDFHMAC._hmacr1 c C s$ | j r
td| _ t|| j| j| jS r2 )r) r r# r' rJ r( r3 r
r
r r4 y s zConcatKDFHMAC.deriver5 c C s t | ||std S r- r7 r8 r
r
r r9 s zConcatKDFHMAC.verify)N)r: r; r< r r= r> r? r@ rA rB r+ r rI rJ r4 r9 r
r
r
r rC Z s rC )r r? Z cryptographyr Zcryptography.exceptionsr r Z cryptography.hazmat.primitivesr r r Z"cryptography.hazmat.primitives.kdfr r> rA r r= r@ r CallableZ HashContextr# r$ rC r
r
r
r s&