a ù!f½ã@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|dksJ‚|dkr$dn| ¡dd}t|ƒt|ƒS)z2Left encode function as defined in NIST SP 800-185ééøréé)Ú bit_lengthrr ©ÚxÚnum©rúZ/var/www/html/python-backend/venv/lib/python3.9/site-packages/Cryptodome/Hash/cSHAKE128.pyÚ _left_encode+srcCsD|dd>kr|dksJ‚|dkr$dn| ¡dd}t|ƒt|ƒS)z3Right encode function as defined in NIST SP 800-185r r rrr)rr rrrrrÚ _right_encode6srcCs,t|ƒd}|dd>kr tdƒ‚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|r8t|ƒt|ƒ}t|d|dƒ}d|_n d}d|_t | ¡t|dƒtdƒ¡}|rpt d|ƒ‚t |  ¡tj ƒ|_ d|_|r–| |¡|r¤| |¡dS)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__]s.  þÿÿ zcSHAKE_XOF.__init__cCsH|jrtdƒ‚t |j ¡t|ƒtt|ƒƒ¡}|rDt 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ƒ¡}|rDt 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"rr1r)r'rZbfrr.rrrÚreadŠs ýÿzcSHAKE_XOF.readN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r/r&r2rrrrrWsrcCst||d|ƒS)Né©r)r(r)r+rrrÚ_new¦sr9NcCst||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 r7ór8)r(r)rrrÚnew«sr;)NN)ZCryptodome.Util.py3compatrZCryptodome.Util._raw_apirrrrrrr ZCryptodome.Util.numberr ZCryptodome.Hash.keccakr rrrrÚobjectrr9r;rrrrÚs $      O