U Š¾úgëã@s’ddlmZmZddlmZmZmZmZmZm Z m Z ddl m Z ddl mZdd„Zdd„Zd d „Zd d „ZGd d„deƒZdd„Zddd„ZdS)é)ÚbchrÚconcat_buffers)Ú VoidPointerÚ SmartPointerÚcreate_string_bufferÚget_raw_bufferÚc_size_tÚ c_uint8_ptrÚc_ubyte)Ú long_to_bytes)Ú_raw_keccak_libcCsD|dd>kr|dkst‚|dkr$dn| ¡dd}t|ƒt|ƒS)z2Left encode function as defined in NIST SP 800-185ééøréé)ÚAssertionErrorÚ bit_lengthrr ©ÚxÚnum©rú=/tmp/pip-unpacked-wheel-_q8s9isk/Cryptodome/Hash/cSHAKE128.pyÚ _left_encode+srcCsD|dd>kr|dkst‚|dkr$dn| ¡dd}t|ƒt|ƒS)z3Right encode function as defined in NIST SP 800-185r rrrr)rrr rrrrrÚ _right_encode6srcCs.t|ƒd}|dd>kr tdƒ‚tt|ƒ|ƒS)z4Encode string function as defined in NIST SP 800-185rr rz$String too large to encode in cSHAKE)ÚlenÚ ValueErrorrr)rZbitlenrrrÚ _encode_strAs  rcCs.tt|ƒ|ƒ}|t|ƒ||}|d|S)z2Zero pad byte string as defined in NIST SP 800-185ó)rrr)rÚlengthZto_padZnpadrrrÚ_bytepadKsrc@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%r rrrÚname)r)r*r0rrrr(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$rr3r)r)rZbfrr0rrrÚreadŠs ýÿzcSHAKE_XOF.readN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r1r(r4rrrrr Wsr cCst||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 r9ór:)r*r+rrrÚnew«sr=)NN)ZCryptodome.Util.py3compatrrZCryptodome.Util._raw_apirrrrrr r ZCryptodome.Util.numberr ZCryptodome.Hash.keccakr rrrrÚobjectr r;r=rrrrÚs$      O