U %Y @sddlZddlmZddlZddlZddlmZmZddZddZ d"dd Z d d Z d d Z ddZ ddZddZddZddZddZddZddZddZd d!ZdS)#N)sha256) text_type binary_typecCs:|dkr dSt|tr |dSt|tr.|StddSNutf-8z!Must be a string or bytes object.) isinstancerencoder TypeErrorZstring_or_bytesr 3/usr/lib/python3/dist-packages/pymacaroons/utils.pyconvert_to_bytes s   r cCs:|dkr dSt|tr|St|tr.|dStddSr)rrrdecoder r r r r convert_to_strings   rcCsX|dkr d}t|}t|}||kr4t|d|S||krPt|d||S|SdS)N ) bytearraylenbytes)Z byte_stringsizeZ byte_arrayZlengthr r r truncate_or_padsrcCs td|S)Nsmacaroons-key-generator) hmac_digest)keyr r r generate_derived_key,srcCstj||tdS)N)msgZ digestmod)hmacnewrZdigest)rdatar r r r0s rcCst||}t|SN)rbinasciiZhexlify)rrdigr r r hmac_hex8s r!cCst|}t||Sr)rr!)rZ identifierZ derived_keyr r r create_initial_signature=sr"cCs"t||}t||}t|||Sr)rr!)rZdata1Zdata2Zhash1Zhash2r r r hmac_concatBs  r#cCs t||Sr)r!) signatureZ predicater r r sign_first_party_caveatHsr%cCs t|||Sr)r#)r$Zverification_idZ caveat_idr r r sign_third_party_caveatLsr&cCsHt|t|krdSd}t||D]\}}|t|t|AO}q"|dkS)aE Returns True if the two strings are equal, False otherwise. The time taken is independent of the number of characters that match. For the sake of simplicity, this function executes in constant time only when the two strings have the same length. It short-circuits when they have different lengths. Fr)rzipord)Zval1Zval2resultxyr r r equalsPs r,cCs|dt| dS)zAdd padding to base64 encoded bytes. Padding can be removed when sending the messages. @param b bytes to be padded. @return a padded bytes. =)rbr r r add_base64_paddingbsr1cCs&ds d|krt|Stt|SdS)N_-)raw_urlsafe_b64decodebase64Z b64decoder1sr r r raw_b64decodems r8cCstt|dS)ztBase64 decode with added padding and conversion to bytes. @param s string decode @return bytes decoded r)r5Zurlsafe_b64decoder1rr6r r r r4tsr4cCst|dS)z^Base64 encode with padding removed. @param s string decode @return bytes decoded r-)r5Zurlsafe_b64encoderstripr/r r r raw_urlsafe_b64encode}sr:)N)r5ZhashlibrrrZsixrrr rrrrr!r"r#r%r&r,r1r8r4r:r r r r s&