File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/__pycache__/scripts.cpython-310.pyc
o
;jwZ � @ sT d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl m Z ddl
ZddlmZmZmZ e�e�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Zdd� Z dd� Z dd� Z dd� Z dd� Z dd � Z!d d � Z"d d � Z#d!d"� Z$d#d$� Z%d%d&� Z&d'd(� Z'd)d*� Z(d+d,� Z)d-d.� Z*d/d0� Z+d1d2� Z,d3d4� Z-d7d5d6�Z.dS )8zI
This module contains the function calls to execute command line scripts
� N)�randint)�SaltClientError�SaltReqTimeoutError�SaltSystemExitc C s� z| j j}W n
ttfy d}Y nw |tjkr6d}z| jjd }|d�|�7 }W n
ttfy5 Y nw d }|d u r@|r@d}|rJt |t
j dd� |riz t �
� }t�|� W n ty` Y nw t
�tjjj� t
�tjjj� d S )NFz
Exiting gracefully on Ctrl-c�jidz�
This job's jid is: {0}
The minions may not have all finished running and any remaining minions will return upon completion. To look up the return data for this job later, run the following command:
salt-run jobs.lookup_jid {0}z"
Exiting with hard crash on Ctrl-cT��file�flush)�options�
hard_crash�AttributeError�KeyError�signal�SIGINTZ local_clientZpub_data�format�print�sys�stderr� traceback�
format_exc�log�error�exit�salt�defaults� exitcodes�
EX_GENERIC�EX_OK)�client�signum�sigframeZ hardcrashZexit_msgr �trace� r" �@/opt/saltstack/salt/lib/python3.10/site-packages/salt/scripts.py�_handle_signals s< �
�� �r$ c C s^ t �t j�t jt jfv rt � t jt�t| �� t �t j�t ju r-t � t jt�t| �� d S d S �N) r � getsignalr �SIG_DFL�default_int_handler� functools�partialr$ �SIGTERM)r r" r" r# �_install_signal_handlers> s
�r, c C s8 ddl } tdkrtjt tjd< | jj�� }|�� dS )z
Start the salt master.
r N�__main__)�salt.cli.daemons�__name__r �modules�cli�daemonsZMaster�start)r Zmasterr" r" r# � salt_masterI s
r4 c
sr ddl �t�� �jj_ddl�ddl�dd� } t� � � � �fdd�}z�z+�j
j � � s8tj
|t�� fd�}|�� �jj�� }t�tjt�| |�� |�� W nQ tttfy� } zB� jdd � tj d
dd � t� d � d
}|dur}t |d�r}|j � dd
�}t d|�}t�!d|� t"�#|� t$�%�j&j'j(� W Y d}~nd}~ww W � jdd � dS W � jdd � dS � jdd � w )z
Start a minion process
r Nc S s | j �� d S r% )�minion�reload)�manager�sig�framer" r" r# �
handle_hupj s z"minion_process.<locals>.handle_hupc
s� � j dd�rL� �� t�d� zt�� dkr"�jj�� s"t� | d� W n t
yC } zt � d|� t�
�jjj� W Y d}~nd}~ww � j dd�sdS dS )z�
Have the minion suicide if the parent process is gone
NOTE: small race issue where the parent PID could be replace
with another process with same PID!
F��blocking� r z(Minion process encountered exception: %sN)�acquire�release�time�sleep�os�getuid�utils�platform�
is_windows�kill�OSErrorr r �_exitr r r )�
parent_pid�exc��lockr r" r# �suicide_when_without_parento s
� ���z3minion_process.<locals>.suicide_when_without_parent��target�argsTr; z4Fatal functionality error caught by minion handler:
��exc_infoz** Restarting minion **�< �config�random_reauth_delay� z waiting random_reauth_delay %ss))Z
salt._loggingrB �getpidZ_loggingZin_mainprocessZ__pid__r. �salt.utils.platform� threading�RLockrD rE rF �Thread�getppidr3 r1 r2 �Minionr �SIGHUPr) r* r r r r> r �warning�hasattrrU �getr �infor@ rA r r r r �SALT_KEEPALIVE)r: rN �threadr5 rK �delayr" rL r# �minion_processY sF
� �
���� rg c C s ddl } ddl}ddl}ddl}|jj�� |jj�� ddl }ddl
}t dkr0t j
t t j
d< dt jv r;t j�d� |jj�� rM|jj�� }|�� dS dt jv r^|jj�� }|�� dS dd� }| �| j�}| �| j�}d} z1|jtd
d �}|�� | � | jt � ||j �� | � | jt � ||j �� | � | j t � ||j �� W n t y� |jj�� }|�� Y dS w |s�ddl!}t"|j#j$d � d }|�%� | � | j|� | � | j|� |j&|j'j(j)ks�t �*|j&� t+�,d
t-dd� � t.�/� } | j0D ]}
| �1|
� q�t.�2� qq)zR
Start the salt minion in a subprocess.
Auto restart minion on error.
r Nr- � �--disable-keepalivec S s t �| |� dS )zy
Escalate the signal received to the multiprocessing process that
is actually running the minion
N)rB rG )�pidr r r" r" r# �escalate_signal_to_process� s z/salt_minion.<locals>.escalate_signal_to_processFTZMinionKeepAlive)rP �namer5 � rW �
)3r Zsalt.utils.debugrY �salt.utils.processrD �debugZenable_sigusr1_handler�process�notify_systemd�multiprocessingr. r/ r r0 �path�removerE rF r1 r2 r^ r3 �argvr& r r+ �Processrg r) r* rj r_ � Exception� salt.config�_supervisor_drop_privilegesrU �
minion_config�join�exitcoder r rd r r@ rA r �logging� getLogger�handlers�
removeHandler� basicConfig) r r rs r5 rk Zprev_sigint_handlerZprev_sigterm_handler� supervisor_privileges_droppedrq �rlogger�handlerr" r" r# � salt_minion� s|
� � � � �
�r� c
s| ddl }ddl}t�� � � fdd�}zlz)|jj�� s(tj|t� � fd�}|�
� d}d}|j j j
}|jj�� }|�
� W n: ttttfy` } ztjddd � d}|j j j}W Y d}~nd}~w tyv } z d}|j}W Y d}~nd}~ww W � jdd
� n� jdd
� w |du r�t�d � d }|dur�t |d
�r�|j � dd �}t d|�} t� d| � | �!| � n| �!d� t"�#|� dS )z&
Start a proxy minion process
r Nc s` � j dd�r.� �� t�d� zt�| d� W n ty% t�d� Y nw � j dd�sdS dS )z�
Have the minion suicide if the parent process is gone
NOTE: there is a small race issue where the parent PID could be replace
with another process with the same PID!
Fr; r= r i� N)r> r? r@ rA rB rG rH rI )rJ �rM r" r# rN s
��z9proxy_minion_process.<locals>.suicide_when_without_parentrO Fz Proxy Minion failed to start: TrR r; z ** Restarting proxy minion **rT rU rV rW z*Sleeping random_reauth_delay of %s seconds)$r. rY rZ r[ rD rE rF r\ rB r] r3 r r r r1 r2 � ProxyMinionrx r r r r r rd �
SystemExit�coder> r` ra rU rb r rc �putr r )
�queuer rN re Zrestart� proxyminion�statusrK rf Z random_delayr" r� r# �proxy_minion_process sV
�
�����
r� c C s� ddl } ddl}ddl}dtjv rtj�d� |jj�� r)|j j
� � }|� � dS dtj
v r:|j j
� � }|� � dS d} z| �� }W n tyW |j j
� � }|� � Y dS w | jt|fdd�}|� � |stddl}t|jjd � d}z2|�� z|jdd
�}W n ty� |jdkr�Y W dS d }Y nw |dkr�t�|j� t�|� W n
t y� Y dS w t � � }|j D ]}|� |� q�t �!� q=) z
Start a proxy minion.
r Nrh ri FTr� )rP rQ rl �proxy)�blockrT )"rs r. rY r rt ru rD rE rF r1 r2 r� r3 rv �Queuerx rw r� ry rz rU Z proxy_configr| rb r} r r@ rA �KeyboardInterruptr~ r r� r� r� ) rs r r� r� r� rq Z
restart_delayr� r� r" r" r# �
salt_proxyV sd
��
� �
�r� c C s` ddl } | jj�� ddl} t�� }z
| jj� � }|�
� W dS t y/ t� |d� Y dS w )z
Start the salt syndic.
r N� )
ro rD rq rr r. rB rX r1 r2 ZSyndicr3 r� rG )r rj Zsyndicr" r" r# � salt_syndic� s �r� c
C sf ddl } z| jj�� }t|� |�� W dS ty2 } ztj� d|� d�� W Y d}~dS d}~ww )z7
Manage the authentication keys with salt-key.
r NzError: �
)
Z salt.cli.keyr1 �keyZSaltKeyr, �runrx r r �write)r r �errr" r" r# �salt_key� s "��r� c C s( ddl } | jj�� }t|� |�� dS )�V
Publish commands to the salt system from the command line on the
master.
r N)Z salt.cli.cpr1 ZcpZ SaltCPClir, r� �r r r" r" r# �salt_cp� s r� c C �> ddl } dtjv rtj�d� | jj�� }t|� |�� dS )zi
Directly call a salt command in the modules, does not require a running
salt minion to run.
r Nrh ) Z
salt.cli.callr rt ru r1 �callZSaltCallr, r� r� r" r" r# � salt_call� �
r� c C s> ddl } dtjv rtj�d� | jj�� }t|� |�� dS )z-
Execute a salt convenience routine.
r Nrh )Z salt.cli.runr rt ru r1 r� ZSaltRunr, r� r" r" r# �salt_run� s
r� c C s� ddl } dtjv rtj�d� z| jj�� }t|� |�� W dS t y` } z4t
t |�tj dd� z|j
jr@t�� }t�|� W n ttfyL Y nw t�| jjj� W Y d}~dS d}~ww )z%
Execute the salt-ssh system
r Nrh Tr )Z salt.cli.sshr rt ru r1 ZsshZSaltSSHr, r� r r �strr r
r r r r r r r
r r r r )r r r� r! r" r" r# �salt_ssh� s(
�� ��r� c
C s� z
ddl } ddl} W n( ty2 } z ddl} t�d|� td� t�| j j
j � W Y d}~nd}~ww dtj v r>tj �
d� | jj�� }t|� |�� dS )z*
The main function for salt-cloud
r Nz Error importing salt cloud: %sz*salt-cloud is not available in this systemrh )Z
salt.cloudZsalt.cloud.cli� ImportError�salt.defaults.exitcodesr r r r r r r �EX_UNAVAILABLErt ru Zcloudr1 Z SaltCloudr, r� )r �er r" r" r# �
salt_cloud� s ��
r� c C s4 ddl } | jj�� ddl} | jj�� }|�� dS )z(
The main function for salt-api
r N) ro rD rq rr Z salt.cli.apir1 ZapiZSaltAPIr3 )r Zsapir" r" r# �salt_api s
r� c C r� )r� r Nrh ) Z
salt.cli.saltr rt ru r1 r ZSaltCMDr, r� r� r" r" r# � salt_main r� r� c C s ddl } | jj�� }|�� dS )z]
The main function for spm, the Salt Package Manager
.. versionadded:: 2015.8.0
r N)Z salt.cli.spmr1 �spmZSPMr� )r r� r" r" r# �salt_spm+ s r� c C s$ ddl }|jjj| ||||d� dS )zb
Quickstart for developing on the saltstack installation
.. versionadded:: 2016.11.0
r N)� extensionrl � description�salt_dir�merge)Zsalt.utils.extendrD �extendr� )r� rl r� r� r� r r" r" r# � salt_extend7 s
�r� c C s� g } t tjt �D ]}|�d�r| �|dd� � q ttj�dk r8d}| D ] }|d|� �7 }q%t|� t� d� tjd }|| vrIdtjd< t
}nd |� �tjd< tj� d� t tjt d
|� ��}|� dS ) z=
Change the args and redirect to another salt script
r r= Nrm z4Must pass in a salt command, available commands are:r� rW r zsalt-Zsalt_)
�dirr r0 r/ �
startswith�append�lenrv r r r� �pop�getattr)ZavailZfun�msg�cmdZs_funr" r" r# �
salt_unityH s&
�
r� c C sF | d d � }d}| D ]}d|v rd}q
d| v r!|s!|� d|� �� |S )NFz--targetT�installz --target=)r� )rQ rP �new_argsZtarget_in_args�argr" r" r# � _pip_argsb s � r� c C s: | � � }d| v r|� tj� | d � �|d< |S ||d< |S )N�
PYTHONPATH)�copyrB �pathsep)�env�extrasZnew_envr" r" r# �_pip_environmentm s �r� c C s4 t �t�� tjW d � S 1 sw Y d S r% )�
contextlib�suppressr r �RELENVr" r" r" r# �_get_onedir_env_pathv s �r� c C s� ddl } tjdd� }d}t|�D ]?\}}|rd}q|dv r.|d t|�k r.||d S |�d�r=|�dd�d S |�d�rPt|�d krP|d d� S qtj� d
�}|r[|S | j
j S ) a�
Return the salt config directory to use when the keepalive supervisor
reads minion or proxy options.
Respects ``-c``/``--config-dir`` on the command line first, then the
``SALT_CONFIG_DIR`` env var, then falls back to the compiled-in
default. Keeping this lookup simple and ``argparse``-free avoids
pulling in the full option parser (which the child process already
runs) and keeps the parent's privilege drop cheap.
r NrW F)�-cz --config-dirz
--config-dir=�=r� rm �SALT_CONFIG_DIR) Z
salt.syspathsr rv � enumerater� r� �splitrB �environrb ZsyspathsZ
CONFIG_DIR)r rv Z skip_next�idxr� Zenv_dirr" r" r# �_supervisor_config_dir} s$
� r� c C s� ddl }ddl}t� }tj�||�}z| |�}W n ty+ tjd||dd� Y dS w |� d�}|s5dS ||j
j � � kr?dS |j
j
�|� dS )a~
Drop privileges in the salt-minion / salt-proxy keepalive supervisor.
The supervisor forks a child process that runs the actual minion;
only the child needs root to perform ``verify_env`` chowns and to
create the pidfile, and the child drops to the configured user via
``salt.cli.daemons.Minion._real_start``. The parent only needs to
forward signals to the child and restart it on
``SALT_KEEPALIVE``-flavoured exits -- neither requires privileges.
Leaving the parent as root meant an unprivileged minion process tree
still had a privileged root process at its head, defeating the
purpose of configuring ``user: <something other than root>``. See
issue #68115.
``config_loader`` is ``salt.config.minion_config`` or
``salt.config.proxy_config``; ``config_name`` is the corresponding
config file basename (``minion`` or ``proxy``).
r NzDSupervisor could not load %s config from %s; skipping privilege dropTrR �user)�salt.utils.user�salt.utils.verifyr� rB rt r| rx r rp rb rD r� �get_user�verify�
check_user)Z
config_loaderZ config_namer �
config_dir� config_file�optsr� r" r" r# rz � s* ��
rz c
C sh ddl }ddl}ddl}t� }|du r$tdtjdd� t�|jj j
� n
t |dj tj
� �}| r8tj�| d�}n'tj�d�rLtj�d�}tj�|d�}ntj�d �rYtj�d �}n|jjd
}|j�|�}|�d �}|jj�� }|r�|d kr�||kr�|jj�|� ttj� � |�} t tj d
d� |�}
tj ddgt tj d
d� |� } t j!| dd| d�} t�| j"� dS )z'
Proxy to current python's pip
r Nzi'salt-pip' is only meant to be used from a Salt onedir. You probably want to use the system 'pip` binary.Tr z extras-{}.{}r5 r� ZSALT_MINION_CONFIGZ conf_filer� �rootrW z-mZpipF)�shell�checkr� )#ry r� r� r� r r r r r r r r� r � version_inforB rt r| r� rb rU ZDEFAULT_MINION_OPTSr{ rD r� r� r� r� r� r� r� rv �
executable�
subprocessr� �
returncode)
r� r Z relenv_pathr� r� Zsalt_config_dirr� r� � current_userr� rQ Zcommand�retr" r" r# �salt_pip� sF �
��r� r% )/�__doc__r� r) r~ rB r r� r rZ r@ r �randomr r� r Zsalt.exceptionsr r r r r/ r r$ r, r4 rg r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rz r� r" r" r" r# �<module> sP
' EnJD
+