U
    ¶[Öe  ã                   @   sP   d dl mZ d dlmZ d dlmZmZ dZG dd„ deƒZG dd„ deƒZ	d	S )
é    )Ú_evp_pkey_derive)Úserialization)ÚX25519PrivateKeyÚX25519PublicKeyé    c                   @   s8   e Zd Zdd„ Zejejedœdd„Zedœdd„Z	d	S )
Ú_X25519PublicKeyc                 C   s   || _ || _d S ©N©Ú_backendÚ	_evp_pkey©ÚselfÚbackendÚevp_pkey© r   úm/var/www/html/services/stratfitenv/lib/python3.8/site-packages/cryptography/hazmat/backends/openssl/x25519.pyÚ__init__   s    z_X25519PublicKey.__init__)ÚencodingÚformatÚreturnc                 C   sV   |t jjks|t jjkr@|t jjk	s0|t jjk	r8tdƒ‚|  ¡ S | j ||| | jd ¡S )Nz3When using Raw both encoding and format must be Raw)	r   ÚEncodingÚRawÚPublicFormatÚ
ValueErrorÚ_raw_public_bytesr
   Z_public_key_bytesr   )r   r   r   r   r   r   Úpublic_bytes   s&    
ÿ
þ
ÿ
þÿ    ÿz_X25519PublicKey.public_bytes©r   c                 C   s|   | j j d¡}| j j | j|¡}| j  |dk¡ | j  |d | j jjk¡ | j j |d | 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   -   s     ÿ ÿz"_X25519PublicKey._raw_public_bytesN)
Ú__name__Ú
__module__Ú__qualname__r   r   r   r   Úbytesr   r   r   r   r   r   r      s   ür   c                   @   sZ   e Zd Zdd„ Zedœdd„Zeedœdd„Zej	ej
ejed	œd
d„Zedœdd„ZdS )Ú_X25519PrivateKeyc                 C   s   || _ || _d S r   r	   r   r   r   r   r   ;   s    z_X25519PrivateKey.__init__r   c                 C   sz   | j  ¡ }| j j || j¡}| j  |dk¡ | j j || j jj¡}| j  || j jjk¡ | 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_key?   s    
 ÿ ÿz_X25519PrivateKey.public_key)Úpeer_public_keyr   c                 C   s"   t |tƒstdƒ‚t| j| j|ƒS )Nz(peer_public_key must be X25519PublicKey.)Ú
isinstancer   Ú	TypeErrorr   r
   r   )r   r/   r   r   r   ÚexchangeL   s    
z_X25519PrivateKey.exchange)r   r   Úencryption_algorithmr   c                 C   sd   |t jjks|t jjkrL|t jjk	s<|t jjk	s<t|t jƒsDtdƒ‚|  ¡ S | j	 
|||| | jd ¡S )NzcWhen using Raw both encoding and format must be Raw and encryption_algorithm must be NoEncryption())r   r   r   r   ÚPrivateFormatr0   ZNoEncryptionr   Ú_raw_private_bytesr
   Z_private_key_bytesr   )r   r   r   r3   r   r   r   Úprivate_bytesR   s2    
ÿ
þ
ÿ
þ ÿýÿ     ÿz_X25519PrivateKey.private_bytesc              	   C   s|   | j  ¡ }| j j || j| j jj| j jjd| j jj| j jj¡}| j  |dk¡ | 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   r5   n   s    
ù	z$_X25519PrivateKey._raw_private_bytesN)r&   r'   r(   r   r   r.   r)   r2   r   r   r4   ZKeySerializationEncryptionr6   r5   r   r   r   r   r*   :   s   ûr*   N)
Z*cryptography.hazmat.backends.openssl.utilsr   Zcryptography.hazmat.primitivesr   Z0cryptography.hazmat.primitives.asymmetric.x25519r   r   r9   r   r*   r   r   r   r   Ú<module>   s
   )