o Ìn~bQ#ã@s~ddlmZmZmZmZmZmZmZddlm Z ddl m Z ddl m Z dd„ZdZdZd Zd ZGd d „d eƒZddd„Zd S)é)Ú VoidPointerÚ SmartPointerÚcreate_string_bufferÚget_raw_bufferÚc_size_tÚ c_uint8_ptrÚc_ubyte)Ú long_to_bytes)Úbchré)Ú_raw_keccak_libcCs$|dkrdSt|ƒ}|tt|ƒƒS)Nró)r r Úlen)ÚxÚS©rúI/usr/local/lib/python3.10/dist-packages/Cryptodome/Hash/KangarooTwelve.pyÚ_length_encode*sréééc@sReZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z ddd„Z dS)ÚK12_XOFzeA KangarooTwelve hash object. Do not instantiate directly. Use the :func:`new` function. cCsb|dkrd}|tt|ƒƒ|_t|_d|_| ¡|_d|_d|_ d|_ d|_ |r/|  |¡dSdS)Nór) rrÚ_customÚ SHORT_MSGÚ_stateÚ_paddingÚ_create_keccakÚ_hash1Ú_length1Ú_hash2Ú_length2Ú_ctrÚupdate©ÚselfÚdataÚcustomrrrÚ__init__?s ÿzK12_XOF.__init__cCs@tƒ}t | ¡tdƒtdƒ¡}|rtd|ƒ‚t| ¡tj ƒS)Né é z+Error %d while instantiating KangarooTwelve) rr Z keccak_initZ address_ofrrÚ ValueErrorrÚgetZkeccak_destroy)r%ÚstateÚresultrrrrVs þÿzK12_XOF._create_keccakcCs2t | ¡t|ƒtt|ƒƒ¡}|rtd|ƒ‚dS)Nz,Error %d while updating KangarooTwelve state)r Z keccak_absorbr,rrrr+)r%r&Úhash_objr.rrrÚ_update`s  þÿÿzK12_XOF._updatecCs<t|ƒ}t | ¡|t|ƒt|ƒ¡}|rtd|ƒ‚t|ƒS)Nz-Error %d while extracting from KangarooTwelve)rr Zkeccak_squeezer,rrr+r)r%r/ÚlengthÚpaddingZbfrr.rrrÚ_squeezehs ýÿzK12_XOF._squeezecCs"t | ¡¡}|rtd|ƒ‚dS)Nz-Error %d while resetting KangarooTwelve state)r Z keccak_resetr,r+)r%r/r.rrrÚ_resetts ÿÿzK12_XOF._resetc Csè|jtkr tdƒ‚|jtkr-|jt|ƒ}|t|jƒdkr*||_| ||j¡|St |_|jt krŽt |ƒ}|jdks=J‚t t|ƒd|jƒ}| |d|…|j¡|j|7_|jdkr`|S|jdksgJ‚d}| ||j¡|jd7_|  ¡|_ d|_d|_t|_| ||d…¡S|jtks•J‚d}t|ƒ}t |ƒ}||kròt |d|j|ƒ}| |||…|j ¡|j||7_|}|jdkrî| |j dd ¡} | | |j¡|jd7_| |j ¡d|_|jd7_||ks£|S) aHash the next piece of data. .. note:: For better performance, submit chunks with a length multiple of 8192 bytes. Args: data (byte string/byte array/memoryview): The next chunk of the message to hash. z/You cannot call 'update' after the first 'read'i Nsérr r)é )rÚ SQUEEZINGÚ TypeErrorrrrrr0rÚ LONG_MSG_S0Ú memoryviewÚminrr r!r"Ú LONG_MSG_SXr#r3r4) r%r&Z next_lengthZdata_memZdtcÚdividerÚindexZlen_dataZ new_indexÚcv_irrrr#zsX       ózK12_XOF.updatecCsd}|jtkr| |j|j¡d|_t|_|jtkr)| |j¡d}|jt ks)J‚|jt krw|s6| |j¡|j dkra|  |j dd¡}| ||j¡|j d7_ | |j ¡d|_ |jd7_t|jdƒd}| ||j¡d |_t|_|  |j||j¡S) ad Produce more bytes of the digest. .. note:: You cannot use :meth:`update` anymore after the first call to :meth:`read`. Args: length (integer): the amount of bytes this method must return :return: the next piece of XOF output (of the given length) :rtype: byte string FéTrr)r6r sÿÿé)rrr0rrrr7r9r#r<r!r3r rr4r"r)r%r1Zcustom_was_consumedr?ÚtrailerrrrÚreadÃs0       z K12_XOF.readNrcCst|ƒ||ƒS)N)Útyper$rrrÚnewósz K12_XOF.new)Nr) Ú__name__Ú __module__Ú __qualname__Ú__doc__r(rr0r3r4r#rCrErrrrr9s  I0rNcCs t||ƒS)agReturn a fresh instance of a KangarooTwelve object. Args: data (bytes/bytearray/memoryview): Optional. The very first chunk of the message to hash. It is equivalent to an early call to :meth:`update`. custom (bytes): Optional. A customization byte string. :Return: A :class:`K12_XOF` object )r)r&r'rrrrE÷s rE)NN)ZCryptodome.Util._raw_apirrrrrrrZCryptodome.Util.numberr ZCryptodome.Util.py3compatr Zkeccakr rrr9r<r7ÚobjectrrErrrrÚs$    ?