U g@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. cCsft}t|t|td}|r0td|t|tj |_ d|_ ||_ ||_ |rb||dS)N z'Error %d while instantiating TurboSHAKEF)rr Z keccak_initZ address_ofrr ValueErrorrgetZkeccak_destroy_state _is_squeezing _capacity_domainupdate)selfcapacitydomain_separationdatastateresultrA/tmp/pip-unpacked-wheel-_q8s9isk/Cryptodome/Hash/TurboSHAKE128.py__init__s zTurboSHAKE.__init__cCsB|jrtdt|jt|tt|}|r>t 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 Z keccak_absorbrrrrlenr)rrrrrrr%s  zTurboSHAKE.updatecCsFd|_t|}t|j|t|t|j}|r>t 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 Zkeccak_squeezerrrrrrr)rlengthZbfrrrrrread7s zTurboSHAKE.readNcCst||j|j|S)N)typerr)rrrrrnewRszTurboSHAKE.newcCs*t|j}|r td|d|_dS)Nz)Error %d while resetting TurboSHAKE stateF)r Z keccak_resetrrrr)rrrrr_resetUs zTurboSHAKE._reset)N) __name__ __module__ __qualname____doc__rrr"r$r%rrrrr s  r cKsF|dd}d|kr dks.ntd||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 )kwargsrrrrrr$]s  r$N)ZCryptodome.Util._raw_apirrrrrrrZCryptodome.Util.numberr ZCryptodome.Util.py3compatr Zkeccakr objectr r$rrrrs $   Q