U
    §òcc‘  ã                   @   sj   d dl Z d dlmZ d dlmZ d dlmZmZ e jrBd dl	m
Z
 dZG dd„ deƒZG d	d
„ d
eƒZdS )é    N)Ú_evp_pkey_derive)Úserialization)ÚX25519PrivateKeyÚX25519PublicKey)ÚBackendé    c                   @   s>   e Zd Zddœdd„Zejejedœdd„Zedœd	d
„Z	dS )Ú_X25519PublicKeyr   ©Úbackendc                 C   s   || _ || _d S ©N©Ú_backendÚ	_evp_pkey©Úselfr
   Úevp_pkey© r   úe/var/www/html/project/venv/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   1   s     ÿ ÿz"_X25519PublicKey._raw_public_bytesN)
Ú__name__Ú
__module__Ú__qualname__r   r   r   r   Úbytesr   r   r   r   r   r   r      s   ür   c                   @   s`   e Zd Zddœ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 )Ú_X25519PrivateKeyr   r	   c                 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_keyC   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   r1   r   r   r   ÚexchangeP   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   ÚPrivateFormatr2   ZNoEncryptionr   Ú_raw_private_bytesr   Z_private_key_bytesr   )r   r   r   r5   r   r   r   Úprivate_bytesV   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   r7   r   s    
ù	z$_X25519PrivateKey._raw_private_bytesN)r(   r)   r*   r   r   r0   r+   r4   r   r   r6   ZKeySerializationEncryptionr8   r7   r   r   r   r   r,   >   s   ûr,   )ÚtypingZ*cryptography.hazmat.backends.openssl.utilsr   Zcryptography.hazmat.primitivesr   Z0cryptography.hazmat.primitives.asymmetric.x25519r   r   ÚTYPE_CHECKINGZ,cryptography.hazmat.backends.openssl.backendr   r;   r   r,   r   r   r   r   Ú<module>   s   )