File: //var/opt/nydus/ops/cryptography/hazmat/backends/openssl/__pycache__/dh.cpython-39.pyc
a
�,�h�/ � @ s� d dl Z d dlmZmZ d dlmZ d dlmZ e jrBd dl m
Z
dd�dd �Z dd
d �d d
�Z G dd
� d
ej
�Zed�dd�ZG dd� dej�ZG dd� dej�ZdS )� N)�UnsupportedAlgorithm�_Reasons)�
serialization)�dh)�Backendr ��backendc 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_cdatar �lib�ffi� param_cdata�qZq_dup�res� r �Z/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/cryptography/hazmat/backends/openssl/dh.py�_dh_params_dup s
r �
_DHParameters)r �returnc C s t | |�}t||�S �N)r r )r r r r r r �_dh_cdata_to_parameters" s
r c @ sP e Zd Zdd�dd�Zejd�dd�Zejd�dd �Ze j
e j e d
�d d �Z
d
S )r r r c C s || _ || _d S r )�_backend� _dh_cdata)�selfr r r r r �__init__( s z_DHParameters.__init__�r c 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_key> s z"_DHParameters.generate_private_key��encoding�formatr c C s& |t jju rtd��|t jjur(td��| jj� d�}| jj
� | j | jjj
|| jjj
� |d | jjj
kr�| jj
js�tdtj��|t jju r�|d | jjj
kr�| jj
j}q�| jj
j}n>|t jju r�|d | jjj
kr�| jj
j}q�| jj
j}ntd��| j�� }||| j �}| j�|dk� | j�|�S )Nz!OpenSSH encoding is not supportedz%Only PKCS3 serialization is supportedr
r �'DH X9.42 serialization is not supportedz/encoding must be an item from the Encoding enumr )r �EncodingZOpenSSH� TypeError�ParameterFormatZPKCS3�
ValueErrorr! r
r r r r" r � Cryptography_HAS_EVP_PKEY_DHXr r �UNSUPPORTED_SERIALIZATIONZPEMZPEM_write_bio_DHxparamsZPEM_write_bio_DHparamsZDERZ Cryptography_i2d_DHxparams_bioZi2d_DHparams_bioZ_create_mem_bio_gcr Z
_read_mem_bio)r# r0 r1 r Z write_bioZbior r r r �parameter_bytesA s: ����
z _DHParameters.parameter_bytesN)�__name__�
__module__� __qualname__r$ r r* r, � DHPrivateKeyr. r r3 r5 �bytesr9 r r r r r ' s �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_bitsl s r? c @ s� e Zd Zdd�dd�Zeed�dd��Zejd�dd �Z ej
e d
�d d �Z e
d
d�dd�Zej
d�dd�Zejd�dd�Zejejeje d�dd�Zd
S )�
_DHPrivateKeyr r c C s& || _ || _|| _| j j�|�| _d S r )r! r" � _evp_pkeyr ZDH_size�_key_size_bytes�r# r r �evp_pkeyr r r r$ t s z_DHPrivateKey.__init__r% c C s t | j| j�S r )r? r! r" r- r r r �key_sizez 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_numbers~ 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� _DHPublicKeyr4 r! r ZEVP_PKEY_CTX_newrA r
r r r ZEVP_PKEY_CTX_freeZEVP_PKEY_derive_initZEVP_PKEY_derive_set_peer�_exchange_assertr ZEVP_PKEY_derive�bufferrB �len)r# rO �ctxr Zkeylen�buf�key�padr r r �exchange� s6
�� �
z_DHPrivateKey.exchangeN)�okr c C s |s| j �� }td|��d S )NzError computing shared key.)r! Z_consume_errors_with_textr6 )r# r[ Zerrors_with_textr r r rS � 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 rJ r r r Z DH_set0_keyZ_dh_cdata_to_evp_pkeyrR )r# r rM Z pub_key_dupr rD 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)r0 r1 �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 r2 )r �
PrivateFormatZPKCS8r6 r! r r7 r
r r r" r r r r8 Z_private_key_bytesrA )r# r0 r1 r_ r r r r �
private_bytes� s2 �
���z_DHPrivateKey.private_bytes)r: r; r< r$ �property�intrE r rK rN � DHPublicKeyr> rZ �boolrS r\ � DHParametersr^ r r3 r` ZKeySerializationEncryptionra r r r r r@ s s $�r@ c @ sb e Zd Zdd�dd�Zeed�dd��Zejd�dd �Z ej
d�d
d �Z e j
e jed �d
d�ZdS )rR r r c C s& || _ || _|| _t| j | j�| _d S r )r! r" rA r? �_key_size_bitsrC r r r r$ � s z_DHPublicKey.__init__r% c C s | j S r )rg r- r r r rE 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& rF )
r! r
r r r r" r r r) rJ r rL r* )r# r' r( r r+ rM r r r rH s, ���z_DHPublicKey.public_numbersc C s t | j| j�S r r] 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 r2 )r � PublicFormatZSubjectPublicKeyInfor6 r! r r7 r
r r r" r r r r8 Z_public_key_bytesrA )r# r0 r1 r r r r � public_bytes# s( �
�� �z_DHPublicKey.public_bytesN)r: r; r< r$ rb rc rE r rL rH rf r^ r r3 rh r> ri r r r r rR � s �rR )�typingZcryptography.exceptionsr r Z cryptography.hazmat.primitivesr Z)cryptography.hazmat.primitives.asymmetricr �
TYPE_CHECKINGZ,cryptography.hazmat.backends.openssl.backendr r r rf r rc r? r= r@ rd rR r r r r �<module> s E