3
“k®ZE
ã @ s² d dl Z d dlZd dlZddlmZmZmZ yd dlmZm Z W n e
k
rT Y nX dd„ Z dd„ Z d d
„ Z
d d „ Zd
d„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ ZdS )é Né )Ú binary_typeÚbytes_from_intÚ text_type)Údecode_dss_signatureÚencode_dss_signaturec C s. t | tƒr| jdƒS t | tƒr"| S tdƒ‚d S )Nzutf-8zExpected a string value)Ú
isinstancer Údecoder Ú TypeError)Úvalue© r ú/usr/lib/python3.6/utils.pyÚ
force_unicode s
r c C s. t | tƒr| jdƒS t | tƒr"| S tdƒ‚d S )Nzutf-8zExpected a string value)r r Úencoder r
)r r r r
Ú force_bytes s
r c C sB t | tƒr| jdƒ} t| ƒd }|dkr8| dd| 7 } tj| ƒS )NÚasciié r ó =)r r r ÚlenÚbase64Zurlsafe_b64decode)ÚinputZremr r r
Úbase64url_decode! s
r c C s t j| ƒjddƒS )Nr ó )r Zurlsafe_b64encodeÚreplace)r r r r
Úbase64url_encode- s r c C s0 | dk rt dƒ‚t| ƒ}t|ƒdkr(d}t|ƒS )Nr zMust be a positive integeró )Ú
ValueErrorr r r )ÚvalÚ int_bytesr r r
Úto_base64url_uint1 s r c C sJ t | tƒr| jdƒ} t| ƒ}tjdt|ƒ |ƒ}tdjdd„ |D ƒƒdƒS )Nr z%sBÚ c S s g | ] }d | ‘qS )z%02xr )Ú.0Zbyter r r
ú
D s z'from_base64url_uint..é ) r r r r ÚstructÚunpackr ÚintÚjoin)r ÚdataZbufr r r
Úfrom_base64url_uint= s
r) c C sV |s| S y| j ƒ }|j|ƒ W n2 ttfk
rP } ztd| ƒ‚W Y d d }~X nX |S )Nz-original and updates must be a dictionary: %s)ÚcopyÚupdateÚAttributeErrorr r
)ZoriginalZupdatesZmerged_optionsÚer r r
Ú
merge_dictG s r. c C s$ dd| | f }t j|jdƒƒ}|S )Nz%0*xé r )ÚbinasciiZa2b_hexr )ZnumÚ num_bytesZ
padded_hexZ
big_endianr r r
Únumber_to_bytesT s r2 c C s t tj| ƒdƒS )Nr# )r& r0 Zb2a_hex)Ústringr r r
Úbytes_to_numberZ s r4 c C s2 |j }|d d }t| ƒ\}}t||ƒt||ƒ S )Né é )Úkey_sizer r2 )Zder_sigÚcurveÚnum_bitsr1 ÚrÚsr r r
Úder_to_raw_signature^ s r< c C sT |j }|d d }t| ƒd| kr*tdƒ‚t| d |… ƒ}t| |d … ƒ}t||ƒS )Nr5 r6 r/ zInvalid signature)r7 r r r4 r )Zraw_sigr8 r9 r1 r: r; r r r
Úraw_to_der_signatureg s r= )r r0 r$ Úcompatr r r Z/cryptography.hazmat.primitives.asymmetric.utilsr r Ú ImportErrorr r r r r r) r. r2 r4 r< r= r r r r
Ú s$