golden hour
/opt/alt/python37/lib/python3.7/site-packages/ssa/__pycache__
⬆️ Go Up
Upload
File/Folder
Size
Actions
__init__.cpython-37.pyc
148 B
Del
OK
agent.cpython-37.pyc
2.27 KB
Del
OK
db.cpython-37.pyc
3.62 KB
Del
OK
manager.cpython-37.pyc
11.95 KB
Del
OK
Edit: manager.cpython-37.pyc
B �z d�- � @ 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
Save