([fddlZddlmZGddeZGddeeZGdd eeZGd d ZGd d ZGddeZ GddeZ dS)N)ABCMetac2eZdZdZdZdZdZdZdZdZ dZ dS)JWEAlgorithmBasez+Base interface for all JWE algorithms. NJWEalgctNNotImplementedError)selfraw_datas O/var/www/piapp/venv/lib/python3.11/site-packages/authlib/jose/rfc7516/models.py prepare_keyzJWEAlgorithmBase.prepare_key!!ctr r )r enc_algkeys rgenerate_presetz JWEAlgorithmBase.generate_presetrr) __name__ __module__ __qualname____doc__ EXTRA_HEADERSname descriptionalgorithm_typealgorithm_locationrrrrrrsVM DKN""""""""rr) metaclassc eZdZdZddZdZdS) JWEAlgorithmzInterface for JWE algorithm conforming to RFC7518. JWA specification (RFC7518) SHOULD implement the algorithms for JWE with this base implementation. Nctr r )r rheadersrpresets rwrapzJWEAlgorithm.wraprrctr r )r rekr$rs runwrapzJWEAlgorithm.unwraprrr )rrrrr&r)rrrr"r"sA"""""""""rr"c0eZdZdZddZdZddZddZdS)$JWEAlgorithmWithTagAwareKeyAgreementzInterface for JWE algorithm with tag-aware key agreement (in key agreement with key wrapping mode). ECDH-1PU is an example of such an algorithm. Nctr r )r rr sender_keyr%s r!generate_keys_and_prepare_headerszFJWEAlgorithmWithTagAwareKeyAgreement.generate_keys_and_prepare_headers%rrctr r )r rr$rr-epkcektags ragree_upon_key_and_wrap_cekz@JWEAlgorithmWithTagAwareKeyAgreement.agree_upon_key_and_wrap_cek(rrctr r )r rr$rr-r%s rr&z)JWEAlgorithmWithTagAwareKeyAgreement.wrap+rrctr r )r rr(r$rr-r2s rr)z+JWEAlgorithmWithTagAwareKeyAgreement.unwrap.rrr )rrrrr.r3r&r)rrrr+r+!si"""""""""""""""""rr+cDeZdZdZdZdZdZdZdZdZ dZ dZ dZ dZ dS) JWEEncAlgorithmNrencc:tj|jdzSN)osurandomCEK_SIZEr s r generate_cekzJWEEncAlgorithm.generate_cek;sz$-1,---rc:tj|jdzSr:)r<r=IV_SIZEr?s r generate_ivzJWEEncAlgorithm.generate_iv>sz$,!+,,,rcZt|dz|jkrtddS)Nr;zInvalid "iv" size)lenrB ValueError)r ivs rcheck_ivzJWEEncAlgorithm.check_ivAs0 r77Q;$, & &011 1 ' &rct)aEncrypt the given "msg" text. :param msg: text to be encrypt in bytes :param aad: additional authenticated data in bytes :param iv: initialization vector in bytes :param key: encrypted key in bytes :return: (ciphertext, tag) r )r msgaadrGrs rencryptzJWEEncAlgorithm.encryptEs "!rct)a8Decrypt the given cipher text. :param ciphertext: ciphertext in bytes :param aad: additional authenticated data in bytes :param iv: initialization vector in bytes :param tag: authentication tag in bytes :param key: encrypted key in bytes :return: message r )r ciphertextrKrGr2rs rdecryptzJWEEncAlgorithm.decryptPs "!r)rrrrrrrrBr>r@rCrHrLrOrrrr7r72s~ DKNGH...---222 " " " " " " " "rr7c*eZdZdZdZdZdZdZdZdS)JWEZipAlgorithmNrzipctr r r ss rcompresszJWEZipAlgorithm.compresscrrctr r rTs r decompresszJWEZipAlgorithm.decompressfrr) rrrrrrrrVrXrrrrQrQ]sG DKN""""""""rrQc>eZdZdZfdZdZedZxZS)JWESharedHeaderziShared header object for JWE. Combines protected header and shared unprotected header together. ci}|r|||r||t||r|ni|_|r|ni|_dSr )updatesuper__init__ protected unprotected)r r_r`obj __class__s rr^zJWESharedHeader.__init__os{  " JJy ! ! !  $ JJ{ # # # &/7R*5=;;2rcd|||j|dSr )r\r_)r additions rupdate_protectedz JWESharedHeader.update_protectedys0 H h'''''rct||r|S||d|dS)Nr_r`) isinstanceget)clsras r from_dictzJWESharedHeader.from_dict}sC c3   Js377;'')?)?@@@r) rrrrr^re classmethodrj __classcell__rbs@rrZrZjss>>>>>(((AA[AAAAArrZc"eZdZdZfdZxZS) JWEHeaderzHeader object for JWE. Combines protected header, shared unprotected header and specific recipient's unprotected header together. ci}|r|||r|||r||t||r|ni|_|r|ni|_|r|ni|_dSr )r\r]r^r_r`header)r r_r`rqrarbs rr^zJWEHeader.__init__s  " JJy ! ! !  $ JJ{ # # #   JJv    &/7R*5=;;2 &.ffB r)rrrrr^rlrms@rrorosB / / / / / / / / /rro) r<abcrrr"r+r7rQdictrZrorrrrtsj """""""""""""""#w"""""""""+;w"""""("("("("("("("("V " " " " " " " "AAAAAdAAA4//////////r