a
¹aÉ+ ã @ s€ d dl mZmZ d dlmZ d dlmZ dd„ Zdd„ ZG dd „ d ej ƒZ
e d
œd d „Z G d
d„ dej
ƒZG dd„ dejƒZdS )é )ÚUnsupportedAlgorithmÚ_Reasons)Ú
serialization)Údhc C sŒ |j }|j}| | ¡}| ||jk¡ | ||j¡}|jrˆ| d¡}| | |j||j¡ |
|d ¡}| ||j||j¡}| |dk¡ |S ©Nú BIGNUM **r é ) Ú_libÚ_ffiZ DHparams_dupÚopenssl_assertÚNULLÚgcZDH_freeZCRYPTOGRAPHY_IS_LIBRESSLÚnewÚ DH_get0_pqgÚBN_dupZ DH_set0_pqg)Údh_cdataÚbackendÚlibZffiÚ param_cdataÚqZq_dupÚres© r úM/usr/lib64/python3.9/site-packages/cryptography/hazmat/backends/openssl/dh.pyÚ_dh_params_dup s
r c C s t | |ƒ}t||ƒS ©N)r Ú
_DHParameters)r r r r r r Ú_dh_cdata_to_parameters s
r c @ sJ e Zd Zdd„ Zejdœdd„Zejdœdd„Ze j
e j e dœd d
„Z
d S ) r c C s || _ || _d S r )Ú_backendÚ _dh_cdata)Úselfr r r r r Ú__init__# s z_DHParameters.__init__©Úreturnc C sÄ | j j d¡}| j j d¡}| j j d¡}| j j | j|||¡ | j |d | j jjk¡ | j |d | j jjk¡ |d | j jjkrŒd }n| j |d ¡}t j
| j |d ¡| j |d ¡|dS )Nr r ©ÚpÚgr ) r r
r r r r r r Ú
_bn_to_intr ÚDHParameterNumbers)r r$ r% r Úq_valr r r Úparameter_numbers' s ýz _DHParameters.parameter_numbersc C s | j | ¡S r )r Zgenerate_dh_private_key©r r r r Úgenerate_private_key8 s z"_DHParameters.generate_private_key©ÚencodingÚformatr" c C s~ |t jjurtdƒ‚| jjjsl| jj d¡}| jj | j
| jjj || jjj ¡ |d | jjj krlt dt
jƒ‚| j ||| j
¡S )Nz%Only PKCS3 serialization is supportedr r ú'DH X9.42 serialization is not supported)r ÚParameterFormatZPKCS3Ú
ValueErrorr r Ú Cryptography_HAS_EVP_PKEY_DHXr
r r r r r r ÚUNSUPPORTED_SERIALIZATIONZ_parameter_bytes©r r- r. r r r r Úparameter_bytes; s
üþz _DHParameters.parameter_bytesN)Ú__name__Ú
__module__Ú __qualname__r r r' r) Ú DHPrivateKeyr+ r ÚEncodingr0 Úbytesr5 r r r r r " s ür r! c C sL | j d¡}| j ||| j j| j j¡ | |d | j jk¡ | j |d ¡S )Nr r )r
r r r r r Z BN_num_bits)r r r$ r r r Ú_get_dh_num_bitsS s r< c @ sŠ e Zd Zdd„ Zeedœdd„ƒZejdœdd„Z ej
e dœd d
„Z d d „ Z
ej
dœd
d„Zejdœdd„Zejejeje dœdd„ZdS )Ú
_DHPrivateKeyc C s& || _ || _|| _| j j |¡| _d S r )r r Ú _evp_pkeyr ZDH_sizeÚ_key_size_bytes©r r r Úevp_pkeyr r r r [ s z_DHPrivateKey.__init__r! c C s t | j| jƒS r )r< r r r* r r r Úkey_sizea s z_DHPrivateKey.key_sizec C sT | j j d¡}| j j d¡}| j j d¡}| j j | j|||¡ | j |d | j jjk¡ | j |d | j jjk¡ |d | j jjkrŒd }n| j |d ¡}| j j d¡}| j j d¡}| j j | j||¡ | j |d | j jjk¡ | j |d | j jjk¡ t
j t
j t
j
| j |d ¡| j |d ¡|d| j |d ¡d| j |d ¡dS )Nr r r# ©r) Úy)Úpublic_numbersÚx)r r
r r r r r r r& Ú DH_get0_keyr ÚDHPrivateNumbersÚDHPublicNumbersr' )r r$ r% r r( Úpub_keyZpriv_keyr r r Úprivate_numberse s2 ýú÷z _DHPrivateKey.private_numbers)Úpeer_public_keyr" c C sV t |tƒstdƒ‚| jj | j| jjj¡}| j || jjjk¡ | jj
|| jjj ¡}| jj |¡}| j |dk¡ | jj
||j¡}| |dk¡ | jj d¡}| jj || jjj|¡}| |dk¡ | j |d dk¡ | jj d|d ¡}| jj |||¡}| j |dk¡ | jj ||d ¡d d … }| jt|ƒ }|dkrRd| | }|S )Nz%peer_public_key must be a DHPublicKeyr zsize_t *r zunsigned char[]ó )Ú
isinstanceÚ _DHPublicKeyÚ TypeErrorr r ZEVP_PKEY_CTX_newr> r
r r r
ZEVP_PKEY_CTX_freeZEVP_PKEY_derive_initZEVP_PKEY_derive_set_peerÚ_exchange_assertr ZEVP_PKEY_deriveÚbufferr? Úlen)r rL Úctxr ZkeylenÚbufÚkeyZpadr r r Úexchange s6
ÿÿ ÿ
z_DHPrivateKey.exchangec C s |s| j ¡ }td|ƒ‚d S )NzError computing shared key.)r Z_consume_errors_with_textr1 )r ÚokZerrors_with_textr r r rQ ¥ s
þz _DHPrivateKey._exchange_assertc C sº t | j| jƒ}| jj d¡}| jj | j|| jjj¡ | j |d | jjjk¡ | jj |d ¡}| j || jjjk¡ | jj
||| jjj¡}| j |dk¡ | j |¡}t | j||ƒS r )
r r r r
r r rG r r r Z DH_set0_keyZ_dh_cdata_to_evp_pkeyrO )r r rJ Z pub_key_dupr rA r r r Ú
public_key s ÿ ÿ z_DHPrivateKey.public_keyc C s t | j| jƒS r ©r r r r* r r r Ú
parameters¾ s z_DHPrivateKey.parameters)r- r. Úencryption_algorithmr" c C s† |t jjurtdƒ‚| jjjsl| jj d¡}| jj | j
| jjj || jjj ¡ |d | jjj krlt dt
jƒ‚| j |||| | j| j
¡S )Nz0DH private keys support only PKCS8 serializationr r r/ )r Ú
PrivateFormatZPKCS8r1 r r r2 r
r r r r r r r3 Z_private_key_bytesr> )r r- r. r\ r r r r Ú
private_bytesÁ s2 ÿ
üþúz_DHPrivateKey.private_bytesN)r6 r7 r8 r ÚpropertyÚintrB r rH rK Ú DHPublicKeyr; rW rQ rY Ú DHParametersr[ r r: r] ZKeySerializationEncryptionr^ r r r r r= Z s $ûr= c @ s\ e Zd Zdd„ Zeedœdd„ƒZejdœdd„Z ej
dœdd „Z e j
e jed
œd d „Zd
S )rO c C s& || _ || _|| _t| j | jƒ| _d S r )r r r> r< Ú_key_size_bitsr@ r r r r ä s z_DHPublicKey.__init__r! c C s | j S r )rc r* r r r rB ê s z_DHPublicKey.key_sizec C s | j j d¡}| j j d¡}| j j d¡}| j j | j|||¡ | j |d | j jjk¡ | j |d | j jjk¡ |d | j jjkrŒd }n| j |d ¡}| j j d¡}| j j | j|| j jj¡ | j |d | j jjk¡ t
j t
j | j |d ¡| j |d ¡|d| j |d ¡dS )Nr r r# rC )
r r
r r r r r r r& rG r rI r' )r r$ r% r r( rJ r r r rE î s, ÿýúz_DHPublicKey.public_numbersc C s t | j| jƒS r rZ r* r r r r[ s z_DHPublicKey.parametersr, c C s‚ |t jjurtdƒ‚| jjjsl| jj d¡}| jj | j
| jjj || jjj ¡ |d | jjj krlt dt
jƒ‚| j ||| | jd ¡S )Nz>DH public keys support only SubjectPublicKeyInfo serializationr r r/ )r Ú PublicFormatZSubjectPublicKeyInfor1 r r r2 r
r r r r r r r3 Z_public_key_bytesr> r4 r r r Ú public_bytes
s( ÿ
üþ ÿz_DHPublicKey.public_bytesN)r6 r7 r8 r r_ r` rB r rI rE rb r[ r r: rd r; re r r r r rO ã s ürO N)Zcryptography.exceptionsr r Z cryptography.hazmat.primitivesr Z)cryptography.hazmat.primitives.asymmetricr r r rb r r` r< r9 r= ra rO r r r r Ú s 1