File: //lib64/python3.6/site-packages/M2Crypto/__pycache__/RSA.cpython-36.pyc
3
6�\�8 � @ s d dl mZ d dlZd dlmZmZmZmZ ejrZd dl m
Z
m Z m Z m
Z
mZmZmZmZ G dd� de�Zeje� ejZejZejZejZG dd� de�ZG d d
� d
e�Zd d � Z ej fd
d�Z e fdd�Z ej fdd�Z!ej fdd�Z"ej fdd�Z#dd� Z$dd� Z%dd � Z&dS ) � )�absolute_importN)�BIO�Err�m2�util)�Any�AnyStr�Callable�Dict�List�IO�Optional�Tuplec @ s e Zd ZdS )�RSAErrorN)�__name__�
__module__� __qualname__� r r �/usr/lib64/python3.6/RSA.pyr s r c @ s� e Zd ZdZejZd1dd�Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdejfdd�Zdejfdd�ZeZdejfdd�Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd2d(d)�Zd3d*d+�Z d4d,d-�Z d5d.d/�Z d0S )6�RSAz
RSA Key Pair.
r c C s" t j|�std��|| _|| _dS )zG
:param rsa: binary representation of OpenSSL RSA type
z'rsa' type errorN)r Zrsa_type_check�AssertionError�rsa�_pyfree)�selfr r r r r �__init__ s z RSA.__init__c C s t | dd�r| j| j� d S )Nr r )�getattr� m2_rsa_freer )r r r r �__del__) s z RSA.__del__c C s t tj| j�d> �S )N� )�intr Zrsa_sizer )r r r r �__len__. s z RSA.__len__c C s0 |dkrt j| j�S |dkr(t j| j�S t�d S )N�e�n)r � rsa_get_er � rsa_get_n�AttributeError)r �namer r r � __getattr__2 s
zRSA.__getattr__c C s( | j � std��tj| j�tj| j�fS )Nzkey is not initialised)� check_keyr r r# r r$ )r r r r �pub; s zRSA.pubc C s | j � std��tj| j||�S )Nzkey is not initialised)r( r r Zrsa_public_encryptr )r �data�paddingr r r �public_encrypt@ s zRSA.public_encryptc C s | j � std��tj| j||�S )Nzkey is not initialised)r( r r Zrsa_public_decryptr )r r* r+ r r r �public_decryptE s zRSA.public_decryptc C s | j � std��tj| j||�S )Nzkey is not initialised)r( r r Zrsa_private_encryptr )r r* r+ r r r �private_encryptJ s zRSA.private_encryptc C s | j � std��tj| j||�S )Nzkey is not initialised)r( r r Zrsa_private_decryptr )r r* r+ r r r �private_decryptO s zRSA.private_decryptZ aes_128_cbcc C s^ |dkr t j| j|j� |�S tt |d�}|dkr>td| ��n|� }t j| j|j� ||�S dS )aC
Save the key pair to an M2Crypto.BIO.BIO object in PEM format.
:param bio: M2Crypto.BIO.BIO object to save key to.
:param cipher: Symmetric cipher to protect the key. The default
cipher is 'aes_128_cbc'. If cipher is None, then
the key is saved in the clear.
:param callback: A Python callable object that is invoked
to acquire a passphrase with which to protect
the key. The default is
util.passphrase_callback.
Nznot such cipher %s)r Zrsa_write_key_no_cipherr �_ptrr r Z
rsa_write_key)r �bio�cipher�callbackZciphr r r � save_key_bioT s zRSA.save_key_bioc C s& t j|d��}| j|||�S Q R X dS )a%
Save the key pair to a file in PEM format.
:param file: Name of file to save key to.
:param cipher: Symmetric cipher to protect the key. The default
cipher is 'aes_128_cbc'. If cipher is None, then
the key is saved in the clear.
:param callback: A Python callable object that is invoked
to acquire a passphrase with which to protect
the key. The default is
util.passphrase_callback.
�wbN)r �openfiler4 )r �filer2 r3 r1 r r r �save_keyo s z RSA.save_keyc C s t j� }| j|||� |j� S )zB
Returns the key(pair) as a string in PEM format.
)r � MemoryBufferr4 �read)r r2 r3 r1 r r r �as_pem� s z
RSA.as_pemc C s t j| j|j� �S )z�
Save the key pair to an M2Crypto.BIO.BIO object in DER format.
:param bio: M2Crypto.BIO.BIO object to save key to.
)r Zrsa_write_key_derr r0 )r r1 r r r �save_key_der_bio� s zRSA.save_key_der_bioc C s" t j|d�� }| j|�S Q R X dS )zj
Save the key pair to a file in DER format.
:param file: Filename to save key to
r5 N)r r6 r< )r r7 r1 r r r � save_key_der� s zRSA.save_key_derc C s t j| j|j� �S )z�
Save the public key to an M2Crypto.BIO.BIO object in PEM format.
:param bio: M2Crypto.BIO.BIO object to save key to.
)r �rsa_write_pub_keyr r0 )r r1 r r r �save_pub_key_bio� s zRSA.save_pub_key_bioc
C s* t j|d��}tj| j|j� �S Q R X dS )zq
Save the public key to a file in PEM format.
:param file: Name of file to save key to.
r5 N)r r6 r r> r r0 )r r7 r1 r r r � save_pub_key� s zRSA.save_pub_keyc C s t j| j�S )a�
Validate RSA keys.
It checks that p and q are in fact prime, and that n = p*q.
:return: returns 1 if rsa is a valid RSA key, and 0 otherwise.
-1 is returned if an error occurs while checking the key.
If the key is invalid or an error occurred, the reason
code can be obtained using ERR_get_error(3).
)r Z
rsa_check_keyr )r r r r r( � s z
RSA.check_key�sha1� c C sB t t|d�}|dkr td| ��tj| j||� |�}| j|tj�S )a�
Signs a digest with the private key using RSASSA-PSS
:param digest: A digest created by using the digest method
:param salt_length: The length of the salt to use
:param algo: The hash algorithm to use
Legal values like 'sha1','sha224', 'sha256',
'ripemd160', and 'md5'.
:return: a string which is the signature
Nznot such hash algorithm %s)r r r Zrsa_padding_add_pkcs1_pssr r. �
no_padding)r �digest�algo� salt_length�hash� signaturer r r �sign_rsassa_pss� s
zRSA.sign_rsassa_pssc C sD t t|d�}|dkr td| ��| j|tj�}tj| j|||� |�S )a�
Verifies the signature RSASSA-PSS
:param data: Data that has been signed
:param signature: The signature signed with RSASSA-PSS
:param salt_length: The length of the salt that was used
:param algo: The hash algorithm to use
Legal values are for example 'sha1','sha224',
'sha256', 'ripemd160', and 'md5'.
:return: 1 or 0, depending on whether the signature was
verified or not.
Nznot such hash algorithm %s)r r r r- rC Zrsa_verify_pkcs1_pssr )r r* rH rE rF rG Zplain_signaturer r r �verify_rsassa_pss� s
zRSA.verify_rsassa_pssc C s2 t td| d�}|dkr"td|��tj| j||�S )aT
Signs a digest with the private key
:param digest: A digest created by using the digest method
:param algo: The method that created the digest.
Legal values like 'sha1','sha224', 'sha256',
'ripemd160', and 'md5'.
:return: a string which is the signature
�NID_Nzunknown algorithm)r r �
ValueErrorZrsa_signr )r rD rE � digest_typer r r �sign� s
zRSA.signc C s4 t td| d�}|dkr"td|��tj| j|||�S )a�
Verifies the signature with the public key
:param data: Data that has been signed
:param signature: The signature signed with the private key
:param algo: The method use to create digest from the data
before it was signed. Legal values like
'sha1','sha224', 'sha256', 'ripemd160', and 'md5'.
:return: 1 or 0, depending on whether the signature was
verified or not.
rK Nzunknown algorithm)r r rL Z
rsa_verifyr )r r* rH rE rM r r r �verify s
z
RSA.verifyN)r )rA rB )rA rB )rA )rA ) r r r �__doc__r Zrsa_freer r r r r' r) r, r- r. r/ r �passphrase_callbackr4 r8 Zsave_pemr; r< r= r? r@ r( rI rJ rN rO r r r r r s4
r c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�RSA_pubz0
Object interface to an RSA public key.
c C s |dkrt d��n
|| j|< d S )Nr! r" z0use factory function new_pub_key() to set (e, n))r! r" )r �__dict__)r r& �valuer r r � __setattr__% s
zRSA_pub.__setattr__c G s t d��d S )Nz!RSA_pub object has no private key)r )r �argvr r r r. , s zRSA_pub.private_encryptc G s t d��d S )Nz!RSA_pub object has no private key)r )r rV r r r r/ 0 s zRSA_pub.private_decryptc O s
| j |�S )z*
Save public key to file.
)r@ )r r7 �args�kwr r r r8 4 s zRSA_pub.save_keyc O s
| j |�S )z)
Save public key to BIO.
)r? )r r1 rW rX r r r r4 ; s zRSA_pub.save_key_bioc C s t j| j�S )N)r Zrsa_check_pub_keyr )r r r r r( F s zRSA_pub.check_keyN)
r r r rP rU r. r/ r8 r4 r( r r r r rR s rR c C s t tj� ��d S )N)r r Zget_error_messager r r r � rsa_errorK s rY c C s& ddddg}|j || � |j� dS )z)
Default callback for gen_key().
�.�+�*�
N)�write�flush)�pr" �outZchr r r �keygen_callbackP s rb c C s t tj| ||�d�S )a�
Generate an RSA key pair.
:param bits: Key length, in bits.
:param e: The RSA public exponent.
:param callback: A Python callable object that is invoked
during key generation; its usual purpose is to
provide visual feedback. The default callback is
keygen_callback.
:return: M2Crypto.RSA.RSA object.
� )r r Zrsa_generate_key)�bitsr! r3 r r r �gen_keyZ s re c
C s t j| �� }t||�S Q R X dS )aa
Load an RSA key pair from file.
:param file: Name of file containing RSA public key in PEM format.
:param callback: A Python callable object that is invoked
to acquire a passphrase with which to unlock the
key. The default is util.passphrase_callback.
:return: M2Crypto.RSA.RSA object.
N)r r6 � load_key_bio)r7 r3 r1 r r r �load_keym s
rg c C s( t j| j� |�}|dkr t� t|d�S )a�
Load an RSA key pair from an M2Crypto.BIO.BIO object.
:param bio: M2Crypto.BIO.BIO object containing RSA key pair in PEM
format.
:param callback: A Python callable object that is invoked
to acquire a passphrase with which to unlock the
key. The default is util.passphrase_callback.
:return: M2Crypto.RSA.RSA object.
Nrc )r Z rsa_read_keyr0 rY r )r1 r3 r r r r rf ~ s rf c C s t j| �}t||�S )a^
Load an RSA key pair from a string.
:param string: String containing RSA key pair in PEM format.
:param callback: A Python callable object that is invoked
to acquire a passphrase with which to unlock the
key. The default is util.passphrase_callback.
:return: M2Crypto.RSA.RSA object.
)r r9 rf )�stringr3 r1 r r r �load_key_string� s
ri c C s t j| ��
}t|�S Q R X dS )z�
Load an RSA public key from file.
:param file: Name of file containing RSA public key in PEM format.
:return: M2Crypto.RSA.RSA_pub object.
N)r r6 �load_pub_key_bio)r7 r1 r r r � load_pub_key� s rk c C s& t j| j� �}|dkr t� t|d�S )z�
Load an RSA public key from an M2Crypto.BIO.BIO object.
:param bio: M2Crypto.BIO.BIO object containing RSA public key in PEM
format.
:return: M2Crypto.RSA.RSA_pub object.
Nrc )r Zrsa_read_pub_keyr0 rY rR )r1 r r r r rj � s
rj c C s( | \}}t j� }t j|||� t|d�S )a�
Instantiate an RSA_pub object from an (e, n) tuple.
:param e: The RSA public exponent; it is a string in OpenSSL's MPINT
format - 4-byte big-endian bit-count followed by the
appropriate number of bits.
:param n: The RSA composite of primes; it is a string in OpenSSL's
MPINT format - 4-byte big-endian bit-count followed by the
appropriate number of bits.
:return: M2Crypto.RSA.RSA_pub object.
rc )r Zrsa_newZ
rsa_set_enrR )Ze_nr! r" r r r r � new_pub_key� s rl )'Z
__future__r �sysZM2Cryptor r r r Zpy27plusZtypingr r r r
r r r
r � Exceptionr Zrsa_initrC Z
pkcs1_paddingZsslv23_paddingZpkcs1_oaep_padding�objectr rR rY �stdoutrb re rQ rg rf ri rk rj rl r r r r �<module> s. (
,