| Current File : //lib64/python3.9/site-packages/cryptography/hazmat/backends/openssl/__pycache__/dh.cpython-39.pyc |
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 �|d�S )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 �d�S )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|� }|dk�rRd| | }|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 �d�S )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 �<module> s 1