a
Bg< @ s d Z ddlZddlZddlmZ ddlmZmZmZ ddl Z
ddl m Z ddl
mZmZ ddlmZmZmZ ddlmZmZ dd lmZ dd
lmZmZm Z e e Z G d d d ej Z!G d
d de!eZ"G dd de"Z#G dd dZ$G dd de$e"Z%G dd de$e"Z&G dd de"Z'e"(e#dej)Z*e"(e#dej+Z,e"(e#dej-Z.e"(e%d ej)Z/e"(e%d ej+Z0e"(e%d ej-Z1e"(e&d ej)Z2e"(e&d ej+Z3e"(e&d!ej-Z4e"(e'd"ej)Z5e"(e'd#ej+Z6e"(e'd$ej-Z7dS )%zZJSON Web Algorithms.
https://tools.ietf.org/html/draft-ietf-jose-json-web-algorithms-40
N)Hashable)AnyCallableDict)default_backend)hasheshmac)ecpaddingrsa)decode_dss_signatureencode_dss_signature)
HashAlgorithm)errors
interfacesjwkc @ s e Zd ZdZdS )JWAzJSON Web Algorithm.N)__name__
__module__ __qualname____doc__ r r ./usr/lib/python3.9/site-packages/josepy/jwa.pyr s r c @ s e Zd ZU dZi Zeed f ed< eed< eddddZ ee
dd d
Z e d d d
Z
ed d dddZed ddZeed dddZejeeedddZejeeee
dddZed d d ZdS ) JWASignaturez-Base class for JSON Web Signature Algorithms.
SIGNATURESktyN)namereturnc C s
|| _ d S Nr )selfr r r r __init__& s zJWASignature.__init__)otherr c C s t |tstS | j|jkS r )
isinstancer NotImplementedr )r r" r r r __eq__) s
zJWASignature.__eq__)r c C s t | j| jfS r )hash __class__r r r r r __hash__. s zJWASignature.__hash__)
signature_clsr c C s || j |j< |S )z(Register class for JSON deserialization.)r r )clsr* r r r register1 s zJWASignature.registerc C s | j S r r r( r r r to_partial_json7 s z JWASignature.to_partial_json)jobjr c C s
| j | S r )r )r+ r. r r r from_json: s zJWASignature.from_jsonkeymsgr c C s
t dS ) Sign the ``msg`` using ``key``.NNotImplementedError)r r1 r2 r r r sign> s zJWASignature.signr1 r2 sigr c C s
t dS )z-Verify the ``msg`` and ``sig`` using ``key``.Nr4 )r r1 r2 r8 r r r verifyC s zJWASignature.verifyc C s | j S r r r( r r r __repr__H s zJWASignature.__repr__)r r r r r r str__annotations__r r! boolr% intr) classmethodr, r- r/ abcabstractmethodbytesr6 r9 r: r r r r r s
r c sX e Zd ZejZeeg ef d fdd Z e
e
e
dddZ e
e
e
e ddd Z
ZS )
_JWAHSr hash_c s t | | | _d S r superr! r& r r rE r' r r r! O s z_JWAHS.__init__r0 c C s& t j|| jt d}|| | S )NZbackend)r HMACr& r updatefinalize)r r1 r2 Zsignerr r r r6 S s
z _JWAHS.signr7 c
C sn t j|| jt d}|| z|| W n8 tjjyd } z t j
|dd W Y d }~dS d }~0 0 dS d S )NrJ Texc_infoF) r rK r& r rL r9 cryptography
exceptionsInvalidSignatureloggerdebug)r r1 r2 r8 Zverifiererrorr r r r9 X s
z
_JWAHS.verify)r r r r ZJWKOctr r; r r r! rB r6 r= r9
__classcell__r r rI r rC L s rC c @ sV e Zd ZU ejZeZee d< eZ
e e d< e j
eedddZe jeeedddZd S )
_JWARSAr
r& r0 c
C s z| || j| jW S tyP } z$tj|dd tdW Y d}~nHd}~0 ty } z(tj|dd tt |W Y d}~n
d}~0 0 dS )r3 TrN %Public key cannot be used for signingN)
r6 r
r& AttributeErrorrS rT r Error
ValueErrorr; r r1 r2 rU r r r r6 i s z _JWARSA.signr7 c
C sZ z| ||| j| j W n8 tjjyP } z tj|dd W Y d}~dS d}~0 0 dS dS ),Verify the ``msg` and ``sig`` using ``key``.TrN NF)r9 r
r& rP rQ rR rS rT )r r1 r2 r8 rU r r r r9 t s z_JWARSA.verifyN)r r r r ZJWKRSAr r$ r
r r< r& r r Z
RSAPrivateKeyrB r6 Z RSAPublicKeyr= r9 r r r r rW d s
rW c s. e Zd Zeeg ef dd fdd Z ZS )_JWARSNr rE r c s" t | t | _| | _d S r )rG r! r
ZPKCS1v15r& rH rI r r r! s
z_JWARS.__init__r r r r; r r r! rV r r rI r r^ s r^ c s. e Zd Zeeg ef dd fdd Z ZS )_JWAPSNr_ c s4 t | tjt| tjjd| _| | _d S )N)ZmgfZ salt_length)rG r! r
ZPSSZMGF1Z
MAX_LENGTHr& rH rI r r r! s z_JWAPS.__init__r` r r rI r ra s ra c s e Zd ZejZeeg ef d fdd Z e
j e e dddZ
e
j e e dddZe
je e ed d
d Ze
je e ed d
dZ ZS )_JWAECrD c s t | | | _d S r rF rH rI r r r! s z_JWAEC.__init__r0 c C sB | ||}t|\}}tj|j}|j|dd|j|dd S )r3 big)length byteorder)_signr r JWKECexpected_length_for_curvecurveto_bytes)r r1 r2 r8 ZdrZdsrd r r r r6 s z _JWAEC.signc
C s z| |t| jW S tyR } z$tj|dd tdW Y d }~nHd }~0 t y } z(tj|dd tt
|W Y d }~n
d }~0 0 d S )NTrN rX ) r6 r ECDSAr& rY rS rT r rZ r[ r; r\ r r r rf s z _JWAEC._signr7 c C s^ t j|j}t|d| kr"dS ttj|d| ddtj||d dd}| |||S )r] Fr rc )re N) r rg rh ri lenr
r>
from_bytes_verify)r r1 r2 r8 Zrlenasn1sigr r r r9 s z
_JWAEC.verify)r1 r2 rp r c
C s\ z| ||t| j W n8 tjjyR } z tj|dd W Y d }~dS d }~0 0 dS d S )NTrN F) r9 r rk r& rP rQ rR rS rT )r r1 r2 rp rU r r r ro s z_JWAEC._verify)r r r r rg r r; r r r! r ZEllipticCurvePrivateKeyrB r6 rf ZEllipticCurvePublicKeyr= r9 ro rV r r rI r rb s
rb HS256HS384HS512RS256RS384RS512PS256PS384PS512ES256ES384ES512)8r r@ Zloggingcollections.abcr typingr r r Zcryptography.exceptionsrP Z cryptography.hazmat.backendsr Z cryptography.hazmat.primitivesr r Z)cryptography.hazmat.primitives.asymmetricr r
r Z/cryptography.hazmat.primitives.asymmetric.utilsr r
Z%cryptography.hazmat.primitives.hashesr Zjosepyr r r Z getLoggerr rS ZJSONDeSerializabler r rC rW r^ ra rb r, ZSHA256rq ZSHA384rr ZSHA512rs rt ru rv rw rx ry rz r{ r| r r r r s>
,1