Current File : //opt/alt/python37/lib/python3.7/site-packages/ssa/autotracing/__pycache__/config.cpython-37.pyc
B
r�b& � @ s d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z m Z m
Z
mZmZmZ ddlmZ ddlmZ dd lmZmZmZmZ d
d lmZ d
d l m Z d
d
l m Z m Z m!Z!m"Z"m#Z# e�$d�Z%e&d�dd�Z'G dd� de �Z(eG dd� d��Z)G dd� de�Z*G dd� de*�Z+G dd� de*�Z,e)d�d d �Z-d5e.e e+e,f d �d d �Z/d6e.e.dd!�d"d#�Z0d7e.e&dd%�d&d'�Z1d8e.e&dd%�d(d)�Z2d9e.ee
e.ee. f d*�d+d,�Z3ee
e.e,f d�d-d.�Z4ee. d�d/d0�Z5dd�d1d2�Z6dd�d3d4�Z7dS ):zF
This module contains a config parser for cloudlinux-xray-autotracing
� N)� ConfigParser)� dataclass)�Enum)�Any�Union�Tuple�Iterator�List�Optional)�ClPwd)�cpusers)�is_cl_solo_edition�
CLEditions�CLEditionDetectionError�SupportedEditions� )� flag_file)�SSAError)�umask_0�set_privileges�is_xray_user_agent_active� xray_version�$is_io_throttling_detection_availablezautotracing.config)�returnc
C sj y t jdd�} t�d| � W n4 tk
rP } zt�dt|j�� dS d}~X Y nX | dkr^dS | tj j
kS )zACurrently Auto tracing feature is not supported on Shared editionT)�skip_marker_checkzCurrent edition: %sz Unable to detect edition: %sFN) r �get_cl_edition�logger�infor �error�str�messager �SHARED�value)�edition�e� r% �G/opt/alt/python37/lib/python3.7/site-packages/ssa/autotracing/config.py�is_edition_supported* s r' c @ s e Zd ZdZdZdZdS )�Statusz
Autotracing statuses
�enabled�disabledN)�__name__�
__module__� __qualname__�__doc__�ENABLED�DISABLEDr% r% r% r& r( H s r( c @ s* e Zd ZU dZeed< eed< eed< dS )�Userz
User container
�uid�name�homeN)r+ r, r- r. �int�__annotations__r r% r% r% r& r1 P s
r1 c @ sb e Zd ZdZdZdd�dd�Zeedd�dd �Zeed
�d d �Zedd
�dd�Z ed�dd�Z
dS )�AutotracingConfigz)
Autotracing basic config parser
�confN)r c C s( t j�| j�}t j�|�s$t �|� dS )zh
If subdirectory location for autotracing config file does not exist,
create it
N)�os�path�dirname� config_file�exists�mkdir)�selfZ subdir_pathr% r% r& �check_config_dir` s z"AutotracingConfig.check_config_dir)�keyr" r c C s< || | j |< | �� t| jd��}| �|� W dQ R X dS )zD
Set given config item 'key' to given value 'value'
�wN)� main_sectionr@ �openr< �write)r? rA r" �
configfiler% r% r& �set_config_valuei s z"AutotracingConfig.set_config_value)rA r c C s | � | j� | | j | S )zD
Set given config item 'key' to given value 'value'
)�readr< rC )r? rA r% r% r& �get_config_valuer s z"AutotracingConfig.get_config_value)r" r c C s | � d|� dS )z"
Set given status
�statusN)rG )r? r" r% r% r& �
set_statusy s z AutotracingConfig.set_statusc C s
| � d�S )z"
Set given status
rJ )rI )r? r% r% r& �
get_status s z AutotracingConfig.get_status) r+ r, r- r. rC r@ r rG rI rK rL r% r% r% r& r7 Z s r7 c s e Zd ZdZ� fdd�Z� ZS )�AdminLevelConfigz Admin level autotracing configc s& ddi}d| _ t� j|| jdd� d S )NrJ r* z /usr/share/clos_ssa/autotracingF)�default_section�strict)r< �super�__init__rC )r? �defaults)� __class__r% r& rQ � s zAdminLevelConfig.__init__)r+ r, r- r. rQ �
__classcell__r% r% )rS r& rM � s rM c s& e Zd ZdZed�� fdd� Z� ZS )�UserLevelConfigz User level autotracing config)�
configpathc s2 dt � �� i}|� d�| _t� j|| jdd� d S )NrJ z/.ssa/autotracingF)rN rO )rM rL r< rP rQ rC )r? rV rR )rS r% r& rQ � s zUserLevelConfig.__init__)r+ r, r- r. r rQ rT r% r% )rS r& rU � s rU c C s t �t�� �} t| j| j| j�S )z*
Get current user and his details
)�pwd�getpwuidr9 �getuidr1 �pw_uid�pw_name�pw_dir)Zpw_entryr% r% r& �who_am_i� s r] )� user_homer c C sD t � }|jdkr&| r t| �}q@t� }nt� r8t|j�}ntd��|S )zF
Initialize correct config file instance depending on context
r z�Auto tracing management is not available. Reason: X-Ray End-User plugin is not enabled, please contact your system administrator for help.)r] r2 rU rM r r4 r )r^ Z current_userZ
conf_instancer% r% r& �config_instance� s
r_ )r" �userr c
C s� |r�yt �|�}W n4 tk
rF } ztd|� d��|�W dd}~X Y nX y0t|j|jd�� t|j�� | � W dQ R X W q� t
k
r� } zt|j �|�W dd}~X Y q�X n t d�� t� � | � W dQ R X dS )z
zUser 'z ' not foundN)�
target_uid�
target_gid� )
rW �getpwnam�KeyErrorr r rZ �pw_gidr_ r\ rK �PermissionError�strerrorr )r" r` Zpw_datar$ r% r% r& rG � s $
rG F)�username�mode_allr c C s* t � |r| dkrt� ttjj| � dS )z�
Enable autotracing.
If username is given, the user's config is changed in Admin's mode.
Perform some misconfiguration checks before enabling and
do not enable if some of them appear
N)�misconfiguration_checks�remove_custom_users_configsrG r( r/ r" )ri rj r% r% r& �enable� s rm c C s$ |r| dkrt � ttjj| � dS )ze
Disable autotracing.
If username is given, the user's config is changed in Admin's mode
N)rl rG r( r0 r" )ri rj r% r% r& �disable� s rn )ri r c
C s� y
t � W n, tk
r6 } ztjj|jfS d}~X Y nX | dk r�ytt� �| ��� � dfS tj
k
r� } ztt |��|�W dd}~X Y nX t � � � dfS )zk
Get status of autotracing.
If username is given, the status for a particular user is returned
N)
rk r r( r0 r" �reasonrU r � get_homedirrL �NoSuchUserExceptionr rM )ri r$ r% r% r& rJ � s
rJ c c sL xFt � D ]<} yt� �| �}W n tjk
r4 wY nX | t|�fV qW dS )zT
Iterator over all users on the server along with their autotracing configs
N)r r rp rq rU )r` Z_homedirr% r% r& � user_configs� s rr c C s dd� t � D �S )zGet list of disabled usersc S s$ g | ] \}}|� � tjjkr|�qS r% )rL r( r0 r" )�.0ri Zuserconfr% r% r& �
<listcomp> s z"disabled_users.<locals>.<listcomp>)rr r% r% r% r& �disabled_users s ru c C sF x@t � D ]6\} }tj�|j�rt�|j� t�tj�|j�� qW dS )z,
Remove custom users configurations
N)rr r9 r: �isfiler< �remove�rmdirr; )�_Zuser_config_pathr% r% r& rl s rl c C s� d} t � s t| ddd� dd��tdd�sFt� sFt| d d
d� dd��t� d krft| d d
d� dd��tj�t�s�t| ddd� dd��d S )z(Additional checks for known malfunctionsz@%(reason)s. You should %(fix)s in order to get Auto Tracing workz>Your current CL edition is unsupported by Auto Tracing featurez+use either CL Shared Pro or CL Solo edition)ro �fix�warning)�flagT)�skip_jwt_checkz1Your kernel does not support throttling detectionzupdate the kernelNzX-Ray is not installedz
install X-Rayz Slow Site Analyzer is disabledz enable it) r' r r
r r r9 r: rv r )Z _templater% r% r& rk s"
rk )N)N)NF)NF)N)8r. �loggingr9 rW � configparserr � dataclassesr �enumr �typingr r r r r r
�clcommon.clpwdr Zclcommon.cpapir �clcommon.lib.cleditionr
r r r �internal.constantsr �internal.exceptionsr �internal.utilsr r r r r � getLoggerr �boolr' r( r1 r7 rM rU r] r r_ rG rm rn rJ rr ru rl rk r% r% r% r& �<module>
s<
,
"
Mini Shell