File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/cli/__pycache__/daemons.cpython-310.pyc
o
�N�g Q �
@ s| d Z ddlZddlZddlZddlm mZ ddlmZm Z m
Z
ddl m Z ddl
mZ ddlmZ ejdded d
d � ejd d
ed
d � ejd ded
d � zddlZddlmZ ddlmZmZmZ W n e y Z ze j d dkru� W Y dZ [ ndZ [ ww e� e �Z!G dd� d�Z"G dd� dejj#j$e"�Z%G dd� dejj#j&e"�Z'G dd� dejj#j(e"�Z)G dd� dejj#j*e"�Z+dS ) z
Make me some salt!
� N)�SaltClientError�SaltSystemExit�get_error_message)�
migrations)�is_junos)�
HAS_PSUTIL�once� z^(salt|salt\.(.*))$T)�append�ignorez>With-statements now directly support multiple context managerszV^Module backports was already imported from (.*), but (.*) is being added to sys.path$)�
ip_bracket)�
check_user�
verify_env�
verify_socketzNo module named _msgpackc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� DaemonsMixinz1
Uses the same functions for all daemons
c C s2 | j d �� dv rt�d| j d | jj� dS dS )zq
Verify and display a nag-messsage to the log if vulnerable hash-type is used.
:return:
Z hash_type)�md5�sha1z_IMPORTANT: Do not use %s hashing algorithm! Please set "hash_type" to sha256 in Salt %s config!N)�config�lower�log�warning� __class__�__name__��self� r �D/opt/saltstack/salt/lib/python3.10/site-packages/salt/cli/daemons.py�verify_hash_type= s ��z DaemonsMixin.verify_hash_typec C s t �d|| jj� dS )zN
Say daemon starting.
:param action
:return:
z%s the Salt %sN�r �infor r )r �actionr r r �action_log_infoK s z DaemonsMixin.action_log_infoc C � t �d| jj� dS )z8
Say daemon starting.
:return:
zThe Salt %s is starting upNr r r r r �start_log_infoT � zDaemonsMixin.start_log_infoc C r" )z=
Say daemon shutting down.
:return:
zThe Salt %s is shut downNr r r r r �shutdown_log_info\ r$ z DaemonsMixin.shutdown_log_infoc C s$ t �d| jjt|�� | �|� dS )z~
Log environment failure for the daemon and exit with the error code.
:param error:
:return:
z'Failed to create environment for %s: %sN)r � exceptionr r r �shutdown�r �errorr r r �environment_failured s �z DaemonsMixin.environment_failureN) r �
__module__� __qualname__�__doc__r r! r# r% r* r r r r r 8 s r c �F e Zd ZdZ� fdd�Z� fdd�Z� fdd�Zd � fd
d � Z� ZS )
�Masterz!
Creates a master server
c s. t | jd�r| jj�||� t� �||� d S )N�process_manager)�hasattr�masterr0 �_handle_signals�super�r �signumZsigframe�r r r r3 z s zMaster._handle_signalsc s� t � �� z�| jd r�| jd tj�| jd d�tj�| jd d�tj�| jd d�tj�| jd d�tj�| jd d�| jd tj�| jd d �tj�| jd d
�| jd | jd | jd
| jd g
}t|| jd | jd | jd | jd d� t�| jd
�D ]}t�tj�| jd
|�� q�W n t y� } z | �
|� W Y d}~nd}~ww | � d� t | jd | jd | jd �s�| �
dd� t| jd �| jd< t�| j� ddl}|j�| j�| _| �� | �� |jj�� dS )z�
Run the preparation sequence required to start a salt master server.
If sub-classed, don't **ever** forget to run:
super(YourSubClass, self).prepare()
r �pki_dirZminionsZ minions_preZminions_deniedZminions_autosignZminions_rejected�cachedirZjobs�proc�sock_dirZ token_dirZ
syndic_dirZsqlite_queue_dir�user�permissive_pki_access�root_dir�Z
permissiver> r8 Nz
Setting upZ interfaceZ publish_portZret_port� z#The ports are not available to bindr )r4 �preparer �os�path�joinr �listdir�remove�OSErrorr* r! r r' r r �
migrate_pathsZ salt.masterr2 r/ �daemonize_if_required� set_pidfile�utils�processZnotify_systemd)r �v_dirsZ syndic_filer) �saltr7 r r rA � sZ
�� ���
� zMaster.preparec s< t � �� t| jd �r | �d� | �� | j�� dS dS )z�
Start the actual master.
If sub-classed, don't **ever** forget to run:
super(YourSubClass, self).start()
NOTE: Run any required code before calling `super()`.
r< � Starting upN)r4 �startr
r r! r r2 r r7 r r rP � s
�z Master.startr Nc s8 | � � d}|dur|| }n|�� }t� �||� dS )zM
If sub-classed, run any shutdown operations on this method.
z The salt master is shutdown. N)r% �stripr4 r' )r �exitcode�exitmsg�msgr7 r r r' � s
zMaster.shutdown�r N� r r+ r, r- r3 rA rP r' �
__classcell__r r r7 r r/ s s =r/ c sV e Zd ZdZ� fdd�Z� fdd�Z� fdd�Zdd � Zd
d � Zd� fdd� Z � Z
S )�Minionz
Create a minion server
c s* t | jd�r | j�|� t� �||� d S )N�stop)r1 �minionrY r4 r3 r5 r7 r r r3 � s zMinion._handle_signalsc
s� t � �� zd| jd rh| j�d�}|r2d|v r tj�|�}tj�|�s1tj�tj�| jd �|�}ntj�tj�| jd �d�}| jd | jd | jd | jd |g}t || jd
| jd | jd | jd d
� W n t
y� } z | � |� W Y d}~nd}~ww t �
d| jd � t�| j� tr�| �� r�ts�| �� r�| �d� | �d� | j�d��� }z&ddl}| �� | �� | j�d�dkr�|j�| j� |j�| j�| _W dS t y� t j ddd� | �d� Y dS w )z�
Run the preparation sequence required to start a salt minion.
If sub-classed, don't **ever** forget to run:
super(YourSubClass, self).prepare()
r �default_include�*� conf_filezminion.dr8 r9 r; �extension_modulesr< r= r> r? Nz Setting up the Salt Minion "%s"�id�'An instance is already running. Exiting� � transportr � master_type�funcz4An error occured while setting up the minion managerT)�exc_info) r4 rA r �getrB rC �dirname�isabsrD r rG r* r r r rH r Z claim_process_responsibility�
check_runningr! r' r � salt.minionrI rJ rZ �eval_master_funcZ
MinionManager� Exceptionr) )r �confdrM r) rb rN r7 r r rA � sp
�������� ��
��zMinion.preparec
sV t � �� z| �� W dS ty* } z| jjr W Y d}~qW Y d}~dS d}~ww )z�
Start the actual minion.
If sub-classed, don't **ever** forget to run:
super(YourSubClass, self).start()
NOTE: Run any required code before calling `super()`.
TN)r4 rP � _real_startr �options�daemon�r �excr7 r r rP 7 s
�
���z Minion.startc
C s� z#t | jd �r | �d� | �� | j�� | jjr!td��W d S W d S tt fy^ } z-| �d� t
|t�rAt � d� | �
� nt �|� | �
|j� W Y d }~d S W Y d }~d S d }~ww )Nr< rO z"Minion could not connect to Master�Stopping�Exiting on Ctrl-c)r
r r! r rZ �tune_in�restartr �KeyboardInterruptr �
isinstancer r r' r) �coder( r r r rn K s&
�
���zMinion._real_startc
C s� z!| � � t| jd �r tjtjj | jjd< | j� � W dS W dS t
t fy\ } z-| � d� t
|t
�r?t�d� | �� nt�|� | �|j� W Y d}~dS W Y d}~dS d}~ww )a�
Start the actual minion as a caller minion.
cleanup_protecteds is list of yard host addresses that should not be
cleaned up this is to fix race condition when salt-caller minion starts up
If sub-classed, don't **ever** forget to run:
super(YourSubClass, self).start()
NOTE: Run any required code before calling `super()`.
r< Z__rolers rt N)rA r
r �kindsZAPPL_KIND_NAMESZ applKindsZcallerrZ �optsZcall_inrw r r! rx r r r' r) ry )r Zcleanup_protectedsrr r r r �call\ s$
��
���z Minion.callr Nc sP | � d� t| d�rt| jd�r| j�� t� �|d�| jj|p!d�� � � dS )�}
If sub-classed, run any shutdown operations on this method.
:param exitcode
:param exitmsg
�
Shutting downrZ �destroy�The Salt {} is shutdown. {}r N)
r! r1 rZ r r4 r' �formatr r rQ �r rR rS r7 r r r' y s
��zMinion.shutdownrU ) r r+ r, r- r3 rA rP rn r| r' rW r r r7 r rX � s O rX c r. )
� ProxyMinionz&
Create a proxy minion server
c s | j �|� t� �||� d S )N)rZ rY r4 r3 r5 r7 r r r3 � s zProxyMinion._handle_signalsc
s� t � �� t� s| jjs| �d� zd| jd rt| j�d�}|r>d|v r)tj �
|�}tj � |�s=tj � tj �
| jd �|�}ntj � tj �
| jd �d�}| jd | jd | jd | jd
|g}t
|| jd | jd | jd
| jd d� W n ty� } z | �|� W Y d}~nd}~ww | �d�| jd �� t�| j� | �� r�| �d� | �d� ddl}| �� | �� | j�d�dkr�|j�| j� |j�| j�| _dS )z�
Run the preparation sequence required to start a salt proxy minion.
If sub-classed, don't **ever** forget to run:
super(YourSubClass, self).prepare()
z salt-proxy requires --proxyidr r[ r\ r] zproxy.dr8 r9 r; r^ r< r= r> r? N�Setting up "{}"r_ r` ra r rc rd ) r4 rA r �valuesZproxyidr) r rf rB rC rg rh rD r rG r* r! r� r rH ri r' rj rI rJ rZ rk ZProxyMinionManager)r rm rM r) rN r7 r r rA � s^
��������
zProxyMinion.preparec
s� t � �� z#t| jd �r#| �d� | �� | j�� | jjr&t d��W dS W dS t
t fyc } z-| �d� t |t
�rFt
�d� | �� nt
�|� | �|j� W Y d}~dS W Y d}~dS d}~ww )z�
Start the actual proxy minion.
If sub-classed, don't **ever** forget to run:
super(YourSubClass, self).start()
NOTE: Run any required code before calling `super()`.
r< z The Proxy Minion is starting upz(Proxy Minion could not connect to MasterzProxy Minion Stoppingrt N)r4 rP r
r r! r rZ ru rv r rw r rx r r r' r) ry rq r7 r r rP � s(
�
���zProxyMinion.startr Nc sp t | d�r d| jjv r | jjd jd }| jjd | | jj� | �d� t� �|d�| jj |p1d��
� � dS )r} rZ Z proxymodulez .shutdownr~ r� r N) r1 rZ r{ Zloaded_base_namer! r4 r' r� r r rQ )r rR rS Zproxy_fnr7 r r r' s
��zProxyMinion.shutdownrU rV r r r7 r r� � s Rr� c s: e Zd ZdZ� fdd�Z� fdd�Zd
� fdd � Z� ZS ) �Syndicz
Create a syndic server
c
s� t � �� z(| jd r,t| jd | jd | jd g| jd | jd | jd | jd d� W n tyD } z | �|� W Y d }~nd }~ww | �d
�| jd �� d d l}| � � |j
� | j�| _ | �
� d S )
z�
Run the preparation sequence required to start a salt syndic minion.
If sub-classed, don't **ever** forget to run:
super(YourSubClass, self).prepare()
r r8 r; r^ r< r= r> r? Nr� r_ r )r4 rA r r rG r* r! r� rj rI rZ Z
SyndicManager�syndicrJ )r r) rN r7 r r rA ' s.
��� �� zSyndic.preparec sf t � �� t| jd �r1| �d� | �� z| j�� W dS ty0 | �d� | � � Y dS w dS )z�
Start the actual syndic.
If sub-classed, don't **ever** forget to run:
super(YourSubClass, self).start()
NOTE: Run any required code before calling `super()`.
r< rO rs N)
r4 rP r
r r! r r� ru rw r' r r7 r r rP I s
��z Syndic.startr Nc s0 | � d� t� �|d�| jj|pd��� � dS )r} r~ r� r N)r! r4 r' r� r r rQ r� r7 r r r' ] s
��zSyndic.shutdownrU )r r+ r, r- rA rP r' rW r r r7 r r� s
"r� ),r- �loggingrB �warningsZsalt.utils.kindsrK rz Zsalt.exceptionsr r r Z
salt.utilsr Zsalt.utils.platformr Zsalt.utils.processr �filterwarnings�DeprecationWarning� UserWarningZsalt.utils.parsersrN Zsalt.utils.networkr Zsalt.utils.verifyr
r r � ImportErrorrr �args� getLoggerr r r �parsersZMasterOptionParserr/ ZMinionOptionParserrX ZProxyMinionOptionParserr� ZSyndicOptionParserr� r r r r �<module> sl � �� ���
;
�
g
�
6
�
�