3
l�_� � @ s� d dl mZmZmZ d dlmZ d dlmZ d dlm Z d dl
m Z m Z dZ
eje �G dd� de��Zeje �G d d
� d
e��Zd S ) � )�absolute_import�division�print_function)�utils)�_evp_pkey_derive)�
serialization)�X25519PrivateKey�X25519PublicKey� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_X25519PublicKeyc C s || _ || _d S )N)�_backend� _evp_pkey)�self�backend�evp_pkey� r � /usr/lib64/python3.6/x25519.py�__init__ s z_X25519PublicKey.__init__c C sV |t jjks|t jjkr@|t jjk s0|t jjk r8td��| j� S | jj||| | jd �S )Nz3When using Raw both encoding and format must be Raw) r �Encoding�Raw� PublicFormat�
ValueError�_raw_public_bytesr Z_public_key_bytesr
)r �encoding�formatr r r � public_bytes s z _X25519PublicKey.public_bytesc C s| | j jjd�}| j jj| j|�}| j j|dk� | j j|d | j jjk� | j jj|d | j jj �}| j jj
||�d d � S )Nzunsigned char **r
r ) r �_ffi�new�_libZ EVP_PKEY_get1_tls_encodedpointr
�openssl_assert�NULL�gcZ OPENSSL_free�buffer)r Zucharpp�res�datar r r r , s
z"_X25519PublicKey._raw_public_bytesN)�__name__�
__module__� __qualname__r r r r r r r r s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �_X25519PrivateKeyc C s || _ || _d S )N)r r
)r r r r r r r ; s z_X25519PrivateKey.__init__c C sz | j j� }| j jj|| j�}| j j|dk� | j jj|| j jj�}| j j|| j jjk� | j jj || j jj
�}t | j |�S )N� ) r �_create_mem_bio_gcr Zi2d_PUBKEY_bior
r Zd2i_PUBKEY_bior r r! Z
EVP_PKEY_freer )r �bior# r r r r �
public_key? s
z _X25519PrivateKey.public_keyc C s" t |t�std��t| j| j|�S )Nz(peer_public_key must be X25519PublicKey.)�
isinstancer � TypeErrorr r r
)r Zpeer_public_keyr r r �exchangeL s
z_X25519PrivateKey.exchangec C sf |t jjks|t jjkrN|t jjk s>|t jjk s>t|t j� rFtd��| j� S | j j
|||| | j d �S )NzcWhen using Raw both encoding and format must be Raw and encryption_algorithm must be NoEncryption()) r r r r Z
PrivateFormatr- Z NoEncryptionr �_raw_private_bytesr Z_private_key_bytesr
)r r r Zencryption_algorithmr r r �
private_bytesR s z _X25519PrivateKey.private_bytesc C s| | j j� }| j jj|| j| j jj| j jjd| j jj| j jj�}| j j|dk� | j j|�}| j jt |�dk� |t
d � S )Nr r) �0 ) r r* r Zi2d_PKCS8PrivateKey_bior
r r r Z
_read_mem_bio�len�_X25519_KEY_SIZE)r r+ r# Zpkcs8r r r r0 i s
z$_X25519PrivateKey._raw_private_bytesN)r% r& r' r r, r/ r1 r0 r r r r r( 9 s
r( N)Z
__future__r r r Z cryptographyr Z*cryptography.hazmat.backends.openssl.utilsr Z cryptography.hazmat.primitivesr Z0cryptography.hazmat.primitives.asymmetric.x25519r r r4 Zregister_interface�objectr r( r r r r � s %