File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/hashutils.cpython-310.pyc
o
�N�g� � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddl Zddl
m Z e d�dd� �Z e d�dd � �Z
d
d � Zd d
� Ze d�dd� �Ze d�dd� �Ze d�dd� �Ze d�dd� �Ze d�dd � �Ze d �d d � �Ze d �e d �d)d"d ���Ze d#�d*d%d&��ZG d'd(� d(�ZdS )+z-
A collection of hashing and encoding utils.
� N)� jinja_filterZ
base64_encodec C �6 t jjjt�t jj�| ��t jj�� rdd�S dd�S )z�
Encode a string as base64 using the "modern" Python interface.
Among other possible differences, the "modern" encoder does not include
newline ('\n') characters in the encoded output.
�utf8N��encoding) �salt�utils� stringutils�
to_unicode�base64� b64encode�to_bytes�platform�
is_windows��instr� r �H/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/hashutils.py�base64_b64encode � ��r Z
base64_decodec C sV t �tjj�| ��}ztjjj|tjj�� rdd�W S dd�W S t y* | Y S w )zM
Decode a base64-encoded string using the "modern" Python interface.
r Nr )
r � b64decoder r r r
r
r r �UnicodeDecodeError)r �decodedr r r �base64_b64decode s �� �r c C r )a
Encode a byte-like object as base64 using the "modern" Python interface.
Among other possible differences, the "modern" encoder includes
a newline ('\n') character after every 76 characters and always
at the end of the encoded string.
r Nr ) r r r r
r � encodebytesr
r r r r r r �base64_encodestring- r r c C sZ t jj�| �}t�|�}zt jjj|t jj�� r dd�W S dd�W S t y, | Y S w )zW
Decode a base64-encoded byte-like object using the "modern" Python interface.
r Nr )
r r r r
r � decodebytesr
r r r )r Zbvaluer r r r �base64_decodestring; s
�� �r �md5c C �" t jj�t�t jj�| ���� �S )z1
Generate an md5 hash of a given string.
)r r r r
�hashlibr r
� hexdigestr r r r �
md5_digestI � �r" �sha1c C s t �tjj�| ���� S )z2
Generate an sha1 hash of a given string.
)r r$ r r r r
r! r r r r � sha1_digestS s r% �sha256c C r )z3
Generate a sha256 hash of a given string.
)r r r r
r r& r
r! r r r r �
sha256_digest[ r# r' �sha512c C r )z2
Generate a sha512 hash of a given string
)r r r r
r r( r
r! r r r r �
sha512_digeste r# r) �hmacc C sP t jj�| �}t jj�|�}t jj�|�}t�||tj�}t� |�
� �}||kS )z�
Verify a challenging hmac signature against a string / shared-secret
Returns a boolean if the verification succeeded or failed.
) r r r r
r* �newr r& r r �digest)�string�
shared_secretZchallenge_hmac�msg�keyZ challenge� hmac_hashZ
valid_hmacr r r �hmac_signatureo s r2 � hmac_computec C s4 t jj�| �}t jj�|�}t�||tj��� }|S )z
Create an hmac digest.
) r r r r
r* r+ r r& r! )r- r. r/ r0 r1 r r r r3 } s Zrand_str� random_hash� �c( c C s8 |sd}t t|�}|tjj�tt�� � d| �����
� S )zG
Return a hash of a randomized data from random.SystemRandom()
r r ) �getattrr r r r r
�str�random� SystemRandom�randintr! )�size� hash_typeZhasherr r r r4 � s
��Z file_hashsum� c s� t t|�r
tt|�p d}|du rtd|� ���tjj�| d��!�|� }t� �fdd�d�D ]}|� |� q-|�
� W d � S 1 sCw Y dS )ay
Get the hash sum of a file
This is better than ``get_sum`` for the following reasons:
- It does not read the entire file into memory.
- It does not return a string on error. The returned value of
``get_sum`` cannot really be trusted since it is vulnerable to
collisions: ``get_sum(..., 'xyz') == 'Hash xyz not supported'``
N�Invalid hash type: �rbc s
�� � �S �N)�readr ��
chunk_size�ifiler r �<lambda>� s
zget_hash.<locals>.<lambda>� ) �hasattrr r6 �
ValueErrorr r �files�fopen�iter�updater! )�path�formrC r< Zhash_obj�chunkr rB r �get_hash� s $�rP c @ s* e Zd ZdZd dd�Zdd� Zdd � Zd
S ) �DigestCollectorz3
Class to collect digest of the file tree.
r& r= c C s> t t|�r tt|�� p d| _| jdu rtd|� ���|| _dS )z@
Constructor of the class.
:param form:
Nr> )rG r r6 �_DigestCollector__digestrH �_DigestCollector__buff)�selfrN Zbuffr r r �__init__� s
zDigestCollector.__init__c s\ t jj�|d�� � t� �fdd�d�D ]}�j�|� qW d � dS 1 s'w Y dS )ze
Update digest with the file content by path.
:param path:
:return:
r? c s � � �j�S r@ )rA rS r �rD rT r r rE � s z%DigestCollector.add.<locals>.<lambda>rF N)r r rI rJ rK rR rL )rT rM rO r rV r �add� s
�"�zDigestCollector.addc C s t jj�| j�� tj �S )z/
Get digest.
:return:
)r r r Zto_strrR r! �os�linesep)rT r r r r, � s zDigestCollector.digestN�r& r= )�__name__�
__module__� __qualname__�__doc__rU rW r, r r r r rQ � s
rQ )r5 NrZ )r^ r r r* rX r8 Zsalt.utils.filesr Zsalt.utils.platformZsalt.utils.stringutilsZsalt.utils.decorators.jinjar r r r r r" r% r' r) r2 r3 r4 rP rQ r r r r �<module> sD