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)�X448PrivateKey�
X448PublicKey�8 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_X448PublicKeyc C s || _ || _d S )N)�_backend� _evp_pkey)�self�backend�evp_pkey� r � /usr/lib64/python3.6/x448.py�__init__ s z_X448PublicKey.__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_X448PublicKey.public_bytesc C sp | j jjdt�}| j jjdt�}| j jj| j||�}| j j|dk� | j j|d tk� | j jj|t�d d � S )Nzunsigned char []zsize_t *� r ) r �_ffi�new�_X448_KEY_SIZE�_lib�EVP_PKEY_get_raw_public_keyr
�openssl_assert�buffer)r �buf�buflen�resr r r r + s z _X448PublicKey._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 ) �_X448PrivateKeyc C s || _ || _d S )N)r r
)r r r r r r r 8 s z_X448PrivateKey.__init__c C sd | j jjdt�}| j jjdt�}| j jj| j||�}| j j|dk� | j j|d tk� | j j|�S )Nzunsigned char []zsize_t *r r ) r r r r r r! r
r" Zx448_load_public_bytes)r r$ r% r& r r r �
public_key< s z_X448PrivateKey.public_keyc C s" t |t�std��t| j| j|�S )Nz&peer_public_key must be X448PublicKey.)�
isinstancer � TypeErrorr r r
)r Zpeer_public_keyr r r �exchangeF s
z_X448PrivateKey.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_bytesL s z _X448PrivateKey.private_bytesc C sp | j jjdt�}| j jjdt�}| j jj| j||�}| j j|dk� | j j|d tk� | j jj|t�d d � S )Nzunsigned char []zsize_t *r r ) r r r r r Z EVP_PKEY_get_raw_private_keyr
r" r# )r r$ r% r& r r r r/ c s z"_X448PrivateKey._raw_private_bytesN)r' r( r) r r+ r. r0 r/ r r r r r* 6 s
r* N)Z
__future__r r r Z cryptographyr Z*cryptography.hazmat.backends.openssl.utilsr Z cryptography.hazmat.primitivesr Z.cryptography.hazmat.primitives.asymmetric.x448r r r Zregister_interface�objectr r* r r r r �<module> s #