o Œõ±j¦ã@s<ddlmZddlmZddlmZmZGdd„deƒZdS)é)ÚAES)ÚCounter)Ú symmetricBASEÚ cipherMODEc@s:eZdZejddfdd„Zdd„Zd dd„Zd d d „ZdS) rNé€cCstj|||||ddS)N)Ú segment_size)rÚ__init__)ÚselfÚkeyÚmodeÚIVr©r ú—/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/unicrypto/backends/pycryptodomex/AES.pyrsz AES.__init__c Cs<|jtjkrt |jtj¡|_dS|jtjkr%tj|jtj |j d|_dS|jtj krDtj|jtj t jdtj|j ddddd|_dS|jtjkrZtj|jtj|j |jd|_dS|jtjkrntj|jtj|j d|_dS|jtjkr„tj|jtj|j |jd |_dS|jtjkrštj|jtj|j |jd |_dStd ƒ‚) N)ÚivrÚbigF)Ú byteorderÚsigned)Ú initial_value)Úcounter)rr)Úmac_lenzUnknown cipher mode!)r rÚECBÚ_pyCryptodomeAESÚnewr ÚMODE_ECBÚ_cipherÚCBCÚMODE_CBCr ÚCTRÚMODE_CTRrÚintÚ from_bytesÚCFBÚMODE_CFBrÚOFBÚMODE_OFBÚCCMÚMODE_CCMÚGCMÚMODE_GCMÚ Exception)r r r rÚ setup_cipher s   2       zAES.setup_ciphercCs<|jtjks |jtjkr|j |¡|j |¡S|j |¡S©N)r rr%r'rÚupdateÚencrypt_and_digestÚencrypt)r ÚdataÚaadr r rr.s   z AES.encryptcCs>|jtjks |jtjkr|j |¡|j ||¡S|j |¡Sr+)r rr%r'rr,Údecrypt_and_verifyÚdecrypt)r r/r0Úmacr r rr2#s  z AES.decryptr+)NN) Ú__name__Ú __module__Ú __qualname__rrrr*r.r2r r r rrs  rN)ÚCryptodome.CipherrrÚCryptodome.UtilrÚunicrypto.symmetricrrr r r rÚs