Edit File: algorithms.cpython-37.pyc
B �b�[4 � @ s� d dl Z d dlZd dlZddlmZmZ ddlmZ ddlm Z m Z m Z m Z m Z mZmZmZ y�d dlmZ d dlmZmZmZ d dlmZmZmZmZm Z m Z m Z m Z d d l m!Z!m"Z" d d l#m$Z$m%Z% d d l&m'Z' d d l(m)Z) d Z*W n e+k r� dZ*Y nX e,ddddddddddg �Z-dd� Z.G dd � d e/�Z0G d d � d e0�Z1G d d � d e0�Z2e*�r�G d!d"� d"e0�Z3G d#d$� d$e0�Z4G d%d&� d&e3�Z5dS )'� N� )�constant_time_compare� string_types)�InvalidKeyError)�base64url_decode�base64url_encode�der_to_raw_signature� force_bytes� force_unicode�from_base64url_uint�raw_to_der_signature�to_base64url_uint)�hashes)�load_pem_private_key�load_pem_public_key�load_ssh_public_key)� RSAPrivateKey� RSAPublicKey�RSAPrivateNumbers�RSAPublicNumbers�rsa_recover_prime_factors� rsa_crt_dmp1� rsa_crt_dmq1� rsa_crt_iqmp)�EllipticCurvePrivateKey�EllipticCurvePublicKey)�ec�padding)�default_backend)�InvalidSignatureTF�RS256�RS384�RS512�ES256�ES384�ES521�ES512�PS256�PS384�PS512c C s� t � ttj�ttj�ttj�d�} tr�| �ttj�ttj�ttj�ttj�ttj�ttj�ttj�t t j�t t j�t t j�d� � | S )zE Returns the algorithms that are implemented by the library. )ZnoneZHS256ZHS384ZHS512) r r! r" r# r$ r% r& r' r( r) ) � NoneAlgorithm� HMACAlgorithm�SHA256�SHA384�SHA512� has_crypto�update� RSAAlgorithm� ECAlgorithm�RSAPSSAlgorithm)Zdefault_algorithms� r4 �?/opt/alt/python37/lib/python3.7/site-packages/jwt/algorithms.py�get_default_algorithms& s" r6 c @ s@ e Zd ZdZdd� Zdd� Zdd� Zedd � �Zed d � �Z d S ) � AlgorithmzH The interface for an algorithm used to sign and verify tokens. c C s t �dS )z� Performs necessary validation and conversions on the key and returns the key value in the proper format for sign() and verify(). N)�NotImplementedError)�self�keyr4 r4 r5 � prepare_keyF s zAlgorithm.prepare_keyc C s t �dS )zn Returns a digital signature for the specified message using the specified key value. N)r8 )r9 �msgr: r4 r4 r5 �signM s zAlgorithm.signc C s t �dS )zz Verifies that the specified digital signature is valid for the specified message and key values. N)r8 )r9 r< r: �sigr4 r4 r5 �verifyT s zAlgorithm.verifyc C s t �dS )z7 Serializes a given RSA key into a JWK N)r8 )�key_objr4 r4 r5 �to_jwk[ s zAlgorithm.to_jwkc C s t �dS )zb Deserializes a given RSA key from JWK back into a PublicKey or PrivateKey object N)r8 )�jwkr4 r4 r5 �from_jwkb s zAlgorithm.from_jwkN) �__name__� __module__� __qualname__�__doc__r; r= r? � staticmethodrA rC r4 r4 r4 r5 r7 B s r7 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r* zZ Placeholder for use when no signing or verification operations are required. c C s |dkr d }|d k r t d��|S )N� z*When alg = "none", key value must be None.)r )r9 r: r4 r4 r5 r; o s zNoneAlgorithm.prepare_keyc C s dS )N� r4 )r9 r< r: r4 r4 r5 r= x s zNoneAlgorithm.signc C s dS )NFr4 )r9 r< r: r> r4 r4 r5 r? { s zNoneAlgorithm.verifyN)rD rE rF rG r; r= r? r4 r4 r4 r5 r* j s r* c @ sZ e Zd ZdZejZejZej Z dd� Z dd� Z e dd� �Ze dd � �Zd d � Zd d � ZdS )r+ zf Performs signing and verification operations using HMAC and the specified hash function. c C s || _ d S )N)�hash_alg)r9 rK r4 r4 r5 �__init__� s zHMACAlgorithm.__init__c s6 t � �� ddddg}t� fdd�|D ��r2td��� S )Ns -----BEGIN PUBLIC KEY-----s -----BEGIN CERTIFICATE-----s -----BEGIN RSA PUBLIC KEY-----s ssh-rsac s g | ] }|� k�qS r4 r4 )�.0Z string_value)r: r4 r5 � <listcomp>� s z-HMACAlgorithm.prepare_key.<locals>.<listcomp>zdThe specified key is an asymmetric key or x509 certificate and should not be used as an HMAC secret.)r �anyr )r9 r: Zinvalid_stringsr4 )r: r5 r; � s zHMACAlgorithm.prepare_keyc C s t �ttt| ���dd��S )N�oct)�k�kty)�json�dumpsr r r )r@ r4 r4 r5 rA � s zHMACAlgorithm.to_jwkc C s, t �| �}|�d�dkr td��t|d �S )NrR rP zNot an HMAC keyrQ )rS �loads�getr r )rB �objr4 r4 r5 rC � s zHMACAlgorithm.from_jwkc C s t �||| j��� S )N)�hmac�newrK �digest)r9 r< r: r4 r4 r5 r= � s zHMACAlgorithm.signc C s t || �||��S )N)r r= )r9 r< r: r> r4 r4 r5 r? � s zHMACAlgorithm.verifyN)rD rE rF rG �hashlib�sha256r, �sha384r- �sha512r. rL r; rH rA rC r= r? r4 r4 r4 r5 r+ s r+ c @ sZ e Zd ZdZejZejZejZdd� Zdd� Z e dd� �Z e dd � �Z d d � Z d d � ZdS )r1 z~ Performs signing and verification operations using RSASSA-PKCS-v1_5 and the specified hash function. c C s || _ d S )N)rK )r9 rK r4 r4 r5 rL � s zRSAAlgorithm.__init__c C s� t |t�st |t�r|S t |t�r~t|�}y.|�d�rFt|t� d�}nt|d t� d�}W q� t k rz t |t� d�}Y q�X nt d��|S )Ns ssh-rsa)�backend)�passwordr_ z Expecting a PEM-formatted key.) � isinstancer r r r � startswithr r r � ValueErrorr � TypeError)r9 r: r4 r4 r5 r; � s zRSAAlgorithm.prepare_keyc C s� d }t | dd �r�| �� }ddgtt|jj��tt|jj��tt|j��tt|j��tt|j ��tt|j ��tt|j ��tt|j ��d� }nBt | dd �r�| �� }ddgtt|j��tt|j��d�}nt d��t�|�S )N�private_numbers�RSAr= ) rR �key_ops�n�e�d�p�q�dp�dq�qir? )rR rg rh ri zNot a public or private key)�getattrre r r �public_numbersrh ri rj rk rl �dmp1�dmq1�iqmpr rS rT )r@ rW �numbersr4 r4 r5 rA � s* zRSAAlgorithm.to_jwkc s� yt �| �� W n tk r* td��Y nX � �d�dkrBtd��d� k�rhd� k�rhd� k�rhd� krptd ��d d d d dg}� fdd�|D �}t|�}|r�t|�s�td��tt� d �t� d ��}|�rt t� d �t� d �t� d �t� d �t� d �t� d �|d�}nHt� d �}t |j ||j �\}}t |||t ||�t||�t||�|d�}|�t� �S d� k�r�d� k�r�tt� d �t� d ��}|�t� �S td��d S )NzKey is not valid JSONrR rf zNot an RSA keyrj ri rh Zothz5Unsupported RSA private key: > 2 primes not supportedrk rl rm rn ro c s g | ] }|� k�qS r4 r4 )rM Zprop)rW r4 r5 rN s z)RSAAlgorithm.from_jwk.<locals>.<listcomp>z@RSA key must include all parameters if any are present besides d)rj rk rl rr rs rt rq zNot a public or private key)rS rU rc r rV rO �allr r r r rh ri r r r Z private_keyr Z public_key) rB Z other_propsZ props_foundZany_props_foundrq ru rj rk rl r4 )rW r5 rC � sT zRSAAlgorithm.from_jwkc C s |� |t�� | �� �S )N)r= r �PKCS1v15rK )r9 r< r: r4 r4 r5 r= 8 s zRSAAlgorithm.signc C s6 y |� ||t�� | �� � dS tk r0 dS X d S )NTF)r? r rw rK r )r9 r< r: r> r4 r4 r5 r? ; s zRSAAlgorithm.verifyN)rD rE rF rG r r, r- r. rL r; rH rA rC r= r? r4 r4 r4 r5 r1 � s $ ?r1 c @ sB e Zd ZdZejZejZejZdd� Zdd� Z dd� Z dd � Z d S ) r2 zr Performs signing and verification operations using ECDSA and the specified hash function c C s || _ d S )N)rK )r9 rK r4 r4 r5 rL K s zECAlgorithm.__init__c C s� t |t�st |t�r|S t |t�r~t|�}y,|�d�rFt|t� d�}nt|t� d�}W q� t k rz t |d t� d�}Y q�X nt d��|S )Ns ecdsa-sha2-)r_ )r` r_ z Expecting a PEM-formatted key.) ra r r r r rb r r r rc r rd )r9 r: r4 r4 r5 r; N s zECAlgorithm.prepare_keyc C s" |� |t�| �� ��}t||j�S )N)r= r �ECDSArK r �curve)r9 r< r: �der_sigr4 r4 r5 r= f s zECAlgorithm.signc C s\ yt ||j�}W n tk r$ dS X y |�||t�| �� �� dS tk rV dS X d S )NFT)r ry rc r? r rx rK r )r9 r< r: r> rz r4 r4 r5 r? k s zECAlgorithm.verifyN) rD rE rF rG r r, r- r. rL r; r= r? r4 r4 r4 r5 r2 B s r2 c @ s e Zd ZdZdd� Zdd� ZdS )r3 zA Performs a signature using RSASSA-PSS with MGF1 c C s* |� |tjt�| �� �| jjd�| �� �S )N)�mgf� salt_length)r= r �PSS�MGF1rK � digest_size)r9 r< r: r4 r4 r5 r= | s zRSAPSSAlgorithm.signc C sJ y0|� ||tjt�| �� �| jjd�| �� � dS tk rD dS X d S )N)r{ r| TF)r? r r} r~ rK r r )r9 r< r: r> r4 r4 r5 r? � s zRSAPSSAlgorithm.verifyN)rD rE rF rG r= r? r4 r4 r4 r5 r3 w s r3 )6r[ rX rS �compatr r � exceptionsr �utilsr r r r r r r r Z cryptography.hazmat.primitivesr Z,cryptography.hazmat.primitives.serializationr r r Z-cryptography.hazmat.primitives.asymmetric.rsar r r r r r r r Z,cryptography.hazmat.primitives.asymmetric.ecr r Z)cryptography.hazmat.primitives.asymmetricr r Z cryptography.hazmat.backendsr Zcryptography.exceptionsr r/ � ImportError�setZrequires_cryptographyr6 �objectr7 r* r+ r1 r2 r3 r4 r4 r4 r5 �<module> s6 ( ( (4 5