File: //opt/alt/python37/lib/python3.7/site-packages/cllimits/__pycache__/cagefs_lib.cpython-37.pyc
B
�[ ���j3� @ 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