cT ` d dl Z d dlZd dlZd dlZd dlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
ddlmZ dd lmZ d
Zej ej d e Z ej e Z G d d
Z G d d ZddZdeddfdZ deddfdZ dS ) N)TemporaryDirectory )helpers)plesk)
FtpBackupBase)Flags) safe_move)scanz.restore-infected~c e Zd Zd Zd Zd ZdS )FileDatac R t j | | _ t j | j rPt j | j }|j | _ t j
|j | _ d S d | _ d | _ d S N)
ospathabspathfilenameexistsstatst_sizesizer DateTime
fromtimestampst_mtimemtime)selfr r s M/opt/imunify360/venv/lib/python3.11/site-packages/restore_infected/restore.py__init__zFileData.__init__ sy 11
7>>$- ( ( 74=))D DI )77
FFDJJJ DI DJJJ c V | j | j | j f|j |j |j fk S r )r r r )r others r __eq__zFileData.__eq__ s0 4:t}5 J K N:
r c L d | j j | j S )Nz
<{0}({1})>)format __class____name__r r s r __repr__zFileData.__repr__& s ""4>#:DMJJJr N)r&
__module__ __qualname__r r" r( r r r
r
sF
K K K K Kr r
c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS ) InfectedListc D d |D | _ || _ || _ d S )Nc , g | ]}t | S r+ )r
.0fs r
z)InfectedList.__init__.., s 111ahqkk111r )files restore_dir scan_func)r r4 r5 r6 s r r zInfectedList.__init__+ s* 115111
&"r c * t | j S r )boolr4 r' s r __bool__zInfectedList.__bool__0 DJ r c * t | j S r )iterr4 r' s r __iter__zInfectedList.__iter__3 r: r c |j || j fi |}| }| | }| ||| S )V
Return list of cured files. This files are removed from InfectedList
)restorer5 keysr6 _build_cured_list)r backup
check_listkw check_map scan_liststill_infecteds r checkzInfectedList.check6 sZ #FN:t/?FF2FF NN$$ 22%%iNKKKr c ( K t j }| dt j |j || j fi | d{V }| }| | d{V }| ||| S )r? N)
asyncioget_event_looprun_in_executor functoolspartialr@ r5 rA r6 rB )r rC rD rE looprF rG rH s r async_checkzInfectedList.async_checkA s %'' ..
D,<
@B
NN$$ #~~i 8 8888888%%iNKKKr c h t | t | z
}g }|D ]}|| | j t | |f J# t
$ r6 t d fd| j D | _ Y w xY w|S )Nz)Skipping restoration of a changed file %sc * g | ]}|j k
|S r+ r )r1 r2 dsts r r3 z2InfectedList._build_cured_list..] s% I I IAqzS7H7Ha7H7H7Hr )setr4 remover
append
ValueErrorloggerwarning)r rF rG rH cured
cured_listsrcrU s @r rB z InfectedList._build_cured_listR s I^!4!44
.
.CC.C
.
!!(3--000 !!3*----
J
J
J? J I I I I I I
J s 'A//=B/.B/N) r& r) r* r r9 r= rI rQ rB r+ r r r- r- * st # # #
L L LL L L" r r- c x t } | r%t j | t }t j |d t
| S # t $ rY}t j t | dd
| ddg }t | |d }~ww xY w)NT)exist_ok)dir z There is a file in place of {}z7Most probably it is done to prevent `restore` to be runz,Remove this file to be able to restore again) RESTORE_DIRr r joinRESTORE_DIR_NAMEmakedirsr FileExistsErrorlinesepstrr$ )tmp_dirr5 emessages r prep_restore_dirrm c s K3 B ',,w0@AAK
K$//// "k2222 . . .*// FF , 3 3K @ @ E : #
g&&A-.s =A
B9 AB44B9.c t | }t ||j | }| || } | pi } g }
g } | j dd|i| d|dD ]K} t
j rt | t r%g }
|D ]} | |j
}nk# t $ r Y *t $ rS t d | | t u rt d Y w xY w||k r|
| |
rb |j | |
fi |}|D ]P\ }} t' || |
| ,# t( $ r | | Y Mw xY w| |s nM| d |D | |
| fS ) Nrj untilFasync_rj zHUnexpected end of file reached during processing backup: {}. Skipping...z5Plesk multivolume backups currently are not supportedc g | ] }|j
S r+ rT r0 s r r3 z$restore_infected.. 555!1:555r r+ )rm r- name pre_backupsbackupsr DisableFtpBackups
isinstancer file_datar FileNotFoundErrorEOFErrorrZ r[ r$ r rX rI r PermissionErrorcloseextendcleanup)backendr4 rp r6 pre_restore_hookrj rE r5
infected_listargssuccessfailedrC rD
infected_file backup_file restore_listr^ rU s r restore_infectedr v s #7333K (8)DDM ue , ,D :2DG
F!'/ (-w % % " z&-'H'H
* + +M
$ . .}/E F F $
&&,fVnn e##NNO
++ k * * * * (.=.vzHHRHHL (
(
(S( c3''' NN3'''' ' ' ' 'MM#&&&&&'
E MM55}555666 OO F?s+ B
D,ADDE'' F F c K t | }t ||j | }| ||d d {V } | pi } g }
g } t j } | j dd|i| d|d d {V D ]}
t j rt |
t r%g }|D ]i} | d t j
|
j |j d {V }n# t $ r Y Iw xY w||k rT| | j|rh |j |
|fi | d {V }|D ]P\ }} t' || |
| ,# t( $ r | | Y Mw xY w|
|s n| d |D | d {V |
| fS )Nro T)rr rp rq c g | ] }|j
S r+ rT r0 s r r3 z*async_restore_infected.. rt r r+ )rm r- ru rv rK rL rw r rx ry r rM rN rO rz r r{ rX rQ r r} r~ r r )r r4 rp r6 r rj rE r5 r r r r rP rC rD r r r r^ rU s r async_restore_infectedr s #7333K (8)DDM$$UE$$?? ? ? ? ? ? ?D :2DG
F ! # #D' (,g # # " z&-'H'H
* + +M
$($8$8 % (-*@ % % %
++ k * * * * (!: !:
" "&(" " L )
(
(S( c3''' NN3'''' ' ' ' 'MM#&&&&&'
E MM55}555666
//
F?s$ (9C""
C/.C/.E E65E6r ) rK rN loggingr tempfiler rb r backup_backendsr backup_backends_libr configr safe_fileopsr r
re r
expanduserrd rc getLoggerr& rZ r
r- rm r r r+ r r r s ' ' ' ' ' ' " " " " " " . . . . . . # # # # # # & g c3C!D!DEE 8 $ $K K K K K K K K,6 6 6 6 6 6 6 6r3 3 3 3,
< < < <D
: : : : : :r