Current File : //opt/alt/python37/lib/python3.7/site-packages/ssa/__pycache__/manager.cpython-37.pyc
B
r�b�- � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ddl
m Z m Z ddl
mZ ddlmZ dd lmZ dd
lmZ dd lmZ G d d
� d
�Zdd�dd�ZdS )zA
This module contains classes implementing SSA Manager behaviour
� N)�iglob)�is_cl_solo_edition� )�load_validated_parser�load_configuration)� flag_file)�SSAManagerError)� ssa_version)�
AutoTracer)�
DecisionMakerc @ s� e Zd ZdZdd� Zedd�dd��Zeed�dd ��Z ee
d�d
d ��Z ee
d�d d
��Z e
e
d�dd�Zd?eejd�dd�Ze
dd�dd�Zdd�dd�Zdd�dd�Zdd�d d �Zdd�d d �Zdd�d d!�Zeed"�d#d$�Zed�d%d&�Zed'd(�d)d*�Z d'd�d+d,�Z ed�d-d.�Z d'd�d/d0�Z d'd�d1d2�Z d'd�d3d4�Z!ed�d5d6�Z"d'd�d7d8�Z#d'd�d9d:�Z$dd�d;d<�Z%d'd�d=d>�Z&d'S )@�Managerz
SSA Manager class.
c C s0 t �d�| _d| _d| _d| _tttj f| _
d S )N�managerz clos_ssa.ini)Zphp44Zphp51Zphp52Zphp53zphp\d+-imunifyz php-internal)z /opt/alt/php[0-9][0-9]/link/confz./var/cagefs/*/*/etc/cl.php.d/alt-php[0-9][0-9]z+/opt/cpanel/ea-php[0-9][0-9]/root/etc/php.dz$/opt/plesk/php/[0-9].[0-9]/etc/php.dz'/usr/local/php[0-9][0-9]/lib/php.conf.dzM/usr/share/cagefs/.cpanel.multiphp/opt/cpanel/ea-php[0-9][0-9]/root/etc/php.dzA/usr/share/cagefs-skeleton/usr/local/php[0-9][0-9]/lib/php.conf.d) �logging� getLogger�logger�
ini_file_name� substrings_to_exclude_dir_paths�wildcard_ini_locations�OSError�
ValueError�
subprocess�SubprocessError�subprocess_errors)�self� r �</opt/alt/python37/lib/python3.7/site-packages/ssa/manager.py�__init__! s
zManager.__init__zjson str)�returnc O s* ddi}|� dd� |�� D �� t�|�S )z@
Form a success json response with given kwargs
�result�successc S s i | ]\}}||�qS r r )�.0�k�vr r r �
<dictcomp>: s z$Manager.response.<locals>.<dictcomp>)�update�items�json�dumps)�args�kwargsZ raw_responser r r �response4 s zManager.responsec C s t j�t�S )z
Is SSA enabled
)�os�path�isfiler )r r r r �_enabled= s zManager._enabledc C s ddhS )zK
Configuration settings required Request Processor restart
�requests_duration� ignore_listr )r r r r �_restart_required_settingsD s z"Manager._restart_required_settingsc C s dddddhS )N� correlation�correlation_coefficient�request_number�time�domains_numberr )r r r r �solo_filtered_settingsK s z Manager.solo_filtered_settings)�settingsr c C s | j �|�S )z�
SSA Agent requires restart in case of changing these configuration:
- requests_duration
- ignore_list
)r1 � intersection)r r8 r r r �_restart_requiredP s zManager._restart_requiredF)�commandr c
C s y0t jdd|gdd|d�}| j�d|� d�� W n� t jk
r� } zj| jjd|j� d|j� d |jpf|j � �|j|j|j|j d
�d � t
d|j� d|j� d |jp�|j � ���W d d }~X Y nZ | j k
�r } z8| jjd
|� d�dt |�id � t
d
|� d|� ���W d d }~X Y nX |S )z�
Run /sbin/service utility to make given operation with SSA Agent service
:command: command to invoke
:check_retcode: whether to run with check or not
:return: subprocess info about completed process
z
/sbin/servicez ssa-agentT)�capture_output�text�checkz
ssa-agent z
succeededz
SSA Agent z failed with code z: )�cmd�retcode�stdout�stderr)�extraNzFailed to run z command for SSA Agent�errz for SSA Agent: )
r �runr �info�CalledProcessError�errorr? �
returncoderA rB r r �str)r r; �
check_retcoder �er r r �run_service_utilityX s( 4$zManager.run_service_utility)r( r c
C s� t � }|�|� y |�� W nH tk
rd } z*| jjddt|�id� td|� ���W dd}~X Y nX | �|�r~| j ddd� | �
� S ) z�
Change SSA config and restart it.
:args: dict to override current option values
:return: JSON encoded result of the action
z Failed to update SSA config filerD )rC z"Failed to update SSA config file: N�restartT)rK ) r �overrideZwrite_ssa_confr r rH rJ r r: rM r* )r r( �configrL r r r �
set_configu s
zManager.set_configc s> t � }tdd�r2� fdd�|�� D �}� j|d�S � j|d�S )zV
Get current SSA config.
:return: JSON encoded current config
T)�skip_jwt_checkc s i | ]\}}|� j kr||�qS r )r7 )r �key�value)r r r r# � s z&Manager.get_config.<locals>.<dictcomp>)rP )r r r% r* )r Z full_configZfiltered_configr )r r �
get_config� s
zManager.get_configc C s | j r
dnd}| j|d�S )zY
Get current status of SSA.
:return: JSON encoded current status
�enabled�disabled)Z
ssa_status)r. r* )r �statusr r r �get_ssa_status� s zManager.get_ssa_statusc C s2 | j rtddd��| �� | �� | �� | �� S )a�
Enable SSA:
- add clos_ssa extension for each PHP version on server
- add clos_ssa extension into cagefs for each user and each ver
- start SSA Agent (if it is not already started)
- restart Apache (etc.) and FPM, reset CRIU images
- create flag_file indicating that SSA is enabled successfully
:return: JSON encoded current status
zSSA is already enabled�warning)�flag)r. r �
generate_inis�start_ssa_agent� create_flagr* )r r r r �
enable_ssa� s
zManager.enable_ssac C s2 | j stddd��| �� | �� | �� | �� S )a{
Disable SSA:
- remove clos_ssa extension for each PHP version on server
- remove clos_ssa extension from cagefs for each user and each ver
- stop SSA Agent
- restart Apache (etc.) and FPM, reset CRIU images
- remove flag_file indicating that SSA is enabled
:return: JSON encoded current status
zSSA is already disabledrZ )r[ )r. r �remove_clos_inis�stop_ssa_agent� remove_flagr* )r r r r � disable_ssa� s
zManager.disable_ssac C s> dd� t � �� D �}| j|t� | jr(dnd| �� t� �� d�S )z�
Get SSA statistics.
Includes:
- config values
- version
- SSA status (enabled|disabled)
- SSA Agent status (active|inactive)
:return: JSON encoded current statistics
c S s i | ]\}}t |��� |�qS r )rJ �lower)r rS rT r r r r# � s z%Manager.get_stats.<locals>.<dictcomp>rV rW )rP �versionrX � agent_statusZ autotracing)r r% r* r r. �status_ssa_agentr
� get_stats)r Z_configr r r rh � s
zManager.get_stats)�dir_pathr c s � fdd�| j D �}|S )z6
Checking for substrings in a string.
c s g | ]}t �|� �r|�qS r )�re�search)r Z substring)ri r r �
<listcomp>� s z+Manager.unused_dir_path.<locals>.<listcomp>)r )r ri �resr )ri r �unused_dir_path� s zManager.unused_dir_pathc c s8 x2| j D ](}x"t|�D ]}| �|�r&q|V qW qW dS )zr
Generator of existing paths (matching known wildcard locations)
for additional ini files
N)r r rn )r �locationri r r r �existing_paths� s
zManager.existing_pathsN)�ini_pathr c C s0 t tj�|| j�d��}|�d� W dQ R X dS )zB
Enable SSA extension for single ini_path (given)
�wzextension=clos_ssa.soN)�openr+ r, �joinr �write)r rq Zinir r r �generate_single_ini� s zManager.generate_single_inic C s8 | j �d� x| �� D ]}| �|� qW | j �d� dS )zj
Place clos_ssa.ini into each existing Additional ini path,
including cagefs ones
z Generating clos_ssa.ini files...z Finished!N)r rF rp rv )r rq r r r r\ � s zManager.generate_inisc c sD x>| � � D ]2}x,t�|�D ] }| j|krtj�||�V qW q
W dS )zr
Generator function searching for clos_ssa.ini files
in all existing Additional ini paths
N)rp r+ �listdirr r, rt )r rq �namer r r �find_clos_inis� s
zManager.find_clos_inisc C s8 | j �d� x| �� D ]}t�|� qW | j �d� dS )z8
Remove all gathered clos_ssa.ini files
z Removing clos_ssa.ini files...z Finished!N)r rF ry r+ �unlink)r Zclos_inir r r r` s zManager.remove_clos_inisc C s2 | � d�}|jr | j ddd� n| j ddd� dS )ze
Start SSA Agent service
or restart it if it is accidentally already running
rX �startT)rK rN N)rM rI )r rf r r r r]
s
zManager.start_ssa_agentc C s" | � d�}|js | j ddd� dS )z`
Stop SSA Agent service
or do nothing if it is accidentally not running
rX �stopT)rK N)rM rI )r rf r r r ra s
zManager.stop_ssa_agentc C s, y| j ddd� W n tk
r& dS X dS )z:
Get SSA Agent status: active or inactive
rX T)rK �inactive�active)rM r )r r r r rg s
zManager.status_ssa_agentc C s. t td�� W dQ R X | j�dt� d�� dS )zE
Create a flag file indicating successful enablement
rr Nz
Flag file z created)rs r r rF )r r r r r^ ( s
zManager.create_flagc
C sf y"t �t� | j�dt� d�� W n> tk
r` } z | j�dt� dt|�� �� W dd}~X Y nX dS )z:
Remove a flag file indicating enablement
z
Flag file z removedz removal failed: N)r+ rz r r rF r rZ rJ )r rL r r r rb 0 s
zManager.remove_flagc C s t � �� }| jf |�S )zG
Get last report.
:return: JSON encoded report
)r Zget_json_reportr* )r Zreportr r r �
get_report; s
zManager.get_reportc C s | j r| �� dS )z@
Regenerates clos_ssa inis while SSA is enabled
N)r. r\ )r r r r �regenerate_inisC s zManager.regenerate_inis)F)'�__name__�
__module__� __qualname__�__doc__r � staticmethodr* �property�boolr. �setr1 r7 �dictr: rJ r �CompletedProcessrM rQ rU rY r_ rc rh �listrn rp rv r\ ry r` r] ra rg r^ rb r r� r r r r r s>
r zManager instance)r c C s t � S )zk
Factory function for appropriate manager initialization
:return: appropriate manager instance
)r r r r r �initialize_managerK s r� )r� r& r r+ rj r �globr �clcommon.lib.cleditionr Z
configurationr r �internal.constantsr �internal.exceptionsr �internal.utilsr Zmodules.autotracerr
Zmodules.decision_makerr r r� r r r r �<module>
s" 1
Mini Shell