o j@sdddlmZmZmZmZmZmZmZddlm Z ddl m Z ddl m Z GdddeZdd Zd S) ) VoidPointer SmartPointercreate_string_bufferget_raw_bufferc_size_t c_uint8_ptrc_ubyte) long_to_bytes)bchr)_raw_keccak_libc@s:eZdZdZddZddZddZd d d Zd d ZdS) TurboSHAKEzaA TurboSHAKE hash object. Do not instantiate directly. Use the :func:`new` function. cCsjt}t|t|td}|rtd|t|tj |_ d|_ ||_ ||_ |r3||dSdS)N z'Error %d while instantiating TurboSHAKEF)rr keccak_init address_ofrr ValueErrorrgetkeccak_destroy_state _is_squeezing _capacity_domainupdate)selfcapacitydomain_separationdatastateresultr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/Cryptodome/Hash/TurboSHAKE128.py__init__s  zTurboSHAKE.__init__cCsB|jrtdt|jt|tt|}|rt d||S)zContinue 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 TurboSHAKE state) r TypeErrorr keccak_absorbrrrrlenr)rrrrrr r%s  zTurboSHAKE.updatecCsFd|_t|}t|j|t|t|j}|rt d|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 TurboSHAKE) rrr keccak_squeezerrrrrrr)rlengthbfrrrrr read7s zTurboSHAKE.readNcCst||j|j|SN)typerr)rrrrr newRszTurboSHAKE.newcCs*t|j}|rtd|d|_dS)Nz)Error %d while resetting TurboSHAKE stateF)r keccak_resetrrrr)rrrrr _resetUs  zTurboSHAKE._resetr)) __name__ __module__ __qualname____doc__r!rr(r+r-rrrr r s  r cKsF|dd}d|krdksntd||d}td||dS) aCreate a new TurboSHAKE128 object. Args: domain (integer): Optional - A domain separation byte, between 0x01 and 0x7F. The default value is 0x1F. data (bytes/bytearray/memoryview): Optional - The very first chunk of the message to hash. It is equivalent to an early call to :meth:`update`. :Return: A :class:`TurboSHAKE` object domainr z&Incorrect domain separation value (%d)r )r)rrr )kwargsrrrrr r+]s  r+N)Cryptodome.Util._raw_apirrrrrrrCryptodome.Util.numberr Cryptodome.Util.py3compatr keccakr objectr r+rrrr s $    Q