File: //usr/lib/python3/dist-packages/josepy/__pycache__/jwk.cpython-38.pyc
U
��]�$ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlm Z ddl
m Z ddl
m Z ddl
mZ ddl
mZ ddlmZmZmZ e�e�ZG d d
� d
ej�ZejG d d � d e��ZejG d
d� de��ZejG dd� de��Z dS )�
JSON Web Key.� N)�default_backend)�hashes)�
serialization)�ec)�rsa)�errors� json_util�utilc @ sd e Zd ZdZdZi ZdZeZdddd�Z e
j fdd �Z e
jd
d � �Zedd d
��Zeddd��ZdS )�JWKr Zkty� N)�,�:T)�indentZ
separatorsZ sort_keysc sP t j|� t� d�}|�tjt� fdd�t�� � � �D ��f� j
�� � � |� � S )zgCompute JWK Thumbprint.
https://tools.ietf.org/html/rfc7638
:returns: bytes
)�backendc 3 s$ | ] \}}|� j kr||fV qd S �N)�required)�.0�k�v��selfr �,/usr/lib/python3/dist-packages/josepy/jwk.py� <genexpr>3 s
�z!JWK.thumbprint.<locals>.<genexpr>)
r ZHashr �update�json�dumps�dict�six� iteritemsZto_json� _thumbprint_json_dumps_params�encode�finalize)r Z
hash_functionZdigestr r r �
thumbprint) s ��
zJWK.thumbprintc C s
t � �dS )ziGenerate JWK with public key.
For symmetric cryptosystems, this would return ``self``.
N��NotImplementedErrorr r r r �
public_key8 s zJWK.public_keyc C s� |d krt � n|}i }tjtjfD ]L}z||||�W S tttjjfk
rl } z |||< W 5 d }~X Y q"X q"tj tj
fD ]H}z|||�W S ttjjfk
r� } z |||< W 5 d }~X Y q|X q|t � d�
|���d S )Nz Unable to deserialize key: {0})r r Zload_pem_private_keyZload_der_private_key�
ValueError� TypeError� cryptography�
exceptionsZUnsupportedAlgorithmZload_pem_public_keyZload_der_public_keyr �Error�format)�cls�data�passwordr r* �loader�errorr r r �_load_cryptography_keyA s* �� �� zJWK._load_cryptography_keyc
C s� z| � |||�}W n> tjk
rP } z t�d|� t|d� W Y �S d}~X Y nX | jtk r~t|| j �s~t�d�
|j | j ���t �
| j�D ] }t||j �r�||d� S q�t�d�
|j ���dS )a� Load serialized key as JWK.
:param str data: Public or private key serialized as PEM or DER.
:param str password: Optional password.
:param backend: A `.PEMSerializationBackend` and
`.DERSerializationBackend` provider.
:raises errors.Error: if unable to deserialize, or unsupported
JWK algorithm
:returns: JWK of an appropriate type.
:rtype: `JWK`
z,Loading symmetric key, asymmetric failed: %s��keyNz"Unable to deserialize {0} into {1}zUnsupported algorithm: {0})r2 r r+ �logger�debug�JWKOct�typ�NotImplemented�
isinstance�cryptography_key_typesr, � __class__r Z
itervalues�TYPES)r- r. r/ r r4 r1 Zjwk_clsr r r �load[ s" � � zJWK.load)NN)NN)�__name__�
__module__� __qualname__�__doc__�type_field_namer= r; r9 r r r ZSHA256r# �abc�abstractmethodr& � classmethodr2 r> r r r r r s �
r c @ sJ e Zd ZdZdZejejfZde j
ddfZ dd� Z e
dd � �Zd
d � Zd S )
�JWKESz<ES JWK.
.. warning:: This is not yet implemented!
ZESZcrv�x�yc C s
t � �d S r r$ r r r r �fields_to_partial_json� s z JWKES.fields_to_partial_jsonc C s
t � �d S r r$ �r- �jobjr r r �fields_from_json� s zJWKES.fields_from_jsonc C s
t � �d S r r$ r r r r r& � s zJWKES.public_keyN)r? r@ rA rB r8 r ZEllipticCurvePublicKeyZEllipticCurvePrivateKeyr; r rC r rJ rF rM r& r r r r rG { s �
rG c @ s> e Zd ZdZdZdZdejfZdd� Z e
dd� �Z d d
� Z d S ) r7 zSymmetric JWK.�octr3 r c C s dt �| j�iS )Nr )r �encode_b64joser4 r r r r rJ � s z JWKOct.fields_to_partial_jsonc C s | t �|d �d�S )Nr r3 )r �decode_b64joserK r r r rM � s zJWKOct.fields_from_jsonc C s | S r r r r r r r& � s zJWKOct.public_keyN)
r? r@ rA rB r8 � __slots__r rC r rJ rF rM r& r r r r r7 � s
r7 c st e Zd ZdZdZejejfZdZ de
j dfZ � fdd�Z
edd � �Zed
d � �Zd d
� Zedd� �Zdd� Z� ZS )�JWKRSAz�RSA JWK.
:ivar key: :class:`~cryptography.hazmat.primitives.asymmetric.rsa.RSAPrivateKey`
or :class:`~cryptography.hazmat.primitives.asymmetric.rsa.RSAPublicKey` wrapped
in :class:`~josepy.util.ComparableRSAKey`
ZRSAr3 �e�nc s@ d|kr*t |d tj�s*t�|d �|d< tt| �j||� d S )Nr4 )r: r
ZComparableRSAKey�superrR �__init__)r �args�kwargs�r< r r rV � s
�zJWKRSA.__init__c C s. dd� }t �t�|t|�dd� �d����S )zQEncode Base64urlUInt.
:type data: long
:rtype: unicode
c S s t | �d rd| S | S )N� �0)�len)�argr r r �_leading_zeros� s z,JWKRSA._encode_param.<locals>._leading_zerosrZ N�L)r rO �binasciiZ unhexlify�hex�rstrip)r- r. r^ r r r �
_encode_param� s �zJWKRSA._encode_paramc C s: zt t�t�|��d�W S tk
r4 t�� �Y nX dS )zDecode Base64urlUInt.� N)�intr` Zhexlifyr rP r' r ZDeserializationError)r- r. r r r �
_decode_param� s zJWKRSA._decode_paramc C s t | �| j�� d�S )Nr3 )�typer4 r& r r r r r& � s zJWKRSA.public_keyc
s@ � �fdd�dD �\}}t j||d�}d�kr<|�t� �}n�� ��d �}d�kszd�kszd�kszd �kszd
�kszd �kr�t�fd d�d
D �� \}}} }
} } tdd� | D ��r�t�d�| ���t� fdd�| D ��\}}} }
} n6t � |||�\}}t �
||�} t � ||�}
t � ||�} t �
|||| |
| |��t� �}� |d�S )Nc 3 s | ]}� � �| �V qd S r �rf �r rH rK r r r � s z*JWKRSA.fields_from_json.<locals>.<genexpr>�rT rS )rS rT �d�p�q�dp�dq�qiZothc 3 s | ]}� � |�V qd S r )�getri )rL r r r � s )rl rm rn ro rp c s s | ]}|d kr|V qd S r r )r Zparamr r r r � s z(Some private parameters are missing: {0}c 3 s | ]}� � |�V qd S r rh ri )r- r r r � s r3 )r ZRSAPublicNumbersr& r rf �tupler r+ r, Zrsa_recover_prime_factorsZ rsa_crt_dmp1Z rsa_crt_dmq1Z rsa_crt_iqmpZRSAPrivateNumbersZ private_key)
r- rL rT rS �public_numbersr4 rk rl rm rn ro rp Z
all_paramsr rK r rM � sP ��� ��� � ��zJWKRSA.fields_from_jsonc s� t � jjtj�r*� j�� }|j|jd�}n>� j�� }� j� � �� }|j|j|j
|j |j |j
|j|jd�}t� fdd�t�|�D ��S )Nrj )rT rS rk rl rm rn ro rp c 3 s | ]\}}|� � |�fV qd S r )rc )r r4 �valuer r r r s �z0JWKRSA.fields_to_partial_json.<locals>.<genexpr>)r: r4 Z_wrappedr � RSAPublicKeyrs rT rS Zprivate_numbersr& rk rl rm Zdmp1Zdmq1Ziqmpr r r )r ZnumbersZparamsZprivateZpublicr r r rJ � s&
�
�
�z JWKRSA.fields_to_partial_json)r? r@ rA rB r8 r ru Z
RSAPrivateKeyr; rQ r rC r rV rF rc rf r& rM rJ �
__classcell__r r rY r rR � s
&rR ) rB rD r` r ZloggingZcryptography.exceptionsr) r Z cryptography.hazmat.backendsr Z cryptography.hazmat.primitivesr r Z)cryptography.hazmat.primitives.asymmetricr r Zjosepyr r r
Z getLoggerr? r5 ZTypedJSONObjectWithFieldsr �registerrG r7 rR r r r r �<module> s(
g