HEX

File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/files.cpython-310.pyc
o

;jZ��@s�dZddlZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlZddl
ZddlZddlZddlZddlZddlmZmZmZddlmZzddlZdZWneyidZYnweed�rsejZnddlZGdd	�d	�Z d
d�Ze�!e"�Z#dZ$dZ%d
e%Z&dZ'ddddddd�Z(dd�e(�)�D�Z*dd�Z+dd�Z,dd�Z-dd�Z.d\d!d"�Z/d#d$�Z0d]d%d&�Z1ej2d^d)d*��Z3ed^d+d,��Z4d-d.�Z5ej2d/d0��Z6d1d2�Z7ej2d3d4��Z8ej2d5d6��Z9d_d7d8�Z:d9d:�Z;d;d<�Z<ed=�d>d=��Z=d`d?d@�Z>dAdB�Z?d]dCdD�Z@edE�dadGdH��ZAedI�dJdK��ZBdLdM�ZCedN�dOdN��ZDdPdQ�ZEdRdS�ZFdTdU�ZGdVdW�ZHd]dXdY�ZIdZd[�ZJdS)bz"
Functions for working with files
�N)�CommandExecutionError�
FileLockError�MinionError)�jinja_filterTF�asynccontextmanagerc@s(eZdZdZdd�Zdd�Zdd�ZdS)	�_AsyncGeneratorContextManagerzA
        Async context manager backport for Python < 3.7
        cCs||i|��|_dS�N)�_gen)�self�func�args�kwargs�r�D/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/files.py�__init__.sz&_AsyncGeneratorContextManager.__init__c�s.�z	|j��IdHWStytd�d�w)Nzgenerator didn't yield)r	�	__anext__�StopAsyncIteration�RuntimeError)r
rrr�
__aenter__1s�
�z(_AsyncGeneratorContextManager.__aenter__c
�s��|durz
|j��IdHWtd��tyYdSw|dur$|�}z|j�|||�IdHWdSty<YdStyd}z||urNWYd}~dSt|t�r_|j|ur_WYd}~dS�d}~wty{}z||urvWYd}~dS�d}~ww)Nzgenerator didn't stopTF)r	rrr�athrow�
isinstance�	__cause__�	Exception)r
�exc_type�	exc_value�	traceback�excrrr�	__aexit__7s:����
���z'_AsyncGeneratorContextManager.__aexit__N)�__name__�
__module__�__qualname__�__doc__rrrrrrrr)s
rcst����fdd��}|S)z8
        asynccontextmanager decorator backport
        cst�||�Sr)r)rr
�rrr�helperYsz#asynccontextmanager.<locals>.helper)�	functools�wraps)rr#rr"rrTs)��file)�http�https�ftpZswiftZs3)�saltr'�__salt.tmp.��`�@�8�(� )�sha512�sha384�sha256�sha224�sha1�md5cCsi|]\}}||�qSrr)�.0�x�yrrr�
<dictcomp>osr<c
CsHzt|�WdSty#}ztjd|tjd�WYd}~dSd}~ww)z 
    Remove temporary files
    z+Exception while removing temp directory: %s)Zexc_info_on_loglevelN)�rm_rfr�log�error�logging�DEBUG)�tmprrrr�__clean_tmprs���rCcCsJ|��}dD]}|�d|�rdSqdD]
}|�d|�r"|SqdS)zH
    Guess an archive type (tar, zip, or rar) by its file extension
    )
�tarztar.gzZtgzztar.bz2Ztbz2Ztbzztar.xzZtxzztar.lzmaZtlz�.rD)�zipZrarN)�lower�endswith)�nameZendingrrr�guess_archive_type�s��rJcOsPd|vrd|d<|�dd�}tj|i|��\}}|dur ||fSt�|�~|S)a
    Helper function which does exactly what ``tempfile.mkstemp()`` does but
    accepts another argument, ``close_fd``, which, by default, is true and closes
    the fd before returning the file path. Something commonly done throughout
    Salt's code.
    �prefixr,�close_fdTF)�pop�tempfile�mkstemp�os�close)rr
rLZfd_Zf_pathrrrrO�s
rOc
Cs�tjj�|�D];\}}}|�|d��tj�}tj�||�}tj�	|�s(t�
|�|D]}tj�|||�}tj�||�}	t�||	�q*qdS)z�
    Recursively copy the source directory to the destination,
    leaving files with the source does not explicitly overwrite.

    (identical to cp -r on a unix machine)
    r&N)
r+�utils�pathZos_walk�replace�lstriprP�sep�join�exists�makedirs�shutil�copyfile)
�source�dest�root�_�filesZpath_from_sourceZtarget_directoryrIZfile_path_from_source�target_pathrrr�recursive_copy�s
��rbr&c	Cs�tj�|�s
td|����tj�tj�|��std|����tj�|�}tj�tj�|��}t||d�}t	�
||�d}|rEtj�|d�}|dksO|dkrZ|rZtj�|�rZt
||�|dksd|dkre|re	d}tjj��s~zt�|�}Wn	ty}Ynwzt	�||�Wnty�t|��w|dur�t�||j|j�t�||j�tjj�d	�}	|	r�d
}
zddlm}|��}
Wntt fy�Ynw|
d
kr�t!tj"d��}|	|g}
t#j$|
||d�Wd�n1s�wYtj�|�r�t|�dSdS)zj
    Copy files from a source to a destination in an atomic way, and if
    specified cache the file.
    z%[Errno 2] No such file or directory: )rK�dirr&Zfile_backupZminionZbothZmasterNZ
restoreconFr)�selinuxZ	Enforcing�w)�stdout�stderr)%rPrS�isfile�OSError�isdir�dirname�basename�abspathrOrZr[rWrX�
backup_minionr+rR�platform�
is_windows�stat�moverrC�chown�st_uid�st_gid�chmod�st_mode�whichZsalt.modulesrdZ
getenforce�ImportErrorr�fopen�devnull�
subprocess�call)r\r]Zbackup_modeZcachedir�bname�dname�tgt�bkroot�fstatZrcon�policyrdZdev_null�cmdrrrr[�sb
�����r[c
Cs�z	t�||�WdStyT}z?|jtjkr�zt�|�Wn#tyB}z|jtjkr8td|�d|j����WYd}~nd}~wwt�||�WYd}~dSd}~ww)z�
    On Windows, os.rename() will fail with a WindowsError exception if a file
    exists at the destination path. This function checks for this error and if
    found, it deletes the destination path first.
    zError: Unable to remove �: N)	rP�renameri�errno�EEXIST�remove�ENOENTr�strerror)�src�dstrrrrr�s �����r�cCsv|dur
t|t�r|f}nd}|j|vrdS|jtjkr#t|�d���|jtjkr0td|����td�|j||j���)z�
    Common code for raising exceptions when reading a file fails

    The ignore argument can be an iterable of integer error codes (or a single
    integer error code) that should be ignored.
    Nrz does not existzPermission denied reading from z(Error {} encountered reading from {}: {})r�intr�r�rZEACCES�formatr�)rrS�ignorerrr�process_read_exceptions 
�

��r��皙�����?c#sF�t|t�s
td��|dur|d}�durt���d}d�fdd�	}z�z�zt�|�}Wn#tyO}z|jtjkrE|d�	|j||j
��WYd}~nd}~wwt�|j�s^|d|�d	��tj
tjBtjB}	t���|kr�zt�||	�}
Wn/ty�}z#|jtjkr�|d
�	|j||j
��t�d||�t�|�WYd}~n'd}~wwt�|
d��Wd�n1s�wYt�d
|�d}dVnt���|kso|d�	||��Wn#ty��ty�}z|d|�d|���WYd}~nd}~wwW|�rt�|�t�d||�dSdS|�r"t�|�t�d||�ww)�J
    Obtain a write lock. If one exists, wait for it to release first
    �path must be a stringN�.wFc�t|�d���z'
        Raise a FileLockError
        ��
time_start�r��msgZracer�rr�_raise_error=�zwait_lock.<locals>._raise_error�.Error {} encountered checking file lock {}: {}�lock_fn � exists and is not a file�/Error {} encountered obtaining file lock {}: {}�(Lock file %s exists, sleeping %f secondsre�Write lock %s obtainedT�DTimeout of {} seconds exceeded waiting for lock_fn {} to be released�&Error encountered obtaining file lock r��Write lock for %s (%s) released�F)r�strr�timerPrqrir�r�r�r��S_ISREGrw�O_CREAT�O_EXCL�O_WRONLY�openr�r>�trace�sleep�fdopenrr��rSZlock_fn�timeoutr�r�Z
obtained_lockr�Z	lock_statrZ
open_flagsZfh_rr�r�	wait_lock0s��

����
�������� ���

�
�r�csT�t|t�s
td��|dur|d}�durt���d}d�fdd�	}z�z�zt�|�}Wn#tyO}z|jtjkrE|d�	|j||j
��WYd}~nd}~wwt�|j�s^|d|�d	��tj
tjBtjB}	t���|kr�zt�||	�}
Wn5ty�}z)|jtjkr�|d
�	|j||j
��t�d||�tjjj�|�IdHWYd}~n'd}~wwt�|
d��Wd�n1s�wYt�d
|�d}dVnt���|kso|d�	||��Wn$ty��t�y}z|d|�d|���WYd}~nd}~wwW|�rt�|�t�d||�dSdS|�r)t�|�t�d||�ww)r�r�Nr�Fcr�r�r�r�r�rrr��r�z await_lock.<locals>._raise_errorr�r�r�r�r�rer�Tr�r�r�r�r�)rr�rr�rPrqrir�r�r�r�r�rwr�r�r�r�r�r>r�r+�extZtornado�genr�r�rr�r�rr�r�
await_lock�s��

����
��"������ ���

�
�r�cCst�d�}t�|�|S)z#
    Returns the current umask
    r)rP�umask)�retrrr�	get_umask�s

r�c	csN�|dustjj��rdVdSt�|�}zdVWt�|�dSt�|�w)zM
    Temporarily set the umask and restore once the contextmanager exits
    N)r+rRrorprPr�)�maskZ	orig_maskrrr�	set_umask�s�

r�cOs�z|ddvrt|d�d���Wn	tyYnwd}|�dd�rtt|�dkrLt|�}d|dvrK|d�dd�|d<d|dvrK|dd7<n(|�d	�rpd|d	vro|d	�dd�|d	<d|d	vro|d	d7<nd
|d	<d|vr�d}t|�dkr�t|�}d|dvr�d
}|�d	d�r�d|d	vr�d
}|s�t|d<|s�|�dd�s�d|d<|�d�dkr�d|�d	d�vr�t�	d�t
j|d<t|i|��}t
�r�ztj}Wnty�d}Ynwt�|��tj�}t�|��tj||B�|S)a�
    Wrapper around open() built-in to set CLOEXEC on the fd.

    This flag specifies that the file descriptor should be closed when an exec
    function is invoked;

    When a file descriptor is allocated (as with open or dup), this bit is
    initially cleared on the new file descriptor, meaning that descriptor will
    survive into the new program after exec.

    NB! We still have small race condition between open and fcntl.
    r)r��z# is not a permitted file descriptorN�binaryr��b�t�mode�rb�encodingFT�newliner&�	bufferingzaLine buffering (buffering=1) isn't supported in binary mode, the default buffer size will be used)�	TypeError�
IndexErrorrM�len�listrT�get�__salt_system_encoding__r>�debug�io�DEFAULT_BUFFER_SIZEr��is_fcntl_available�fcntl�
FD_CLOEXEC�AttributeError�fileno�F_GETFD�F_SETFD)rr
r��f_handler�Z	old_flagsrrrrz�sn
���
��
��
�rzc/s���d�dd�}�d}t|g��Ri|���M}z3tdd�r;tj}�r3t�fdd�|D��r3tj}t�|��|�|VWtdd�rMt�|��tj�ntdd�r]t�|��tj�wwWd�dS1siwYdS)	z;
    Shortcut for fopen with lock and context manager.
    rr�N�waT�Zcheck_sunoscsg|]}|�dv�qS)rr)r9�write�rrr�
<listcomp>Nszflopen.<locals>.<listcomp>)	rzr�r��LOCK_SH�any�LOCK_EX�flockr��LOCK_UN)rr
�filename�writingr�Z	lock_typerr�r�flopenCs&�

�
���"�r�c	os��|�dd�}|�dd�}|�dd�}t|i|���M}|d}t�|�}ttd�rC|j|ks3|j|krCdd	�||fD�rCt�|||�|dur\t�|j	�}||kr\t�
||j	|A|B�|VWd�dS1sjwYdS)
a�
    Shortcut for fopen with extra uid, gid, and mode options.

    Supported optional Keyword Arguments:

    mode
        Explicit mode to set. Mode is anything os.chmod would accept
        as input for mode. Works only on unix/unix-like systems.

    uid
        The uid to set, if not set, or it is None or -1 no changes are
        made. Same applies if the path is already owned by this uid.
        Must be int. Works only on unix/unix-like systems.

    gid
        The gid to set, if not set, or it is None or -1 no changes are
        made. Same applies if the path is already owned by this gid.
        Must be int. Works only on unix/unix-like systems.

    �uid����gidr�NrrscSsg|]}|dkr|�qS)r�r�r9�irrrr�xszfpopen.<locals>.<listcomp>)rMrzrPrq�hasattrrtrurs�S_IMODErwrv)	rr
r�r�r�r�rSZd_statZ	mode_partrrr�fpopenWs$�

�"�r�c
cs:�|durt�}zt�|�}Wnty*}z|dur||�WYd}~dSd}~ww|rHt�|�}|jdkrH|j|jf}||vrCdS|�|�gg}	}
|D]}tj�	||�}tj�
|�rd|	�|�qO|
�|�qO|rr||	|
fV|	D]}tj�	||�}
|s�tj�|
�s�t
|
||||�EdHqt|s�||	|
fVdSdS)z�
    A clone of the python os.walk function with some checks for recursive
    symlinks. Unlike os.walk this follows symlinks by default.
    Nr)�setrP�listdirrirq�st_ino�st_dev�addrSrWrj�append�islink�	safe_walk)�top�topdown�onerror�followlinks�_seen�names�err�status�node�dirs�nondirsrI�	full_path�new_pathrrrr��sD���



��r�cCs&zt�|�WdStyYdSw)z
    Safely remove a file
    N)rPr�ri)r�rrr�safe_rm�s
�rcCsrdd�}tj�|�stj�|�st�|�dStjj��r0z	tjj	�
|�}Wn	ty/Ynwtj
||d�dS)zj
    Platform-independent recursive delete. Includes code from
    http://stackoverflow.com/a/2656405
    cSs6tjj��rt�|tj�st�|tj	�||�dS�)a/
        Error handler for `shutil.rmtree`.

        If the error is due to an access error (read only file)
        it attempts to add write permission and then retries.

        If the error is for another reason it re-raises the error.

        Usage : `shutil.rmtree(path, onerror=onerror)`
        N)
r+rRrorprP�access�W_OKrvrq�S_IWUSR)rrS�exc_inforrr�_onerror�szrm_rf.<locals>._onerror)r�N)rPrSr�rjr�r+rRrorp�stringutilsZ
to_unicoder�rZ�rmtree)rSr	rrrr=�s�r=�is_emptycCs(z	t�|�jdkWStyYdSw)z
    Is a file empty?
    rF)rPrq�st_sizeri)r�rrrr�s
�cCs|r
tjj��r
dStS)z�
    Simple function to check if the ``fcntl`` module is available or not.

    If ``check_sunos`` is passed as ``True`` an additional check to see if host is
    SunOS is also made. For additional information see: http://goo.gl/159FF8
    F)r+rRroZis_sunos�	HAS_FCNTLr�rrrr��sr�cCsBdd�}t|t�st�d|t|d��dd��Stjd||tjd�S)a�
    Input the basename of a file, without the directory tree, and returns a safe name to use
    i.e. only the required characters are converted by urllib.parse.quote
    If the input is a PY2 String, output a PY2 String. If input is Unicode output Unicode.
    For consistency all platforms are treated the same. Hard coded to utf8 as its ascii compatible
    windows is \ / : * ? " < > | posix is /

    .. versionadded:: 2017.7.2

    :codeauthor: Damon Atkins <https://github.com/damon-atkins>
    cSstjj|�d�dd�S)Nrr&)�safe)�urllib�parse�quote�group)Zre_objrrr�_replace	sz$safe_filename_leaf.<locals>._replacez[\\:/*?"<>|]�utf8�ascii�backslashreplace)�flags)rr��re�sub�encode�UNICODE)Z
file_basenamerrrr�safe_filename_leaf�s
�rcCsh|stj}t|t�st|t�r|�d�}tj�|�\}}|�dd�|�|�D��}|r2|�||g�}|S)a3
    Input the full path and filename, splits on directory separator and calls safe_filename_leaf for
    each part of the path. dir_sep allows coder to force a directory separate to a particular character

    .. versionadded:: 2017.7.2

    :codeauthor: Damon Atkins <https://github.com/damon-atkins>
    rcSsg|]}t|��qSr)r)r9Zfile_sectionrrrr�)sz!safe_filepath.<locals>.<listcomp>)	rPrVrr�rrS�
splitdriverW�rsplit)Zfile_path_nameZdir_sep�driverSrrr�
safe_filepaths	
�r!Zis_text_file�cs�dd��d��fdd�tdd�D��d}z|�|�}Wn1tyMzt|d	��
}|�|�}Wd
�n1s9wYWntyJYYdSwYnwd|vrTdS|sXd
Sz|�d�Wd
StyiYnw|�d
|�}t	t
|��t
|�dkS)a
    Uses heuristics to guess whether the given file is text or binary,
    by reading a single block of bytes from the file.
    If more than 30% of the chars in the block are non-text, or there
    are NUL ('') bytes in the block, assume this is a binary file.
    cSs
t|f�Sr)�bytes)r:rrr�int2byte9s
zis_text.<locals>.int2byte�c3s�|]}�|�VqdSrrr��r$rr�	<genexpr><s�zis_text.<locals>.<genexpr>r2�s

	r�NF�Tzutf-8g333333�?)rW�range�readr�rzri�decode�UnicodeDecodeError�	translate�floatr�)�fp_�	blocksizeZtext_characters�blockZfp2_Znontextrr&r�is_text0s6	"�����	
�r3Zis_bin_filec	Cs�tj�|�sdSz=t|d��-}z|�d�}|�t�}tjj	�
|�WWd�WSty:YWd�WdSw1s>wYWdStyOYdSw)z�
    Detects if the file is a binary, returns bool. Returns True if the file is
    a bin, False if the file is not and None if the file is not available.
    Fr��NT)
rPrSrhrzr+r,r�r+rRr
�	is_binaryr-ri)rSr0�datarrrr5Xs"

����r5c
CsFzt�|�WdSty"}z|jtjkr�WYd}~dSd}~ww)zS
    Runs os.remove(path) and suppresses the OSError if the file doesn't exist
    N)rPr�rir�r�)rSrrrrr�ls���r��
list_filescCsht�}|�|�t|�D]#\}}}|D]}|�tj�||��q|D]}|�tj�||��q"qt|�S)zS
    Return a list of all files found under directory (and its subdirectories)
    )r�r�r�rPrSrWr�)�	directoryr�r^rr`rIrrrr7ws
�c	Cs,z	t|�dd�WSttfyYdSw)zj
    Convert the st_mode value from a stat(2) call (as returned from os.stat())
    to an octal mode.
    ���Nr&)�octr�r��r�rrr�st_mode_to_octal�s
�r<cCsF|durdSt|t�st|�}|�dd�}|�d��d��d��d�S)z�
    Return a mode value, normalized to a string and containing a leading zero
    if it does not have one.

    Allow "keep" as a valid mode (used by file state/module to preserve mode
    from the Salt fileserver in file states).
    NZ0o�0�"�'�)rr�rT�striprU�zfillr;rrr�normalize_mode�s
rCcCs\dddddd�}t|�}t�d|�}|std��t|�d��}d	|�|�d�d
�}||S)z/
    Convert human-readable units to bytes
    r�r��r@r�)�K�M�G�T�Pz^(\d+)([KMGTP])?$zFSize must be all digits, with an optional unit type (K, M, G, T, or P)ir)r�r�match�
ValueErrorr�rr�)Z
human_sizeZsize_exp_mapZhuman_size_strrJZsize_numZunit_multiplierrrr�human_size_to_bytes�s�rLc	Cs,tj�|�\}}tjj��r|�dd�}n|dd�}tjj��s&t�|�}t	t
t��d��dd�}tjj��r@t�d�}nt�d�}|dd	��|�d|d	d���}tj�
|||�d|���}tj�tj�|��svt�tj�|��t�||�tjj��s�t�||j|j�t�||j�dSdS)
z%
    Backup a file on the minion
    �:r_r�Ni@Bi����z%a_%b_%d_%H-%M-%S_%Yz%a_%b_%d_%H:%M:%S_%Yr9)rPrS�splitr+rRrorprTrqr�r�r��strftimerWrjrkrYrZr[rsrtrurvrw)	rSr�rr~Zsrc_dirr��msecsZstampZbkpathrrrrn�s&

"�rncCsFtjd|dd��}tj�|j���Wd�S1swYdS)z�
    Detect case insensitivity on a system.

    Returns:
        bool: Flag to indicate case insensitivity

    .. versionadded:: 3004

    ZTmPT)rKrc�deleteN)rNZNamedTemporaryFilerPrSrXrIrG)rSZtmp_filerrr�case_insensitive_filesystem�s
$�rRcCs�dd�}dd�}dd�}dd�}tj�|�std	��zt|d
��
}|�d�}Wd�n1s0wYWntyAtd
��w||�}|rJ|S||�rPdS||�rVtS||�r\dStd��)ao
    Detect a file's encoding using the following:
    - Check for Byte Order Marks (BOM)
    - Check for UTF-8 Markers
    - Check System Encoding
    - Check for ascii

    Args:

        path (str): The path to the file to check

    Returns:
        str: The encoding of the file

    Raises:
        CommandExecutionError: If the encoding cannot be detected
    cSs0z
|�d�t�d�WdStyYdSw)N�ASCIIzFound ASCIIFT)r,r>r�r-��_datarrr�check_ascii�s
��z!get_encoding.<locals>.check_asciic
Ss�dtjj�tj�fdtjj�tj�fdtjj�tj�fdtjj�tj�fdtjj�tj	�fdtjj�d�fdtjj�d�fdtjj�d	�fdtjj�d
�fdtjj�d�fg
}|D]\}}|�
|�rlt�d|�|SqYd
S)Nz	UTF-32-BEz	UTF-32-LEz	UTF-16-BEz	UTF-16-LE�UTF-8zUTF-7z+/v8-z+/v8z+/v9z+/v+z+/v/zFound BOM for %sF)
r+rRr
�to_bytes�codecs�BOM_UTF32_BE�BOM_UTF32_LE�BOM_UTF16_BE�BOM_UTF16_LE�BOM_UTF8�
startswithr>r�)rUZboms�	_encodingZbomrrr�	check_bom�s"�
�zget_encoding.<locals>.check_bomcSs&z|�d�}WdStyYdSw)NrWFT)r,r-)rU�decodedrrr�check_utf8_markers���z(get_encoding.<locals>.check_utf8_markerscSs&z|�t�WdStyYdSw)NFT)r,r�r-rTrrr�check_system_encodingrdz+get_encoding.<locals>.check_system_encodingz
Not a filer�r4NzFailed to open filerWrSzCould not detect file encoding)rPrSrhrrzr+rir�)rSrVrarcrer0r6r�rrr�get_encoding�s0
���rf)r&r&r)Nr�r�N)TNTNr�)r")Kr!rY�
contextlibr�r�r@rPrrZrqr|rNr��urllib.parserZsalt.ext.tornado.genr+Zsalt.utils.pathZsalt.utils.platformZsalt.utils.stringutilsZsalt.exceptionsrrrZsalt.utils.decorators.jinjarr�rryr�rr$r�	getLoggerrr>ZLOCAL_PROTOSZ
REMOTE_PROTOSZVALID_PROTOSZTEMPFILE_PREFIXZHASHES�itemsZ
HASHES_REVMAPrCrJrOrbr[r�r��contextmanagerr�r�r�r�rzr�r�r�rr=rr�rr!r3r5r�r7r<rCrLrnrRrfrrrr�<module>s��
+
�
D
XQ	
O


-1
$


'