OnC d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlmZm Z d dl
m Z m Z d dl
mZ d dlmZ e j e ZdZd Zd
Zed d gZd
ZddZd Zde eef fdZ dedeeef fdZ dS ) N)glob)Path)TupleUnion) CharFieldModel)
FilenameField) HostingPanelz.imunify.quarantinedz/var/imunify360 quarantinedz/var/wwwz/home*c 2 G fddt }|S )zl
Model stub for migration because we can't use migrator.orm[] due to
custom field FilenameField
c Z e Zd Z G fdd Z ed Z e ZdS ) get_model..MalwareHitc e Zd ZdZ ZdS )"get_model..MalwareHit.Meta malware_hitsN)__name__
__module__ __qualname__db_tabledatabasedbs e/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/migrations/160_remove_quarantine.pyMetar s %HHHH r F)nullN)r r r r r orig_filer statusr s r
MalwareHitr s` "Mu--- r r )r )r r s ` r get_modelr s? U r Fc |rd S |pt }t | }| |j t
k }|D ]8}t
|j \ } } || | 9t D ]B}
t t j
|
t D ]
} || Cd S N)delete_quarantine_folderr selectwherer QUARANTINED find_quarr delete_instanceQUARANTINE_PARENTSr ospathjoin QUAR_NAME) _migratorr fakedelete_function___modelr hitpath_to_deleteparents r migrater7 ) s &A)AO h E,,..&&u|{'BCCK %cm 4 4''' % , , "27<< #B#B C C , ,N ON + + + + ,, ,r c d S r" )r1 r2 s r rollbackr: ? s Dr quarantine_pathc t | } | j t k rK| | k r5t d| t
j | d d S d S d S )Nz Deleting quarantine folder %sT)
ignore_errors)r namer- resolveloggerinfoshutilrmtree)r; s r r# r# C sr ?++O )) 6 6 8 8 8 8 3_EEE
oT:::::: *) 8 8r sourcereturnc t | }t t t z | t d f}d}d}|j D ]P} t j | j }|} n # t $ r Y >t $ r |cY c S w xY w| |j dk r|S | | | z } t |j }n# t t f$ r |cY S w xY w | | }n# t" $ r |cY S w xY w|t z |fS )zy
Find file in quarantine by source path.
This function is copied from agent code since it is to be removed.
/Nroot)r DEF_QUARr- relative_toparentspwdgetpwuidstatst_uidFileNotFoundErrorKeyErrorpw_namer? r
base_home_dirpw_dir RuntimeError
ValueError) rD filedefault_resultuserr6 r+ resolved_placebase_dirrelatives r r' r' M s <>// << | , !--h77 i ))s< +B
B' B'&B'$&D D! D!%D; ; E
E
)FN) loggingr* rL rB r pathlibr typingr r peeweer r $defence360agent.model.simplificationr +defence360agent.subsys.panels.hosting_panelr
getLoggerr r@ r- rI r& r) r r7 r: strr# r' r9 r r re s\
# # # # # # # # ? > > > > > D D D D D D 8 $ $ "
H5 ", , , ,, ;eCI.> ; ; ; ;'*c '*eD$J / '* '* '* '* '* '*r