ZSs} d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
m Z m Z d dl
mZ d dlmZ d dlmZ ej d Ze G d d Z G d
d ZdS ) N) dataclass)OptionalDictList)drop_user_privileges) socket_utils)run_in_cagefs_if_neededz^[1-9][0-9]*\Zc e Zd ZU eed< eed< eed< eed< eed< eed< dZe e ed< dZ
e e ed < dS )
EnableFeatureTaskuiddomainwp_pathfeature
ignore_errorstimeout_timestampN advice_idpid) __name__
__module__ __qualname__int__annotations__strlistboolr r r (py/clwpos/daemon_subscription_handler.pyr r s HHH
KKK
LLL MMM $Ix}### C# r r c e Zd ZdZdZdZdZddZd Zd Z d Z
dd
e j d e j fd Z d
e j d
e
defdZd
e j dedefdZdS )PendingSubscriptionWatcherzr
Listen to set-suite and user events in order to automatically
enable module when suite is purchased.
Nc l t | _ |pt j t | _ d S N)dict_pending_enable_taskslogging getLoggerr _logger)selfloggers r __init__z#PendingSubscriptionWatcher.__init__: s- JN" <!28!>>F L Df M M M M, U U U L " " $@AB18
U
U
U
U
U
U
U
U O O O L " "#M N N N N N O =#((4..99999s ,A C,&B'C Cc | |j r-| dddt |j g d S |j rl|j D ]b}| j d| ddd|d|j d |j g}|j r| d
| | cd S t j d d S ) Nz&/opt/alt/php-xray/cl-smart-advice-userapplyz --advice_idz*activate feature in background: feature=%szcloudlinux-awp-userenablez --featurez--domainz --wp-pathz--ignore-errorsz;Task does not have any advice or domain specified, skipping) r rH r r
r r* r6 r r appendr( )r+ taskfcmds r _run_enable_commandz.PendingSubscriptionWatcher._run_enable_commanda s > X % %8' s4> 2 2'
[
X\
/
/ !!"NPQRRR)8 % 2JJ 0111))#....
/
/
LV W W W W Wr client_socket_obj
daemon_socketc | | t | j D ]\ }}t | D ]}t j }|rh| j d| ||_ t j |d | j d| |
| || | ddl m
} t t j |j j dd ||j fd|j D }|D ] } | j d| fd |j D |_ |j s-t j d t j d d S | | t j d # t j d w xY w|| j v r| j | s| j |= d
d i}
t- j ||
d S ) Nz!background process forked: pid=%dr z#background process finished: pid=%d)get_allowed_modulesT)effective_or_realset_envc g | ]}|v| S r r .0rN allowed_featuress r
zEPendingSubscriptionWatcher.suite_allowed_callback.. s$ %[%[%[AJZAZAZaAZAZAZr zCunable to activate feature in background: feature=%s is not allowedc g | ]}|v | S r r rX s r r[ zEPendingSubscriptionWatcher.suite_allowed_callback.. s$ 'Y'Y'Ya1HXCXCXCXCXCXr resultsuccess)r9 r r' r/ r1 forkr* r6 r waitpidr4 closeclwpos.feature_suitesrT r pwdgetpwuidr pw_namer _exitrP r (send_dict_to_socket_connection_and_close) r+ rQ rR r tasksrM fprT
disallowedrN responserZ s @r suite_allowed_callbackz1PendingSubscriptionWatcher.suite_allowed_callbackv so ##%%% t 9 ? ? A ABB $ 4 $ 4JC
U
$
$WYY $L%%&I2NNN !DHJr1%%%L%%&KRPPPLL&&&&$(4 ) / / 1 1 1)//111MMMMMM,S\$(-C-C-K?CTS S S S+>+>tx+H+H(%[%[%[%[ %[%[%[
!+ x xA L - -.suv w w w w'Y'Y'Y'Y4<'Y'Y'Y #| # HQKKK " 00666 d0009STW9X0.s3
i
=>OQYZZZZZs CG*?G**H user_requestr c T | ddlm}m} ddlm} d |D }|d d }|D ]&} | |vr dd| z d }
t j ||
d S '| d
} | Et | } t | s dd d }
t j ||
d S |d } | 4t | | j k r dd
d }
t j ||
d S | r3d| d v r ddd }
t j ||
d S |d }
t |
| d uk r ddd }
t j ||
d S |
r@ || }d || D }|
|vr ddd }
t j ||
d S || j vr
g | j |< | j | }|D ]u}|j |
k rh|j | k r]t' |j t' | k r8|j | k r-|j |d k r ddi}
t j ||
d S vt | | j k r ddd }
t j ||
d S | j | t3 ||
| ||d | t5 t7 j | j z | j d| j | ddi}
t j ||
d S )Nr )get_main_username_by_uid userdomains)ALL_OPTIMIZATION_FEATURESc 6 h | ]}| S r )to_interface_name)rY rN s r zFPendingSubscriptionWatcher.add_pending_upgrade_task.. s$ SSSA!--//SSSr r ,errorzunknown feature: %s)r] contextr zinvalid advice_id formatr zwp_path is too longz../z1invalid wp_path: must not contain '..' componentsr
z'domain and wp_path must be set togetherc h | ]\ }}|S r r )rY d_s r rt zFPendingSubscriptionWatcher.add_pending_upgrade_task.. s @@@$!QA@@@r z"domain does not belong to the userr r] r^ ztoo many pending tasks)r r
r r r r r z7Successfully added pending upgrade subscription task %s) r9 clcommon.cpapiro rp %clwpos.optimization_features.featuresrq splitr rg getr
_ADVICE_ID_REmatchlen_MAX_WP_PATH_LENGTHr r' r
r setr r r _MAX_PENDING_TASKS_PER_UIDrL r r r0 _FEATURE_STATUS_CHECK_TIMEOUTr* r6 )r+ rQ rm r ro rp rq known_featuresrequested_featuresrN rk r r r
username user_domainspendingexistings r add_pending_upgrade_taskz3PendingSubscriptionWatcher.add_pending_upgrade_task s ##%%%HHHHHHHHSSSSSSSS9RSSS )) 4 : :3 ? ?# A&&,3@UXY@Y!Z!Z EFWYabbb '
!$$[11 II &&y11
,3@Z![![ EFWYabbb y) 3w<<$2J#J#J(/OQYZZZZZr r c | | j |g }dt t fd|D d}t j || d S )Nr^ c 3 * K | ]
}|j v V d S r% )r )rY r8 r s r zEPendingSubscriptionWatcher.get_upgrade_task_status.. s+ b bL<+? ? b b b b b br )r] r )r9 r' r r anyr rg )r+ rQ r r r7 rk s ` r get_upgrade_task_statusz2PendingSubscriptionWatcher.get_upgrade_task_status s ##%%% 266sB??
C b b b bTa b b b b bcc
=>OQYZZZZZr r% )r r r __doc__r r r r- r9 rH rP socketrl r& r r r r r r r r r . s$ %(!!# = = = =i i i(
:
:
:X X X, ?C-[ -[
-[.4m-[ -[ -[ -[^U[&- U[W[ U[be U[ U[ U[ U[n[ [S [[^ [ [ [ [ [ [r r )r( r1 rc rer r> r0 dataclassesr typingr r r clcommon.clpwdr clwposr clwpos.utilsr compiler r r r r r r s9
! ! ! ! ! ! ' ' ' ' ' ' ' ' ' ' / / / / / / 0 0 0 0 0 0
,--
4V[ V[ V[ V[ V[ V[ V[ V[ V[ V[r