3
y�_& � @ s� d dl mZmZmZ d dlmZmZ d dlmZ d dl m
Z
m Z m Z m
Z
eje �G dd� de��Zeje
�G dd� de��Zd S )
� )�absolute_import�division�print_function)�
exceptions�utils)�
serialization)�Ed25519PrivateKey�Ed25519PublicKey�_ED25519_KEY_SIZE�_ED25519_SIG_SIZEc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�_Ed25519PublicKeyc C s || _ || _d S )N)�_backend� _evp_pkey)�self�backend�evp_pkey� r � /usr/lib64/python3.6/ed25519.py�__init__ s z_Ed25519PublicKey.__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 _Ed25519PublicKey.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�newr
�_lib�EVP_PKEY_get_raw_public_keyr �openssl_assert�buffer)r �buf�buflen�resr r r r * s z#_Ed25519PublicKey._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 �verify4 s
z_Ed25519PublicKey.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 ) �_Ed25519PrivateKeyc C s || _ || _d S )N)r
r )r r r r r r r L s z_Ed25519PrivateKey.__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# Zed25519_load_public_bytes)r r$ r% r& r r r r �
public_keyP s z _Ed25519PrivateKey.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 r r+ ZEVP_DigestSignr# )r r, r- r& r$ r% r r r �sign[ s$ z_Ed25519PrivateKey.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_bytesr s z _Ed25519PrivateKey.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 r6 � s z%_Ed25519PrivateKey._raw_private_bytesN)r/ r0 r1 r r3 r4 r7 r6 r r r r r2 J s
r2 N)Z
__future__r r r Z cryptographyr r Z cryptography.hazmat.primitivesr Z1cryptography.hazmat.primitives.asymmetric.ed25519r r r
r Zregister_interface�objectr r2 r r r r � s 8