xC ( d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z d dl m
Z
mZmZmZmZ ddlmZ G d d e Z G d
d e Z G d d
e Z G d de Z G d de Z G d de Z G d d Z G d d Z G d d Z G d d Z G d d e Z G d d e Z d Z!d! Z"d" Z#d# Z$d$ Z%d% Z&d& Z' G d' d(ej( Z(e(j) Z*dS )) N)abstractmethod)ThreadPoolExecutor)suppress)BinaryIODictLiteralSetTuple )helpersc e Zd ZdS ) BackendErrorN)__name__
__module__ __qualname__ Y/opt/imunify360/venv/lib/python3.11/site-packages/restore_infected/backup_backends_lib.pyr r s Dr r c e Zd ZdZdS )BackendNotAuthorizedErrorzR
Backup backend was't initialized properly and can't be used without auth
Nr r r __doc__r r r r r s Dr r c e Zd ZdZdS )BackendNonApplicableErrorzABackup couldn't be used on the systems with current configurationNr r r r r r s KKDr r c e Zd ZdZdS )BackendClientRequiredErrorz4Backup provider requires Client soft to be installedNr r r r r r s >>>>r r c e Zd ZdZdS )NoSuchUserErrorz.Backup file owner is not present on the systemNr r r r r r # s 8888r r c e Zd ZdZdS )UnsupportedBackupErrorz,Backend can't recognize a backup file formatNr r r r r r ' s 6666r r c $ e Zd ZdZd Zd Zd ZdS ) BaseResourcez(
Base class for backup resource
c | || _ || _ t | | _ t | j | _ d S N)pathresource _split_pathprefixlen
prefix_lenselfr% r& s r __init__zBaseResource.__init__0 s4
!(++ dk**r c f d | j j t | j S )Nz
<{0} for {1}>)format __class__r reprr& r, s r __repr__zBaseResource.__repr__6 s. %%dn&=&*4=&9&9; ; ;r c L | j t | d | j k S r$ )r( r' r* r, r% s r
is_relatedzBaseResource.is_related: s$ {k$ / /0@ 0@ AAAr N)r r r r r- r3 r6 r r r r" r" + sP + + + ; ; ;B B B B Br r" c e Zd ZdZd Zd ZdS )FileDataz:
Class to manipulate of file from backup resource
c || _ | | \ | _ | _ | _ | _ | _ || _ d S r$ )r& infosizemtimeuidgidmodefilename)r, r& r@ s r r- zFileData.__init__D sC
?G}} @
@
< 4:tx49 !
r c 6 d | j j t | j t | j t | j t | j t | j t | j d| j
z S )NzQ<{0}(resource={1}, filename={2}, size={3}, mtime={4}, uid={5}, gid={6}, mode={7}>z0o%03o) r/ r0 r r1 r& r@ r; r< r= r> r? r2 s r r3 zFileData.__repr__L su
**0&'T]##T]##TYTZ TXTX49$ + +
r N)r r r r r- r3 r r r r8 r8 ? s< ! ! !
r r8 c e Zd Zd Zd Zd Zd Zd Z ej d d Z
d
d
e e d e
eef fd ZdS )
BackupBasec " || _ || _ d S r$ )r% created)r, r% rE s r r- zBackupBase.__init__] s r c " | j |j k S r$ )rE )r, others r __lt__zBackupBase.__lt__b s |em++r c \ d | j j t | S )Nz
<{0}({1})>)r/ r0 r strr2 s r r3 zBackupBase.__repr__e s# ""4>#:CIIFFFr c | j S r$ )r% r2 s r __str__zBackupBase.__str__h s
yr c B | j D ]}| d S r$ ) resourcesclose)r, r& s r rO zBackupBase.closek s0 H NN r Nmaxsizec | j D ])}| | rt || c S *t t | r$ )rN r6 r8 FileNotFoundErrorr1 r+ s r file_datazBackupBase.file_datap sZ 0 0H""4((
0 $/////
0 T
+++r /itemsreturnc t j | s t j | }i }|D ]}t j ||j t j }t j | }t j
| |j |j 5 }t |d 5 }t j
| D ]} | | d d d n # 1 swxY w Y d d d n # 1 swxY w Y t j }
|j } t j ||
| f t j ||j |j t j ||j |j ||< |S )Nwb)osr% isabsabspathjoinr@ lstripsepdirnamer mkdirr& openreadwriteDateTimenow timestampr< utimechownr= r> chmodr? ) r, rV destinationresultitem target_name
target_dirfileobjtargetchunkatimer< s r restorezBackupBase.restorex s! w}}[)) 7 '//+66K 0 0D ',,T] 1 1"& 9 9 K 55J M* % % % ##DM22
(gtT@ @
( $\' 2 2 ( (ELL''''(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
(
( $((**4466EJ((**E H[5%. 1 1 1 H[$(DH 5 5 5 H[$) , , ,"&-F;
s6 D<+-D% D<%D) )D<,D) -D<<E E )rU )r r r r- rH r3 rL rO functools lru_cacherT r r8 r rJ rt r r r rC rC \ s
, , ,G G G
Y&&&, , '&, S] S#X r rC c e Zd ZdZdZd ZdS )FileResourceMixinz@
Mixin class for file-based resources (tar, tar.gz, gz)
Nc X | j r"| j d | _ d S d S r$ )rp rO r2 s r rO zFileResourceMixin.close s5 < L DLLL r )r r r r rp rO r r r rx rx s4 G r rx c R e Zd ZdZd Z ej d d Zd ZdS )TarResourceMixinz5
Mixin class for tar resources (tar, tar.gz)
c l | j st | j | _ | | S r$ )rp tarfile_openr% _normalize_pathr5 s r _prepzTarResourceMixin._prep s1 | 3' 22DL##D)))r NrP c 2 | | } | j | }n*# t $ r t t | w xY w|j t j |j
|j |j |j
fS r$ )r rp getmemberKeyErrorrS r1 r; r re
fromtimestampr< r= r> r? )r, r% tar_pathtar_infos r r: zTarResourceMixin.info s ::d## 0|--h77HH 0 0 0#DJJ// / 0
w / = =hn M M hlHM; ;s 2 'Ac | | } | j | S # t $ r t t | t j $ r"}d|j v rt |j Y d }~d S d }~ww xY w)Nzunexpected end of data)
r rp extractfiler rS r1 tarfile ReadErrorargsEOFError)r, r% r es r rb zTarResourceMixin.open s ::d## (<++H55 5 0 0 0#DJJ// / ( ( ('1611 '' 211111 (s 1 4B%BB) r r r r r ru rv r: rb r r r r{ r{ sh * * * Y&&& ; ; '& ; ( ( ( ( (r r{ c p e Zd ZdZeeed Zd fd Zd Z d Z
fdZ xZ S )
FtpBackupBasez-
Base class for backup on FTP server
c d S r$ r )clss r FTP_DIR_NAMEzFtpBackupBase.FTP_DIR_NAME s
r Nc t || || _ |p t j d | _ t j | j | j | _ d S )Nz~/)
superr- ftprZ r%
expandusertmp_dirr] r ftp_dir)r, r r% rE r r0 s r r- zFtpBackupBase.__init__ s`
w''' :"'"4"4T":": w||DL$2CDD r c t j t | j | j t j j S r$ )rZ r% r] rJ r r^ r_ r2 s r rL zFtpBackupBase.__str__ s4 w||CMM49+;+;BGK+H+HIIIr c t j | j | j | j | j S # t j $ r t j d| j z Y nw xY wdS )zZ
:raises helpers.IsNotDirError:
:raises helpers.DirNotEmptyError:
z Error retrieving data from %sN)r ra r r retriever% FtpErrorwarningr2 s r _retrievezFtpBackupBase._retrieve s}
dl### H8$$TY == = H H H O ;dh F G G G G G Hts $A +A.-A.c t t t 5 t j | j d d d d S # 1 swxY w Y d S r$ )r rO r rS shutilrmtreer )r, r0 s r rO zFtpBackupBase.close s
'
(
( ( ( M$, ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s A A!$A!r$ )
r r r r property classmethodr r r- rL r rO
__classcell__)r0 s @r r r s
^ [ X
E E E E E EJ J J ( ( ( ( ( ( ( ( (r r c | t j } | t j }t | S r$ )rstriprZ r_ splittuple)r% path_lists r r' r' s6 ;;rv D
26""I r c fd}|S )Nc t j | } t dt j | _ | S )N
__signature__)ru update_wrappergetattrinspect signaturer )wrapperwrappeds r decoratorzwraps..decorator s@ *7G<< ' (/(9'(B(B!D !D r r )r r s ` r wrapsr s$
r c J t fd }d|_ |S )Nc | S r$ r )r fs r r zextra..wrapper s q$xr T)r extra)r r s ` r r r s:
1XX X GM Nr c \ fdt fd d_ S )Nc K t j t j t j g| R i | d {V S r$ )asyncioget_event_looprun_in_executor asyncableexecutorru partial)r kwargsfuncs r coroutine_functionz%asyncable..coroutine_function sj +--== d 4T 4 4 4V 4 4
r c | dj rEt t d st d t _ | i |}j |_ |S | i |S )Nasync_r ) max_workers)popr hasattrr r r r )r r coroutiner r r s r r zasyncable..wrapper
s| ::h / / 9j11
G%7A%F%F%F "**D;F;;I%)%6I " tT$V$$$r F)r r )r r r s `@@r r r sa
4[[% % % % % % [% GN Nr c fd}|S )Nc R t fd _ S )Nc ` t j j s | i |S r$ )rZ r% existstoken)r r excr r s r r zC_backend_checker_decorator..real_decorator..wrapper s5 7>>'-00
1d%f%% %r )r r )r r r r% s `@r real_decoratorz2_backend_checker_decorator..real_decorator sF q & & & & & &
&
r r )r% r r s `` r _backend_checker_decoratorr s* r c < t | t | S r$ )r r )
token_path error_msgs r backend_auth_requiredr * s$ %j&? &J&J L L Lr c < t | t | S r$ )r r ) client_pathr s r backup_client_requiredr / s$ %k&@&K&K M M Mr c d e Zd Zi ej j ddiZe dded fd ZdS ) TarFilezstdzstdopenrNr? r wxc |dvrt d ddlm}m} n"# t $ r t j d w xY w ||p|||| } | j |||fi |} nc# |t t f$ r0 |
|dk rt j d t $ r |
w xY wd | _
| S )
zhOpen zstd compressed tar archive name for reading or writing.
Appending is not allowed.
r z mode must be 'r', 'w' or 'x'r ) ZstdErrorZstdFilez pyzstd module is not available)level_or_option zstd_dictr znot a zstd fileF)
ValueErrorpyzstdr r ImportErrorr CompressionErrortaropenOSErrorr rO r Exception _extfileobj)
r namer? rp r r r r r ts
r r zTarFile.zstdopen: s9 & & ;<< < M 2 2 2 2 2 2 2 2 2 M M M*+KLL L M ( Ot +
D$::6::AA7H- MMOOOs{{ '(9::: MMOOO
s =A# #A C)r NNN) r r r r r OPEN_METHr r r r r r r r 4 sy
/
#
I
(+
& & m$& & & [& & &r r )+r ru r rZ r r abcr concurrent.futuresr
contextlibr typingr r r r r
r r r r r r r r r" r8 rC rx r{ r r' r r r r r r r rb r} r r r r sU
1 1 1 1 1 1 6 6 6 6 6 6 6 6 6 6 6 6 6 6 9
? ? ? ? ? ? ? ?9 9 9 9 9l 9 9 97 7 7 7 7\ 7 7 7B B B B B B B B(
:6 6 6 6 6 6 6 6r
( ( ( ( (( ( ( (F%( %( %( %( %(J %( %( %(P * L L L
M M M
- - - - -go - - -` | r