관리-도구
편집 파일: manager.cpython-37.pyc
B � 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 )NZmanagerz 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) �loggingZ getLogger�logger� ini_file_name� substrings_to_exclude_dir_paths�wildcard_ini_locations�OSError� ValueError� subprocessZSubprocessError�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 Zrequests_durationZ ignore_listr )r r r r �_restart_required_settingsD s z"Manager._restart_required_settingsc C s dddddhS )NZ correlationZcorrelation_coefficientZrequest_numberZtimeZdomains_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 )r, � intersection)r r. 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)Zcapture_output�textZcheckz ssa-agent z succeededz SSA Agent z failed with code z: )�cmdZretcode�stdout�stderr)�extraNzFailed to run z command for SSA Agent�errz for SSA Agent: ) r �runr �infoZCalledProcessError�errorr3 � returncoder4 r5 r r �str)r r1 � 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 filer7 )r6 z"Failed to update SSA config file: N�restartT)r= ) r �overrideZwrite_ssa_confr r r: r< r r0 r? r' )r r% �configr> 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)Zskip_jwt_checkc s i | ]\}}|� j kr||�qS r )r- )r �key�value)r r r r � s z&Manager.get_config.<locals>.<dictcomp>)rB )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 disabledrK )rL )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 )r< �lower)r rD rE r r r r � s z%Manager.get_stats.<locals>.<dictcomp>rG rH )rB �versionrI � agent_statusZ autotracing)r r" r' r r+ �status_ssa_agentr � get_stats)r Z_configr r r rY � 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)rZ r r � <listcomp>� s z+Manager.unused_dir_path.<locals>.<listcomp>)r )r rZ �resr )rZ 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 r_ )r �locationrZ 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 rb 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 r9 ra rg )r rb r r r rM � 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)ra r( �listdirr r) re )r rb �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 r9 rj r( �unlink)r Zclos_inir r r rQ 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 rI �startT)r= r@ N)r? r; )r rW r r r rN 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 rI �stopT)r= N)r? r; )r rW r r r rR 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 rI T)r= ZinactiveZactive)r? r )r r r r rX 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 rc Nz Flag file z created)rd r r r9 )r r r r rO ( 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( rk r r r9 r rK r< )r r> r r r rS 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+ rM )r r r r �regenerate_inisC s zManager.regenerate_inis)F)'�__name__� __module__� __qualname__�__doc__r � staticmethodr' �property�boolr+ �setr, r- �dictr0 r< r ZCompletedProcessr? rC rF rJ rP rT rY �listr_ ra rg rM rj rQ rN rR rX rO rS rn ro 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 rz )rs r# r r( r[ r Zglobr Zclcommon.lib.cleditionr Z configurationr r Zinternal.constantsr Zinternal.exceptionsr Zinternal.utilsr Zmodules.autotracerr Zmodules.decision_makerr r rz r r r r �<module> s" 1