Edit File: dh.cpython-36.pyc
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 �<module> s -v