File: //lib/python3/dist-packages/cryptography/hazmat/backends/openssl/__pycache__/dh.cpython-38.pyc
U
�e�]>* � @ s� d dl mZmZmZ d dlmZ d dlmZmZ d dl m
Z
d dl m Z dd� Z
dd � Ze�e j�G d
d � d e��Zd d
� Zdd� Ze�e j�G dd� de��Ze�e j�G dd� de��ZdS )� )�absolute_import�division�print_function)�utils)�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_freeZ"CRYPTOGRAPHY_OPENSSL_LESS_THAN_102�new� DH_get0_pqg�BN_dupZ DH_set0_pqg)�dh_cdata�backend�libZffi� param_cdata�qZq_dup�res� r �I/usr/lib/python3/dist-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 @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r c C s || _ || _d S r )�_backend� _dh_cdata)�selfr r r r r �__init__& s z_DHParameters.__init__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_keyc C s~ |t jjk rtd��| 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 ZParameterFormatZPKCS3�
ValueErrorr! r
� Cryptography_HAS_EVP_PKEY_DHXr r r r" r r r �UNSUPPORTED_SERIALIZATIONZ_parameter_bytes�r# �encoding�formatr r r r �parameter_bytes> s* �
���z _DHParameters.parameter_bytesN)�__name__�
__module__� __qualname__r$ r+ r- r5 r r r r r $ s r c C s, |j }|�| d �|j|j�� td��d S )Nr z.Public key value is invalid for this exchange.)r
r Z_lib_reason_matchZ
ERR_LIB_DHZDH_R_INVALID_PUBKEYr/ )�errorsr r r r r � _handle_dh_compute_key_errorU 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_bitsa s
�r; c @ sH e Zd Zdd� Zedd� �Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dS )�
_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$ l s z_DHPrivateKey.__init__c C s t | j| j�S r )r; r! r" r, r r r �key_sizer 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 ZDHPrivateNumbers�DHPublicNumbersr) )r# r& r' r r* �pub_keyZpriv_keyr r r �private_numbersv s2 ���z _DHPrivateKey.private_numbersc C s� | j j�d| j�}| j j�d�}| j j�|j|| j jj� | j �|d | j jjk� | j j� ||d | j�}|dkr�| j �
� }t || j �S | j �|dk� | j j� |�d |� }| jt
|� }|dkr�d| | }|S d S )Nzunsigned char[]r r ���r � )r! r r r> r
rF r" r r ZDH_compute_keyZ_consume_errorsr: �buffer�len)r# Zpeer_public_keyZbufrH r r9 �keyZpadr r r �exchange� s( ��
z_DHPrivateKey.exchangec 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
rF r r r Z DH_set0_keyZ_dh_cdata_to_evp_pkey� _DHPublicKey)r# r rH Z pub_key_dupr r@ 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.parametersc C s� |t jjk rtd��| 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 Z
PrivateFormatZPKCS8r/ r! r
r0 r r r r" r r r r1 Z_private_key_bytesr= )r# r3 r4 Zencryption_algorithmr r r r �
private_bytes� s. �
���z_DHPrivateKey.private_bytesN) r6 r7 r8 r$ �propertyrA rI rO rQ rS rT r r r r r< j s
r< c @ s8 e Zd Zdd� Zedd� �Zdd� Zdd� Zd d
� Zd S ) rP c C s& || _ || _|| _t| j | j�| _d S r )r! r" r= r; �_key_size_bitsr? r r r r$ � s z_DHPublicKey.__init__c C s | j S r )rV r, r r r rA � 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% rB )
r! r r r
r r" r r r( rF r rG r) )r# r& r' r r* rH r r r rD � s. ���z_DHPublicKey.public_numbersc C s t | j| j�S r rR r, r r r rS � s z_DHPublicKey.parametersc C s� |t jjk rtd��| 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 Z PublicFormatZSubjectPublicKeyInfor/ r! r
r0 r r r r" r r r r1 Z_public_key_bytesr= r2 r r r � public_bytes s. �
���z_DHPublicKey.public_bytesN) r6 r7 r8 r$ rU rA rD rS rW r r r r rP � s
rP N)Z
__future__r r r Z cryptographyr Zcryptography.exceptionsr r Z cryptography.hazmat.primitivesr Z)cryptography.hazmat.primitives.asymmetricr r r Zregister_interfaceZ DHParametersWithSerialization�objectr r: r; Z DHPrivateKeyWithSerializationr< Z DHPublicKeyWithSerializationrP r r r r �<module> s
0
n