HEX

File: //lib64/python3.6/site-packages/M2Crypto/__pycache__/EVP.cpython-36.pyc
3

YݧZh6�@s
ddlmZddlZddlmZmZmZmZmZej	rJddl
mZmZm
Z
ejd�ZGdd�de�Zeje�dd	�ZGd
d�de�ZGdd
�d
e�Zddd�ZGdd�de�ZGdd�de�Zejfdd�Zejfdd�Zejfdd�Zejfdd�Zejfdd�ZdS) �)�absolute_importN)�BIO�Err�RSA�m2�util)�AnyStr�Optional�CallableZEVPc@seZdZdS)�EVPErrorN)�__name__�
__module__�__qualname__�rr�/usr/lib64/python3.6/EVP.pyrsrcCstj||||�S)a%
    Derive a key from password using PBKDF2 algorithm specified in RFC 2898.

    :param password: Derive the key from this password.
    :param salt:     Salt.
    :param iter:     Number of iterations to perform.
    :param keylen:   Length of key to produce.
    :return:         Key.
    )rZpkcs5_pbkdf2_hmac_sha1)Zpassword�salt�iterZkeylenrrr�pbkdf2src@s:eZdZdZejZdd�Zdd�Zdd�Z	dd	�Z
e
Zd
S)�
MessageDigestz
    Message Digest
    cCsHtt|d�}|dkr"tj|�|_n|�|_tj�|_tj|j|j�dS)N)�getattrrZget_digestbyname�md�
md_ctx_new�ctxZdigest_init)�self�algorrrr�__init__,s
zMessageDigest.__init__cCst|dd�r|j|j�dS)Nr)r�m2_md_ctx_freer)rrrr�__del__8szMessageDigest.__del__cCstj|j|�S)zw
        Add data to be digested.

        :return: -1 for Python error, 1 for success, 0 for OpenSSL failure.
        )rZ
digest_updater)r�datarrr�update=szMessageDigest.updatecCstj|j�S)N)rZdigest_finalr)rrrr�finalFszMessageDigest.finalN)rr
r�__doc__r�md_ctx_freerrrrr �digestrrrrr&s	rc@s@eZdZejZd
dd�Zdd�Zdd�Zdd	�Z	d
d�Z
e
ZdS)�HMAC�sha1cCsFtt|d�}|dkrtd|��|�|_tj�|_tj|j||j�dS)Nzunknown algorithm)rr�
ValueErrorrZhmac_ctx_newr�	hmac_init)r�keyrrrrrrQs

z
HMAC.__init__cCst|dd�r|j|j�dS)Nr)r�m2_hmac_ctx_freer)rrrrrZszHMAC.__del__cCstj|j||j�dS)N)rr'rr)rr(rrr�reset_sz
HMAC.resetcCstj|j|�dS)N)rZhmac_updater)rrrrrrcszHMAC.updatecCstj|j�S)N)rZ
hmac_finalr)rrrrr gsz
HMAC.finalN)r%)rr
rrZ
hmac_ctx_freer)rrr*rr r#rrrrr$Ms
	r$r%cCs.tt|d�}|dkrtd|��tj|||��S)Nzunknown algorithm)rrr&�hmac)r(rrrrrrr+ns
r+c@s>eZdZejZddd�Zdd�Zd	d
�Zdd�Z	dd
d�Z
dS)�Cipherr�md5�12345678�c
Cs�tt|d�}
|
dkrtd|��|
�|_|r`tt|d�}|dkrHtd|��tj|j|�||||�}tj�|_tj|j|j|||�|j|	�~dS)Nzunknown cipherzunknown message digest)	rrr&�cipherZbytes_to_keyZcipher_ctx_newrZcipher_init�set_padding)rZalgr(Ziv�opZkey_as_bytes�dr�i�paddingr0Zkmdrrrrzs



zCipher.__init__cCst|dd�r|j|j�dS)Nr)r�m2_cipher_ctx_freer)rrrrr�szCipher.__del__cCstj|j|�S)N)rZ
cipher_updater)rrrrrr�sz
Cipher.updatecCstj|j�S)N)rZcipher_finalr)rrrrr �szCipher.finalcCstj|j|�S)z*
        Actually always return 1
        )rZcipher_set_paddingr)rr5rrrr1�szCipher.set_paddingN)rr-r.r/r/)r/)rr
rrZcipher_ctx_freer6rrrr r1rrrrr,vs
r,c@s�eZdZdZejZejZd-dd�Z	dd�Z
d	d
�Zdd�Zd.d
d�Z
dd�Zdd�Zdd�ZeZeZdd�Zdd�Zdd�Zd/dd�Zdd�Zd ejfd!d"�Zd ejfd#d$�Zd ejfd%d&�Zd'd(�Zd)d*�Zd+d,�ZdS)0�PKeyz
    Public Key
    Nrr%cCs4|dk	r||_||_ntj�|_d|_|j|�dS)Nr/)�pkey�_pyfreerZpkey_new�_set_context)rr8r9rrrrr�s
z
PKey.__init__cCs4t|dd�r|j|j�t|dd�r0|j|j�dS)Nr9rr)r�m2_pkey_freer8rr)rrrrr�szPKey.__del__cCs|jS)N)r8)rrrr�_ptr�sz	PKey._ptrcCs4tt|d�}|dkrtd|��|�|_tj�|_dS)Nzunknown message digest)rrr&rrr)rrZmdarrrr:�s

zPKey._set_contextcCs|j|�dS)zj
        Reset internal message digest context.

        :param md: The message digest algorithm.
        N)r:)rrrrr�
reset_context�szPKey.reset_contextcCstj|j|j�dS)z9
        Initialise signing operation with self.
        N)r�	sign_initrr)rrrrr>�szPKey.sign_initcCstj|j|�dS)zZ
        Feed data to signing operation.

        :param data: Data to be signed.
        N)r�sign_updater)rrrrrr?�szPKey.sign_updatecCstj|j|j�S)zD
        Return signature.

        :return: The signature.
        )r�
sign_finalrr8)rrrrr@�szPKey.sign_finalcCstj|j|j�dS)zH
        Initialise signature verification operation with self.
        N)r�verify_initrr)rrrrrA�szPKey.verify_initcCstj|j|�S)z�
        Feed data to verification operation.

        :param data: Data to be verified.
        :return: -1 on Python error, 1 for success, 0 for OpenSSL error
        )r�
verify_updater)rrrrrrB�szPKey.verify_updatecCstj|j||j�S)z�
        Return result of verification.

        :param sign: Signature to use for verification
        :return: Result of verification: 1 for success, 0 for failure, -1 on
                 other error.
        )r�verify_finalrr8)rZsignrrrrC�s	zPKey.verify_finalr/cCs4|r tj|j|j�}|r0d|_ntj|j|j�}|S)a�
        Assign the RSA key pair to self.

        :param rsa: M2Crypto.RSA.RSA object to be assigned to self.

        :param capture: If true (default), this PKey object will own the RSA
                        object, meaning that once the PKey object gets
                        deleted it is no longer safe to use the RSA object.

        :return: Return 1 for success and 0 for failure.
        r)rZpkey_assign_rsar8�rsar9Z
pkey_set1_rsa)rrDZcapture�retrrr�
assign_rsas
zPKey.assign_rsacCstj|j�}tj|d�}|S)zd
        Return the underlying RSA key if that is what the EVP
        instance is holding.
        r/)rZ
pkey_get1_rsar8rZRSA_pub)rZrsa_ptrrDrrr�get_rsaszPKey.get_rsaZaes_128_cbccCs&tj|d��}|j|||�SQRXdS)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�openfile�save_key_bio)r�filer0�callback�biorrr�save_key&sz
PKey.save_keycCsX|dkrtj|j|j�|�Stt|d�}|dkr<td|��tj|j|j�|�|�SdS)aA
        Save the key pair to the M2Crypto.BIO object 'bio' in PEM format.

        :param bio: M2Crypto.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.
        Nzno such cipher %s)rZpkey_write_pem_no_cipherr8r<rr&Zpkey_write_pem)rrMr0rL�protorrrrJ:szPKey.save_key_biocCstj�}|j|||�|j�S)a�
        Return key in PEM format in a string.

        :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.
        )r�MemoryBufferrJ�read_all)rr0rLrMrrr�as_pemSszPKey.as_pemcCstj|j�}tj|�}|j�S)z6
        Return key in DER format in a string
        )rZpkey_as_derr8rrPrQ)rZbufrMrrr�as_deres
zPKey.as_dercCstj|j�S)z6
        Return the size of the key in bytes.
        )rZ	pkey_sizer8)rrrr�sizensz	PKey.sizecCstj|j�S)z3
        Return the modulus in hex format.
        )rZpkey_get_modulusr8)rrrr�get_modulususzPKey.get_modulus)Nrr%)r%)r/) rr
rr!rZ	pkey_freer;r"rrrr<r:r=r>r?r@rr rArBrCrFrGr�passphrase_callbackrNrJrRrSrTrUrrrrr7�s4


		


	r7c
Cs0tj|d��}tj|j|�}WdQRXt|d�S)a7
    Load an M2Crypto.EVP.PKey from file.

    :param file: Name of file containing the key in PEM format.

    :param callback: A Python callable object that is invoked
                     to acquire a passphrase with which to protect the
                     key.

    :return: M2Crypto.EVP.PKey object.
    �rNr/)rrIr�
pkey_read_pemrMr7)rKrLrM�cptrrrr�load_key}s
rZcCstj|j�|�}t|d�S)aO
    Load an M2Crypto.EVP.PKey from an M2Crypto.BIO object.

    :param bio: M2Crypto.BIO object containing the key in PEM format.

    :param callback: A Python callable object that is invoked
                     to acquire a passphrase with which to protect the
                     key.

    :return: M2Crypto.EVP.PKey object.
    r/)rrXr<r7)rMrLrYrrr�load_key_bio�s
r[cCs.tj|j�|�}|dkr$ttj���t|d�S)a^
    Load an M2Crypto.EVP.PKey from a public key as a M2Crypto.BIO object.

    :param bio: M2Crypto.BIO object containing the key in PEM format.

    :param callback: A Python callable object that is invoked
                     to acquire a passphrase with which to protect the
                     key.

    :return: M2Crypto.EVP.PKey object.
    Nr/)rZpkey_read_pem_pubkeyr<rrZ	get_errorr7)rMrLrYrrr�load_key_bio_pubkey�s
r\cCstj|�}t||�S)a7
    Load an M2Crypto.EVP.PKey from a string.

    :param string: String containing the key in PEM format.

    :param callback: A Python callable object that is invoked
                     to acquire a passphrase with which to protect the
                     key.

    :return: M2Crypto.EVP.PKey object.
    )rrPr[)�stringrLrMrrr�load_key_string�s
r^cCstj|�}t||�S)aG
    Load an M2Crypto.EVP.PKey from a public key as a string.

    :param string: String containing the key in PEM format.

    :param callback: A Python callable object that is invoked
                     to acquire a passphrase with which to protect the
                     key.

    :return: M2Crypto.EVP.PKey object.
    )rrPr\)r]rLrMrrr�load_key_string_pubkey�s
r_)r%) Z
__future__rZloggingZM2CryptorrrrrZpy27plusZtypingrr	r
Z	getLogger�logr&rZevp_initr�objectrr$r+r,r7rVrZr[r\r^r_rrrr�<module>s&


'!
*^