관리-도구
편집 파일: config.cpython-37.pyc
B � dG( � @ 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 Zuser_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@ , "