File: //opt/alt/python37/lib/python3.7/site-packages/clselect/__pycache__/utils.cpython-37.pyc
B
���^�6 � @ s� d dl mZ d dl mZ d dl mZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m Z d dl
mZmZmZmZ d dlmZ d dlmZ d d lmZmZmZ d d
lmZ d d l m Z d d
l m Z dZ dGdd�Z!dHdd�Z"dd� Z#dd� Z$dd� Z%dd� Z&dd � Z'd d � Z(d d � Z)d!d"� Z*dId$d%�Z+dJd&d'�Z,dKd)d*�Z-dLd+d,�Z.d-d.� Z/d/d0� Z0d1d2� Z1d3d4� Z2d5d6� Z3d7d8� Z4d9d:� Z5dMd=d>�Z6d?d@� Z7dAdB� Z8dCdD� Z9dNdEdF�Z:dS )O� )�print_function)�absolute_import)�divisionN)�time)�AnyStr�List�Dict�Optional)� resolve_username_and_doc_root)�CP_NAME)�NoDomain� NotSupported�
IncorrectData)� mod_makedirs� )�clprint)�ClSelectExceptz/sbin/cagefs_enter_userc
C s� y0t j| td�t jt jd|dd�}|�� \}}W n: ttfk
rj } zt� | d t
|���W dd}~X Y nX |j dkr�t� |p�dd�
| �|f ��|S )z�
Runs external process and returns output
:param cmd: command and arguments as a list
:param env_data
:return string
z /dev/nullT)�stdin�stdout�stderr� close_fds�env�textr Nz output of the command: %s
%s� )�
subprocess�Popen�open�PIPEZSTDOUT� communicate�OSError�IOErrorr �FileProcessError�str�
returncode�ExternalProgramFailed�join)�cmd�env_data�output�std_out�std_err�e� r, �?/opt/alt/python37/lib/python3.7/site-packages/clselect/utils.py� run_command# s &
r. c
C sz y2t j| td�t jt j|d||d�}|�� \}}W n: ttfk
rl } zt�| d t |���W dd}~X Y nX |j
||fS )a�
Runs external process and returns output.
Differs from subprocess.check_output, run_command above, and check_output below in that it does not
throw an exception if process's return code != 0
:param cmd: command and arguments as a list
:param env_data
:param preexec_fn: Pre-exec function. None if don't need
:param cwd: Directory name to set as current
:return Cortege: (ret_code, stdout, stderr)
z /dev/nullT)r r r �
preexec_fnr �cwdr r N) r r r r r r r r r! r" r# )r& r' r/ r0 �processr) r* r+ r, r, r- �run_command_full< s &r2 c C s> d}y t | �}|�� }|�� W n ttfk
r8 Y nX |S )z8
Reads file contents and returns it as a string
� )r �read�closer r )�filename�defaults_contents�fr, r, r- �read_file_as_stringW s r9 c O s t | |� d S )N)� check_output)�args�kwargsr, r, r- �
check_calle s r= c
O s� |� dd�}|�d�}|� dd�}|� dd�}|r@| |7 } |d= y"tjf | tjtjdd�|��}W n: ttfk
r� } zt�| d t |���W dd}~X Y nX |�
� \}} |r�|� |� |� � |dk r�|�
� |jr�t�|p�| p�|��|S ) av
check_output(val1, val2, val3, arg4=val4)
equivalent
check_output(args=(val1, val2, val3), arg4=val4)
or equivalent
check_output(val1, args=(val2, val3), arg4=val4)
DON'T USE check_output((val1, val2, val3), arg4=val4)
:param tuple *args: arguments for command line
:param dict **kwargs: parameters for subprocess.Popen
:return str:
�errorNr; r( �waitT)r; r r r r )�pop�getr r r r r r r! r" r �writer5 r? r# r$ )
r; r< r> Zargs_from_kwargsr( r? �pr+ r r r, r, r- r: i s*
&
r: c s � fdd�t �� �D �S )Nc s& g | ] }t j�t j�� |��r|�qS r, )�os�path�isdirr% )�.0�d)rE r, r- �
<listcomp>� s z list_dirs.<locals>.<listcomp>)rD �listdir)rE r, )rE r- � list_dirs� s rK c c sR | V xF|D ]>\}}| t |� V x&|D ] }| d�tt ||f�� V q(W q W d S )N�.)r" r% �map)rE �versions�majorZminors�minorr, r, r- �
versioning� s
rQ c
C sR yt �| � W n> tk
rL } z |jtjkr:t j�| �r:n� W d d }~X Y nX d S )N)rD �makedirsr �errno�EEXISTrE rF )rE �excr, r, r- �mkdir_p� s rV c C sf t j�t�| �j�}t j�t j�||��}t||�d �t j �}t
||f�s^t � d||f ��||fS )z�
Returns absolute and relative to user's home directories paths,
after eliminating all symlinks
:param user: Unix user whose home contains the directory
:param directory: The directory to obtain absolute and relative paths for
� z&Directory '%s' not in user home ('%s'))
rD rE �realpath�pwd�getpwnam�pw_dirr% � s_partition�lstrip�sep�allr � WrongData)�user� directoryZuser_dirZabs_dirZrel_dirr, r, r- � get_abs_rel� s rc c C s t j�d�S )Nz /var/.cagefs)rD rE rF r, r, r, r- � in_cagefs� s rd �rc C s. d}t | ||d�}z |�� }W d|�� X |S )zg
read file and close
:param path: file path
:param mode: reading mode
:return str:
N)�mode�errors)r r4 r5 )rE rf rg �data�streamr, r, r- � file_read� s
rj c C s, g }t | |d�}z |�� }W d|�� X |S )zE
read litle file and close
:param path:
:return str:
)rg N)r � readlinesr5 )rE rg rh ri r, r, r- �file_readlines� s
rl �wc C s, t | ||d�}z|�|� W d|�� X dS )zN
write litle data to file and close
:param path:
:return str:
)rg N)r rB r5 )rE �linerf rg ri r, r, r- �
file_write� s ro c C s, t | ||d�}z|�|� W d|�� X dS )zN
write litle data to file and close
:param path:
:return str:
)rg N)r �
writelinesr5 )rE �linesrf rg ri r, r, r- �file_writelines� s rr c C sd t jd dkr&t jd dkr&| �|�S | �|d�}t|�dkrR|d ||d f}n|d ddf}|S )z�
str.partition for all python versions
:param s: string to parse
:param sep: separator to split by
:return: cortege - see str.partition function for python 2.5+
r rW r � r3 )�sys� version_info� partition�split�len)�sr^ Zs_partsZ ret_corteger, r, r- r\ � s
r\ c C s^ yt |d�}W n tk
r" dS X d}t�| �}x |D ]}|�|�r8|d7 }q8W |�� |S )z"
find regex count in file
re Nr r )r r �re�compile�searchr5 )�patternrE r8 ZcounterZregexrn r, r, r- �grep s
r~ c C s ddl m} || �S )z�
Determines is user should be skipped in selectorctl work
:param username: user name to check
:return: True - user should be skipped, False - not
r )�is_admin)�clcommon.cpapir )�usernamer r, r, r- �user_should_be_skipped s r� c
C sb t j�| �s^yt| d� W nB ttfk
r\ } z td| dt|�� t� d� W d d }~X Y nX d S )Ni� zError: failed to create�:r )
rD rE rF r r r �printr" rt �exit)Zdest_dirr+ r, r, r- �make_dir$ s r� c C s@ yt �| � W n tk
r" Y nX t j�| �r<t�| d� d S )NT)rD �unlinkr rE rF �shutil�rmtree)rE r, r, r- �remove_file_or_dir- s r� c C s� yJt j�|�r4t �|�| krHt �|� t �| |� nt|� t �| |� W nB tk
r� } z$td|d| dt |�� t
� d� W d d }~X Y nX d S )Nz Error: failed to create symlinkz->r� r ) rD rE �islink�readlinkr� �symlinkr� r r� r" rt r� )�src�dstr+ r, r, r- � make_symlink6 s
r� c C s t j| dddd�S )N� )�,z: T)�indent�
separators� sort_keys)�json�dumps)rh r, r, r- � pretty_jsonD s r� r Fc C s� | }d}yt | |d�\}}W n� tk
rT t�|dd�| |�d�� t�d� Y np tk
r� |rhn t�|ddt d�� t�d� Y n8 t k
r� t�|dd|| f d�� t�d� Y nX ||fS ) a�
Safely resolve username and doc_root by domain,
or resolve document root by username,
or resolve document root and username by effective uid
:param pass_not_supported_panel: pass not supported panel for php selector
:param msg_format: error messages' format (json or text)
:param user: str -> name of unix user
:param domain: str -> domain of panel user
:return: tuple -> user, doc_root
N)ra �domain�ERRORz(No domain found for user: {}, domain: {})�status�messager z6NodeJs/Ruby/Python selector not supported for %s panelz%Domain %s is not owned by the user %s)
r
r r �
print_diag�formatrt r� r
r r )ra r� Z
msg_formatZpass_not_supported_panelZ result_userZresult_doc_rootr, r, r- �$safely_resolve_username_and_doc_rootI s6 r� c C sL t | |�d }x*|D ]"}t | |�d }||kr|| S qW td�|���dS )a
Transform all directories to realpath, compare and return config value
:param user: str -> unix user - owner of the config
:param directory: str -> wanted directory key
:param config: dict -> config to get value from,
should have directories as keys
r z%Config does not contain directory: {}N)rc �KeyErrorr� )ra rb �config�
abs_directory�conf_dir� abs_conf_dirr, r, r- �get_using_realpath_keys~ s
r� c C sb t | |�d }d}t|�� �}x,|D ]$}t | |�d }||kr$||= d}q$W |s^td�|���dS )a
Transform all directories to realpath, compare and remove directory key
:param user: str -> unix user - owner of the config
:param directory: str -> wanted directory key
:param config: dict -> config to remove key from,
should have directories as keys
r FTz%Config does not contain directory: {}N)rc �list�keysr� r� )ra rb r� r� Zremoved_at_least_oneZ config_keysr� r� r, r, r- �delete_using_realpath_keys� s
r� c C s t | |�d t | |�d kS )a#
Checks that two paths in user home directory are the same after
stripping symlinks
:param user: Unix user whose home directory contains both paths
:type path1: str
:type path2: str
:return: True if paths are canonically the same otherwise False
:rtype: bool
r )rc )ra �path1�path2r, r, r- �realpaths_are_equal� s
r� c
C s� i }|s i }t | |g| }tj||tjtjdd�}|�� \}}|j} |sf| dkrfd�|�dt� d�}n|�� }||d< | |d< | r�dnd |d
< |S ) z=
Run any process using the utility cagefs_enter_user
T)r r r r � z�User's process "{}" failed and didn't return anything. It could mean that user has too small limit of PMEM for running of cloudlinux-selector. Please, increase the limit to user and try run process againr> )�details�result� timestampr( r# FZfailed) �CAGEFS_ENTER_USER_BINr r r r r# r� r �strip)
ra Zpath_to_utilityZargs_for_utilityZenv_varsr� r& r1 r �_Zretcoder, r, r- �run_process_in_cagefs� s( r� )N)NNN)re N)N)rm N)rm N)NNr F)N);�
__future__r r r rS rD rY rz r rt r� r� r �typingr r r r Zclcommon.clpwdr
r� r Z clcommon.cpapi.cpapiexceptionsr r
r Zclcommon.utilsr �
clselectprintr �clselectexceptr r� r. r2 r9 r= r: rK rQ rV rc rd rj rl ro rr r\ r~ r� r� r� r� r� r� r� r� r� r� r, r, r, r- �<module> sZ
#
5