← Back
Editing: jws.cpython-38.pyc
U ��]�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|kr$d|krt �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|kr*|� | j�st�|t| j�d� S |S )zEncoder.r N)� startswithr �AssertionError�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 s^ t |t| ��s td�t|����| �� }|�� }t|��|�rFtd��|�|� t| �f |�S )Nz Header cannot be added to: {0}z+Addition of overlapping headers not defined)� isinstance�type� TypeError�formatr5 �set� intersection�update)r0 �otherZnot_omitted_selfZnot_omitted_otherr r r �__add__R s � zHeader.__add__c C s | j dkrt�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 sJ zt dd� | D ��W S tjjk rD } zt�|��W 5 d }~X Y nX d S )Nc s s, | ]$}t �tj�tjjt�|���V qd S r+ )r ZComparableX509rC rD Zload_certificaterE rA � b64decoderG r r r r1 s � �z Header.x5c.<locals>.<genexpr>)�tuplerC rD 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* r5 r> r@ r r( r r r r r & sN � � � � � 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 rI r r r rT � s zSignature.protectedc C s t �| ��d�S rX )r rQ r rI r r r rT � s c s8 d|kr| � |�}tt| �jf |� | jjd k s4t�d S )NrS )�_with_combined�superrR �__init__rS r r )r0 �kwargs�� __class__r r r] � s zSignature.__init__c C sZ d|ks t �|�d| jd j�}|�d| jd j�}|rJ|| j�|� }n|}||d< |S )NrS rV rT )r �getr4 r% � header_cls� json_loads)r r^ rV rT rS r r r r[ � s zSignature._with_combinedc C s t �|�d��d t �|� S )NrY � .)r rB r )r rT �payloadr r r �_msg� s �zSignature._msgNc C s8 |dkr| j �� n|}| j jj|j| j| �| j|�d�S )zEVerify. :param JWK key: Key used for verification. N)�key�sig�msg)rS r@ r �verifyrg rW rf rT )r0 re rg r r r rj � s �zSignature.verifyc K s� t ||j�st�|}||d< |r,|�� |d< t|��| jj�sBt�|�| jj�sTt�i }|D ]} | |kr\|�| �|| < q\|r�| jf |�� � } nd} | jf |�} |� |j | � | |��} | | | | d�S )z;Sign. :param JWK key: Key for signature. r r rU )rT rV rW ) r6 �ktyr � public_keyr: �issubsetrb r4 �popZ json_dumps�signrg rf ) r re rg r Z include_jwk�protectr^ Z header_paramsZprotected_paramsrV rT rW r r r ro � s" zSignature.signc s$ t t| ��� }|d �� s |d= |S )NrV )r\ rR �fields_to_partial_jsonr5 )r0 �fieldsr_ r r rq � s z Signature.fields_to_partial_jsonc s8 t t| ��|�}| �|�}d|d �� kr4t�d��|S )Nr rS zalg not present)r\ rR �fields_from_jsonr[ r5 r r )r �jobjrr Zfields_with_combinedr_ r r rs � s zSignature.fields_from_json)N)r r r r r rb � __slots__r rM rT rO rV rQ rZ rW r( r r] r r[ rf rj � frozensetro rq rs � __classcell__r r r_ r rR � s@ � � � rR 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. �re � signaturesNc s t � �fdd��jD ��S )�Verify.c 3 s | ]}|� �j� �V qd S r+ )rj re �r- rh �rg r0 r r r1 s z JWS.verify.<locals>.<genexpr>)�allrz )r0 rg r r} r rj s z JWS.verifyc K s | || j jf d|i|��fd�S )�Sign.re ry )� signature_clsro )r re r^ r r r ro s �zJWS.signc C s t | j�dkst�| jd S )zPGet a singleton signature. :rtype: :class:`JWS.signature_cls` � r )r rz r r/ r r r rW s z JWS.signaturec C s\ t | j�dkst�d| jj�� ks&t�t�| jj� d��d t�| j � d t�| jj� S )z7Compact serialization. :rtype: bytes r� r rY rd ) r rz r rW rV r5 r rB rT r re r/ r r r � to_compact s � �� ��zJWS.to_compactc C sh z|� d�\}}}W n tk r2 t�d��Y nX | jt�|��d�t�|�d�}| t�|�|fd�S )zACompact deserialization. :param bytes compact: rd zOCompact JWS serialization should comprise of exactly 3 dot-separated componentsrY )rT rW ry )�split� ValueErrorr r r� r rJ r )r �compactrT re rW rh r r r � from_compact) s � �zJWS.from_compactTc C sR | j s t�t�| j�}|rBt| j �dkrB| j d �� }||d< |S || j d�S d S )Nr� r re ry )rz r r rZ re r �to_partial_json)r0 Zflatre Zretr r r r� < s �zJWS.to_partial_jsonc sv d|kr d|kr t �d��nVd|krF� t�|�d��� j�|�fd�S � t�|d �t� fdd�|d D ��d�S d S )NrW rz zFlat mixed with non-flatre ry c 3 s | ]}� j �|�V qd S r+ )r� rO r| �r r r r1 S s �z JWS.from_json.<locals>.<genexpr>)r r r rQ rn r� rO rK )r rt r r� r rO J s � ��z JWS.from_json)N)T)r r r r ru rR r� rj r ro �propertyrW r� r� r� rO r r r r rx � s rx 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�� |jdkr.g |_|jr@|j�d� t j t j �� � � ||j t|j�d�}|jr�t�|�� �d�� nt�|�� � dS )r Nr )re rg r rp rY )r rk �loadrg �read�closerp r� �appendrx ro �sys�stdinr r: r3 �print_r� r Zjson_dumps_pretty)r �argsrg rh r r r ro Z s �zCLI.signc C s� |j r t�tj�� �� �}nJzt�tj�� �}W n4 tj k rd } zt � |� W Y �dS d}~X Y nX |j dk r�|j dk s~t�|j �|j �� ��� }|j �� nd}tj�|j� |j|d� S )r{ ���N)rg )r� rx r� r� r� r� r rc r r? r3 r� rg rk r r� rl r� �stdout�writere rj )r r� rh rL rg r r r rj l s z CLI.verifyc C s t j�|�S r+ )r rN rO �r �argr r r � _alg_type� s z CLI._alg_typec C s |t jjkst�|S r+ )rR rb r4 r r� r r r � _header_type� s zCLI._header_typec C s |t jjkst�t jj| S r+ )r rP ZTYPESr r� r r r � _kty_type� s z CLI._kty_typeNc C s� |dkrt 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)�actionro )�funcz-kz--key�rbT)r7 Zrequiredz-az--alg)r7 r% z-pz --protectr� )r� r7 rj Fz--kty)r� �argv�argparse�ArgumentParser� add_argumentZadd_subparsersZ add_parserZ set_defaultsro ZFileTyper� r ZRS256r� rj r� � parse_argsr� )r r� �parserZ subparsersZ parser_signZ parser_verifyZparsedr r r �run� sP � � � � � zCLI.run)N) r r r r r ro rj r� r� r� r� r r r r r� W s r� �__main__)r r� rA r� rC r3 Zjosepyr r r r r r �objectr ZJSONObjectWithFieldsr rR rx r� r �exitr� r r r r �<module> s `r_V
Save File
Cancel