~hddlZddlmZGddeZGddeeZGdd eeZGd d ZGd d ZGddeZ GddeZ y)N)ABCMetac0eZdZdZdZdZdZdZdZdZ dZ y)JWEAlgorithmBasez&Base interface for all JWE algorithms.NJWEalgctNNotImplementedError)selfraw_datas W/opt/mcp/mcp-sentiment/venv/lib/python3.12/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_locationrrrrrrs)0M DKN""rr) metaclassceZdZdZddZdZy) 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"s ""rr"c.eZdZdZddZdZddZddZy)$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,s "!rctr r )r rr$rr-r%s rr&z)JWEAlgorithmWithTagAwareKeyAgreement.wrap1rrctr r )r rr(r$rr-r2s rr)z+JWEAlgorithmWithTagAwareKeyAgreement.unwrap4rrr )rrrrr.r3r&r)rrrr+r+#s "" ""rr+cBeZdZdZdZdZdZdZdZdZ dZ dZ dZ dZ y) JWEEncAlgorithmNrenccFtj|jdzSN)osurandomCEK_SIZEr s r generate_cekzJWEEncAlgorithm.generate_cekAszz$--1,--rcFtj|jdzSr:)r<r=IV_SIZEr?s r generate_ivzJWEEncAlgorithm.generate_ivDszz$,,!+,,rcPt|dz|jk7r tdy)Nr;zInvalid "iv" size)lenrB ValueError)r ivs rcheck_ivzJWEEncAlgorithm.check_ivGs' r7Q;$,, &01 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.encryptKs "!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.decryptVs "!r)rrrrrrrrBr>r@rCrHrLrOrrrr7r78s: DKNGH.-2 " "rr7c(eZdZdZdZdZdZdZdZy)JWEZipAlgorithmNrzipctr r r ss rcompresszJWEZipAlgorithm.compressirrctr r rTs r decompresszJWEZipAlgorithm.decompresslrr) rrrrrrrrVrXrrrrQrQcs! DKN""rrQc8eZdZdZfdZdZedZxZS)JWESharedHeaderziShared header object for JWE. Combines protected header and shared unprotected header together. ci}|r|j||r|j|t| ||r|ni|_|r||_yi|_yr )updatesuper__init__ protected unprotected)r r_r`obj __class__s rr^zJWESharedHeader.__init__vsO  JJy !  JJ{ # &/R*5;2rc\|j||jj|yr )r\r_)r additions rupdate_protectedz JWESharedHeader.update_protecteds  H h'rclt||r|S||jd|jdS)Nr_r`) isinstanceget)clsras r from_dictzJWESharedHeader.from_dicts0 c3 J377;')?@@r) rrrrr^re classmethodrj __classcell__rbs@rrZrZps( >(AArrZc"eZdZdZfdZxZS) JWEHeaderzHeader object for JWE. Combines protected header, shared unprotected header and specific recipient's unprotected header together. ci}|r|j||r|j||r|j|t| ||r|ni|_|r|ni|_|r||_yi|_yr )r\r]r^r_r`header)r r_r`rqrarbs rr^zJWEHeader.__init__sh  JJy !  JJ{ #  JJv  &/R*5;2 &f B r)rrrrr^rlrms@rroros / /rro) r<abcrrr"r+r7rQdictrZrorrrrtsd """ "#w ""+;w"*("("V " "AdA6//r