File: //opt/alt/python37/lib/python3.7/site-packages/ssa/autotracing/__pycache__/config.cpython-37.pyc
B
�BgdG( � @ s d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z m Z m
Z
mZmZmZ ddlmZ ddlmZ dd lmZ dd
lmZmZmZ d d lm Z d d
l m Z d dl 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 � d e+�Z-e*d�d d �Z.d8e/e e,e-f d �d d!�Z0d9e/e/dd"�d#d$�Z1d:e/e'dd&�d'd(�Z2d;e/e'dd&�d)d*�Z3d<e/ee
e/ee/ f d+�d,d-�Z4e
d�d.d/�Z5ee
e/e-f d�d0d1�Z6ee/ d�d2d3�Z7dd�d4d5�Z8dd�d6d7�Z9dS )=zF
This module contains a config parser for cloudlinux-xray-autotracing
� N)� ConfigParser)� dataclass)�Enum)�Any�Union�Tuple�Iterator�List�Optional)�ClPwd)�cpusers)�CPAPIException)�
CLEditions�CLEditionDetectionError�SupportedEditions� )� flag_file)�SSAError)�umask_0�set_privileges�is_xray_user_agent_active� xray_version�is_kernel_version_supportedzautotracing.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^ � 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" ra �pw_datar$ r% r% r& rG � s $
rG F)�username�mode_allr c
C s\ y
t � W n( tk
r2 } z
|j}W dd}~X Y nX d}|rJ| dkrJt� ttjj| � |S )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_checksr �reason�remove_custom_users_configsrG r( r/ r" )rk rl r$ �issuer% r% r& �enable� s
rq 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)ro rG r( r0 r" )rk rl r% r% r& �disable� s rr )rk r c
C s� y
t � W n( tk
r2 } z
|j}W dd}~X Y nX d}| dk r�ytt� �| ���� dfS tjk
r� } ztt|��|�W dd}~X Y nX t � �� |fS )zk
Get status of autotracing.
If username is given, the status for a particular user is returned
N)
rm r rn rU r � get_homedirrL �NoSuchUserExceptionr rM )rk r$ rp r% r% r& rJ � s
rJ c C s" yt � S tk
r t� S X dS )zP
Get panel users via cpapi, ignoring exceptions like NotSupported, etc.
N)r r
�tupler% r% r% r& � _panel_users s rv 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)rv r rs rt rU )ra Z_homedirr% r% r& � user_configs s rw c C s dd� t � D �S )zGet list of disabled usersc S s$ g | ] \}}|� � tjjkr|�qS r% )rL r( r0 r" )�.0rk Zuserconfr% r% r& �
<listcomp> s z"disabled_users.<locals>.<listcomp>)rw r% r% r% r& �disabled_users s rz c C s� x�t � D ]�\} }t�| �}yNt|j|jd��4 tj�|j �r\t�
|j � t� tj� |j �� W dQ R X W q t
k
r� } zt|j�|�W dd}~X Y qX qW dS )z,
Remove custom users configurations
)rb rc N)rw rW re r rZ rg r9 r: �isfiler< �remove�rmdirr; rh r ri )ra �user_config_pathrj r$ r% r% r& ro " s
"ro c C s� d} t � s t| ddd� dd��t� s<t| ddd� dd��t� d kr\t| d
d d� dd��tj�t�s~t| d d
d� 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)rn �fix�warning)�flagz1Your 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 r9 r: r{ r )Z _templater% r% r& rm 3 s
rm )N)N)NF)NF)N):r. �loggingr9 rW � configparserr � dataclassesr �enumr �typingr r r r r r
�clcommon.clpwdr �clcommon.cpapir Z clcommon.cpapi.cpapiexceptionsr
�clcommon.lib.cleditionr r r �internal.constantsr �internal.exceptionsr �internal.utilsr r r r r � getLoggerr �boolr' r( r1 r7 rM rU r] r r` rG rq rr rJ rv rw rz ro rm r% r% r% r& �<module>
s@
,
"