o j@s&ddlmZmZmZddlmZmZmZdZdZ ddZ e fddZ e fd d Z e e_ e e_e e_d d Ze fd dZe fddZee_ ee_ee_ddZe fddZe fddZee_ ee_ee_GdddeZGdddeZGdddeZdZefddZee fd d!Zee fd"d#Zd$S)%)AESBlockModeOfOperationAESSegmentModeOfOperationAESStreamModeOfOperation)append_PKCS7_paddingstrip_PKCS7_padding to_bufferablenonedefaultcCs|dkrdSdS)Nselfsizer r /root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/unicrypto/backends/pure/external/AES/blockfeeder.py_block_can_consume8s rcCsr|tkr t|}n|tkrt|dkrtdntdt|dkr4||dd||ddS||S)Nr #invalid data length for final blockinvalid padding option )PADDING_DEFAULTr PADDING_NONElen Exceptionencryptrdatapaddingr r r_block_final_encrypt=s   $ rcCsD|tkr t||S|tkrt|dkrtd||Std)Nr rr)rrdecryptrrrrr r r_block_final_decryptMs  rcCs|jt||jSN) segment_bytesintr r r r_segment_can_consume`sr#cCN|tkrtdtd|jt||j}|t|}||dt|SNrr )rrchrr!rrrrrr faux_paddingpaddedr r r_segment_final_encryptd  r*cCr$r%)rrr&r!rrrr'r r r_segment_final_decryptmr+r,cCs|Sr r r r r r_stream_can_consume}sr-cC|ttfvr td||SNr)rrrrrr r r_stream_final_encrypt  r0cCr.r/)rrrrrr r r_stream_final_decryptr1r2c@s&eZdZdZefddZdddZdS) BlockFeederzThe super-class for objects to handle chunking a stream of bytes into the appropriate block size for the underlying mode of operation and applying (or stripping) padding, as necessary.cCs&||_||_||_td|_||_dS)N)_mode_feed_finalr_buffer_padding)rmodefeedfinalrr r r__init__s   zBlockFeeder.__init__NcCs|jdur td|dur||j|j}d|_|S|jt|7_td}t|jdkrZ|jt|jd}|dkr?|S|||jd|7}|j|d|_t|jdks.|S)aProvide bytes to encrypt (or decrypt), returning any bytes possible from this or any previous calls to feed. Call with None or an empty string to flush the mode of operation and return any final bytes; no further calls to feed may be made.Nzalready finished feederr4r r ) r8 ValueErrorr7r9rrr5 _can_consumer6)rrresult can_consumer r rr;s zBlockFeeder.feedr )__name__ __module__ __qualname____doc__rr=r;r r r rr3s r3c@eZdZdZefddZdS) Encrypterz ||}|s n ||}||q|}||dS)zGUses feeder to read and convert from in_stream and write to out_stream.TN)readr;write)feeder in_stream out_stream block_sizechunk convertedr r r _feed_streams   rXcCt||d}t||||dS)zCEncrypts a stream of bytes from in_stream to out_stream using mode.rN)rGrX)r:rSrTrUr encrypterr r rencrypt_stream r\cCrY)zCDecrypts a stream of bytes from in_stream to out_stream using mode.rZN)rNrX)r:rSrTrUr decrypterr r rdecrypt_streamr]r_N)AESrrrutilrrrrrrrrr?rIrOr#r*r,r-r0r2objectr3rGrN BLOCK_SIZErXr\r_r r r rs:       *