,IVsa d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
d dl m Z dZ
dZ ee Z G d d Ze
d efd
ZdS ) N)Iterator)contextmanager)BinaryIOz/var/lve/snapshotsz .snapshotc e Zd ZdZd dededdfdZdedefd Ze d
Z
deddfd Z de e fd Z
d
edz dedz de e fdZd
edz dedz dee fdZdedefdZde fdZdefdZdedefdZdedefdZdeddfdZe dededz fd Ze d!d e e d
edz dedz de e fd Ze d edefd ZdS )"Snapshotaj
Class responsible for loading and saving snapshot files for the interval.
The files will be saved in a format of: `/var/lve/snapshots/[uid]/dump_time.snapshot`
dump_time is the timestamp/integer.
The directories `/var/lve/snapshots/[uid]` and the dump files themselves will be owned by user.
They will not be readable by other users.
incident
compresslevelreturnNc T || _ || _ t j d | _ d S )Nz lib-snapshot)r
r logging getLoggerlog)selfr r
s /builddir/build/BUILDROOT/lve-stats-4.3.2-1.el9.x86_64/opt/cloudlinux/venv/lib/python3.11/site-packages/lvestats/lib/snapshot.py__init__zSnapshot.__init__# s' *
$^44 datac |d }|J | | t j t j | | j }| | 5 }| | | j
d|j d d d n # 1 swxY w Y |j S )N dump_timezSnapshot dumped to file %s) _replace_unicode_datazlibcompressjsondumpsencoder
create_filewriter debugname)r r dump_datejson_compressedfs r savez
Snapshot.save( s % $$$
""4((( -
4(8(8(?(?(A(A4CUVV
i
(
( AA
GGO $ $ $ HNN7 @ @ @ A A A A A A A A A A A A A A A v
s 56B77B;>B;c \ t | t r| dd S | S )Nzutf-8replace)
isinstancebytesdecode)objs r _to_unicodezSnapshot._to_unicode4 s- c5 ! ! 2::gy11 1
r c Z g }| dg D ]7}| t t | j | 8||d< g }| dg D ]7}| t t | j | 8||d< d S )Nsnap_sql snap_http)getappendlistmapr+ )r r u_queriesqueryu_urlshttps r r z Snapshot._replace_unicode_data: s XXj"-- A AE T#d&6">"> ? ? @ @ @ @$ZHH["-- = =D MM$s4#3T : :;; < < < < "[r c | }t j | rt j | S g S N)get_dirospathisdirlistdir)r dir_s r
get_file_listzSnapshot.get_file_listE s: ||~~
7== $:d## # r from_tsto_tsc T | | || S )a
Return ordered list of timestamps when snapshots for this user were created.
:param from_ts: Start timestamp for filtering snapshots (inclusive).
If None, starts from the earliest snapshot.
:param to_ts: End timestamp for filtering snapshots (inclusive).
If None, includes all snapshots up to the latest.
:return: List of timestamps ordered for the given period.
)snapshot_filterr? )r r@ rA s r get_ts_listzSnapshot.get_ts_listK s( ##D$6$6$8$8'5IIIr c
# | K | || D ]!} | | | }t j sHt | j d 5 | | }ddd n # 1 swxY w Y n| | }t j | }|V # t t f$ rZ}| j
d| | | t | Y d}~d}~ww xY wdS )a
Get snapshots for a given period.
Yields one snapshot at a time for memory efficiency.
:param from_ts: Start timestamp for filtering snapshots (inclusive).
If None, starts from the earliest snapshot.
:param to_ts: End timestamp for filtering snapshots (inclusive).
If None, includes all snapshots up to the latest.
:return: Iterator of snapshot dictionaries.
uidNzCorrupted file: %s (%s))rD
get_file_name
ts_to_namer: geteuiddrop_privilegesr read_file_contentr loads
ValueErrorOSErrorr warningstr)r r@ rA tsfilenamecontent
snapshot_dataves r
get_snapshotszSnapshot.get_snapshots[ s ""7E22 B
--doob.A.ABB z|| ?( u)=>> C C"&"8"8"B"BC C C C C C C C C C C C C C C # 4 4X > >G $
7 3 3
##### (
-&&tr':':;;GG
# s= AC2B CB CB 1CD9 AD44D9rR c t |d 5 }| }d d d n # 1 swxY w Y t j | }nl# t j $ rZ t j || j }t |d 5 }| | d d d n # 1 swxY w Y Y nw xY w| S )Nrbwb) openreadr
decompresserrorr r
r r) )r rR r# rS compressed_contents r rK zSnapshot.read_file_content sB
(D
!
! QffhhG ,og..GGz , , ,!% w8J!K!K h%%
,*+++
,
,
,
,
,
,
,
,
,
,
,
,
,
,
, , ~~ sA 266A 9B< B." B<.B2 2B<5B2 6B<;B<c t t | | j d | j d S )zZ
Load all snapshots for given incident
:return: list of snapshots
incident_start_timeincident_end_time)r1 rV r r s r get_incident_snapshotsz Snapshot.get_incident_snapshots s2
D&&t}5J'KT][nMoppqqqr c | t j t t | j d S )NrF )r: r; join
SNAPSHOT_PATHrP r rb s r r9 zSnapshot.get_dir s' w||M3t}U/C+D+DEEEr r c f t j | | S r8 )r: r; re r9 )r r s r rG zSnapshot.get_file_name s w||DLLNND111r r! c * | }t j | s t j | t j |d t j || j d d n@# t $ r3}| j
d|t | Y d}~nd}~ww xY w| |
| }t | j d 5 t |d } t j | d n@# t $ r3}| j
d|t | Y d}~nd}~ww xY wddd n # 1 swxY w Y |S ) z
create file, change its ownership & permissions if needed. Create directories if needed as well
:param dump_date: int timestamp used as file name
:return: open File object
i rF r z Unable to create dir %s (%s)NrY z&Unable to set file permissions %s (%s))r9 r: r; existsmakedirschmodchownr rN r r] rP rG rH rJ rZ fchmodfileno)r r! r> e file_namefile_s r r zSnapshot.create_file s ||~~w~~d## M
M D!!! u%%%t}U 3Q7777
M
M
M =tSVVLLLLLLLL
M&&ty'A'ABB T]5 1
2
2 \ \D))E
\ %,,..%0000
\
\
\ GTWXYTZTZ[[[[[[[[
\ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ sT A
B
B=
)B88B=F'D<;F<
E9)E4/F4E99FF F c h | }t j | }| | }| || }||k rt j |d dS |D ]<}t j | | | =dS )z
Delete old snapshots. If there are no more
:param to_ts: up to which timestamp to remove snapshots
:return: None
)rA T)
ignore_errorsN) r9 r: r= rC shutilrmtreeremoverG rH )r rA _dirfiles
all_snapshots ts_to_removerQ s r
delete_oldzSnapshot.delete_old s ||~~
4 ,,U33
++E+?? L ( ( M$d 3 3 3 3 3 3"
C
C $,,T__R-@-@AABBBB
C
Cr rr c | t rD| dt }| r t | S # t
$ r Y nw xY wd S Nr )endswith SNAPSHOT_EXTSNAPSHOT_EXT_LENisdigitintrM )rr rQ s r get_tszSnapshot.get_ts sn >>, ' ' q***+Bzz||
r77N! Dts A
A A ry c |d}| t j }g }| D ]C}t | }|%||cxk r|k rn .| | Dt | S r~ )sysmaxsizer r r0 sorted)ry r@ rA resultrR rQ s r rC zSnapshot.snapshot_filter s ?G =KE " "H **B~'R"8"8"8"85"8"8"8"8"8
b!!!f~~ r rQ c 0 t | t z S r8 )rP r )rQ s r rH zSnapshot.ts_to_name s 2ww %%r )r )NN)__name__
__module__ __qualname____doc__dictr r rP r$ staticmethodr+ r r1 r? rD r rV rK rc r9 rG r r r| r rC rH r r r r s 5 5 5c 5$ 5 5 5 5
#
\
#$ #4 # # # #tCy JtJ TzJ
c J J J J &t& Tz&
$ & & & &P # # r r r r rF F F F F2# 2# 2 2 2 2 S X >C C C C C C c cDj \ # Cy t Tz
c \ &s &s & & & \& & &r r rF c # , K t j t j t j }}}t j d d }t j g t j | t j | d V t j | t j | t j | d S # t j | t j | t j | w xY w)Nnobody ) r: getuidgetgid getgroupspwdgetpwnam setgroupssetegidseteuid)rF old_uidold_gid
old_groupsgids r rJ rJ s #%9;; R\^^jWG
,x
#CLJsOOOJsOOO!
7
7
Z
7
7
Z s C >D)r r
r: r ru r r collections.abcr
contextlibr typingr rf r lenr r r rJ r r r r s
$ $ $ $ $ $ % % % % % % $
3|$$ Q& Q& Q& Q& Q& Q& Q& Q&h ! ! ! ! ! ! !r