B
    ó±ô`C  ã               @   sŠ   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eƒZG d	d
„ d
eƒZ	G dd„ de
ƒZe jd ejedœdd„ZdS )é    N)Úhashes)ÚAsymmetricPadding)Úrsac               @   s   e Zd ZdZdS )ÚPKCS1v15zEMSA-PKCS1-v1_5N)Ú__name__Ú
__module__Ú__qualname__Úname© r
   r
   ú`/home/dcms/DCMS/lib/python3.7/site-packages/cryptography/hazmat/primitives/asymmetric/padding.pyr      s   r   c               @   s   e Zd Zeƒ ZdZdd„ ZdS )ÚPSSzEMSA-PSSc             C   sF   || _ t|tƒs"|| jk	r"tdƒ‚|| jk	r<|dk r<tdƒ‚|| _d S )Nzsalt_length must be an integer.r   z$salt_length must be zero or greater.)Ú_mgfÚ
isinstanceÚintÚ
MAX_LENGTHÚ	TypeErrorÚ
ValueErrorZ_salt_length)ÚselfÚmgfÚsalt_lengthr
   r
   r   Ú__init__   s    

zPSS.__init__N)r   r   r   Úobjectr   r	   r   r
   r
   r
   r   r      s   r   c               @   s*   e Zd ZdZdejeje dœdd„Z	dS )ÚOAEPzEME-OAEPÚMGF1)r   Ú	algorithmÚlabelc             C   s*   t |tjƒstdƒ‚|| _|| _|| _d S )Nz*Expected instance of hashes.HashAlgorithm.)r   r   ÚHashAlgorithmr   r   Ú
_algorithmZ_label)r   r   r   r   r
   r
   r   r   '   s
    zOAEP.__init__N)
r   r   r   r	   r   r   ÚtypingÚOptionalÚbytesr   r
   r
   r
   r   r   $   s   r   c               @   s"   e Zd Zeƒ Zejdœdd„ZdS )r   )r   c             C   s   t |tjƒstdƒ‚|| _d S )Nz*Expected instance of hashes.HashAlgorithm.)r   r   r   r   r   )r   r   r
   r
   r   r   8   s    zMGF1.__init__N)r   r   r   r   r   r   r   r   r
   r
   r
   r   r   5   s   r   )zrsa.RSAPrivateKeyzrsa.RSAPublicKey)ÚkeyÚhash_algorithmÚreturnc             C   sF   t | tjtjfƒstdƒ‚| jd d }||j d }|dksBt‚|S )Nz(key must be an RSA public or private keyé   é   é   r   )r   r   ZRSAPrivateKeyZRSAPublicKeyr   Zkey_sizeÚdigest_sizeÚAssertionError)r!   r"   Zemlenr   r
   r
   r   Úcalculate_max_pss_salt_length?   s    r)   )r   Zcryptography.hazmat.primitivesr   Z*cryptography.hazmat.primitives._asymmetricr   Z)cryptography.hazmat.primitives.asymmetricr   r   r   r   r   r   ÚUnionr   r   r)   r
   r
   r
   r   Ú<module>   s   