o Ìn~b½ã@sŽddlmZddlmZmZmZmZmZmZm Z ddl m Z ddl m Z dd„Zdd„Zd d „Zd d „ZGd d„deƒZdd„Zddd„ZdS)é)Úbchr)Ú VoidPointerÚ SmartPointerÚcreate_string_bufferÚget_raw_bufferÚc_size_tÚ c_uint8_ptrÚc_ubyte)Ú long_to_bytes)Ú_raw_keccak_libcCsD|dd>kr |dks J‚|dkrdn| ¡dd}t|ƒt|ƒS)z2Left encode function as defined in NIST SP 800-185ééøréé)Ú bit_lengthrr ©ÚxÚnum©rúD/usr/local/lib/python3.10/dist-packages/Cryptodome/Hash/cSHAKE128.pyÚ _left_encode+órcCsD|dd>kr |dks J‚|dkrdn| ¡dd}t|ƒt|ƒS)z3Right encode function as defined in NIST SP 800-185r r rrr)rr rrrrrÚ _right_encode6rrcCs,t|ƒd}|dd>krtdƒ‚t|ƒ|S)z4Encode string function as defined in NIST SP 800-185rr r z$String too large to encode in cSHAKE)ÚlenÚ ValueErrorr)rZbitlenrrrÚ _encode_strAs   rcCs,t|ƒ|}|t|ƒ||}|d|S)z2Zero pad byte string as defined in NIST SP 800-185ó)rr)rÚlengthZto_padZnpadrrrÚ_bytepadKs  rc@s(eZdZdZdd„Zdd„Zdd„ZdS) Ú cSHAKE_XOFz]A cSHAKE hash object. Do not instantiate directly. Use the :func:`new` function. c Cs¬tƒ}|s|rt|ƒt|ƒ}t|d|dƒ}d|_nd}d|_t | ¡t|dƒtdƒ¡}|r8t d|ƒ‚t |  ¡tj ƒ|_ d|_|rK| |¡|rT| |¡dSdS)Ni@réééz#Error %d while instantiating cSHAKEF)rrrÚ_paddingr Z keccak_initZ address_ofrr rrÚgetZkeccak_destroyÚ_stateÚ _is_squeezingÚupdate) ÚselfÚdataÚcustomÚcapacityÚfunctionÚstateZ prefix_unpadÚprefixÚresultrrrÚ__init__]s0  þÿÿ ÿzcSHAKE_XOF.__init__cCsH|jrtdƒ‚t |j ¡t|ƒtt|ƒƒ¡}|r"t d||j fƒ‚|S)z¼Continue hashing of a message by consuming the next chunk of data. Args: data (byte string/byte array/memoryview): The next chunk of the message being hashed. z/You cannot call 'update' after the first 'read'z Error %d while updating %s state) r&Ú TypeErrorr Z keccak_absorbr%r$rrrrÚname)r(r)r/rrrr'xs  þÿzcSHAKE_XOF.updatecCsLd|_t|ƒ}t |j ¡|t|ƒt|jƒ¡}|r"t d||j fƒ‚t |ƒS)ah Compute the next piece of XOF output. .. 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 Tz!Error %d while extracting from %s) r&rr Zkeccak_squeezer%r$rr r#rr2r)r(rZbfrr/rrrÚreadŠs ýÿzcSHAKE_XOF.readN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r0r'r3rrrrrWs  rcCst||d|ƒS)Né©r)r)r*r,rrrÚ_new¦sr:NcCst||ddƒS)azReturn a fresh instance of a cSHAKE128 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 bytestring (``S`` in SP 800-185). :Return: A :class:`cSHAKE_XOF` object r8ór9)r)r*rrrÚnew«sr<)NN)ZCryptodome.Util.py3compatrZCryptodome.Util._raw_apirrrrrrr ZCryptodome.Util.numberr ZCryptodome.Hash.keccakr rrrrÚobjectrr:r<rrrrÚs $      O