U g @sZddlmZddlmZddlmZddZdZdZdZ d Z Gd d d e Z dd dZ d S)) long_to_bytes)bchr) TurboSHAKE128cCs$|dkr dSt|}|tt|S)Nr)rrlen)xSr B/tmp/pip-unpacked-wheel-_q8s9isk/Cryptodome/Hash/KangarooTwelve.py_length_encode$sr c@s2eZdZdZddZddZddZd d d ZdS) K12_XOFzeA KangarooTwelve hash object. Do not instantiate directly. Use the :func:`new` function. cCsb|dkr d}|tt||_t|_d|_tjdd|_d|_ d|_ d|_ d|_ |r^| |dS)Nrdomainr)r r_custom SHORT_MSG_state_paddingrnew_hash1_length1_hash2_length2_ctrupdateselfdatacustomr r r __init__9szK12_XOF.__init__c Cs|jtkrtd|jtkrX|jt|}|t|jdkrR||_|j||St |_|jt krt |}|jdkszt t t|d|j}|j|d||j|7_|jdkr|S|jdkst d}|j||jd7_t jdd|_d|_d |_t|_|||dS|jtks,t d}t|}t |}||krt |d|j|}|j||||j||7_|}|jdkr@|jd } |j| |jd 7_|jd|_|jd 7_q@|S) a Hash the next piece of data. .. note:: For better performance, submit chunks with a length multiple of 8192 bytes. Args: data (byte string/byte array/memoryview): The next chunk of the message to hash. z/You cannot call 'update' after the first 'read'i Ns rrr )r SQUEEZING TypeErrorrrrrrr LONG_MSG_S0 memoryviewAssertionErrorminrrrrr LONG_MSG_SXread_reset) r r!Z next_lengthZdata_memZdtcdividerindexZlen_dataZ new_indexcv_ir r r rQsV           zK12_XOF.updatecCsd}|jtkr(|j|jd|_t|_|jtkrP||jd}|jtksPt |jtkr|sj||j|j dkr|j d}|j||j d7_ |j d|_ |jd7_t|jdd}|j|d|_t|_|j|j_|j |S) ad Produce more bytes of the digest. .. 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 FTrr&rs)rrrrrrr'r)r-r+rrr.rr/rr Z_domain)r lengthZcustom_was_consumedr2trailerr r r r.s2           z K12_XOF.readNrcCst|||S)N)typerr r r rsz K12_XOF.new)Nr)__name__ __module__ __qualname____doc__r#rr.rr r r r r3s I1rNcCs t||S)agReturn a fresh instance of a KangarooTwelve 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 byte string. :Return: A :class:`K12_XOF` object )r)r!r"r r r rsr)NN)ZCryptodome.Util.numberrZCryptodome.Util.py3compatrrr rr)r-r'objectrrr r r r s