3
¥K´^Z  ã               @   s„   d dl mZmZmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZmZ dZejeƒG dd„ deƒƒZejeƒG d	d
„ d
eƒƒZdS )é    )Úabsolute_importÚdivisionÚprint_function)Úutils)Ú_evp_pkey_derive)Úserialization)ÚX25519PrivateKeyÚX25519PublicKeyé    c               @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )Ú_X25519PublicKeyc             C   s   || _ || _d S )N)Ú_backendÚ	_evp_pkey)ÚselfÚbackendÚevp_pkey© r   úO/tmp/pip-unpacked-wheel-vvkwn1hz/cryptography/hazmat/backends/openssl/x25519.pyÚ__init__   s    z_X25519PublicKey.__init__c             C   st   |t jjks|t jjkr@|t jjk	s0|t jjk	r8tdƒ‚| jƒ S |t jkr^|t jjk	r^tdƒ‚| jj	||| | j
d ƒS )Nz3When using Raw both encoding and format must be Rawz?format must be SubjectPublicKeyInfo when encoding is PEM or DER)r   ÚEncodingÚRawÚPublicFormatÚ
ValueErrorÚ_raw_public_bytesÚ_PEM_DERZSubjectPublicKeyInfor   Z_public_key_bytesr   )r   ÚencodingÚformatr   r   r   Úpublic_bytes   s    
z_X25519PublicKey.public_bytesc             C   s|   | j jjdƒ}| j jj| j|ƒ}| j j|dkƒ | j j|d | j jjkƒ | j jj|d | j jj	ƒ}| j jj
||ƒd d … S )Nzunsigned char **r
   r   )r   Ú_ffiÚnewÚ_libZEVP_PKEY_get1_tls_encodedpointr   Úopenssl_assertÚNULLÚgcZOPENSSL_freeÚbuffer)r   ZucharppÚresÚdatar   r   r   r   4   s    
z"_X25519PublicKey._raw_public_bytesN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r   r      s   r   c               @   s4   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ ZdS )Ú_X25519PrivateKeyc             C   s   || _ || _d S )N)r   r   )r   r   r   r   r   r   r   C   s    z_X25519PrivateKey.__init__c             C   sz   | j jƒ }| j jj|| jƒ}| j j|dkƒ | j jj|| j jjƒ}| j j|| j jjkƒ | j jj	|| j jj
ƒ}t| j |ƒS )Né   )r   Ú_create_mem_bio_gcr   Zi2d_PUBKEY_bior   r    Zd2i_PUBKEY_bior   r!   r"   ZEVP_PKEY_freer   )r   Úbior$   r   r   r   r   Ú
public_keyG   s    
z_X25519PrivateKey.public_keyc             C   s"   t |tƒstdƒ‚t| j| j|ƒS )Nz(peer_public_key must be X25519PublicKey.)Ú
isinstancer	   Ú	TypeErrorr   r   r   )r   Zpeer_public_keyr   r   r   ÚexchangeT   s    
z_X25519PrivateKey.exchangec             C   s‚   |t jjks|t jjkrN|t jjk	s>|t jjk	s>t|t jƒ rFtdƒ‚| jƒ S |t j	krl|t jj
k	rltdƒ‚| jj|||| jd ƒS )NzcWhen using Raw both encoding and format must be Raw and encryption_algorithm must be NoEncryption()z0format must be PKCS8 when encoding is PEM or DER)r   r   r   r   ZPrivateFormatr.   ZNoEncryptionr   Ú_raw_private_bytesr   ZPKCS8r   Z_private_key_bytesr   )r   r   r   Zencryption_algorithmr   r   r   Úprivate_bytes\   s    
z_X25519PrivateKey.private_bytesc             C   s|   | j jƒ }| j jj|| j| j jj| j jjd| j jj| j jjƒ}| j j|dkƒ | j j|ƒ}| j jt	|ƒdkƒ |t
 d … S )Nr   r*   é0   )r   r+   r   Zi2d_PKCS8PrivateKey_bior   r   r!   r    Z_read_mem_bioÚlenÚ_X25519_KEY_SIZE)r   r,   r$   Zpkcs8r   r   r   r1   y   s    
z$_X25519PrivateKey._raw_private_bytesN)r&   r'   r(   r   r-   r0   r2   r1   r   r   r   r   r)   A   s
   r)   N)Ú
__future__r   r   r   Zcryptographyr   Z*cryptography.hazmat.backends.openssl.utilsr   Zcryptography.hazmat.primitivesr   Z0cryptography.hazmat.primitives.asymmetric.x25519r   r	   r5   Zregister_interfaceÚobjectr   r)   r   r   r   r   Ú<module>   s   .