o
    ªŒc   ã                   @   sn   d dl Z d dlmZ d dlmZ d dlmZmZmZm	Z	 e j
r%d dlmZ G dd„ deƒZG dd	„ d	eƒZdS )
é    N)Ú
exceptions)Úserialization)ÚEd25519PrivateKeyÚEd25519PublicKeyÚ_ED25519_KEY_SIZEÚ_ED25519_SIG_SIZE)ÚBackendc                   @   sT   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ededdfdd„Z
dS )Ú_Ed25519PublicKeyÚbackendr   c                 C   ó   || _ || _d S ©N©Ú_backendÚ	_evp_pkey©Úselfr
   Zevp_pkey© r   úPusr/lib/python3.10/site-packages/cryptography/hazmat/backends/openssl/ed25519.pyÚ__init__   ó   
z_Ed25519PublicKey.__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_Ed25519PublicKey.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 []úsize_t *é   r   )	r   Ú_ffiÚnewr   Ú_libÚEVP_PKEY_get_raw_public_keyr   Úopenssl_assertÚbuffer©r   ÚbufÚbuflenÚresr   r   r   r   0   ó   ÿz#_Ed25519PublicKey._raw_public_bytesÚ	signatureÚdataNc                 C   s¬   | j j ¡ }| j  || j jjk¡ | j j || j jj¡}| j j || j jj| j jj| j jj| j	¡}| j  |dk¡ | j j 
||t|ƒ|t|ƒ¡}|dkrT| j  ¡  tj‚d S )Nr"   )r   r%   ÚEVP_MD_CTX_newr'   r#   ÚNULLÚgcÚEVP_MD_CTX_freeZEVP_DigestVerifyInitr   ZEVP_DigestVerifyÚlenZ_consume_errorsr   ZInvalidSignature)r   r.   r/   Ú
evp_md_ctxr,   r   r   r   Úverify:   s(   
ÿûÿ
þz_Ed25519PublicKey.verify©r
   r   )Ú__name__Ú
__module__Ú__qualname__r   r   r   r   Úbytesr   r   r6   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 )Ú_Ed25519PrivateKeyr
   r   c                 C   r   r   r   r   r   r   r   r   Q   r   z_Ed25519PrivateKey.__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ed25519_load_public_bytes)r   r*   r+   r,   r   r   r   r   Ú
public_keyU   s   ÿz_Ed25519PrivateKey.public_keyr/   c                 C   sð   | j j ¡ }| j  || j jjk¡ | j j || j jj¡}| j j || j jj| j jj| j jj| j	¡}| j  |dk¡ | j j 
dt¡}| j j 
dt|ƒ¡}| j j ||||t|ƒ¡}| j  |dk¡ | j  |d tk¡ | j j ||d ¡d d … S )Nr"   zunsigned char[]r!   r   )r   r%   r0   r'   r#   r1   r2   r3   ZEVP_DigestSignInitr   r$   r   r4   ZEVP_DigestSignr(   )r   r/   r5   r,   r*   r+   r   r   r   Úsign`   s*   
ÿûÿz_Ed25519PrivateKey.signr   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   ÚPrivateFormatÚ
isinstanceZNoEncryptionr   Ú_raw_private_bytesr   Z_private_key_bytesr   )r   r   r   r?   r   r   r   Úprivate_bytesw   s   ÿÿÿÿz _Ed25519PrivateKey.private_bytesc                 C   r   r    )	r   r#   r$   r   r%   ZEVP_PKEY_get_raw_private_keyr   r'   r(   r)   r   r   r   rB   “   r-   z%_Ed25519PrivateKey._raw_private_bytesNr7   )r8   r9   r:   r   r   r=   r;   r>   r   r   r@   ZKeySerializationEncryptionrC   rB   r   r   r   r   r<   P   s    
þýü
ûr<   )ÚtypingÚcryptographyr   Zcryptography.hazmat.primitivesr   Z1cryptography.hazmat.primitives.asymmetric.ed25519r   r   r   r   ÚTYPE_CHECKINGZ,cryptography.hazmat.backends.openssl.backendr   r	   r<   r   r   r   r   Ú<module>   s   <