3
�6�gS" � @ sP d dl Z d dlZd dlZd dlZd dlZd dlZd dlT d dlmZ dZd Z dZ
dZ dZ dZ
eje Zde_d e_d e_d
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zd>dd�Zd?dd �Z d d � Z d d � Z d!d"� Z d#d$� Z d%d&� Z!d'd(� Z"d)d*� Z#d+d*� Z#d,d-� Z$d.d/� Z%d0d1� Z&d2d3� Z'd4d5� Z(d6d7� Z)d8d9� Z*d:d;� Z+d<d=� Z,dS )@� N)�*)�PIPEz0.0.2� � � � z/usr/local/lswsc C s t jddd� d S )Nz%(asctime)s.%(msecs)03dz [%(levelname)s] %(message)sz%Y-%m-%d %H:%M:%S)�formatZdatefmtz3%(asctime)s.%(msecs)03d [%(levelname)s] %(message)s)�loggingZ basicConfig� r
r
�"/usr/local/lsws/lsns/bin/common.py� init_logging s r c C s t jS )N)�this�
serverRootr
r
r
r � server_root s r c C s
| t _d S )N)r
r )�rootr
r
r �set_server_root s r c C s t � d | S )Nz /lsns/conf/)r )�fileonlyr
r
r �
get_conf_file s r c C s t � d | S )Nz
/lsns/bin/)r )r r
r
r � get_bin_file s r c C s t j| � tjd� d S )Nr )r �error�sys�exit)�msgr
r
r � fatal_error" s
r c C s dddddgS )N�cpu�io�iops�mem�tasksr
r
r
r
r � get_options& s r Fc C s | j � rdytjt| ��}W q� tk
r` } z*|s@td| |f � tjd| |f � dS d }~X q�X nVytj| �}W nF tk
r� } z*|s�td| |f � tjd| |f � dS d }~X nX |j t
� k � r�|s�td|j t
� f � tjd|j t
� f � dS |dfS ) Nz Error getting UID for %s: %sFz Error getting name for %s: %sz#Specified uid: %d < minimum uid: %dT)NF)NF)NF) �isdigit�pwd�getpwuid�int� Exceptionr r �debug�getpwnam�pw_uid� get_min_uid)Zparam�no_fatal� user_info�errr
r
r �get_user) s* r, c C s2 g }x(| D ] }t ||�\}}|r
|j|� q
W |S )N)r, �append)�uidsr) �usersZuidr* Zgotr
r
r � get_usersA s
r0 c C s t jjd�rdS dS )Nz/etc/plesk-releaseTF)�os�path�existsr
r
r
r � get_pleskI s r4 c C s t � r
dS dS )Ni' i� )r4 r
r
r
r �get_def_min_uidN s r5 c C s t jdkrt jS td�} yt| d�}W nP tk
rv } z4t jd kr^tjd| |t� f � dt _t� t _t jS d }~X nX y |j � }W nR tk
r� } z6t jd kr�tjd| |t� f � dt _t
t� �}W Y d d }~X nX |j � t |�t _tj
dt j � t jS )Nr z lsns.conf�rz8Error opening %s: %s, continuing with default min uid %dTz8Error reading %s: %s, continuing with default min uid %dzUsing min uid: %d)r
�min_uidr �openr$ �loggedr �infor5 �readline�str�closer# r% )Zfullfile�fr+ Zuidstrr
r
r r( S s,
r( c C s
t � d S )Nz/lsns/conf/lscntr.txt)r r
r
r
r �container_filek s r? c C s
t � d S )Nz/conf/httpd_config.conf)r r
r
r
r �
ols_conf_filen s r@ c C s
t � d S )Nz/conf/httpd_config.xml)r r
r
r
r �lsws_conf_fileq s rA c C s
t � d S )Nz /lsns/conf/ns_disabled_uids.conf)r r
r
r
r �get_disabled_uid_filet s rB c C s
t � d S )Nz /lsns/conf/ns_disabled_uids.conf)r r
r
r
r rB w s c C s
t � d S )Nz/lsns/conf/packages)r r
r
r
r � get_pkg_dirz s rC c C sB t jjt� �s4t jjt� �s$td� t jt� dd� t� d| S )Nz3Missing LiteSpeed high level installation directoryi� )�modez/%s.conf)r1 r2 �isdirrC r r �mkdir)�pkgr
r
r �pkg_to_filename} s
rH c C s
t � d S )Nz"/lsns/conf/redis_package_size.conf)r r
r
r
r �get_package_size_file� s rI c C s t jt� d�dkrtd� d S )Nr z5You must configure LiteSpeed for LiteSpeed Containers)r1 �accessr? r r
r
r
r �ls_ok� s rK c C sX t jd|| f � tjd| gttd�}|jdkrTtd|jjd� d |j jd� � d S )Nz restart_external %s by touch: %sZtouch)�stdout�stderrr z!Error in running: touch, errors: zutf-8� )
r r% �
subprocess�runr �
returncoder rL �decoderM )�fileZdesc�resultr
r
r �touch_restart_external� s
rU c C s
i }y�t jjt� �dkr�tt� d�}tj|�}|j� x`|d D ]T}t jj|�}x"| D ]}|j |krV|||j
< P qVW t jj |�s�t j |� t
|d� q@W |r�t
dd� n*x(| D ] }|j
|kr�q�t
|j d d� q�W W n0 tk
�r } ztd | � W Y d d }~X nX d S )
Nr r6 Z
reset_listz
in lscntr.txtz,/usr/local/lsws/admin/tmp/.lsphp_restart.txtzfor allz/.lsphp_restart.txtzfor homezError managing restart: %s)r1 r2 �getsizer? r8 �json�loadr= �dirname�pw_dir�pw_namer3 rF rU r$ r ) r/ �allZ
users_usedr> �datarS Zdironly�userr+ r
r
r �restart_external� s.
r_ c C s� i } i }t jd�}g }x�|D ]�}t|�dkr<|d d� dks t|�dkrX|d d� dks t|�dkr�|d d� dks |d d� dkr�q d | }t j|�}t|j�d
kr tjd | d t|� � |j |� |||< q W xn|D ]f}d
}x\|�r@d}xL|D ]D} || k�rq�t|�t| �k r�|| d t|�� kr�|j
| � d
}P q�W q�W q�W xL|D ]D}t j || j �}
t j
|| j �} |
| k�r�i | |
< || |
| < �qLW tjdt|� d t| � � || fS )Nz/devr Zloop� Zcdromr Zdm�srz/dev/r z
filename: z statinfo: TFzFinal devices: z Block Devices: )r1 �listdir�len�stat�S_ISBLK�st_moder r% r< r- �remove�major�st_rdev�minor) Z
block_devicesZ statinfo_dev�filesZdevicesrS �filenameZstatinfoZdeviceZretry�innerrh rj r
r
r � get_devices� s@
d
$
rn c C s� | dkr dS t | �}|dkr&d}d}n:|dkr8d}d}n(|dkrJd}d}n|dkr\d}d }n| S t|| �d
k r�d || tt||d
d
��|f }nd || |f }tt|��| kr�|S tjd
tt|��| | f � | S )N� l �Ti @�Gi �Mi �K�
z%d.%d%sz%d%sz%s != %s use %s)�float�roundr# r< �int_num_valuesr r% )�valstr�valZdivisorZ qualifierZstr_numr
r
r �str_num_values� s. ( rz c C s$ | dks| dkrdS | j � r$t| �S t| �dk r<td| � | t| �d j� }|j� sbd}| }n| d t| �d � }y t|�}W n tk
r� td| � Y nX |dkr�d}nD|d kr�d
}n6|d kr�d }n(|d
kr�d}n|dkr�d}n td| � tj d| ||t|| �f � || }t|�S )Nro z-1r r zSInvalid value specification: %s, must be number optionally followed by T, G, M or Kz\Invalid value specification (prefix): %s, must be number optionally followed by T, G, M or Krp l rq i @rr i rs i z\Invalid value specification (suffix): %s, must be number optionally followed by T, G, M or Kzint_num %s, %f * %f = %d���)
r r# rc r �upper�isalpharu r$ r r% )rx �suffix�prefixZprefZ
multiplierZflr
r
r rw � s: rw )F)F)-rW r r1 r! rO r rd r �VERSIONZ
OPTION_CPUZ OPTION_IOZ OPTION_IOPSZ
OPTION_MEMZ OPTION_TASKS�modules�__name__r
r9 r r7 r r r r r r r r, r0 r4 r5 r( r? r@ rA rB rC rH rI rK rU r_ rn rz rw r
r
r
r �<module> sL 0
" |