3
l�_�' � @ s� d dl mZmZmZ d dlmZ d dlmZmZ d dl m
Z
d dl m Z dd� Z
dd � Zeje j�G d
d � d e��Zd d
� Zeje j�G dd� de��Zeje j�G dd� de��ZdS )� )�absolute_import�division�print_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_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 �/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 jjkr�d }n| j j|d �}t j
| j j|d �| j j|d �|d�S )Nz BIGNUM **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 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 �encoding�formatr 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| j�S )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 jjkr�d }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 �d�S )Nz BIGNUM **r )r" 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_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
|� }|dkr�d| | }|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) �buffer�len)r Zpeer_public_keyZbufr= r Zerrors_with_text�keyZpadr 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| j�S )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 jjkr�d }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 �d�S )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| j�S )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 DHParametersWithSerialization�objectr r2 Z DHPrivateKeyWithSerializationr3 Z DHPublicKeyWithSerializationrE r r r r � s -v