3
l_' @ s d dl mZmZmZ d dlmZ d dlmZmZ d dl m
Z
d dl m Z dd Z
dd Zeje jG d
d d eZd d
Zeje jG dd deZeje jG dd deZdS ) )absolute_importdivisionprint_function)utils)UnsupportedAlgorithm_Reasons)
serialization)dhc C s |j }|j}|j| }|j||jk |j||j}|jr|jd}|j | |j||j |j
|d }|j ||j||j}|j|dk |S )Nz BIGNUM **r ) _lib_ffiZ DHparams_dupopenssl_assertNULLgcZDH_freeZCRYPTOGRAPHY_IS_LIBRESSLnew DH_get0_pqgBN_dupZ DH_set0_pqg)dh_cdatabackendlibZffi param_cdataqZq_dupres r /usr/lib64/python3.6/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 @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
r c C s || _ || _d S )N)_backend _dh_cdata)selfr r r r r __init__& s z_DHParameters.__init__c C s | j jjd}| j jjd}| j jjd}| j jj| j||| | j j|d | j jjk | j j|d | j jjk |d | j jjkrd }n| j j|d }t j
| j j|d | j j|d |dS )Nz BIGNUM **r )pgr ) 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 j| S )N)r Zgenerate_dh_private_key)r r r r generate_private_key; s z"_DHParameters.generate_private_keyc C s~ |t jjk rtd| jjjsl| jjjd}| jjj | j
| jjj || jjj |d | jjj krlt dt
j| jj||| j
S )Nz%Only PKCS3 serialization is supportedz BIGNUM **r z'DH X9.42 serialization is not supported)r ZParameterFormatZPKCS3
ValueErrorr r Cryptography_HAS_EVP_PKEY_DHXr r r r r r r UNSUPPORTED_SERIALIZATIONZ_parameter_bytes)r encodingformatr r r r parameter_bytes> s
z _DHParameters.parameter_bytesN)__name__
__module__ __qualname__r! r' r( r. r r r r r $ s r c C sL | j jd}| jj||| j j| j j | j|d | j jk | jj|d S )Nz BIGNUM **r )r r r r r r
Z BN_num_bits)r r r" r r r _get_dh_num_bitsR s r2 c @ sH e Zd Zdd Zedd Zdd Zdd Zd d
Zd d Z d
d Z
dS )
_DHPrivateKeyc C s& || _ || _|| _| j jj|| _d S )N)r r _evp_pkeyr ZDH_size_key_size_bytes)r r r evp_pkeyr r r r! [ s z_DHPrivateKey.__init__c C s t | j| jS )N)r2 r r )r r r r key_sizea s z_DHPrivateKey.key_sizec C sT | j jjd}| j jjd}| j jjd}| j jj| j||| | j j|d | j jjk | j j|d | j jjk |d | j jjkrd }n| j j|d }| j jjd}| j jjd}| j jj | j|| | j j|d | j jjk | j j|d | j jjk t
j t
j t
j
| j j|d | j j|d |d| j j|d d| j j|d dS )Nz BIGNUM **r )r" r# r )r' y)public_numbersx)r r r r r r r
r r$ DH_get0_keyr ZDHPrivateNumbersDHPublicNumbersr% )r r" r# r r& pub_keyZpriv_keyr r r private_numberse s, z _DHPrivateKey.private_numbersc C s | j jjd| j}| j jjd}| j jj|j|| j jj | j j|d | j jjk | j jj ||d | j}|dkr| j j
}t d|nL| j j|dk | j jj |d | }| jt
| }|dkrd| | }|S d S )Nzunsigned char[]z BIGNUM **r r
zKError computing shared key. Public key is likely invalid for this exchange. )r r r r5 r r; r r r
ZDH_compute_keyZ_consume_errors_with_textr) bufferlen)r Zpeer_public_keyZbufr= r Zerrors_with_textkeyZpadr r r exchange s$
z_DHPrivateKey.exchangec C s t | j| j}| jjjd}| jjj| j|| jjj | jj|d | jjjk | jjj |d }| jj|| jjjk | jjj
||| jjj}| jj|dk | jj |}t | j||S )Nz BIGNUM **r r
)
r r r r r r r; r r
r Z DH_set0_keyZ_dh_cdata_to_evp_pkey _DHPublicKey)r r r= Z pub_key_dupr r6 r r r
public_key s z_DHPrivateKey.public_keyc C s t | j| jS )N)r r r )r r r r
parameters s z_DHPrivateKey.parametersc C s |t jjk rtd| jjjsl| jjjd}| jjj | j
| jjj || jjj |d | jjj krlt dt
j| jj|||| | j| j
S )Nz0DH private keys support only PKCS8 serializationz BIGNUM **r z'DH X9.42 serialization is not supported)r Z
PrivateFormatZPKCS8r) r r r* r r r r r r r r+ Z_private_key_bytesr4 )r r, r- Zencryption_algorithmr r r r
private_bytes s*
z_DHPrivateKey.private_bytesN) r/ r0 r1 r! propertyr7 r> rD rF rG rH r r r r r3 Y s r3 c @ s8 e Zd Zdd Zedd Zdd Zdd Zd d
Zd S ) rE c C s& || _ || _|| _t| j | j| _d S )N)r r r4 r2 _key_size_bits)r r r r6 r r r r! s z_DHPublicKey.__init__c C s | j S )N)rJ )r r r r r7 s z_DHPublicKey.key_sizec C s | j jjd}| j jjd}| j jjd}| j jj| j||| | j j|d | j jjk | j j|d | j jjk |d | j jjkrd }n| j j|d }| j jjd}| j jj | j|| j jj | j j|d | j jjk t
j t
j | j j|d | j j|d |d| j j|d dS )Nz BIGNUM **r )r" r# r )r' r8 )
r r r r r r r
r r$ r; r r< r% )r r" r# r r& r= r r r r9 s& z_DHPublicKey.public_numbersc C s t | j| jS )N)r r r )r r r r rG s z_DHPublicKey.parametersc C s |t jjk rtd| jjjsl| jjjd}| jjj | j
| jjj || jjj |d | jjj krlt dt
j| jj||| | jd S )Nz>DH public keys support only SubjectPublicKeyInfo serializationz BIGNUM **r z'DH X9.42 serialization is not supported)r Z PublicFormatZSubjectPublicKeyInfor) r r r* r r r r r r r r+ Z_public_key_bytesr4 )r r, r- r r r r public_bytes s
z_DHPublicKey.public_bytesN) r/ r0 r1 r! rI r7 r9 rG rK r r r r rE s
rE N)Z
__future__r r r Z cryptographyr Zcryptography.exceptionsr r Z cryptography.hazmat.primitivesr Z)cryptography.hazmat.primitives.asymmetricr r r Zregister_interfaceZ DHParametersWithSerializationobjectr r2 Z DHPrivateKeyWithSerializationr3 Z DHPublicKeyWithSerializationrE r r r r s -v