3
l�_ � @ s� d dl mZmZmZ d dlZd dlZd dlmZ d dlm Z ddd�Z
G dd� de �Z G d d
� d
e �Z
G d d � d e �Zejej�G d
d� de ��ZeZejej�G dd� de ��Zejej�G dd� de��Zejej�G dd� de ��ZeZdS )� )�absolute_import�division�print_functionN)�utils)� _get_backendc C s t |�}|j| |�S )N)r Zgenerate_dh_parameters)� generator�key_size�backend� r
�/usr/lib64/python3.6/dh.py�generate_parameters s r c @ sB e Zd Zdd� Zdd� Zdd� Zd dd �Zejd
�Z ejd �Z
dS )
�DHPrivateNumbersc C s6 t |tj�std��t |t�s&td��|| _|| _d S )Nzx must be an integer.z6public_numbers must be an instance of DHPublicNumbers.)�
isinstance�six�
integer_types� TypeError�DHPublicNumbers�_x�_public_numbers)�self�x�public_numbersr
r
r �__init__ s
zDHPrivateNumbers.__init__c C s& t |t�stS | j|jko$| j|jkS )N)r r
�NotImplementedr r )r �otherr
r
r �__eq__! s
zDHPrivateNumbers.__eq__c C s
| |k S )Nr
)r r r
r
r �__ne__* s zDHPrivateNumbers.__ne__Nc C s t |�}|j| �S )N)r Zload_dh_private_numbers)r r r
r
r � private_key- s z DHPrivateNumbers.private_keyr r )N) �__name__�
__module__� __qualname__r r r r r �read_only_propertyr r r
r
r
r r
s
r
c @ sB e Zd Zdd� Zdd� Zdd� Zd dd �Zejd
�Z ejd �Z
dS )
r c C s6 t |tj�std��t |t�s&td��|| _|| _d S )Nzy must be an integer.z5parameters must be an instance of DHParameterNumbers.)r r r r �DHParameterNumbers�_y�_parameter_numbers)r �y�parameter_numbersr
r
r r 6 s
zDHPublicNumbers.__init__c C s& t |t�stS | j|jko$| j|jkS )N)r r r r# r$ )r r r
r
r r B s
zDHPublicNumbers.__eq__c C s
| |k S )Nr
)r r r
r
r r K s zDHPublicNumbers.__ne__Nc C s t |�}|j| �S )N)r Zload_dh_public_numbers)r r r
r
r �
public_keyN s zDHPublicNumbers.public_keyr# r$ )N) r r r r r r r' r r! r% r& r
r
r
r r 5 s
r c @ sN e Zd Zd
dd�Zdd� Zdd� Zddd �Zejd
�Z ejd �Z
ejd �Z dS )r" Nc C sh t |tj� s t |tj� r$td��|d k rBt |tj� rBtd��|dk rRtd��|| _|| _|| _d S )Nzp and g must be integerszq must be integer or None� z!DH generator must be 2 or greater)r r r r �
ValueError�_p�_g�_q)r �p�g�qr
r
r r W s zDHParameterNumbers.__init__c C s2 t |t�stS | j|jko0| j|jko0| j|jkS )N)r r" r r* r+ r, )r r r
r
r r f s
zDHParameterNumbers.__eq__c C s
| |k S )Nr
)r r r
r
r r n s zDHParameterNumbers.__ne__c C s t |�}|j| �S )N)r Zload_dh_parameter_numbers)r r r
r
r �
parametersq s z DHParameterNumbers.parametersr* r+ r, )N)N) r r r r r r r0 r r! r- r. r/ r
r
r
r r" V s
r" c @ s6 e Zd Zejdd� �Zejdd� �Zejdd� �ZdS )� DHParametersc C s dS )z7
Generates and returns a DHPrivateKey.
Nr
)r r
r
r �generate_private_key| s z!DHParameters.generate_private_keyc C s dS )z=
Returns the parameters serialized as bytes.
Nr
)r �encoding�formatr
r
r �parameter_bytes� s z DHParameters.parameter_bytesc C s dS )z/
Returns a DHParameterNumbers.
Nr
)r r
r
r r& � s z DHParameters.parameter_numbersN)r r r �abc�abstractmethodr2 r5 r& r
r
r
r r1 z s r1 c @ sD e Zd Zejdd� �Zejdd� �Zejdd� �Zejdd� �Z d S )
� DHPrivateKeyc C s dS )z6
The bit length of the prime modulus.
Nr
)r r
r
r r � s zDHPrivateKey.key_sizec C s dS )zC
The DHPublicKey associated with this private key.
Nr
)r r
r
r r' � s zDHPrivateKey.public_keyc C s dS )zK
The DHParameters object associated with this private key.
Nr
)r r
r
r r0 � s zDHPrivateKey.parametersc C s dS )zn
Given peer's DHPublicKey, carry out the key exchange and
return shared key as bytes.
Nr
)r Zpeer_public_keyr
r
r �exchange� s zDHPrivateKey.exchangeN)
r r r r6 �abstractpropertyr r7 r' r0 r9 r
r
r
r r8 � s r8 c @ s( e Zd Zejdd� �Zejdd� �ZdS )� DHPrivateKeyWithSerializationc C s dS )z-
Returns a DHPrivateNumbers.
Nr
)r r
r
r �private_numbers� s z-DHPrivateKeyWithSerialization.private_numbersc C s dS )z6
Returns the key serialized as bytes.
Nr
)r r3 r4 Zencryption_algorithmr
r
r �
private_bytes� s z+DHPrivateKeyWithSerialization.private_bytesN)r r r r6 r7 r< r= r
r
r
r r; � s r; c @ sD e Zd Zejdd� �Zejdd� �Zejdd� �Zejdd� �Z d S )
� DHPublicKeyc C s dS )z6
The bit length of the prime modulus.
Nr
)r r
r
r r � s zDHPublicKey.key_sizec C s dS )zJ
The DHParameters object associated with this public key.
Nr
)r r
r
r r0 � s zDHPublicKey.parametersc C s dS )z,
Returns a DHPublicNumbers.
Nr
)r r
r
r r � s zDHPublicKey.public_numbersc C s dS )z6
Returns the key serialized as bytes.
Nr
)r r3 r4 r
r
r � public_bytes� s zDHPublicKey.public_bytesN)
r r r r6 r: r r7 r0 r r? r
r
r
r r> � s r> )N)Z
__future__r r r r6 r Z cryptographyr Z cryptography.hazmat.backendsr r �objectr
r r" Z
add_metaclass�ABCMetar1 Z DHParametersWithSerializationr8 r; r> Z DHPublicKeyWithSerializationr
r
r
r � s$
!!$