�
1�Dg� � � � d dl mZ d dlmZmZmZmZmZmZm Z d dl
m Z d dl m
Z
d� Zd� Zd� Zd� Z G d � d
e� � Zd � Zdd
�Zd S )� )�bchr)� VoidPointer� SmartPointer�create_string_buffer�get_raw_buffer�c_size_t� c_uint8_ptr�c_ubyte)�
long_to_bytes)�_raw_keccak_libc � � | ddz k r| dk sJ �| dk rdn| � � � dz dz }t |� � t | � � z S )z2Left encode function as defined in NIST SP 800-185� � r � � )�
bit_lengthr r ��x�nums �w/builddir/build/BUILD/imunify360-venv-2.4.0/opt/imunify360/venv/lib64/python3.11/site-packages/Crypto/Hash/cSHAKE128.py� _left_encoder + s\ � �
��d��O�O��Q���� &� �A�v�v�!�!�A�L�L�N�N�Q� .�1�4�C� ��9�9�}�Q�'�'� '�'� c � � | ddz k r| dk sJ �| dk rdn| � � � dz dz }t | � � t |� � z S )z3Right encode function as defined in NIST SP 800-185r r r r r )r r r r s r �
_right_encoder 6 s\ � �
��d��O�O��Q���� &� �A�v�v�!�!�A�L�L�N�N�Q� .�1�4�C� �� � �d�3�i�i� '�'r c �z � t | � � dz }|ddz k rt d� � �t |� � | z S )z4Encode string function as defined in NIST SP 800-185r r r z$String too large to encode in cSHAKE)�len�
ValueErrorr )r �bitlens r � _encode_strr A sD � � ��V�V�a�Z�F�
�!�t�)� � ��?�@�@�@� �� � �!� #�#r c �f � t |� � | z }|t |� � |z z
|z }|d|z z S )z2Zero pad byte string as defined in NIST SP 800-185� )r r )r �length�to_pad�npads r �_bytepadr% K sA � � �&�
!�
!�A�
%�F�
�S��[�[�6�)� )�V� 3�D� �G�d�N� "�"r c �$ � e Zd ZdZd� Zd� Zd� ZdS )�
cSHAKE_XOFz]A cSHAKE hash object.
Do not instantiate directly.
Use the :func:`new` function.
c �8 � t � � }|s|r=t |� � t |� � z }t |d|z
dz � � }d| _ n d }d| _ t j |� � � t |dz � � t d� � � � }|rt d|z � � �t |� � � t j � � | _
d| _ |r| � |� � |r| � |� � d S d S )Ni@ r � � � z#Error %d while instantiating cSHAKEF)r r r% �_paddingr � keccak_init�
address_ofr r
r r �get�keccak_destroy�_state�
_is_squeezing�update) �self�data�custom�capacity�function�state� prefix_unpad�prefix�results r �__init__zcSHAKE_XOF.__init__] s/ � ��
�
�� � !�X� !�&�x�0�0�;�v�3F�3F�F�L� �l�T�H�_�q�,@�A�A�F� �D�M�M��F� �D�M� �,�U�-=�-=�-?�-?�-5�h��k�-B�-B�-4�R�[�[�:� :�� � '� � B� %� &� '� '�
'�"�5�9�9�;�;�#2�#A�C� C�� � "��� � � �K�K�� � � � � � �K�K�� � � � � � � r c � � | j rt d� � �t j | j � � � t
|� � t t |� � � � � � }|rt d|| j
fz � � �| 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) r2 � TypeErrorr �
keccak_absorbr1 r/ r r r r �name)r4 r5 r< s r r3 zcSHAKE_XOF.updatex s� � � � � O�� M�N�N� N� �.�t�{���/@�/@�/:�4�/@�/@�/7��D� � �/B�/B�D� D�� � 4� � ?� &�� � 2� 3� 4� 4�
4�� r c � � d| _ t |� � }t j | j � � � |t
|� � t | j � � � � }|rt d|| j
fz � � �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) r2 r r �keccak_squeezer1 r/ r r
r, r rA r )r4 r" �bfrr< s r �readzcSHAKE_XOF.read� s� � � "���"�6�*�*�� �/�� ���0A�0A�03�08��0@�0@�07��
�0F�0F�H� H�� � 4� � @� &�� � 2� 3� 4� 4�
4� �c�"�"�"r N)�__name__�
__module__� __qualname__�__doc__r= r3 rE � r r r' r' W sK � � � � � �� �
� � �6� � �$#� #� #� #� #r r' c �& � t | |d|� � S )N� �r' )r5 r6 r8 s r �_newrN � s � � �d�F�C�� 2� 2�2r Nc �& � t | |dd� � S )az Return 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
rL r rM )r5 r6 s r �newrP � s � � �d�F�C�� -� -�-r )NN)�Crypto.Util.py3compatr �Crypto.Util._raw_apir r r r r r r
�Crypto.Util.numberr �Crypto.Hash.keccakr r r r r% �objectr' rN rP rJ r r �<module>rV s5 ��> '� &� &� &� &� &�8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8� 8�
-� ,� ,� ,� ,� ,� .� .� .� .� .� .�(� (� (�(� (� (�$� $� $� #� #� #�L#� L#� L#� L#� L#�� L#� L#� L#�^3� 3� 3�
.� .� .� .� .� .r