3
y�_� � @ s� d dl mZmZmZ d dlmZmZ d dlmZ d dl m
Z
m Z dZ dZ
eje �G dd� de��Zeje
�G d d
� d
e��Zd S ) � )�absolute_import�division�print_function)�
exceptions�utils)�
serialization)�Ed448PrivateKey�Ed448PublicKey�9 �r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�_Ed448PublicKeyc C s || _ || _d S )N)�_backend� _evp_pkey)�self�backend�evp_pkey� r � /usr/lib64/python3.6/ed448.py�__init__ s z_Ed448PublicKey.__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 _Ed448PublicKey.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�_ED448_KEY_SIZE�_lib�EVP_PKEY_get_raw_public_keyr �openssl_assert�buffer)r �buf�buflen�resr r r r + s z!_Ed448PublicKey._raw_public_bytesc C s� | j jj� }| j j|| j jjk� | j jj|| j jj�}| j jj|| j jj| j jj| j jj| j �}| j j|dk� | j jj
||t |�|t |��}|dkr�| j j � t
j�d S )Nr )r
r! �Cryptography_EVP_MD_CTX_newr# r �NULL�gc� Cryptography_EVP_MD_CTX_freeZEVP_DigestVerifyInitr ZEVP_DigestVerify�lenZ_consume_errorsr ZInvalidSignature)r Z signature�data�
evp_md_ctxr' r r r �verify5 s
z_Ed448PublicKey.verifyN)�__name__�
__module__� __qualname__r r r r/ r r r r r s
r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �_Ed448PrivateKeyc C s || _ || _d S )N)r
r )r r r r r r r M s z_Ed448PrivateKey.__init__c C sz | j jjdt�}| j jjdt�}| j jj| j||�}| j j|dk� | j j|d tk� | j jj|�d d � }| j j |�S )Nzunsigned char []zsize_t *r r )
r
r r r r! r" r r# r$ Zed448_load_public_bytes)r r% r& r' r r r r �
public_keyQ s z_Ed448PrivateKey.public_keyc C s� | j jj� }| j j|| j jjk� | j jj|| j jj�}| j jj|| j jj| j jj| j jj| j �}| j j|dk� | j jj
dt �}| j jj
dt |��}| j jj
||||t |��}| j j|dk� | j j|d t k� | j jj||d �d d � S )Nr zunsigned char[]zsize_t *r )r
r! r( r# r r) r* r+ ZEVP_DigestSignInitr r �_ED448_SIG_SIZEr, ZEVP_DigestSignr$ )r r- r. r' r% r& r r r �sign\ s$ z_Ed448PrivateKey.signc 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
PrivateFormat�
isinstanceZ NoEncryptionr �_raw_private_bytesr
Z_private_key_bytesr )r r r Zencryption_algorithmr r r �
private_bytess s z _Ed448PrivateKey.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 r8 � s z#_Ed448PrivateKey._raw_private_bytesN)r0 r1 r2 r r4 r6 r9 r8 r r r r r3 K s
r3 N)Z
__future__r r r Z cryptographyr r Z cryptography.hazmat.primitivesr Z/cryptography.hazmat.primitives.asymmetric.ed448r r r r5 Zregister_interface�objectr r3 r r r r � s 8