Current File : //snap/core18/2887/usr/lib/python3/dist-packages/jwt/__pycache__/algorithms.cpython-36.pyc
3
�M�bX3 � @ s� d dl Z d dlZd dlZddlmZmZ ddlmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZ y�d dlmZ d dlmZmZmZ d dlmZmZm Z m Z m 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� Z0G dd � d e1�Z2G d d � d e2�Z3G d d � d e2�Z4e,�r�G d!d"� d"e2�Z5G d#d$� d$e2�Z6G d%d&� d&e5�Z7dS )'� 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�
is_pem_format�
is_ssh_key)�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�| jttj�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� r6 �0/usr/lib/python3/dist-packages/jwt/algorithms.py�get_default_algorithms' s" r8 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�keyr6 r6 r7 � prepare_keyG s zAlgorithm.prepare_keyc C s t �dS )zn
Returns a digital signature for the specified message
using the specified key value.
N)r: )r; �msgr< r6 r6 r7 �signN s zAlgorithm.signc C s t �dS )zz
Verifies that the specified digital signature is valid
for the specified message and key values.
N)r: )r; r> r< �sigr6 r6 r7 �verifyU s zAlgorithm.verifyc C s t �dS )z7
Serializes a given RSA key into a JWK
N)r: )�key_objr6 r6 r7 �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)r: )�jwkr6 r6 r7 �from_jwkc s zAlgorithm.from_jwkN)
�__name__�
__module__� __qualname__�__doc__r= r? rA � staticmethodrC rE r6 r6 r6 r7 r9 C s r9 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 )r; r< r6 r6 r7 r= p s
zNoneAlgorithm.prepare_keyc C s dS )N� r6 )r; r> r< r6 r6 r7 r? y s zNoneAlgorithm.signc C s dS )NFr6 )r; r> r< r@ r6 r6 r7 rA | s zNoneAlgorithm.verifyN)rF rG rH rI r= r? rA r6 r6 r6 r7 r, k 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)r; rM r6 r6 r7 �__init__� s zHMACAlgorithm.__init__c C s$ t |�}t|�st|�r td��|S )NzdThe specified key is an asymmetric key or x509 certificate and should not be used as an HMAC secret.)r r r r )r; r< r6 r6 r7 r= � s
zHMACAlgorithm.prepare_keyc C s t jttt| ���dd��S )N�oct)�k�kty)�json�dumpsr
r r )rB r6 r6 r7 rC � s zHMACAlgorithm.to_jwkc C s, t j| �}|jd�dkr td��t|d �S )NrQ rO zNot an HMAC keyrP )rR �loads�getr r )rD �objr6 r6 r7 rE � s
zHMACAlgorithm.from_jwkc C s t j||| j�j� S )N)�hmac�newrM Zdigest)r; r> r< r6 r6 r7 r? � s zHMACAlgorithm.signc C s t || j||��S )N)r r? )r; r> r< r@ r6 r6 r7 rA � s zHMACAlgorithm.verifyN)rF rG rH rI �hashlibZsha256r. Zsha384r/ Zsha512r0 rN r= rJ rC rE r? rA r6 r6 r6 r7 r- � s
r- c @ sZ e Zd ZdZejZejZejZdd� Zdd� Z e
dd� �Z e
dd � �Z d
d � Z
d d
� ZdS )r3 z~
Performs signing and verification operations using
RSASSA-PKCS-v1_5 and the specified hash function.
c C s
|| _ d S )N)rM )r; rM r6 r6 r7 rN � s zRSAAlgorithm.__init__c C s� t |t�st |t�r|S t |t�r~t|�}y.|jd�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)�passwordrZ z Expecting a PEM-formatted key.) �
isinstancer r r r �
startswithr r r �
ValueErrorr � TypeError)r; r< r6 r6 r7 r= � s
zRSAAlgorithm.prepare_keyc C s� d }t | dd �r�| j� }ddgtt|jj��tt|jj��tt|j��tt|j��tt|j ��tt|j
��tt|j ��tt|j ��d�
}nBt | dd �r�| j� }ddgtt|j��tt|j��d�}nt
d��tj|�S )N�private_numbers�RSAr? )
rQ �key_ops�n�e�d�p�q�dp�dq�qirA )rQ rb rc rd zNot a public or private key)�getattrr` r
r
�public_numbersrc rd re rf rg �dmp1�dmq1�iqmpr rR rS )rB rV �numbersr6 r6 r7 rC � s* zRSAAlgorithm.to_jwkc s� yt j| �� W n tk
r* td��Y nX � jd�dkrBtd��d� koXd� koXd� k�rfd� krltd ��d
d d d
dg}� fdd�|D �}t|�}|r�t|� r�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�}|jt� �S d� k�r�d� k�r�tt� d �t� d ��}|jt� �S td��d S )NzKey is not valid JSONrQ ra zNot an RSA keyre rd rc Zothz5Unsupported RSA private key: > 2 primes not supportedrf rg rh ri rj c s g | ] }|� k�qS r6 r6 )�.0Zprop)rV r6 r7 �
<listcomp> s z)RSAAlgorithm.from_jwk.<locals>.<listcomp>z@RSA key must include all parameters if any are present besides d)re rf rg rm rn ro rl zNot a public or private key)rR rT r^ r rU �any�allr r r r rc rd r r r Z private_keyr Z
public_key) rD Z other_propsZ props_foundZany_props_foundrl rp re rf rg r6 )rV r7 rE � sT
zRSAAlgorithm.from_jwkc C s |j |tj� | j� �S )N)r? r �PKCS1v15rM )r; r> r< r6 r6 r7 r? 2 s zRSAAlgorithm.signc C s6 y |j ||tj� | j� � dS tk
r0 dS X d S )NTF)rA r ru rM r! )r; r> r< r@ r6 r6 r7 rA 5 s
zRSAAlgorithm.verifyN)rF rG rH rI r r. r/ r0 rN r= rJ rC rE r? rA r6 r6 r6 r7 r3 � s $ ?r3 c @ sB e Zd ZdZejZejZejZdd� Zdd� Z dd� Z
dd � Z d
S ) r4 zr
Performs signing and verification operations using
ECDSA and the specified hash function
c C s
|| _ d S )N)rM )r; rM r6 r6 r7 rN E s zECAlgorithm.__init__c C s� t |t�st |t�r|S t |t�r~t|�}y,|jd�rFt|t� d�}nt|t� d�}W q� t k
rz t
|d t� d�}Y q�X nt d��|S )Ns ecdsa-sha2-)rZ )r[ rZ z Expecting a PEM-formatted key.) r\ r r r r r] r r r r^ r r_ )r; r< r6 r6 r7 r= H s
zECAlgorithm.prepare_keyc C s" |j |tj| j� ��}t||j�S )N)r? r �ECDSArM r �curve)r; r> r< �der_sigr6 r6 r7 r? ` s zECAlgorithm.signc C s\ yt ||j�}W n tk
r$ dS X y |j||tj| j� �� dS tk
rV dS X d S )NFT)r rw r^ rA r rv rM r! )r; r> r< r@ rx r6 r6 r7 rA e s zECAlgorithm.verifyN) rF rG rH rI r r. r/ r0 rN r= r? rA r6 r6 r6 r7 r4 < s r4 c @ s e Zd ZdZdd� Zdd� ZdS )r5 zA
Performs a signature using RSASSA-PSS with MGF1
c C s* |j |tjtj| j� �| jjd�| j� �S )N)�mgf� salt_length)r? r �PSS�MGF1rM � digest_size)r; r> r< r6 r6 r7 r? v s
zRSAPSSAlgorithm.signc
C sJ y0|j ||tjtj| j� �| jjd�| j� � dS tk
rD dS X d S )N)ry rz TF)rA r r{ r| rM r} r! )r; r> r< r@ r6 r6 r7 rA � s
zRSAPSSAlgorithm.verifyN)rF rG rH rI r? rA r6 r6 r6 r7 r5 q s
r5 )8rY rW rR �compatr r �
exceptionsr Zutilsr r r 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! r1 � ImportError�setZrequires_cryptographyr8 �objectr9 r, r- r3 r4 r5 r6 r6 r6 r7 �<module> s6 0 (
(- 5
Mini Shell