o Df@sddlZddlmZGdddedZGdddeedZGdd d eedZGd d d ZGd d d ZGdddeZ GdddeZ dS)N)ABCMetac@s4eZdZdZdZdZdZdZdZddZ ddZ dS) JWEAlgorithmBasez+Base interface for all JWE algorithms. NJWEalgcCtNNotImplementedError)selfraw_datar T/home/ubuntu/webapp/venv/lib/python3.10/site-packages/authlib/jose/rfc7516/models.py prepare_keyzJWEAlgorithmBase.prepare_keycCrrr)r enc_algkeyr r r generate_presetrz JWEAlgorithmBase.generate_preset) __name__ __module__ __qualname____doc__ EXTRA_HEADERSname descriptionalgorithm_typealgorithm_locationrrr r r r rs r) metaclassc@s"eZdZdZdddZddZdS) JWEAlgorithmzInterface for JWE algorithm conforming to RFC7518. JWA specification (RFC7518) SHOULD implement the algorithms for JWE with this base implementation. NcCrrr)r rheadersrpresetr r r wraprzJWEAlgorithm.wrapcCrrr)r rekrrr r r unwraprzJWEAlgorithm.unwrapr)rrrrr r"r r r r rs  rc@s6eZdZdZd ddZddZd ddZd d d ZdS) $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. NcCrrr)r rr sender_keyrr r r !generate_keys_and_prepare_headers%rzFJWEAlgorithmWithTagAwareKeyAgreement.generate_keys_and_prepare_headerscCrrr)r rrrr$epkcektagr r r agree_upon_key_and_wrap_cek(rz@JWEAlgorithmWithTagAwareKeyAgreement.agree_upon_key_and_wrap_cekcCrrr)r rrrr$rr r r r +rz)JWEAlgorithmWithTagAwareKeyAgreement.wrapcCrrr)r rr!rrr$r(r r r r".rz+JWEAlgorithmWithTagAwareKeyAgreement.unwrapr)rrrrr%r)r r"r r r r r#!s   r#c@sLeZdZdZdZdZdZdZdZddZ ddZ dd Z d d Z d d Z dS)JWEEncAlgorithmNrenccCt|jdSN)osurandomCEK_SIZEr r r r generate_cek;zJWEEncAlgorithm.generate_cekcCr,r-)r/r0IV_SIZEr2r r r generate_iv>r4zJWEEncAlgorithm.generate_ivcCst|d|jkr tddS)Nr.zInvalid "iv" size)lenr5 ValueError)r ivr r r check_ivAszJWEEncAlgorithm.check_ivcCr)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 msgaadr9rr r r encryptEs zJWEEncAlgorithm.encryptcCr)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 ciphertextr<r9r(rr r r decryptPs zJWEEncAlgorithm.decrypt)rrrrrrrr5r1r3r6r:r=r?r r r r r*2s r*c@s,eZdZdZdZdZdZddZddZdS)JWEZipAlgorithmNrzipcCrrrr sr r r compresscrzJWEZipAlgorithm.compresscCrrrrBr r r decompressfrzJWEZipAlgorithm.decompress) rrrrrrrrDrEr r r r r@]s r@cs4eZdZdZfddZddZeddZZS)JWESharedHeaderziShared header object for JWE. Combines protected header and shared unprotected header together. csRi}|r |||r||t||r|ni|_|r$||_dSi|_dSr)updatesuper__init__ protected unprotected)r rJrKobj __class__r r rIos   zJWESharedHeader.__init__cCs|||j|dSr)rGrJ)r additionr r r update_protectedys z JWESharedHeader.update_protectedcCs$t||r|S||d|dS)NrJrK) isinstanceget)clsrLr r r from_dict}s zJWESharedHeader.from_dict) rrrrrIrP classmethodrT __classcell__r r rMr rFjs   rFcs eZdZdZfddZZS) JWEHeaderzHeader object for JWE. Combines protected header, shared unprotected header and specific recipient's unprotected header together. csni}|r |||r|||r||t||r!|ni|_|r(|ni|_|r2||_dSi|_dSr)rGrHrIrJrKheader)r rJrKrXrLrMr r rIs    zJWEHeader.__init__)rrrrrIrVr r rMr rWsrW) r/abcrrrr#r*r@dictrFrWr r r r s  +