o
    ªŒcò  ã                   @   sj   d dl Z d dlmZ d dlmZ d dlmZmZ e jr!d 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)ÚX448PrivateKeyÚX448PublicKey)ÚBackendé8   c                   @   s>   e Zd Zddd„Zdejdejdefdd	„Zdefd
d„Z	dS )Ú_X448PublicKeyÚbackendr   c                 C   ó   || _ || _d S ©N©Ú_backendÚ	_evp_pkey©Úselfr	   Zevp_pkey© r   úMusr/lib/python3.10/site-packages/cryptography/hazmat/backends/openssl/x448.pyÚ__init__   ó   
z_X448PublicKey.__init__ÚencodingÚformatÚreturnc                 C   sV   |t jju s|t jju r |t jjus|t jjurt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_X448PublicKey.public_bytesc                 C   óp   | j j dt¡}| j j dt¡}| j j | j||¡}| j  |dk¡ | j  |d tk¡ | j j |t¡d d … S ©Nzunsigned char []zsize_t *é   r   )	r   Ú_ffiÚnewÚ_X448_KEY_SIZEÚ_libÚEVP_PKEY_get_raw_public_keyr   Úopenssl_assertÚbuffer©r   ÚbufÚbuflenÚresr   r   r   r   0   ó   ÿz _X448PublicKey._raw_public_bytesN©r	   r   )
Ú__name__Ú
__module__Ú__qualname__r   r   r   r   Úbytesr   r   r   r   r   r   r      s    
þý
ür   c                   @   sd   e Zd Zddd„Zdefdd„Zdedefd	d
„Zdej	dej
dejdefdd„Zdefdd„ZdS )Ú_X448PrivateKeyr	   r   c                 C   r
   r   r   r   r   r   r   r   <   r   z_X448PrivateKey.__init__r   c                 C   sz   | j j dt¡}| j j dt¡}| j j | j||¡}| j  |dk¡ | j  |d tk¡ | j j |¡d d … }| j  	|¡S r   )
r   r!   r"   r#   r$   r%   r   r&   r'   Zx448_load_public_bytes)r   r)   r*   r+   r   r   r   r   Ú
public_key@   s   ÿz_X448PrivateKey.public_keyÚpeer_public_keyc                 C   s"   t |tƒs	tdƒ‚t| j| j|ƒS )Nz&peer_public_key must be X448PublicKey.)Ú
isinstancer   Ú	TypeErrorr   r   r   )r   r4   r   r   r   ÚexchangeK   s   
z_X448PrivateKey.exchanger   r   Úencryption_algorithmc                 C   sd   |t jju s|t jju r&|t jjus|t jjust|t jƒs"t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   ÚPrivateFormatr5   ZNoEncryptionr   Ú_raw_private_bytesr   Z_private_key_bytesr   )r   r   r   r8   r   r   r   Úprivate_bytesQ   s   ÿÿÿÿz_X448PrivateKey.private_bytesc                 C   r   r   )	r   r!   r"   r#   r$   ZEVP_PKEY_get_raw_private_keyr   r&   r'   r(   r   r   r   r:   m   r,   z"_X448PrivateKey._raw_private_bytesNr-   )r.   r/   r0   r   r   r3   r1   r7   r   r   r9   ZKeySerializationEncryptionr;   r:   r   r   r   r   r2   ;   s    
þýü
ûr2   )ÚtypingZ*cryptography.hazmat.backends.openssl.utilsr   Zcryptography.hazmat.primitivesr   Z.cryptography.hazmat.primitives.asymmetric.x448r   r   ÚTYPE_CHECKINGZ,cryptography.hazmat.backends.openssl.backendr   r#   r   r2   r   r   r   r   Ú<module>   s   '