o
    pkj^                     @  s   d dl mZ d dlZd dlmZ d dlmZ d dlmZ G dd deZ	G dd	 d	Z
G d
d dZG dd dZG dd deZG dd deZG dd dejdZG dd deZdddZdS )     )annotationsN)hashes)AsymmetricPadding)rsac                   @  s   e Zd ZdZd	ddZdS )
PKCS1v15zEMSA-PKCS1-v1_5otherobjectreturnboolc                 C  s   t |tstS dS )NT)
isinstancer   NotImplementedselfr    r   o/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/cryptography/hazmat/primitives/asymmetric/padding.py__eq__   s   
zPKCS1v15.__eq__Nr   r   r	   r
   )__name__
__module____qualname__namer   r   r   r   r   r      s    r   c                   @     e Zd ZdZdS )
_MaxLengthz Sentinel value for `MAX_LENGTH`.Nr   r   r   __doc__r   r   r   r   r          r   c                   @  r   )_AutozSentinel value for `AUTO`.Nr   r   r   r   r   r      r   r   c                   @  r   )_DigestLengthz#Sentinel value for `DIGEST_LENGTH`.Nr   r   r   r   r   r   "   r   r   c                   @  sN   e Zd ZU e Ze Ze ZdZ	de
d< dd	d
ZdddZedddZdS )PSSzEMSA-PSS(int | _MaxLength | _Auto | _DigestLength_salt_lengthmgfMGFsalt_lengthr	   Nonec                 C  sD   || _ t|ttttfstdt|tr|dk rtd|| _d S )NzBsalt_length must be an integer, MAX_LENGTH, DIGEST_LENGTH, or AUTOr   z$salt_length must be zero or greater.)	_mgfr   intr   r   r   	TypeError
ValueErrorr    )r   r!   r#   r   r   r   __init__-   s   
zPSS.__init__r   r   r
   c                 C  s&   t |tstS | j|jko| j|jkS N)r   r   r   r%   r    r   r   r   r   r   A   s   
z
PSS.__eq__c                 C     | j S r*   r%   r   r   r   r   r!   I      zPSS.mgfN)r!   r"   r#   r   r	   r$   r   r	   r"   )r   r   r   r   
MAX_LENGTHr   AUTOr   DIGEST_LENGTHr   __annotations__r)   r   propertyr!   r   r   r   r   r   &   s   
 

r   c                   @  s@   e Zd ZdZddd	ZdddZedddZedddZdS )OAEPzEME-OAEPr!   r"   	algorithmhashes.HashAlgorithmlabelbytes | Nonec                 C  s*   t |tjs
td|| _|| _|| _d S Nz*Expected instance of hashes.HashAlgorithm.)r   r   HashAlgorithmr'   r%   
_algorithm_label)r   r!   r6   r8   r   r   r   r)   Q   s
   
zOAEP.__init__r   r   r	   r
   c                 C  s2   t |tstS | j|jko| j|jko| j|jkS r*   )r   r5   r   r%   r<   r=   r   r   r   r   r   ^   s   


zOAEP.__eq__c                 C  r+   r*   )r<   r-   r   r   r   r6   h   r.   zOAEP.algorithmc                 C  r+   r*   r,   r-   r   r   r   r!   l   r.   zOAEP.mgfN)r!   r"   r6   r7   r8   r9   r   )r	   r7   r/   )	r   r   r   r   r)   r   r4   r6   r!   r   r   r   r   r5   N   s    


r5   c                   @  s   e Zd ZU ded< dS )r"   r7   r<   N)r   r   r   r3   r   r   r   r   r"   q   s   
 r"   )	metaclassc                   @  s    e Zd ZdddZdd	d
ZdS )MGF1r6   r7   c                 C  s   t |tjs
td|| _d S r:   )r   r   r;   r'   r<   )r   r6   r   r   r   r)   v   s   
zMGF1.__init__r   r   r	   r
   c                 C  s   t |tstS | j|jkS r*   )r   r?   r   r<   r   r   r   r   r   |   s   
zMGF1.__eq__N)r6   r7   r   )r   r   r   r)   r   r   r   r   r   r?   u   s    
r?   key$rsa.RSAPrivateKey | rsa.RSAPublicKeyhash_algorithmr7   r	   r&   c                 C  sF   t | tjtjfstd| jd d }||j d }|dks!J |S )Nz(key must be an RSA public or private key         r   )r   r   RSAPrivateKeyRSAPublicKeyr'   key_sizedigest_size)r@   rB   emlenr#   r   r   r   calculate_max_pss_salt_length   s   rK   )r@   rA   rB   r7   r	   r&   )
__future__r   abccryptography.hazmat.primitivesr   *cryptography.hazmat.primitives._asymmetricr   )cryptography.hazmat.primitives.asymmetricr   r   r   r   r   r   r5   ABCMetar"   r?   rK   r   r   r   r   <module>   s   
(#