@jB Z d dl Z d dlZd dlmZ d dlmZmZmZmZ d dl Z
d dl m Z d dl
mZ d dlmZ d dlmZ d dlmZ d d lmZ d d
lmZ d d lmZmZ d d l m Z d d
l m Z m Z m!Z! d dl"m#Z# d dl$m%Z%m&Z&m'Z' d dl(m)Z) dZ*dZ+dZ,dZ-dZ. G d de/ Z0 G d de/ Z1 G d de Z2dS ) Nchoice)DictListOptionalTuple) sessionmaker)ClPwd) NotSupported)NoDBAccessData)ExternalProgramFailed)LveStatsPlugin) proctitle)
get_chunks reboot_lock) HtpasswdFile) LiteSpeedLiteSpeedExceptionLiteSpeedInvalidCredentials)Snapshot)MySQLOperationalError SQLSnapshot get_lvepsincidenti,
apache litespeedc B e Zd Zd Zd Zd Zd Zed Zd Z dS )LitespeedHelperc ~ d| _ d| _ d | _ d| _ | | _ d| _ d S )NFzlve-stats-admin)
is_running
state_changedforce_litespeedlogingenerate_random_passwordpassword
broken_configselfs /builddir/build/BUILDROOT/lve-stats-5.0.5-1.el8.x86_64/opt/cloudlinux/venv/lib/python3.11/site-packages/lvestats/plugins/generic/snapshot_saver.py__init__zLitespeedHelper.__init__% sE " $ '
5577
" c Z t j }|| j k | _ || _ dS )zCheck litespeed stateN)r is_litespeed_runningr# r$ )r+ litespeed_runnings r, check_litespeed_statez%LitespeedHelper.check_litespeed_state5 s+ % : < < .$/ A+r. c : t t j }n7# t $ r* d| _ t j dt j Y d S w xY w| | j | j |
t j d d S )NTz8Can't change the password. Please, check the file:
'%s'z Password successfully changed.) r r
HTPASSWD_PATH
ValueErrorr) loggingwarningupdater& r( savedebug)r+
passwdfiles r, dump_passwdzLitespeedHelper.dump_passwd; s %i&=>>JJ !%D O WYbYp q q q FF $*dm444
677777s 0AAc t | j | j } | | S # t $ r | w xY w)zGet user data proxy method)r r& r(
get_user_datar r< )r+ usernamer s r, r> z LitespeedHelper.get_user_dataF s` dj$-88 **844 4* s 1 Ac t j t j z dz d fdt d D S )Nz
!@#$%^&*() c 3 6 K | ]}t V d S Nr ).0_charss r, z;LitespeedHelper.generate_random_password..T s) 88ve}}888888r. )string
ascii_lettersdigitsjoinrange)rF s @r, r' z(LitespeedHelper.generate_random_passwordP sF $v}4|Cww8888eBii888888r. c B | j }| j rdS | j | j S |S )z)Get what we must use: litespeed or apacheF)r# r) r% )r+ r0 s r, get_use_litespeedz!LitespeedHelper.get_use_litespeedV s4 $ (5
!
-' '##r. N)
__name__
__module__ __qualname__r- r2 r< r> staticmethodr' rO r. r, r! r! $ sz # # # , , , 8 8 8 9 9 \9
$ $ $ $ $r. r! c R e Zd ZdZi ZdZ e Zd Zd Z d Z
d Z d Z d Z
dS )SnapshotHelperNc p d | _ d| _ d | _ d | _ t j d | _ d S )NF
SnapshotSaver)_mysql_snapshots_enabledlitespeed_died_sql_snapshot_supported_no_operational_errorr6 getLoggerlogr* s r, r- zSnapshotHelper.__init__h s8 (,% #'+$%)"$_55r. c || _ d S rC )rY )r+ mysql_snapshots_enableds r,
set_configzSnapshotHelper.set_configo s (?%%%r. c f | j | S # t j $ r g cY S w xY wrC )clpwd get_namesr
NoSuchUserException)r+ lve_ids r, rd zSnapshotHelper.get_namesr sD :''// /( III s 00c V | j |i di }g }g }| | D ]}|| j |g z
}| }|r} || | z
}P# t $ rV}| j s t j d| d| _ |t j | z
}Y d }~d }~ww xY w|t j | z
}|||fS )NTIDz3Error during getting information from Litespeed: %sT) psgetrd username_dbquery_maprO r> r rZ r6 r7 r Proctitle) r+ rf litespeed_info processesqueriesurlsr?
use_litespeedes r, get_snapshot_dataz SnapshotHelper.get_snapshot_datay sL GKK++//r:: v..
F
FH t044XrBB BG * < < > >M
FJN 8 8 B BBDD) J J J . 3 (]_`aaa.2+I / 1 1 ? ? I IIDDDDDD J +--;;HEEE'4''s B
C9#A C44C9c H t | _ g }|D ]'}|| t | z
}(| | | _ t
| _ d S # t $ r'}| j
d| Y d }~d S d }~ww xY w)Nz.An error occurred while getting processes listexc_info) r
rc rd intretrieve_queriesrk r ri r
r^ r7 )r+ lve_id_list
all_usernamesrf rr s r,
invalidatezSnapshotHelper.invalidate s WW
! 9 9F T^^CKK 8 8 8MM ! !- 0 0
! [ kkDGGG$ [ [ [ H MXY Z Z Z Z Z Z Z Z Z [s A0 0
B!: B B!c | j si S | j 5 }| | }| j s!d| _ | j d d| _ |cd d d S # 1 swxY w Y n# t $ r6}| j dv r#d| _ | j d| Y d }~nMd }~wt t f$ r6}| j dv r#d| _ | j d| Y d }~nd }~ww xY wi S )NTz)SQL snapshot is supported and operational)NTFz2An error occurred while getting MySQL process listru z3SQL snapshot is not supported by this control panel) rY sql_snaprj r[ r^ infor\ r r7 r r )r+ login_names db_requestsresultrr s r, rx z SnapshotHelper.retrieve_queries sz , I a
+$ 553 O37D0HMM"MNNN-1*
% c c c)\99-2* !U`a bbb n- a a a +|;;/4,
S^_
``` a sG A1 AA% A1 %A))A1 ,A)-A1 1
C6;,B,,C6 ,C11C6)rP rQ rR rk ri rc r r} r- ra rd rs r{ rx rT r. r, rV rV b s B E{}}H6 6 6@ @ @ ( ( (* [ [ [ r. rV c e Zd ZdZeZeZdez Z d Z
d Z d Z d Z
d Zd Zd Zd
Zd Zdd
Zd Zd Zd Zd Zd ZdS )rX rA < c i | _ t j d | _ d | _ i | _ i | _ d| _ d| _ t | _
d| _ d| _ d| _
t | _ d S )NrX Tr )incidents_last_tsr6 r] r^ session_snapshots_dataincidents_cache first_runlast_runr! rm snapshots_enabledr`
compresslevelrV _helperr* s r, r- zSnapshotSaver.__init__ sx !# $_55 ! !
- / /!% '+$ %'' r. c
N | d| j | _ t | d| j | _ t | d| j | _ dt | dt
z | _ | | | j _ |
d | dd d
k | _ | d d d
k | _
t t t | d d
d d
| _ | j | j
d S )N server_idperiod_between_incidentsmax_snapshots_per_incidentr snapshots_per_minuteT)force_webserver_messagedisable_snapshotsfalsetruedisable_mysql_snapshotsr r )rj r rw r r DEFAULT_SNAPSHOTS_PER_MINUTE_period_get_webserver_optionrm r% setup_litespeedlowerr r` maxminr r ra )r+ configs r, ra zSnapshotSaver.set_config sV K @@(+FJJ7QSWSp,q,q(r(r%*-fjj9UW[Wv.w.w*x*x'S,BD`!a!a b bb .2.H.H.P.P+ T :::!',?!I!I!O!O!Q!QU[![ '-zz2KW'U'U'['[']']ag'g$ SOQ)G)G%H%H!!L!La P P <=====r. c Z d|v r&|d }|dv rdS |dv rdS dS )z
Check which webserver we must force to use: Apache or Litespeed
:return: None if webserver autodetect
False if apache should be used
True if litespeed should be used
r )on1T)off0FN)r )r+ r options r, r z#SnapshotSaver._get_webserver_option sJ & K(..00F$$t %% utr. c | j t t j | j k t j d S )zBGenerate sqlalchemy Query instance for select incomplete incidentsN)r queryr filterr incident_end_timeis_r* s r, _incomplete_incidents_queryz)SnapshotSaver._incomplete_incidents_query sM |!!(++22 $. 0(2L2P2PQU2V2V
r. c H | t j | j | j z
k }| t j t j i }| j d| | j
d S )Nz%%i old incidents period was finalized) r r r dump_timenowr r8 r r^ r: r commit)r+ not_finalize_incidents_queryfinalized_numbs r, finalize_incidentsz SnapshotSaver.finalize_incidents s '+'G'G'I'I'P'P D,I!I I(
(
$ 6<XZbZl=mnn >OOO r. c
fdt j D }|r j t
t
j j k t
j d t
j
t |
D ]}||j
|_ j nX# t $ rK} j j dt' | Y d }~nd }~ww xY wt | D ]I} j | j | 8# t, $ r Y Dw xY wd S d S )Nc B i | ]\ }}|j j z
k || S rT )r r )rD uidtsr+ s r,
z4SnapshotSaver.save_old_incidents.. s=
R2PTPX[_[xPxKxKxCKxKxKxr. z"Unable to save old incidents: (%s))listr itemsr r r r r r r r in_keysallr Exceptionrollbackr^ errorstrpopr KeyError)r+
old_incidents_incrr r s ` r, save_old_incidentsz SnapshotSaver.save_old_incidents s
#'(>(D(D(F(F#G#G
&L&&x00V *dn< 266t<< ((m.@.@.B.B)C)CDD
SUU E ED .;48-DD** E ##%%%%
M
M
M %%''' CSVVLLLLLLLL
M
M . . 0 011
&**3///(,,S1111 D+ "
s+ CD
E/$AE**E/0G
GGc h | t j |k }|sYt || j | j dd }| j | | j
d|| j | j | j |< |S )Nr r incident_start_timer snapshot_countr z>New incident-period for uid %s started; incident_start_time=%i) r r r r firstr r r addr^ r: r )r+ r r s r, get_incidentzSnapshotSaver.get_incident s //1188 9LMMSSUU
$(H . "& D
L T " " " HNNP
'+h s# r. c | j r? t | j | _ | d| _ d S d S )N)bindF)r r enginer r r* s r, init_sessionzSnapshotSaver.init_session( sO > #9
a%6
a
_```%2> I JJJC HMM# r. c | j sd S | | t | di }| di }| j | | | | j | j
z
| j k r| j | _
t 5 |
| | | | t | D ]\ }}| || d d d n # 1 swxY w Y i |d< i | _ d S d S )N lve_usager )r r r r rj r r r{ cache_snapshotsr r r r r _insert_new_incidents_increment_snapshot_countr r r )r+ lve_datalve_ids
lve_faultsrf r s r, executezSnapshotSaver.executeU s % F x||K4499;;<< \\(B//
((( Z((( 8dm #t| 3 3 HDM
5
5'')))**:666..z:::&*:+;+;+=+=&>&> 5 5NFF$$VV44445
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5 "$HX #%D 4 3s ASQUQY]][[qv
L ! ! ! !
" "r. c fdt | D }|r j | d t | D } j | j d S # t $ r3} j
dt | Y d }~d S d }~ww xY wd S )Nc N i | ]!\ }}|j v||j j d dd"S )r Nr )r r r )rD rf rE r+ s r, r z7SnapshotSaver._insert_new_incidents..y sV
T 111
'+x !^"#%) 211r. c 3 . K | ]}t di |V d S )NrT r )rD r s r, rG z6SnapshotSaver._insert_new_incidents.. s0 UU8++d++UUUUUUr. z Unable to save new incidents: %s) r r r r8 valuesr add_allr r r^ r r )r+ r
new_incidents insert_listrr s ` r, r z#SnapshotSaver._insert_new_incidentsw s
"*"2"2"4"4 5 5
& ' '
6 6 6UU]=Q=Q=S=S8T8TUUUK
& $$[111 ##%%%%%
K
K
K A3q66JJJJJJJJJ
K & &s 7B, ,
C)6(C$$C)c t | D ]3}|| j vr(| j || j | j |< 4d S rC )r r r r rs rm )r+ r rf s r, r z SnapshotSaver.cache_snapshots sc :??,,-- k kFT 111/3|/M/MfVZVi/j/j$V, k kr. c |d }| j |i g g f \ }}}t || j }|d t | j | j |d ||||d} | | n># t $ r1} | j
t | Y d } ~ nd } ~ ww xY w|dxx dz
cc< d S )Nr r )r r r r snap_procsnap_sql snap_http snap_faultsr r ) r rj r r rw r r r9 IOErrorr^ r r )
r+ _incidentr rf rn ro rp snapshot_datarr s
r, r zSnapshotSaver.save_snapshot s 5!#'#7#;#;FRRL#Q#Q 7D Y(:;; U# TX #,-B#C" !
# NN4 # # # HNN3q66 " " " " " " " " #"###q(#####s /B
C 'B;;C N)F)rP rQ rR r DEFAULT_PERIOD_BETWEEN_INCIDENTSr "DEFAULT_MAX_SNAPSHOTS_PER_INCIDENTr r r r- ra r r r r r r r r r r r r r rT r. r, rX rX s I ? !C 00G ( ( ( > > > "
: (# # # S S S. & & &. " " "& & &4k k k ) ) ) ) )r. rX )3r6 rI secretsr typingr r r r sqlalchemy.orm.session
sqlalchemysqlalchemy.ormr clcommon.clpwdr
clcommon.cpapir clcommon.cpapi.cpapiexceptionsr clcommon.utilsr
lvestats.core.pluginr lvestats.lib.commonsr lvestats.lib.commons.funcr r lvestats.lib.commons.htpasswdr lvestats.lib.commons.litespeedr r r lvestats.lib.snapshotr lvestats.lib.ustater r r lvestats.orm.incidentr r r r r r objectr! rV rX rT r. r, r s
. . . . . . . . . . . . ' ' ' ' ' ' ' ' ' ' ' ' 9 9 9 9 9 9 0 0 0 0 0 0 / / / / / / * * * * * * = = = = = = = = 6 6 6 6 6 6 e e e e e e e e e e * * * * * * M M M M M M M M M M * * * * * *#& %' " ;$ ;$ ;$ ;$ ;$f ;$ ;$ ;$|O O O O OV O O Odv) v) v) v) v)N v) v) v) v) v)r.