,ruA% d Z ddlZddlZddlZddlmZ ddlmZmZm Z ddl
m Z m Z ddl
mZ ddlmZmZ ddlZej Zej ej z ej z Zd Zed!d Zed
efd Z d"d eee f d
ee fdZ d
ed
e fdZ ed
ed
e de de fd Z!de de dee ef dee fdZ" e d defd Z#dede fd Z$ddd d ed efd Z%dS )#zF
Mostly copy-pasting from `safe_fileops.py` of the Imunify360 project
N)ProcessPoolExecutor) ExitStackcontextmanagersuppress) lru_cachepartial)Path)OptionalUnionc t j dk rt j g t j | t j | || S )Nr )osgetuid setgroupssetgidsetuid)uidgidfunargss R/opt/imunify360/venv/lib/python3.11/site-packages/restore_infected/safe_fileops.pydropr sI y{{a
RIcNNNIcNNN 3: c # N K t j | || } |V t t 5 t j | ddd dS # 1 swxY w Y dS # t t 5 t j | ddd w # 1 swxY w Y w xY w)zR
Context manager which wraps os.open and close file descriptor at the end
dir_fdN)r
openr OSErrorclose)pathflagsr fds r open_fdr" s#
uV , , ,B
g
HRLLL Xg
HRLLL s@ A" AA A"B$7B B$B B$ B B$namec # K t | t j 5 }t j d | }| |k rt d |V ddd dS # 1 swxY w Y dS )z
Context manager to get a directory file descriptor
It also checks if a directory doesn't contain a symlink in the path
:param name: full directory name
z/proc/self/fd/{}z%Operations on symlinks are prohibitedN)r" r
O_DIRECTORYreadlinkformatPermissionError)r# r reals r open_dir_fdr* ) s
r~ & & &{-44V<<== 4<<!"IJJ J s AA++A/2A/fr c | t | t rt j | || } t | | S )a
Context manager to open file object from file name or from file descriptor
File object extended with 'st' attribute that contains os.stat_result of
the opened file
:param f: file name or file descriptor to open
:param mode: mode for built-in open
:param flags: flags for os.open, ignored if 'f' is a file descriptor
:param dir_fd: directory descriptor, ignored if 'f' is a file descriptor
)r r mode)
isinstancestrr
r )r+ r. r r s r open_fobjr1 8 s? !S 3 GAU6 2 2 2 r c t j | | }t j | |j t j z | |S )Nr )r. r )r
statchmodst_modeS_IRUSR)r# r sts r
make_readabler8 J s; f % % %BHT
T\1&AAAA
Ir r is_safec # K |r3t | || 5 }|dfV ddd dS # 1 swxY w Y dS | |fV dS )z
If is_safe flag is True, open file descriptor using name and dir_fd
If is_safe is False, return name and dir_fd as is
)r r N)r" )r# r r r9 r! s r
safe_tupler; R s
T&
6
6
6 "d(NNN Fls +//src_fdsrc_modedst_f
dst_dir_fdc r t | d 5 }t |dt | 5 }|r
| t j || t |t
r(t
j | | d d d n # 1 swxY w Y d d d d S # 1 swxY w Y d S )Nrbwbr- ) r1 W_FLAGSshutil copyfileobjr/ r0 r
r4 fileno)r< r= r> r? _race_dstsrc_fodst_fos r _moverJ _ s5
64
5FI
tWj- -
5 IKKK66*** eS ! ! 5 HV]]__8 4 4 4 4
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5s5 B,A B B,B B,B B,,B03B0 returnc dt j v rt d S t d S )Ndebianz/var/spool/cron/crontabsz/var/spool/cron)distrolikesplitr r r
_crontab_pathrS s s= 6;==&&((((./// ! " ""r r c J t | j t k S N)r parentrS )r s r
is_crontabrW z s ::
//r ) _race_srcrG srcdstc t j | \ }}t j | \ }}t 5 }| t | } t
|| }
| t |t | } | t | } t | }
|
rht j
| j t j z rBt t 5 t j || d d d n # 1 swxY w Y | t! || t" |
\ }} t% t& |
j |
j t, | |
j || | }t/ d 5 }| | }| d d d n # 1 swxY w Y |r
| t j || |
r>t j ||
j |
j | t j ||
j | d d d d S # 1 swxY w Y d S )Nr rK ) max_workers) r
r rQ r
enter_contextr* r8 r" R_FLAGSrW r3 r5 S_ISVTXr FileNotFoundErrorunlinkr; rC r r st_uidst_gidrJ r submitresultchownr4 )rY rZ rX rG src_dirsrc_namedst_dirdst_namestack
src_dir_fdsrc_str< r? safe_dstr> fnexecutorfutures r safe_moverr ~ s
c**GX
c**GX )?((W)=)=>>
x44$$WXw
%K%KLL((W)=)=>>
c?? 7
++3dlB 7 +,,
7
7 (:6666
7
7
7
7
7
7
7
7
7
7
7
7
7
7
7 " / / xWh ? ?
z M M N
!Q
/
/
/ 8 __R((F MMOOO IKKK
(:.... ? HUFM6= L L L L HUFN: > > > >S)? )? )? )? )? )? )? )? )? )? )? )? )? )? )? )? )? )?s] C I' D?3 I'?E I'E A0I'7*G-! I'-G1 1I'4G1 5A%I''I+.I+rU )r N)&__doc__r
rD r3 concurrent.futuresr
contextlibr r r functoolsr r pathlibr typingr
r rO O_RDONLYr^ O_TRUNCO_CREATO_WRONLYrC r r" r0 r* intr1 r8 boolr; rJ rS rW rr rR r r r s|
2 2 2 2 2 2 : : : : : : : : : : ( ( ( ( ( ( ( ( " " " " " " " "
+
*rz
!BK
/ c sCx # $ S S # c D 5 55 c?5
5 5 5 5( 1 #t # # # # 0S 0T 0 0 0 0 04t -? -? -?3 -?S -? -? -? -? -? -?r