h d Z ddlmZ ddlmZmZ ddlmZ ddlm Z m
Z
m Z ddl m
Z
mZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ eeeg G d d
e Z G d de
ZdS )u
This program is free software: you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License,
or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see .
Copyright © 2019 Cloud Linux Software Inc.
This software is also available under ImunifyAV commercial license,
see
) MessageType)AVFULL)feature)CommonEndpoints
RootEndpointsbind)&MS_CLEAN_REQUIRES_MYIMUNIFY_PROTECTIONcheck_permission)validate_av_plus_license)MalwareCleanupProxy) restore_hits)
MalwareHit)MaliciousEndpointStatusc e Zd Z eddd ed d Z eddd ed d Z eddd ed d
ZdS )
CleanupEndpointsmalware maliciouscleanupFNc K t t | t j ||d }|r7| j t
j ||| d {V d S d S NTuserr hits initiator
standard_onlyr r
r malicious_select_sinkprocess_messager MalwareCleanupTask)selfidsr r r s Z/opt/imunify360/venv/lib/python3.11/site-packages/imav/malwarelib/rpc/endpoints/cleanup.pyr zCleanupEndpoints.cleanup+ s ?FFF*3T4HHH *,,. ""/
z cleanup-allc K t t | t j |d }|r7| j t
j ||| d {V d S d S r r )r# r r r s r% cleanup_allz CleanupEndpoints.cleanup_all9 s ?FFF*dCCC *,,. ""/
r& restore-originalc K t j ||d }t || j d {V \ }}t || S NT)r restorer r r r r )r# r$ r r succededfaileds r% restore_originalz!CleanupEndpoints.restore_originalG sW *3T4HHH!-dDJ!?!???????&&x888r& )FNN)__name__
__module__ __qualname__r r r r( r0 r& r% r r ) s T)[),,
-,
T)[-00
10
T)["455 9 9 9 659 9 9r& r c e Zd Z fdZ eddd d Z eddd ed d Z eddd ed
Z xZ S ) CleanupRootEndpointsc p t | t | _ d S r1 )super__init__r
_proxy)r# sink __class__s r% r: z CleanupRootEndpoints.__init__P s,
)++ r& r r statusc X K | j }|ddi}nd|d}d|iS )Nr> stoppedrunning)r> progressitems)r; get_progress)r# rB r> s r% cleanup_statusz#CleanupRootEndpoints.cleanup_statusT sC ;++-- *FF )x@@F r& r Fc K }t j |d }|rJ|D ]Ifd|D }|r5| j t j || d {V Hd S d S )NTr c * g | ]}|j k
|S r5 )r ).0hitr s r%
z5CleanupRootEndpoints.user_cleanup..d s% E E ESCH4D4DS4D4D4Dr& r )r r r r! r r" )r# r r usersr user_hitss ` r% user_cleanupz!CleanupRootEndpoints.user_cleanup] s *tDDD
E E E ED E E E *44#6!*&**7
r& r) c K t j |d }t || j | d {V \ }}t || S r+ r- )r# r r r. r/ s r% user_restore_originalz*CleanupRootEndpoints.user_restore_originaln sW *dCCC!-dDJ!E!EEEEEEE&&x888r& )F)
r2 r3 r4 r: r rE r rM rO
__classcell__)r= s @r% r7 r7 O s , , , , ,
T)Y))! ! *)!
T)VY''
('
T)V /00 9 9 109 9 9 9 9r& r7 N)__doc__"defence360agent.contracts.messagesr ,defence360agent.feature_management.constantsr r )defence360agent.feature_management.lookupr defence360agent.rpc_tools.lookupr r r %defence360agent.contracts.permissionsr
r "defence360agent.rpc_tools.validater imav.malwarelib.cleanup.cleanerr
imav.malwarelib.cleanup.storager imav.malwarelib.modelr imav.malwarelib.utils.endpointsr r r7 r5 r& r% r\ s~ * ; : : : : : A A A A A A A A = = = = = =
H G G G G G ? ? ? ? ? ? 8 8 8 8 8 8 , , , , , , C C C C C C dV"9 "9 "9 "9 "9 "9 "9 "9J$9 $9 $9 $9 $9= $9 $9 $9 $9 $9r&