Ki4ddlZddlmZGddeZGddeeZGdd eeZGd d ZGd d ZGddeZ GddeZ y)N)ABCMetac4eZdZdZdZdZdZdZdZdZ dZ dZ y)JWEAlgorithmBasez&Base interface for all JWE algorithms.NFJWEalgctNNotImplementedError)selfraw_datas g/home/jay/workspace/scripts/.codegraph-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 description deprecatedalgorithm_typealgorithm_locationrrrrrrs.0M DKJN""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.unwrap rrr )rrrrr'r*r rrr#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.wrap2rrctr r )r rr)r%rr.r3s rr*z+JWEAlgorithmWithTagAwareKeyAgreement.unwrap5rrr )rrrrr/r4r'r*r rrr,r,$s "" ""rr,cBeZdZdZdZdZdZdZdZdZ dZ dZ dZ dZ y) JWEEncAlgorithmNrenccFtj|jdzSN)osurandomCEK_SIZEr s r generate_cekzJWEEncAlgorithm.generate_cekBszz$--1,--rcFtj|jdzSr;)r=r>IV_SIZEr@s r generate_ivzJWEEncAlgorithm.generate_ivEszz$,,!+,,rcPt|dz|jk7r tdy)Nr<zInvalid "iv" size)lenrC ValueError)r ivs rcheck_ivzJWEEncAlgorithm.check_ivHs' 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 msgaadrHrs rencryptzJWEEncAlgorithm.encryptLs "!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 ciphertextrLrHr3rs rdecryptzJWEEncAlgorithm.decryptWs "!r)rrrrrrrrCr?rArDrIrMrPr rrr8r89s: DKNGH.-2 " "rr8c(eZdZdZdZdZdZdZdZy)JWEZipAlgorithmNrzipctr r r ss rcompresszJWEZipAlgorithm.compressjrrctr r rUs r decompresszJWEZipAlgorithm.decompressmrr) rrrrrrrrWrYr rrrRrRds! DKN""rrRc8eZdZdZfdZdZedZxZS)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`raobj __class__s rr_zJWESharedHeader.__init__wsO  JJ{ #  JJy ! &/R*5;2rc\|j||jj|yr )r]r`)r additions rupdate_protectedz JWESharedHeader.update_protecteds  H h'rclt||r|S||jd|jdS)Nr`ra) isinstanceget)clsrbs r from_dictzJWESharedHeader.from_dicts0 c3 J377;')?@@r) rrrrr_rf classmethodrk __classcell__rcs@rr[r[qs( >(AArr[c"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`raheader)r r`rarrrbrcs rr_zJWEHeader.__init__sh  JJ{ #  JJv   JJy ! &/R*5;2 &f B r)rrrrr_rmrns@rrprps / /rrp) r=abcrrr#r,r8rRdictr[rpr rrrusd ""$ "#w ""+;w"*("("V " "AdA6//r