File: //usr/lib/python3/dist-packages/josepy/__pycache__/jws.cpython-39.pyc
a
��]�7 � @ s� d Z ddlZddlZddlZddlZddlZddlmZmZm Z m
Z
m Z m Z G dd� de
�ZG dd� de j�ZG dd � d e j�ZG d
d � d e j�ZG d d
� d
e
�Zedkr�ee�� � dS )zJSON Web Signature.� N)�b64�errors� json_util�jwa�jwk�utilc @ s, e Zd ZdZdZedd� �Zedd� �ZdS )� MediaTypez MediaType field encoder/decoder.z application/c C s( d|vr$d|v rt �d��| j| S |S )zDecoder.�/�;zUnexpected semi-colon)r �DeserializationError�PREFIX��cls�value� r �,/usr/lib/python3/dist-packages/josepy/jws.py�decode s
zMediaType.decodec C s. d|vr*|� | j�sJ �|t| j�d� S |S )zEncoder.r
N)�
startswithr �lenr
r r r �encode s zMediaType.encodeN)�__name__�
__module__� __qualname__�__doc__r � classmethodr r r r r r r s
r c @ s e Zd ZdZejdejjdd�Z ejddd�Z
ejde j jdd�Z ejddd�Z
ejd dd�Zejd
dd d �Zejd
ejdd�Zejdejdd�Zejdejejdd�Zejdejejdd�Zejddd d �Zdd� Zdd� Zdd� Zej dd� �Zej dd � �Zej d d � �Zd S ) �Headera6 JOSE Header.
.. warning:: This class supports **only** Registered Header
Parameter Names (as defined in section 4.1 of the
protocol). If you need Public Header Parameter Names (4.2)
or Private Header Parameter Names (4.3), you must subclass
and override :meth:`from_json` and :meth:`to_partial_json`
appropriately.
.. warning:: This class does not support any extensions through
the "crit" (Critical) Header Parameter (4.1.11) and as a
conforming implementation, :meth:`from_json` treats its
occurrence as an error. Please subclass if you seek for
a different behaviour.
:ivar x5tS256: "x5t#S256"
:ivar str typ: MIME Media Type, inc. :const:`MediaType.PREFIX`.
:ivar str cty: Content-Type, inc. :const:`MediaType.PREFIX`.
�algT)�decoder� omitempty�jku)r r �kid�x5u�x5cr �r �default�x5tzx5t#S256�typ)�encoderr r �cty�critc s t � fdd�t�� j�D ��S )z4Fields that would not be omitted in the JSON object.c 3 s0 | ](\}}|� t� |��s|t� |�fV qd S �N)Zomit�getattr)�.0�nameZfield��selfr r � <genexpr>N s �z%Header.not_omitted.<locals>.<genexpr>)�dict�sixZ iteritems�_fieldsr. r r. r � not_omittedL s
�zHeader.not_omittedc C sb t |t| ��s td�t|����| �� }|�� }t|��|�rFtd��|�|� t| �f i |��S )Nz Header cannot be added to: {0}z+Addition of overlapping headers not defined)�
isinstance�type� TypeError�formatr4 �set� intersection�update)r/ �otherZnot_omitted_selfZnot_omitted_otherr r r �__add__R s �
zHeader.__add__c C s | j du rt�d��| j S )z�Find key based on header.
.. todo:: Supports only "jwk" header parameter lookup.
:returns: (Public) key found in the header.
:rtype: .JWK
:raises josepy.errors.Error: if key could not be found
Nz No key found)r r �Errorr. r r r �find_key` s
zHeader.find_keyc C s t �d��d S )Nz("crit" is not supported, please subclass)r r )Z unused_valuer r r r) o s �z Header.critc C s dd� | D �S )Nc S s& g | ] }t �tj�tjj|j���qS r )�base64� b64encode�OpenSSL�cryptoZdump_certificate�
FILETYPE_ASN1�wrapped�r, Zcertr r r �
<listcomp>y s �
�z Header.x5c.<locals>.<listcomp>r �r r r r r" w s �z
Header.x5cc
C sL zt dd� | D ��W S tjjyF } zt�|��W Y d }~n
d }~0 0 d S )Nc s s, | ]$}t �tj�tjjt�|���V qd S r* )r ZComparableX509rB rC Zload_certificaterD r@ � b64decoderF r r r r0 s
�
�z Header.x5c.<locals>.<genexpr>)�tuplerB rC r> r r )r �errorr r r r" | s �
N) r r r r r �Fieldr � JWASignature� from_jsonr r r �JWKr r! r" �decode_b64joser% Zx5tS256r r r r& r( r) r4 r= r? r r' r r r r r & s>
���
�
�
r c s� e Zd ZdZeZdZejdddd�Z ejdde� ej
d�Z ejd ej ej
d
�Ze jd d � �Z e jd
d � �Z � fdd�Zedd� �Zedd� �Zd dd�Zede� fdd��Z� fdd�Ze� fdd ��Z� ZS ) � Signaturea JWS Signature.
:ivar combined: Combined Header (protected and unprotected,
:class:`Header`).
:ivar unicode protected: JWS protected header (Jose Base-64 decoded).
:ivar header: JWS Unprotected Header (:class:`Header`).
:ivar str signature: The signature.
)�combined� protectedT� r# �header)r r$ r � signature)r r' c C s t �| �d��S �N�utf-8)r �encode_b64joser rH r r r rS � s zSignature.protectedc C s t �| ��d�S rW )r rP r rH r r r rS � s c s< d|vr| � |�}tt| �jf i |�� | jjd us8J �d S )NrR )�_with_combined�superrQ �__init__rR r )r/ �kwargs�� __class__r r r\ � s
zSignature.__init__c C sZ d|vs J �|� d| jd j�}|� d| jd j�}|rJ|| j�|� }n|}||d< |S )NrR rU rS )�getr3 r$ �
header_cls�
json_loads)r r] rU rS rR r r r rZ � s zSignature._with_combinedc C s t �|�d��d t �|� S )NrX � .)r rA r )r rS �payloadr r r �_msg� s �zSignature._msgNc C s8 |du r| j �� n|}| j jj|j| j| �| j|�d�S )zEVerify.
:param JWK key: Key used for verification.
N)�key�sig�msg)rR r? r �verifyrf rV re rS )r/ rd rf r r r ri � s
�zSignature.verifyc K s� t ||j�sJ �|}||d< |r,|�� |d< t|��| jj�sBJ �|�| jj�sTJ �i }|D ]} | |v r\|�| �|| < q\|r�| jf i |���� }
nd}
| jf i |��} |� |j
| � |
|��} | |
| | d�S )z;Sign.
:param JWK key: Key for signature.
r r rT )rS rU rV ) r5 �kty�
public_keyr9 �issubsetra r3 �popZ
json_dumps�signrf re ) r rd rf r Z include_jwk�protectr] Z
header_paramsZprotected_paramsrU rS rV r r r rn � s" zSignature.signc s$ t t| ��� }|d �� s |d= |S )NrU )r[ rQ �fields_to_partial_jsonr4 )r/ �fieldsr^ r r rp � s z Signature.fields_to_partial_jsonc s8 t t| ��|�}| �|�}d|d �� vr4t�d��|S )Nr rR zalg not present)r[ rQ �fields_from_jsonrZ r4 r r )r �jobjrq Zfields_with_combinedr^ r r rr � s
zSignature.fields_from_json)N)r r r r r ra � __slots__r rL rS rN rU rP rY rV r' r r\ r rZ re ri � frozensetrn rp rr �
__classcell__r r r^ r rQ � s: ��
� rQ c @ sd e Zd ZdZdZeZddd�Zedd� �Z e
dd � �Z d
d � Z ed d
� �Z
ddd�Zedd� �ZdS )�JWSzgJSON Web Signature.
:ivar str payload: JWS Payload.
:ivar str signature: JWS Signatures.
�rd �
signaturesNc s t � �fdd��jD ��S )�Verify.c 3 s | ]}|� �j� �V qd S r* )ri rd �r, rg �rf r/ r r r0 � z JWS.verify.<locals>.<genexpr>)�allry )r/ rf r r| r ri s z
JWS.verifyc K s | || j jf d|i|��fd�S )�Sign.rd rx )�
signature_clsrn )r rd r] r r r rn s �zJWS.signc C s t | j�dksJ �| jd S )zPGet a singleton signature.
:rtype: :class:`JWS.signature_cls`
� r )r ry r. r r r rV
s z
JWS.signaturec C s\ t | j�dksJ �d| jj�� vs&J �t�| jj�d��d t�| j � d t�| jj� S )z7Compact serialization.
:rtype: bytes
r� r rX rc )
r ry rV rU r4 r rA rS r rd r. r r r �
to_compact s �
�� ��zJWS.to_compactc C sf z|� d�\}}}W n ty0 t�d��Y n0 | jt�|��d�t�|�d�}| t�|�|fd�S )zACompact deserialization.
:param bytes compact:
rc zOCompact JWS serialization should comprise of exactly 3 dot-separated componentsrX )rS rV rx )�split�
ValueErrorr r r� r rI r )r �compactrS rd rV rg r r r � from_compact) s �
�zJWS.from_compactTc C sR | j s
J �t�| j�}|rBt| j �dkrB| j d �� }||d< |S || j d�S d S )Nr� r rd rx )ry r rY rd r �to_partial_json)r/ Zflatrd Zretr r r r� < s
�zJWS.to_partial_jsonc sv d|v r d|v r t �d��nVd|v rF� t�|�d��� j�|�fd�S � t�|d �t� fdd�|d D ��d�S d S )NrV ry zFlat mixed with non-flatrd rx c 3 s | ]}� j �|�V qd S r* )r� rN r{ �r r r r0 S s �z JWS.from_json.<locals>.<genexpr>)r r r rP rm r� rN rJ )r rs r r� r rN J s � ��z
JWS.from_json)N)T)r r r r rt rQ r� ri r rn �propertyrV r� r� r� rN r r r r rw � s
rw c @ sZ e Zd ZdZedd� �Zedd� �Zedd� �Zedd � �Zed
d � �Z edd
d��Z
d S )�CLIzJWS CLI.c C s� |j j�|j�� �}|j�� |jdu r.g |_|jr@|j�d� t j
t j �� �
� ||j t|j�d�}|jr�t�|�� �d�� nt�|�� � dS )r Nr )rd rf r ro rX )r rj �loadrf �read�closero r� �appendrw rn �sys�stdinr r9 r2 �print_r� r Zjson_dumps_pretty)r �argsrf rg r r r rn Z s
�zCLI.signc
C s� |j r t�tj�� �� �}nHzt�tj�� �}W n2 tj yb } zt
� |� W Y d}~dS d}~0 0 |j dur�|j
dus|J �|j
�|j �� ��� }|j �� nd}tj�|j� |j|d� S )rz N���)rf )r� rw r� r� r� r� r rb r r> r2 r� rf rj r� rk r� �stdout�writerd ri )r r� rg rK rf r r r ri l s
z
CLI.verifyc C s t j�|�S r* )r rM rN �r �argr r r � _alg_type� s z
CLI._alg_typec C s |t jjv sJ �|S r* )rQ ra r3 r� r r r � _header_type� s zCLI._header_typec C s |t jjv sJ �t jj| S r* )r rO ZTYPESr� r r r � _kty_type� s z
CLI._kty_typeNc C s� |du rt jdd� }t�� }|jddd� |�� }|�d�}|j| jd� |jdd t� d
�d d � |jd
d| j
t j d� |jddd| j
d� |�d�}|j| jd� |jdd t� d
�dd � |jd| jdd � |�|�}|�|�S )z Parse arguments and sign/verify.Nr� z --compact�
store_true)�actionrn )�funcz-kz--key�rbT)r6 Zrequiredz-az--alg)r6 r$ z-pz --protectr� )r� r6 ri Fz--kty)r� �argv�argparse�ArgumentParser� add_argumentZadd_subparsersZ
add_parserZ set_defaultsrn ZFileTyper� r ZRS256r� ri r� �
parse_argsr� )r r� �parserZ
subparsersZ parser_signZ
parser_verifyZparsedr r r �run� s4
� �
�
��
zCLI.run)N) r r r r r rn ri r� r� r� r� r r r r r� W s
r� �__main__)r r� r@ r� rB r2 Zjosepyr r r r r r �objectr ZJSONObjectWithFieldsr rQ rw r� r �exitr� r r r r �<module> s `r_V