Your IP : 216.73.216.224
U
�u�iZ# � @ s� d dl Z d dlZd dlZz d dlmZmZmZmZmZ W n e k
rL Y nX ddl
m Z m Z m
Z
mZ ddlmZmZmZmZ ddlmZmZmZmZ ddlmZmZm Z m Z G dd � d e �Z e � Z e j!Z!e j"Z"e j#Z#e j$Z$e j%Z%dS )
� N)�Callable�Dict�List�Optional�Union� )� Algorithm�get_default_algorithms�
has_crypto�requires_cryptography)�Mapping� binary_type� string_types� text_type)� DecodeError�InvalidAlgorithmError�InvalidSignatureError�InvalidTokenError)�base64url_decode�base64url_encode� force_bytes�
merge_dictc @ s� e Zd ZdZd#dd�Zedd� �Zdd� Zd d
� Zd d � Z d$dd�Z
d%dd�Z dd� Z dd� Z
d&dd�ZdhZdd �d d �Zd d � Zd!d"� ZdS )'�PyJWSZJWTNc C sf t � | _|d k rt|�nt| j�| _t| j�� �D ]}|| jkr2| j|= q2|sRi }t| �� |�| _d S )N) r � _algorithms�set� _valid_algs�list�keysr �_get_default_options�options)�self�
algorithmsr �key� r# �-/usr/lib/python3/dist-packages/jwt/api_jws.py�__init__ s �
zPyJWS.__init__c C s ddiS )N�verify_signatureTr# r# r# r# r$ r ' s �zPyJWS._get_default_optionsc C s> || j krtd��t|t�s$td��|| j |< | j�|� dS )zW
Registers a new Algorithm for use when creating and verifying tokens.
z Algorithm already has a handler.z!Object is not of type `Algorithm`N)r �
ValueError�
isinstancer � TypeErrorr �add)r �alg_id�alg_objr# r# r$ �register_algorithm- s
zPyJWS.register_algorithmc C s* || j krtd��| j |= | j�|� dS )z�
Unregisters an Algorithm for use when creating and verifying tokens
Throws KeyError if algorithm is not registered.
zJThe specified algorithm could not be removed because it is not registered.N)r �KeyErrorr �remove)r r+ r# r# r$ �unregister_algorithm: s
zPyJWS.unregister_algorithmc C s
t | j�S )zM
Returns a list of supported values for the 'alg' parameter.
)r r )r r# r# r$ �get_algorithmsF s zPyJWS.get_algorithms�HS256c C s� g }|d krd}|| j kr| j|d�}|rB| j|dd� |�|� ttj|d|d��}|�t|�� |�t|�� d� |�} z$| j
| }
|
� |�}|
� | |�} W n6 t
k
r� ts�|tkr�td| ��ntd ��Y nX |�t| �� d� |�S )
NZnone)�typ�algT��encoding)�,�:)Z
separators�cls� .zFAlgorithm '%s' could not be found. Do you have cryptography installed?�Algorithm not supported)r �
header_typ�_validate_headers�updater �json�dumps�appendr �joinr � prepare_keyZsignr. r
r �NotImplementedError) r �payloadr" � algorithm�headersZ json_encoderZsegments�headerZ json_header�
signing_inputr, � signaturer# r# r$ �encodeL s@
��
��z PyJWS.encode� Tc
K sv t | j|�}|d }|r(|s(t�dt� | �|�\} }
} } | �| � |sZtjdtdd� n|rr| �| |
| | ||� | S )Nr& z�It is strongly recommended that you pass in a value for the "algorithms" argument when calling decode(). This argument will be mandatory in a future version.zSThe verify parameter is deprecated. Please use verify_signature in options instead.� )�
stacklevel)r r �warnings�warn�DeprecationWarning�_loadr= �_verify_signature)
r �jwtr" �verifyr! r �kwargsZmerged_optionsr& rE rI rH rJ r# r# r$ �decode� s( �
� �z PyJWS.decodec C s | � |�d }| �|� |S )z�Returns back the JWT header parameters as a dict()
Note: The signature is not verified so the header parameters
should not be fully trusted until signature verification is complete
rM )rR r= )r rT rG r# r# r$ �get_unverified_header� s
zPyJWS.get_unverified_headerc
C sh t |t�r|�d�}tt|�t�s0td�t���z$|�dd�\}}|� dd�\}}W n t
k
rp td��Y nX z t |�}W n" t t
jfk
r� td��Y nX zt�|�d��}W n. t
k
r� } ztd| ��W 5 d }~X Y nX t |t�s�td��z t |�} W n$ t t
jfk
�r( td ��Y nX z t |�}
W n$ t t
jfk
�rZ td
��Y nX | |||
fS ) Nzutf-8z'Invalid token type. Token must be a {0}r: r zNot enough segmentszInvalid header paddingzInvalid header string: %sz,Invalid header string: must be a json objectzInvalid payload paddingzInvalid crypto padding)r( r rK �
issubclass�typer
r �format�rsplit�splitr' r r) �binascii�Errorr? �loadsrW r ) r rT rI Zcrypto_segmentZheader_segmentZpayload_segmentZ header_datarH �erE rJ r# r# r$ rR � s<
�
z PyJWS._loadc C sr |� d�}|d k r"||kr"td��z.| j| }|�|�}|�|||�sNtd��W n tk
rl td��Y nX d S )Nr4 z&The specified alg value is not allowedz Signature verification failedr; )�getr r rC rU r r. ) r rE rI rH rJ r" r! r4 r, r# r# r$ rS � s
zPyJWS._verify_signatureZb64Fr5 c C s0 d|kr| � |d � |s,d|kr,| �|� d S )N�kid�crit)�
_validate_kid�_validate_crit)r rG r6 r# r# r$ r= � s zPyJWS._validate_headersc C s t |t�std��d S )Nz(Key ID header parameter must be a string)r( r r )r rc r# r# r$ re � s
zPyJWS._validate_kidc C st |d }t |t�r t|�dkr&td��|D ]D}t |t�s@td��|| jkrXtd�|���||kr*td�|���q*d S )Nrd r z/Invalid 'crit' header: must be a non-empty listz-Invalid 'crit' header: values must be stringsz"Unsupported critical extension: {}z/Critical extension '{}' is missing from headers)r( r �lenr �str�_supported_critr[ )r rG rd Zextr# r# r$ rf � s
�zPyJWS._validate_crit)NN)r2 NN)rL TNN)rL N)�__name__�
__module__� __qualname__r< r% � staticmethodr r- r0 r1 rK rW rX rR rS ri r= re rf r# r# r# r$ r s4
�
7 �
! ( �
� r )&r^ r? rO �typingr r r r r � ImportErrorr! r r r
r �compatr r
r r �
exceptionsr r r r Zutilsr r r r �objectr Z_jws_global_objrK rW r- r0 rX r# r# r# r$ �<module> s$ p