Edit File: cagefs_lib.cpython-37.pyc
B 5/9e� � @ sf d dl mZ d dlZd dlZd dlmZ d dlmZ d dl m Z m Z G dd� de�Z G dd � d �Z dS ) � )�absolute_importN)� exec_utility)�FormattedException)� run_command�ExternalProgramFailedc @ s e Zd ZdS )�CageFsExceptionN)�__name__� __module__� __qualname__� r r �py/cllimits/cagefs_lib.pyr s r c @ s� e Zd ZdZd dd�Zedd� �Zdd� Zd d � Zd d d �Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � ZdS ) �CageFsz/usr/sbin/cagefsctlNc C s d | _ d| _|p| �� | _d S )NF)�_cagefs_enabled_users�_is_cagefs_error�_create_dummy_logger�_logger)�self�loggerr r r �__init__ s zCageFs.__init__c C s" t �t�} d| _| �t �� � | S )NF)�loggingZ getLoggerr Z propagateZ addHandlerZ NullHandler)r r r r r s zCageFs._create_dummy_loggerc C s t j�| j�S )z^ Get CageFS presence flag :return: True/False - Cagefs present/absent )�os�path�isfile� _UTILITY_PATH)r r r r �is_cagefs_present' s zCageFs.is_cagefs_presentc C s | � � || jkS )z� Get User status in CageFs :param username: User name :return: True/False - user enabled/disabled in CageFs )� _load_infor )r �usernamer r r �get_user_status/ s zCageFs.get_user_statusFc C sP |s0| � � |r | �|�r dS |s0| �|�s0dS |r:d}nd}| �||g� dS )z� Set user status in CageFs :param str username: User name :param bool status: new user status - True/False --> enable/disable :param bool ignore_cache: ignore data caching :return: None Nz--enablez --disable)r r �_get_cagefsctl_out)r r ZstatusZ ignore_cacheZ cagefsctl_argr r r �set_user_status9 s zCageFs.set_user_statusc C sx | j dk rdS | �� r | jr.tdddid���t� | _ | �dg�}|�d�}x&|D ] }d|kr`qR| j �|�� � qRW dS ) zD Loads users info from Cagefs :return: None Nz%(util)s is disabled�util�CageFS)�message�contextz--list-enabled� z enabled user) r r r r �listr �split�append�strip)r � s_cagefs_outZs_enabled_users_parts�liner r r r P s zCageFs._load_infoc C s | � dg�}|S )z( Just initialize cagefs z--init)r )r �outr r r �initialize_cagefse s zCageFs.initialize_cagefsc C sH yddg}t |�}|S tk rB } ztt|���W d d }~X Y nX d S )N� cagefsctlz --enable-all)r r r �str)r �cmdr+ �errr r r �set_enabled_model s zCageFs.set_enabled_modec C s" | � dg�}|�d�d �� }|S )Nz--display-user-mode�:� )r r& r( )r r+ �moder r r � get_user_modew s zCageFs.get_user_modec C s | � dg� d S )Nz --toggle-mode)r )r r r r �toggle_user_mode} s zCageFs.toggle_user_modec C sn y(ddg}| j �dd�|�� t|�}|S tk rh } z"| j �dt|�� tt|���W d d }~X Y nX d S )Nr- z--enable-cagefsz Running "%s"� z cagefsctl exited with error "%s")r �info�joinr r r. r )r r/ r+ r0 r r r � enable_cagefs� s zCageFs.enable_cagefsc C s� | j �dd�|�� | �� s.tdddid���t| j|dd�\}}}|pJ|}|d ks\d |kr|| j �d ||f� d| _t|��|S ) NzRunning "cagefsctl %s"r7 z%(util)s is not installedr r! )r" r# T)�stderrr zError:z1Cagefs exited with exit code "%s" and output "%s")r r8 r9 r r r r r )r r/ Zret_codeZs_stdoutZs_stderrr) r r r r � s zCageFs._get_cagefsctl_outc C s | � dg�}|S )Nz--rebuild-alt-php-ini)r )r r+ r r r �rebuild_alt_php_ini� s zCageFs.rebuild_alt_php_ini)N)F)r r r r r � staticmethodr r r r r r, r1 r5 r6 r: r r< r r r r r s r )Z __future__r r Zos.pathr Z cllimits.libr Zclcommon.clexceptionr Zclcommon.utilsr r r r r r r r �<module> s