|
Server : Apache System : Linux cvar2.toservers.com 3.10.0-962.3.2.lve1.5.73.el7.x86_64 #1 SMP Wed Aug 24 21:31:23 UTC 2022 x86_64 User : njnconst ( 1116) PHP Version : 8.4.18 Disable Function : NONE Directory : /opt/alt/python37/lib/python3.7/site-packages/ssa/autotracing/__pycache__/ |
Upload File : |
B
��cc( � @ s d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z m Z m
Z
mZmZmZ ddlmZ ddlmZ dd lmZ dd
lmZmZmZmZm Z d d l m 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 �Z0d8e1e e.e/f d �d d!�Z2d9e1e1dd"�d#d$�Z3d:e1e)dd&�d'd(�Z4d;e1e)dd&�d)d*�Z5d<e1ee
e1ee1 f d+�d,d-�Z6e
d�d.d/�Z7ee
e1e/f d�d0d1�Z8ee1 d�d2d3�Z9dd�d4d5�Z:dd�d6d7�Z;dS )=zF
This module contains a config parser for cloudlinux-xray-autotracing
� N)� ConfigParser)� dataclass)�Enum)�Any�Union�Tuple�Iterator�List�Optional)�ClPwd)�cpusers)�CPAPIException)�is_cl_solo_edition� is_container�
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* J s r* c @ s* e Zd ZU dZeed< eed< eed< dS )�Userz
User container
�uid�name�homeN)r- r. r/ r0 �int�__annotations__r! r' r' r' r( r3 R s
r3 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_dirb 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_sectionrB �openr> �write)rA rC r$ �
configfiler' r' r( �set_config_valuek s z"AutotracingConfig.set_config_value)rC r c C s | � | j� | | j | S )zD
Set given config item 'key' to given value 'value'
)�readr> rE )rA rC r' r' r( �get_config_valuet s z"AutotracingConfig.get_config_value)r$ r c C s | � d|� dS )z"
Set given status
�statusN)rI )rA r$ r' r' r( �
set_status{ s z AutotracingConfig.set_statusc C s
| � d�S )z"
Set given status
rL )rK )rA r' r' r( �
get_status� s z AutotracingConfig.get_status) r- r. r/ r0 rE rB r rI rK rM rN r' r' r' r( r9 \ s r9 c s e Zd ZdZ� fdd�Z� ZS )�AdminLevelConfigz Admin level autotracing configc s& ddi}d| _ t� j|| jdd� d S )NrL r, z /usr/share/clos_ssa/autotracingF)�default_section�strict)r> �super�__init__rE )rA �defaults)� __class__r' r( rS � s zAdminLevelConfig.__init__)r- r. r/ r0 rS �
__classcell__r' r' )rU r( rO � s rO c s& e Zd ZdZed�� fdd� Z� ZS )�UserLevelConfigz User level autotracing config)�
configpathc s2 dt � �� i}|� d�| _t� j|| jdd� d S )NrL z/.ssa/autotracingF)rP rQ )rO rN r> rR rS rE )rA rX rT )rU r' r( rS � s zUserLevelConfig.__init__)r- r. r/ r0 r! rS rV r' r' )rU r( rW � s rW c C s t �t�� �} t| j| j| j�S )z*
Get current user and his details
)�pwd�getpwuidr; �getuidr3 �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_ r4 rW rO r r6 r )r` � current_userZ
conf_instancer' r' r( �config_instance� s
rb )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� )
rY �getpwnam�KeyErrorr r r\ �pw_gidrb r^ rM �PermissionError�strerrorr )r$ rc �pw_datar& r' r' r( rI � s $
rI 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_configsrI r* r1 r$ )rm rn r' 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)rp rI r* r2 r$ )rm rn r' r' r( �disable� s rr )rm 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)
ro r r* r2 r$ �reasonrW r � get_homedirrN �NoSuchUserExceptionr! rO )rm r& r' r' r( rL � s
rL 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 rw 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)rw r rt ru rW )rc Z_homedirr' r' r( � user_configs s rx c C s dd� t � D �S )zGet list of disabled usersc S s$ g | ] \}}|� � tjjkr|�qS r' )rN r* r2 r$ )�.0rm Zuserconfr' r' r( �
<listcomp> s z"disabled_users.<locals>.<listcomp>)rx r' r' r' r( �disabled_users s r{ 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
)rd re N)rx rY rg r r\ ri r; r< �isfiler> �remove�rmdirr= rj r rk )rc �user_config_pathrl r& r' r' r( rp s
"rp c C s� d} t � s t| ddd� dd��tdd�sLt� sLt� sLt| d d
d� dd��t� d krlt| 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)rs �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 r r; r< r| r )Z _templater' r' r( ro , s"
ro )N)N)NF)NF)N)<r0 �loggingr; rY � configparserr � dataclassesr �enumr �typingr r r r r r
�clcommon.clpwdr �clcommon.cpapir Z clcommon.cpapi.cpapiexceptionsr
�clcommon.lib.cleditionr r r r r �internal.constantsr �internal.exceptionsr �internal.utilsr r r r r � getLoggerr �boolr) r* r3 r9 rO rW r_ r! rb rI rq rr rL rw rx r{ rp ro r' r' r' r( �<module>
s@
,
"