File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/metaproxy/__pycache__/proxy.cpython-310.pyc
o
�N�g� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Zd dl
Zd dl Zd dl Zd dl
Zd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Zd dl Zd dl Zd dl Zd dl Zd dl!Zd dl"Zd dl#Zd dl$Zd dl%Zd dl&Zd dl'm(Z( d dl)m*Z*m+Z+m,Z,m-Z- d dlm.Z. d dlm/Z/ d dl"m0Z0m1Z1 e�2e3�Z4dd� Z5d d
� Z6d d � Z7d
d� Z8dd� Z9dd� Z:dd� Z;ddd�Z<dS )� N)�DEFAULT_TARGET_DELIM)�CommandExecutionError�CommandNotFoundError�SaltInvocationError�SaltSystemExit)� ProxyMinion)�tagify)�SignalHandlingProcess�default_signalsc c sx � t �d� | jr-|| jd< tjj| j| jd | jd | jd | j�d�d��� V | jd< d | jd vrRd | jvrRd
| jd d d }t � |� d
| _
t d|d��d | jvra| jd d | jd < | j�d�r�t �
| jd �}|�dd� tjjj| j|| j�d�| j�dd
�d�| _nQ| j�d�r�d| jd v r�| jd d | jd< d| jd v r�| j�di �}| jd d }z || | jd< W n ty� } zt � d| jd � W Y d}~nd}~ww | jd d }| �� \| _| _| _| _| jd | jd d� tj�| j�| _tjj| j| jd �| _| �� \| _| _| _| _| j| jjd < | j| jjd < | j| jjd < | jd | jjd < tjj| j| jd!�| _| j| jjd"< | j�� | j j tj j | j| j | jd!� |� d#�| jv�sp|� d$�| jv�r�d%�!|�d& }t � |� d
| _
t d|d��| j�|� d'�d(d)� �� | _"| j|d# }|| j� tjj#| j| jd!�| jd< | �$� | _%tj�&| j�| _&tj'�(| j| j�| _'tjj)j*| j�d*d�d+�} tj+j,| jd, | d-�| _-| j�r| jd �r| �� \| _| _| _| _t.| d.��r| j| j/_| j| j/_t.| d.��s tjj/j0| j| j| jtj+j1d/d0�g| jd1�| _/| jd2 �rJd3| jv �rJ| j/j2d4d3| jd d5d6d5| j�d7d
�d8�id5d9� t �3d:� n| j/j4d4d5d9� | jd; d<k�r�| jd= d>k�r�| j/j2tj+j1d/| jd d?�d@| jd= d5dAd
| jd d5dB�dC�id5d9� | jdD �r�dE| jv �r�| jd | jdE d> k�r�| j/j2tj+j1dFd0�dG| jdH d5dAd
d| jdE d> idC�id5d9� n,| j/j4tj+j1dFd0�d5d9� n | j/j4tj+j1d/| jd d?�d5d9� | j/j4tj+j1dFd0�d5d9� |dI }
|
| jv �r!dJ| jv �r!| j�dKd5��r!| j/j2dLdJ| j�dMdA�d5dAd
dN|idO�id5d9� | j/�5� n| j/j4dLd5d9� | jdP dQd� | jd | _6d5| _7dS )Rz�
Function to finish init after a proxy
minion has finished connecting to a master.
This is primarily loading modules, pillars, etc. (since they need
to know which master they connected to)
z'subclassed LazyLoaded _post_master_init�master�grains�id�saltenv� pillarenv)r r �pillar�proxyz,No proxy key found in pillar or opts for id z. zGCheck your pillar/opts configuration and contents. Salt-proxy aborted.F���)�code�msgZproxy_merge_pillar_in_optsNZ#proxy_merge_pillar_in_opts_strategyZ proxy_deep_merge_pillar_in_opts)ZstrategyZ merge_listsZproxy_mines_pillarZ
mine_intervalZmine_functionszHUnable to merge mine functions from the pillar in the opts, for proxy %s� proxytypezsaltutil.sync_all)r ��utils� __proxy__�__salt__�__ret__�
__pillar__�r � __utils__�.initz .shutdownz=Proxymodule {} is missing an init() or a shutdown() or both. z,Check your proxymodule. Salt-proxy aborted.�.module_executorsc S � g S �N� r" r" r" �H/opt/saltstack/salt/lib/python3.10/site-packages/salt/metaproxy/proxy.py�<lambda>� � z"post_master_init.<locals>.<lambda>�user�r& �cachedir��uid�schedule�alive)�type)Zcleanupr Z mine_enabledz mine.updateZ__mine_intervalT� Zmine_return_job)�function�minutes� jid_include�
maxrunningZ run_on_start�
return_job)Zpersistz Added mine.update to schedulerZ transportZtcpZmaster_alive_intervalr )r- r z
status.master� )r � connected)r/ �secondsr1 r2 r3 �kwargsZmaster_failbackZ master_listZfailbackzstatus.ping_masterZmaster_failback_intervalz.alivezstatus.proxy_reconnectZproxy_keep_aliveZ__proxy_keepaliveZproxy_keep_alive_intervalZ
proxy_name)r/ r0 r1 r2 r3 r7 zsaltutil.sync_grains�base)8�log�debugr5 �opts�saltr Zget_async_pillar�getZcompile_pillar�errorZ_runningr �copy�deepcopy�popr Z
dictupdate�merge� TypeError�
_load_modules� functions� returners�function_errors� executors�loaderr �pack�reload_modulesZio_loopZspawn_callbackZenginesZ
start_enginesZprocess_manager�format�module_executorsr Z_prep_mod_optsZmod_opts�matchersZbeaconsZBeaconr& �get_uid�minion� get_proc_dir�proc_dir�hasattrr+ ZScheduleZ master_eventZadd_job�infoZ
delete_jobZenable_scheduleZ grains_cache�ready) �selfr �errmsgr Zgeneral_proxy_minesZspecific_proxy_minesZterr� fq_proxyname�
proxy_init_fnr* Zproxy_alive_fnr" r" r# �post_master_init; s� �
�
�����
� �����
����
�
�
� ��
�
��
�� ��
���� ���� �
rZ c
C s |r|t jj_|s�| |�}||_t|d�s�|j|d d�\}}}} ||_||_||_ | |_
t j � |j
�|_ t j j|j
|j d�|_|j|d d�\}}}} ||_||_||_ | |_
|j|jjd< |j|jjd< |j|jjd< |j
d |jjd < t j j |j
|jd
�|_ |j |jjd < |j�� |d d
}
|j�|
� d�dd� �� |_|j|
d } | |� t|d�s�t j jj|�dd�d�} t jj|d | d�|_t jjj�|j ��. t |d t �s�t |d t �r�t �!|||� nt �"|||� W d � dS W d � dS 1 �sw Y dS )z�
Handle targeting of the minion.
Calling _thread_multi_return or _thread_return
depending on a single or multiple commands.
rE r )r r r r r r r r r r r r c S r r! r" r" r" r" r# r$ z r% ztarget.<locals>.<lambda>r rR r& Nr' r( r) �fun)#r< �crypt� AsyncAuth� creds_mapr5 rS rD rE rF rG rH rI r r; r rJ rK r= rM r& rO rP rQ rR �ext�tornadoZ
stack_contextZ StackContext�ctx�
isinstance�tuple�listr Z_thread_multi_returnZ_thread_return)
�cls�minion_instancer; �datar5 r^ rE rF rG rH rX rY r* r" r" r# �target< sr
��
��
�
� �$�rh c s t j��j|d �}tjj�d�| j |d �� dt �
� i}|� |� t �
d|d � tjj�|d��}|�tj�|�� W d � n1 sGw Y ddi}|d � |�d
�pet�d
g �pe|�d
d g�}t� �fd d
�|D ��} � �jv s|| du �rܐz`d}
�jr��jd �dd�r��jd �dg �} � dkr�� | vr�d}
�jd �dd�r��jd �dg �} � dkr�� | vr�d}
|
r�td��� �jv r؈j� } tj�| |d |�\}
}n � } |d |}
}d�jj d d< t |t �r�|g}nt |t �r�|�std�|���|�dd��r|d d k�rd |d < t � d |� |D ](}|� d �}|�j!v�r0td |� d!����j!| ||| |
|���du�rB n�qt �t"j#��r�d}i }�D ]<}t |t$��ret |t$��re|� |� n
|�sjg }|�%|� t&|d d"|d# t |�gd$�}d%|i}��'||� |d&7 }�qQ||d%< n�|d%< �jj d �dtj(j)j*�}|tj(j)j*k�r�z
t+�fd'd(�d)D ��}W n t,�y� d}Y nw |�s�tj(j)j-}||d< |tj(j)j*k|d< W �nJ t.�y } z(d*� � d+�}t j/|dd,� |� d-|� �|d%< d.|d/< tj(j)j-|d< W Y d}~�nd}~w t0�y@ } z"t j1d0� |t2j3d1� d2|� �|d%< d.|d/< tj(j)j-|d< W Y d}~n�d}~w t�yq } z%t j1d3� |t2j3d1� d4� � d5|� �|d%< d.|d/< tj(j)j-|d< W Y d}~n�d}~w t4�y� } z(d6�� || j5�p�d�}t j6|t2j3d1� ||d%< d.|d/< tj(j)j-|d< W Y d}~n�d}~w t,�y� d7}t j6|dd,� tjj1j7tj8�9|�||d8� |� d-t:�;� � �|d%< d.|d/< tj(j)j-|d< Y nMw �jd9 � � d:��}|�r��j�<� �|� < ||d%< n#�j�<� �|d%< � �=d;�d }|�j>v �r|d% d<��j>| �7 < d|d< tj(j)j-|d< d.|d/< |d |d< |d |d < |d |d=< d>|v �rE|d> |d>< d?|v �r^t |d? t$��rY|d? |d?< nt �6d@� �j�rk�j?|��@� dA� t |�d%�t ��r�|dB �r�dC�|dB |d% f�|dB< n|d% |dB< t �/dD|� |dB �rt |dB t ��r
dE|v �r�|dE |dE< dF|v �r�|dF |dF< |d# |d#< tA|dB �=dC��D ]F} z#| � dG�} | �jBv �rۈjB| |� n
�jB�<| �} t �1dH| | � W �q� t,�y } zt �CdI|d |� W Y d}~�q�d}~ww dS dS dS )J�g
This method should be used as a threading target, start the actual
minion side execution.
�jidz{}._thread_return {}�pid� Starting a new job with PID %s�w+bN�successFr[ rM Z direct_callc s0 g | ]}|� d ��j v r�j |� d � � ��qS )z.allow_missing_func)rH )�.0�executor)�
function_namerf r" r# �
<listcomp>� s
�z!thread_return.<locals>.<listcomp>Tr �minion_blackout�minion_blackout_whitelist�saltutil.refresh_pillarr ��Minion in blackout mode. Set 'minion_blackout' to False in pillar or grains to resume operations. Only saltutil.refresh_pillar allowed in blackout mode.�argr � __context__�retcodezDWrong executors specification: {}. String or non-empty list expectedZ sudo_user� r ZsudozExecutors list %sz.executez
Executor 'z' is not available�progr
�job�returnr4 c 3 s � | ] }� � |d �V qdS )TN�r= �ro �x)� return_datar" r# � <genexpr> s �
�z thread_return.<locals>.<genexpr>��resultrn zCommand required for 'z ' not found)�exc_infoz: �nested�outz#A command in '%s' had a problem: %s)Zexc_info_on_loglevelzERROR: zProblem executing '%s': %szERROR executing 'z': z%Passed invalid arguments to {}: {}
{}z'The minion function caused an exception)r| zsys.doc�*�.z Possible reasons: '{}'�fun_argsZ master_id�metadataz6The metadata parameter must be a dictionary. Ignoring.��timeout�ret�,zminion return: %s�
ret_config�
ret_kwargs� .returnerz#Returner %s could not be loaded: %s� The return failed for job %s: %s)D�os�path�joinrR r< r �process�appendproctitlerL �__name__�getpid�updater9 rT �files�fopen�write�payload�dumpsr= �getattr�anyrE r5 r; r rP �load_args_and_kwargsrJ rb �strrd �tracerH �types�
GeneratorType�dict�appendr Z _fire_master�defaultsZ exitcodes�EX_OK�all� ExceptionZ
EX_GENERICr r: r r> �logging�DEBUGrC �__doc__�warningZfire_exception�
exceptionsZ MinionError� traceback�
format_excZmissing_fun_string�splitrG � _return_pub�_return_retry_timer�setrF � exception) re rf r; rg �fn_�sdata�fp_r� rH Zallow_missing_funcs�minion_blackout_violation� whitelist�func�argsr7 �name�fname�indZiretZsingle�tagZ
event_datary � func_result�excr Zdocs�mod_name�returnerZ returner_strZ returner_errr" )rq rf r� r# �
thread_return� s� �
�
� � ���
�
��
�
�
�
� ��
���������
�
�
��� ���r� c s t j�|j|d �}tjj�d�| j |d �� dt �
� i}|� |� t �
d|d � tjj�|d��}|�tj�|�� W d � n1 sGw Y |�dd�}t|d �}|rjdg| dg| dg| d
��ni i i d
��td |�D �]} |s�d�d |d | < z�d}
|jr�|jd
�dd�r�|jd
�dg �} |d | dkr�|d | | vr�d}
n$|jd �dd�r�|jd �dg �} |d | dkr�|d | | vr�d}
|
r�td��|j|d | } tj�| |d | |�\}
}d |jj d d< |r�| n|d | � | |
i |���d � < |jj d �dd �}|d k�r<zt � �fdd�dD ��}W n t �y6 d}Y nw |�s<d}|�d � < |d k�d � < W n1 t �y| } z$t � � }t �!d |� |�rh|�d | < n
|�d |d | < W Y d}~nd}~ww |d �d< |d �d < |d �d < qud |v �r�|d �d < |j�r�|j"�|�#� d � |d �r�d!|v �r�|d! �d!< d"|v �r�|d" �d"< t$|d �%d#��D ]4}|d$ �d$< z
|j&|� d%� �� W �q� t �y� } zt �'d&|d |� W Y d}~�q�d}~ww dS dS )'ri rj z{}._thread_multi_return {}rk rl rm N�multifunc_orderedFr[ )r} ry rn r rn r rs rt ru Tr rv rw rx ry r} c 3 s$ � | ]
}�d � � |d�V qdS )r} TNr~ r ��keyr� r" r# r� � s �
�z&thread_multi_return.<locals>.<genexpr>r� r4 z+The minion function caused an exception: %sr� r� r� r� r� r� r� r
r� r� )(r� r� r� rR r< r r� r� rL r� r� r� r9 rT r� r� r� r� r� r= �len�ranger5 r; r rE rP r� rJ r� r� r� r� r� r� r� r� r� rF r> )re rf r; rg r� r� r� r� Z num_funcsr� r� r� r� r� r7 ry r� r� Ztrbr� r" r� r# �thread_multi_returnw s� �
� � �
��
���
�� ��
���r� c C s^ |dur)|d dkr+| � |d �r| �|d � dS | jd r-t�d|d � dS dS dS dS )z`
Verify the publication and then pass
the payload along to _handle_decoded_payload.
N�encZaes�loadZ
zmq_filteringz8Broadcast message received not for this minion, Load: %s)Z _target_loadZ_handle_decoded_payloadr; r9 r� )rV r� r" r" r# �handle_payload� s
��r� c c s` � d|v rt �d|d |d |d � n t �d|d |d � t �d|� t �d| j� | jdurR|d | jv r:dS | j�|d � t| j�| jd krR| j�d
� t |d t
�ru|d d kru| � � \| _ | _
| _| _| j | j_ | j
| j_
| j�d �}|d
kr�ttjj�| j��}||kr�t �d
|d � tjjj�d�V ttjj�| j��}||ks�| }| j�dd�}d�|d �}d}|r�tjj � � r�d}tj j j }t!t"j#t"j$�� t%| j&||| j|| j'|fd�}W d � n1 s�w Y nt(j)| j&|| j|| j'|f|d�}|�r$t!t"j#t"j$�� |�*� W d � n 1 �s w Y n|�*� | j+�,|� dS )zV
Override this method if you wish to handle the decoded data
differently.
r& z(User %s Executing command %s with jid %sr[ rj z Executing command %s with jid %szCommand details %szStarted JIDs: %sNZminion_jid_queue_hwmr zsys.reload_modules�process_count_maxzFMaximum number of processes reached while executing jid %s, waiting...�
�multiprocessingTzProcessPayload(jid={}))rh r� r� )rh r� r� )-r9 rT r: r� Z jid_queuer� r� r; rA rb r� rD rE rF rG rH r+ r= r< r rP Zrunningr� r_ r` �gen�sleeprL �platformZspawning_platformr\ r] r^ r
�signal�SIGINT�SIGTERMr �_targetr5 � threading�Thread�startZsubprocess_list�add) rV rg r� Z
process_count�instanceZmultiprocessing_enabledr� r^ r� r" r" r# �handle_decoded_payload� s� ��
�
��
� ���
��r� c C s� d|vsd|vsd|vsd|vrdS d|v rJ| j �d�|d �d�}|du r(dS |d d v r@|�d
t�}||d |d �s>dS d
S ||d �sHdS d
S | j d |d �sUdS d
S )z/
Verify that the publication is valid.
Ztgtrj r[ rw FZtgt_typez{}_match.matchN)ZgrainZ
grain_pcrer � delimiter)r� zglob_match.matchT)rN r= rL r )rV r� Z
match_funcr� r" r" r# � target_loadQ s" ��r� Tc C s t t| �j|d� dS )z_
Lock onto the publisher. This is the main event loop for the minion
:rtype : None
)r� N)�superr �tune_in)rV r� r" r" r# r� p s r� )T)=r? r� r� r� r� r� r� r< Z salt.beaconsZsalt.cli.daemonsZ salt.clientZ
salt.cryptZsalt.defaults.exitcodesZ salt.enginesZsalt.ext.tornado.genZsalt.ext.tornado.ioloopZ salt.loaderZ salt.minionZ salt.payloadZ salt.pillarZsalt.serializers.msgpackZ
salt.syspathsZsalt.utils.argsZsalt.utils.contextZsalt.utils.dataZsalt.utils.dictupdateZsalt.utils.errorZsalt.utils.eventZsalt.utils.filesZsalt.utils.jidZsalt.utils.minionZsalt.utils.minionsZsalt.utils.networkZsalt.utils.platformZsalt.utils.processZsalt.utils.scheduleZsalt.utils.ssdpZsalt.utils.userZsalt.utils.zeromqZ
salt.defaultsr Zsalt.exceptionsr r r r r r r r
� getLoggerr� r9 rZ rh r� r� r� r� r� r� r" r" r" r# �<module> sp
P lpV