Current File : //opt/alt/python37/lib/python3.7/site-packages/clwpos/__pycache__/daemon.cpython-37.pyc
B
�4Lb i � @ sx d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
m Z m Z m
Z
d dlmZ d dlmZ d dlmZmZ d dlmZ d dlmZ d dlmZm Z m Z m Z d d l m Z m!Z!m"Z"m#Z# d dl$m%Z% d d
l&m'Z' d d l(m)Z) d d l*m+Z+ ed
dddg�Z,edd� �Z-ejd�dd�Z.eed�dd��Z/d e0e
e e0e0f d�dd�Z1G d d � d e'�Z2d d � Z3dS )!� N)�Optional�Dict�Tuple�List)�contextmanager)�
namedtuple)�Lock�Thread)�gettext)�cpusers)�get_process_pid�remove_pid_file�write_pid_file�is_litespeed_running)�is_user_redis_alive�kill_all_users_redises� reload_redis_for_user_thread�
parse_redises)�WposDaemonBase)�WPOS_DAEMON_SOCKET_FILE)�WposDaemonLockError� User_data� redis_pid�lock�last_reload_timec c s t �d�} dV t �| � dS )z,
Context manager for dropping umask
r N)�os�umask)�prev� r �>/opt/alt/python37/lib/python3.7/site-packages/clwpos/daemon.py�_umask_0( s
r )�returnc C s4 t � �$ t�tj�} | �t� | �� | S Q R X dS )zr
Create world-writable socket in given sock_location
or reuse existing one
:return: socket object
N)r �socketZAF_UNIXZbindr Zlisten)Zsockobjr r r �_create_socket2 s
r# )r c c s4 | j dd�sttd�d��z
| V W d| �� X dS )z=
Non-blocking lock implementation for with statement
F)Zblockingz/Can't acquire lock. May be it already acquired.)�messageN)�acquirer �_�release)r r r r �non_blocking_lock? s
r( )�function�input_parametersc C sF |rdd� |� � D �ng }tjdd| f|�dd�}t�|j�� �d S )Nc S s g | ]\}}|� d |� ��qS )�=r )�.0�key�valuer r r �
<listcomp>M s zwhmapi1.<locals>.<listcomp>z/usr/sbin/whmapi1z
--output=jsonT)Zcapture_output�data)�items�
subprocess�run�json�loads�stdout�decode)r) r* Zinput_parameters_as_list�resultr r r �whmapi1L s r9 c s~ e Zd ZdZdZdZdZdZdZdZ eeeee gZ
egZ dZ d Z
� fd
d �Zeeed �d
d�Zeed�dd�Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd@ed!�d"d#�Ze e d$�d%d&��Z e j e eee!e" e!e ef d'�d(d)�Z#e j e"ed*�d+d,�Z$eed-�d.d/�Z%e j e"eed0�d1d2�Z&e j d3�d4d5�Z'e j e"d6�d7d8�Z(e j d3�d9d:�Z)e j d;d'�d<d=�Z*d>d?� Z+� Z,S )A�
WposDaemonzW
AccelerateWP daemon main class. Socket operations, redis process control, etc
z/var/lve/wpos_reload�reloadzget-redis-statuszget-litespeed-statusZphp_get_vhost_versionsZphp_get_installed_versions� � c s2 t � �� i | _g | _d | _d| _d | _d| _d S )Nr T)�super�__init__�_monitoring_users_dictZ_username_list_to_process�_socketZ _reload_uid�_socket_thread�_socket_thread_work)�self)� __class__r r r? k s
zWposDaemon.__init__)�user_uid�is_store_last_reload_timer! c
C s� y�| j �|�}|dkr d}n|j}t�|�j}t| j||�\}}|d dkrP|S |r�| jr^|sdd}nt � � }t
|t � |d�| j |< n|| j kr�| j |= W n6 t k
r� } z| j�
| � dt| �iS d} ~ X Y nX ddiS )aP
Starts/stops redis for user and updates dict for monitoring (self._monitoring_users_dict)
:param user_uid: uid to reload redis
:param is_store_last_reload_time: True - write last reload time to user data, False - write 0
:return: dict {"result": "success"} / {"result": "error", "context": {}}
Nr8 �successr )r r r )r@ �getr �pwd�getpwuid�pw_namer �_loggerZ_is_reload_interval_enabled�timer r � Exception� exception�str)
rD rF rG ZuserdataZ
old_redis_pid�usernameZ
new_redis_pid� result_dictr �er r r �_reload_redis_for_usery s*
z!WposDaemon._reload_redis_for_user)rF r! c
C s� | j | }d}y�t|j��p t| j|�\}}}|s6dS |r>dS t�|�j}t| j|| j | j �\}}|dkrpdS t
||jdd�| j |< W dQ R X W nb t k
r� } z| j� d|j
||� W dd}~X Y n. tk
r� } z| j�|� W dd}~X Y nX dS )a"
One user monitoring actions
:param user_uid: uid to reload redis
:return: User remove flag:
True - user has stopped his redis (or user absent in system), remove it from monitoring list
False - redis still active, should be monitored again
� TFNr )r r r zSLock error: %s. User %s (uid=%d) is working with his redis, skip monitoring actions)r@ r( r r rM rJ rK rL r r r r �debugr$ rO rP ) rD rF � user_datarR �is_redis_aliveZis_user_presentr& r rT r r r �_user_monitoring� s,
zWposDaemon._user_monitoringc C sX g }x.t | j�� �D ] }| �|�}|r|�|� qW x |D ]}|| jkr:| j|= q:W dS )z,
Working redises monitoring
N)�listr@ �keysrZ �append)rD Zuser_ids_list_to_remove�user_idZ is_remover r r �_redises_monitoring� s
z WposDaemon._redises_monitoringc C s^ | j rZd| _ xLt� D ]B}yt�|�j}W n tk
r> wY nX || jkr| �|d� qW dS )zC
Process all requests - start/stop redis for users
FN)Z_reload_config_needr rJ �getpwnamZpw_uid�KeyErrorr@ rU )rD rR r^ r r r �_reload_all_users� s
z WposDaemon._reload_all_usersc C s� d}t � }| j�d|� x0|D ](}| j�|d t|d t� dd�i� q W xt| js�yD| �� | � � | jrlP t
� d� |d7 }|| j kr�| �
� d}W qL tk
r� | jjddd� Y qLX qLW d S )Nr zFound existing redises: %s� )r r r z,Cloudlinux AccelerateWP daemon general errorT)�exc_info)r rM rW r@ �updater r Z
_is_terminate�
_force_reloadrb rN �sleepZ_monitoring_intervalr_ rO rP )rD �iZexisting_redises�itemr r r � _main_cycle� s$
(
zWposDaemon._main_cyclec C s t | jd�| _| j�� dS )z;
Create and start socket processing thread
)�targetN)r �_process_socket_connectionsrB �start)rD r r r � _create_and_start_socket_thread� s z*WposDaemon._create_and_start_socket_threadc
C s� t | j�dk r"| j�d| j� dS tj�t�r6| �� y t � | _
W n< t t fk
r~ } zd}| j�|t
|�� dS d}~X Y nX | j�d� | �� t| j� | �� | �� | �� d| _| j�� | �� t| j� | j�d� dS )z+
Main work daemon function
NzCPID file %s existing. Cloudlinux AccelerateWP daemon already works?z5Can't create AccelerateWP daemon socket. Error is: %sz&Cloudlinux AccelerateWP daemon startedFz)Cloudlinux AccelerateWP daemon terminated)r �
_PID_FILENAMErM �warningr �path�existsr �_remove_socketr# rA �OSError�IOErrorrQ �infoZ_setup_signalsr r; rn rj rC rB �joinr
)rD rT r$ r r r r3 s,
zWposDaemon.runc C sD t j�| j�r@| �� yt �| j� W n ttfk
r> Y nX dS )z%
Force reload daemon
N)r rq �isfile�_WPOS_DAEMON_RELOAD_FILEr; �removert ru )rD r r r rf " s zWposDaemon._force_reloadc
C s~ y>| j �dt� | jdk r$| j�� t�t� | j �dt� W n: ttfk
rx } z| j � dtt
|�� W dd}~X Y nX dS )z-
Remove daemon's socket file
zRemoving socket %sNzSocket %s removedz!Can't remove socket %s. Error: %s) rM rv r rA �closer rz rt ru rp rQ )rD rT r r r rs / s
zWposDaemon._remove_socketF)�gracefulc
C s> t | j�}|�ry | j�d|� t�|tj� W n2 tk
rb } z| j�d||� W dd}~X Y nX d}x�||k r�yt�|d� W n$ tk
r� | j�d|� P Y nX |d7 }t �
d� qjW | j�d|� yt�|tj � W n4 tk
�r } z| j�d||� W dd}~X Y nX |�r$t | j� t
| j� | j�d� dS ) z0
Stops a working daemon process
zKilling process with PID %sz4Process with pid %s is not possible to be killed: %sNr z#Process with pid %s is finally deadrc z:Process with pid %s did not exit in timeout, sigkilling itz&Cloudlinux AccelerateWP daemon stopped)r ro rM rv r �kill�signal�SIGTERMrt rN rg �SIGKILLr r
)rD r| �timeout�pidrT rh r r r �stop< s2
"
"
zWposDaemon.stop)r! c C s dt � � t� d�S )zy
Get litespeed webserver status: running or not.
:return: Dict to send to clwpos-user via socket
rH )r8 � timestamp�status)rN r r r r r �_get_litespeed_statusd s z WposDaemon._get_litespeed_status)�client_socket_objr! c
C s� t �|�}yt�|�j}W n0 tk
rJ t �|dd|it�� d�� dS X yt �|�}W n, t j
t j t j
tj
ttfk
r� d}Y nX |dkr�t �|dd|id�� dS d |kr�t �|d
t�� d �� dS |d | jkr�t �|d t�� d �� dS |d
k�r�d|k�r t �|dt�� d �� |d }yt�|�j}W n2 tk
�rh t �|dd|it�� d�� dS X |d | jk�r�t �|d t�� d �� dS d|||dfS d|||dfS )a,
Validate socket connection. Check:
- root connections
- connection user presense
- command validity
:return: tuple(is_connection_valid, uid, username, user_request_dict, is_root_query)
is_connection_valid:
True - Socket connection valid, should be processed. uid, username and user_request_dict filled
False - invalid connection, should be skipped. uid == -1, username and user_request_dict = None
is_root_query - True - root query, do not check reload interval
zNo user uid=%(uid)d�uid)r8 �contextr� )F���NNFNz5User %(username)s sent invalid query to CLWPOS daemonrR )r8 r� �commandz0Daemon received malformed query (command absent))r8 r� z Daemon received invalid commandr z)Root request to daemon should contain uidTF)� socket_utilsZget_uid_from_socketrJ rK rL ra �(send_dict_to_socket_connection_and_closerN Z'read_unpack_response_from_socket_daemonr" �errorr� r4 �JSONDecodeError�struct�AttributeError�UnicodeDecodeError�_DAEMON_VALID_COMMANDS� _DAEMON_VALID_COMMANDS_FOR_ROOT)rD r� Z_uidrR � user_requestr r r �_validate_socket_connectionl sb
"
z&WposDaemon._validate_socket_connection)r� rR r� c C sv | j �|d�}|dkr.t�|dd|id�� nD|j�8 t| j|�\}}}|d dkr\||d< t�||� W dQ R X dS )z�
Get redis status for user (socket query)
:param client_socket_obj: Client socket connection
:param username: User name
:param uid: User uid
Nz"User %(username)s has no redis yetrR )r8 r� r8 rH r� )r@ rI r� r� r r rM )rD r� rR r� rX rY r& rS r r r � _socket_user_get_redis_status� s z(WposDaemon._socket_user_get_redis_status)r r! c C s |dkpt � � | | jkS )zv
Checks is reload available by checking time
:return: True - available, False - not available
r )rN �_MIN_ALLOWED_RELOAD_PERIOD)rD r r r r �_is_reload_available� s z WposDaemon._is_reload_available)r� rR r� �
is_root_queryc C s� | j �|�}|dkrL| �|d�}t�� |d< t�||� | j�d||� dS |j}|s�| � |�s�t�� d|| j
d�d�}t�||� dS |j �: | �|d�}t�� |d< t�||� | j�d||� W dQ R X dS )a>
Reload redis for user (socket query)
:param client_socket_obj: Client socket connection
:param username: User name
:param uid: User uid
:param is_root_query: True - reload redis by root, do not check reload interval
False - user reload, check reload interval
NTr� z<Cloudlinux AccelerateWP daemon reloaded for user %s (uid=%d)zSCan't reload redis for user %(user)s. Last reload was less than %(sec)s seconds ago)�userZsec)r� r8 r� ) r@ rI rU rN r� r� rM rW r r� r� r )rD r� rR r� r� rX rS r r r r �_socket_user_redis_reload� s& z$WposDaemon._socket_user_redis_reload)r� c C s | � � }t�||� dS )z/
Get litespeed status for user
N)r� r� r� )rD r� r8 r r r � _socket_user_litespeed_status� s z(WposDaemon._socket_user_litespeed_status)r� �accountc s2 d� fdd�t | j�d D �d�}t�||� d S )NrH c s g | ]}|d � kr|�qS )r� r )r, r. )r� r r r/ s zBWposDaemon._socket_user_php_get_vhost_versions.<locals>.<listcomp>�versions)r8 r0 )r9 �%DAEMON_PHP_GET_VHOST_VERSIONS_COMMANDr� r� )rD r� r� �responser )r� r �#_socket_user_php_get_vhost_versions� s z.WposDaemon._socket_user_php_get_vhost_versionsc C s$ dt | j�d d�}t�||� d S )NrH r� )r8 r0 )r9 �)DAEMON_PHP_GET_INSTALLED_VERSIONS_COMMANDr� r� )rD r� r� r r r �'_socket_user_php_get_installed_versions s z2WposDaemon._socket_user_php_get_installed_versionsNc C s� y�| � |�\}}}}}|s dS |d | jkr<| �|||� nn|d | jkr\| �||||� nN|d | jkrv| �|� n4|d | jkr�| �||� n|d | j kr�| �
|� W n4 t j t
jttj tfk
r� | jjddd� Y nX dS )z�
Process client's socket connection (Works in thread)
:param client_socket_obj: Client socket connection
Nr� z"Socket connection processing errorT)rd )r� � _DAEMON_GET_REDIS_STATUS_COMMANDr� �_DAEMON_RELOAD_COMMANDr� �$_DAEMON_GET_LITESPEED_STATUS_COMMANDr� r� r� r� r� r" r� r4 r� r� r� r� rM rP )rD r� Zis_connection_validr� rR r� r� r r r � _handle_client_socket_connection s z+WposDaemon._handle_client_socket_connectionc C s� | j �d� x�| jr�t�| jgg g d�\}}}x\|D ]T}y(|�� \}}t| j|fd�}|�� W q4 t j
k
r� | j j ddd� Y q4X q4W qW | j �d� dS ) z>
Process socket connections (works in thread)
z4Cloudlinux AccelerateWP daemon socket thread startedrc )rk �argszSocket connection errorT)rd z4Cloudlinux AccelerateWP daemon socket thread stoppedN) rM rv rC �selectrA Zacceptr r� rm r" r� rP )rD �readabler& Z _sock_objectr� �tr r r rl + s
z&WposDaemon._process_socket_connections)Fr< )-�__name__�
__module__� __qualname__�__doc__ry r� r� r� r� r� r� r� r� Z_REDIS_SOCKET_WAIT_TIMEOUTr? �int�boolr rU rZ r_ rb rj rn r3 rf rs r� � staticmethod�dictr� r" r r rQ r� r� r� r� r� r� r� r� rl �
__classcell__r r )rE r r: T sL '#"
(J#
r: c C sf t � }| jdkr|�� | jdkr8|jdd� |�� n*| jdkrP|�| j� n| jdkrb|�� d S )Nrm ZrestartF)r| r� r; )r: �actionr3 r� r| r; )Z_optsZdaemonr r r � daemon_main< s
r� )N)4r r~ r2 rN r4 r" r� rJ r� �typingr r r r �
contextlibr � collectionsr Z threadingr r Zclwposr
r& Zclcommon.cpapir Zclcommon.utilsr r
r r Zclwpos.daemon_redis_libr r r r Zclwpos.socket_utilsr� Zclwpos.daemon_baser Zclwpos.constantsr Zclwpos.cl_wpos_exceptionsr r r r# r( rQ r9 r: r� r r r r �<module> s>
k
Mini Shell