U
    /j!                     @  s`  d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ	 d dl
mZmZ d dlmZ d dlmZ G dd	 d	ejd
ZeZee	jj G dd dejd
ZeZee	jj e	jjZe	jjZd*dddd	dddZddddddZddddddZddddddZddddddZddddd d!Z ddddd"d#d$Z!d%Z"dddd&d'd(d)Z#dS )+    )annotationsN)gcd)openssl)_serializationhashes)AsymmetricPadding)utilsc                   @  s   e Zd ZejddddddZeejdddd	Zejd
dddZejdddddddZ	ejddddZ
ejdddddddZejd dddZejdd dddZd S )!RSAPrivateKeybytesr   )
ciphertextpaddingreturnc                 C  s   dS )z3
        Decrypts the provided ciphertext.
        N )selfr   r   r   r   Q/tmp/pip-unpacked-wheel-qzbw3lpx/cryptography/hazmat/primitives/asymmetric/rsa.pydecrypt   s    zRSAPrivateKey.decryptintr   c                 C  s   dS z7
        The bit length of the public modulus.
        Nr   r   r   r   r   key_size   s    zRSAPrivateKey.key_sizeRSAPublicKeyc                 C  s   dS )zD
        The RSAPublicKey associated with this private key.
        Nr   r   r   r   r   
public_key    s    zRSAPrivateKey.public_keyzEasym_utils.Prehashed | hashes.HashAlgorithm | asym_utils.NoDigestInfo)datar   	algorithmr   c                 C  s   dS )z!
        Signs the data.
        Nr   )r   r   r   r   r   r   r   sign&   s    	zRSAPrivateKey.signRSAPrivateNumbersc                 C  s   dS )z/
        Returns an RSAPrivateNumbers.
        Nr   r   r   r   r   private_numbers3   s    zRSAPrivateKey.private_numbers_serialization.Encodingz_serialization.PrivateFormatz)_serialization.KeySerializationEncryption)encodingformatencryption_algorithmr   c                 C  s   dS z6
        Returns the key serialized as bytes.
        Nr   )r   r   r    r!   r   r   r   private_bytes9   s    zRSAPrivateKey.private_bytesc                 C  s   dS z!
        Returns a copy.
        Nr   r   r   r   r   __copy__D   s    zRSAPrivateKey.__copy__dictmemor   c                 C  s   dS z&
        Returns a deep copy.
        Nr   r   r(   r   r   r   __deepcopy__J   s    zRSAPrivateKey.__deepcopy__N)__name__
__module____qualname__abcabstractmethodr   propertyr   r   r   r   r#   r%   r+   r   r   r   r   r	      s"   
r	   )	metaclassc                   @  s   e Zd ZejddddddZeejdddd	Zejd
dddZejddddddZ	ejddddddddZ
ejdddddddZejdddddZejd dd d!Zejd"d d#d$d%Zd&S )'r   r
   r   )	plaintextr   r   c                 C  s   dS )z/
        Encrypts the given plaintext.
        Nr   )r   r3   r   r   r   r   encryptV   s    zRSAPublicKey.encryptr   r   c                 C  s   dS r   r   r   r   r   r   r   \   s    zRSAPublicKey.key_sizeRSAPublicNumbersc                 C  s   dS )z-
        Returns an RSAPublicNumbers
        Nr   r   r   r   r   public_numbersc   s    zRSAPublicKey.public_numbersr   z_serialization.PublicFormat)r   r    r   c                 C  s   dS r"   r   )r   r   r    r   r   r   public_bytesi   s    zRSAPublicKey.public_bytesz+asym_utils.Prehashed | hashes.HashAlgorithmNone)	signaturer   r   r   r   c                 C  s   dS )z5
        Verifies the signature of the data.
        Nr   )r   r9   r   r   r   r   r   r   verifys   s    zRSAPublicKey.verifyz5hashes.HashAlgorithm | asym_utils.NoDigestInfo | None)r9   r   r   r   c                 C  s   dS )z@
        Recovers the original data from the signature.
        Nr   )r   r9   r   r   r   r   r   recover_data_from_signature   s    z(RSAPublicKey.recover_data_from_signatureobjectbool)otherr   c                 C  s   dS )z"
        Checks equality.
        Nr   )r   r>   r   r   r   __eq__   s    zRSAPublicKey.__eq__c                 C  s   dS r$   r   r   r   r   r   r%      s    zRSAPublicKey.__copy__r&   r'   c                 C  s   dS r)   r   r*   r   r   r   r+      s    zRSAPublicKey.__deepcopy__N)r,   r-   r.   r/   r0   r4   r1   r   r6   r7   r:   r;   r?   r%   r+   r   r   r   r   r   U   s&   	
r   r   z
typing.Any)public_exponentr   backendr   c                 C  s   t | | tj| |S )N)_verify_rsa_parametersrust_opensslrsagenerate_private_key)r@   r   rA   r   r   r   rE      s    
rE   r8   )r@   r   r   c                 C  s$   | dkrt d|dk r t dd S )N)   i  zopublic_exponent must be either 3 (for legacy compatibility) or 65537. Almost everyone should choose 65537 here!i   z$key_size must be at least 1024-bits.
ValueError)r@   r   r   r   r   rB      s    rB   )emr   c           	      C  sR   d\}}| | }}|dkrJt ||\}}|||  }||||f\}}}}q|| S )zO
    Modular Multiplicative Inverse. Returns x such that: (x*e) mod m == 1
    )   r   r   )divmod)	rI   rJ   x1Zx2abqrZxnr   r   r   _modinv   s    
rR   )prP   r   c                 C  s"   | dks|dkrt dt|| S )zF
    Compute the CRT (q ** -1) % p value from RSA primes p and q.
    rK   Values can't be <= 1)rH   rR   )rS   rP   r   r   r   rsa_crt_iqmp   s    rU   )private_exponentrS   r   c                 C  s$   | dks|dkrt d| |d  S )zg
    Compute the CRT private_exponent % (p - 1) value from the RSA
    private_exponent (d) and p.
    rK   rT   rG   )rV   rS   r   r   r   rsa_crt_dmp1   s    rW   )rV   rP   r   c                 C  s$   | dks|dkrt d| |d  S )zg
    Compute the CRT private_exponent % (q - 1) value from the RSA
    private_exponent (d) and q.
    rK   rT   rG   )rV   rP   r   r   r   rsa_crt_dmq1   s    rX   )rI   rS   rP   r   c                 C  sL   | dks|dks|dkr t d|d |d  t|d |d  }t| |S )z
    Compute the RSA private_exponent (d) given the public exponent (e)
    and the RSA primes p and q.

    This uses the Carmichael totient function to generate the
    smallest possible working value of the private exponent.
    rK   rT   )rH   r   rR   )rI   rS   rP   Zlambda_nr   r   r   rsa_recover_private_exponent   s    "rY   i  ztuple[int, int])nrI   dr   c                 C  s&  |dks|dkrt ddtd|| | kr4t d|| d }|}|d dkrZ|d }qDd}d}|s|tk rtd| d }|d7 }|}||k rbt||| }	|	dkr|	| d krt|	d| dkrt|	d | }
d}qb|d9 }qqb|st d	t| |
\}}|dks
tt|
|fdd
\}
}|
|fS )z
    Compute factors p and q from the private exponent d. We assume that n has
    no more than two factors. This function is adapted from code in PyCrypto.
    rK   zd, e can't be <= 1   zn, d, e don't match   r   FTz2Unable to compute factors p and q from exponent d.)reverse)	rH   pow_MAX_RECOVERY_ATTEMPTSrandomrandintr   rL   AssertionErrorsorted)rZ   rI   r[   ZktottZspottedtriesrN   kZcandrS   rP   rQ   r   r   r   rsa_recover_prime_factors   s6    
$rh   )N)$
__future__r   r/   ra   typingmathr   Z"cryptography.hazmat.bindings._rustr   rC   Zcryptography.hazmat.primitivesr   r   Z*cryptography.hazmat.primitives._asymmetricr   Z)cryptography.hazmat.primitives.asymmetricr   Z
asym_utilsABCMetar	   ZRSAPrivateKeyWithSerializationregisterrD   r   ZRSAPublicKeyWithSerializationr   r5   rE   rB   rR   rU   rW   rX   rY   r`   rh   r   r   r   r   <module>   s4   ?H 		

