h
d Z ddlZddlZddlZddlmZmZ ddlmZ ddl m
Z
ddl m Z ddl
mZmZ ddlmZ dd lmZ dd
lmZ dd lmZ ej e Z G d d
Zde de fdZ de
fdZ dS )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
N)MalwareMalwareScanIntensity) MessageType)TheSink)resource_limits)MalwareScanResourceType RESTORE_ORIGINAL_INTENSITY_KEY)
MalwareHit)MDS_PATH)MDSDetachedRestoreDir) hash_pathc , e Zd ZdddZdefdZd ZdS )MalwareDatabaseRestoreN)
restore_idc ` || _ || _ |pt j j | _ d S )N)app_namepathuuiduuid4hexr )selfr r r s U/opt/imunify360/venv/lib/python3.11/site-packages/imav/malwarelib/scan/mds/restore.py__init__z MalwareDatabaseRestore.__init__+ s)
$8
(8 work_dirc dt dt | j | j d| j d| j dt |j d| j dt |j dt t j gS ) Nz/opt/ai-bolit/wrapperz --restorez--pathz
--app-namez
--report-filez
--detachedz
--progressz --db-timeout)
r storage_pathr r str report_filer
progress_filer MDS_DB_TIMEOUT)r r s r cmdzMalwareDatabaseRestore.cmd0 sq # DM 2 2 I M $ % % O & ' ' & ' '!
r c K t | j 5 }| | }|j d 5 }|j d 5 }t d| t j |t j t j d||t | t d {V d d d n # 1 swxY w Y d d d n # 1 swxY w Y d d d d S # 1 swxY w Y d S )Nw)modez Running MDS Restore with: %sT)
intensity_cpu intensity_iostart_new_sessionstdoutstderrcwdkey)r r r" log_fileopenerr_fileloggerinfor create_subprocessr CPUIOr r )r r r" l_fe_fs r restorez MalwareDatabaseRestore.restoreD s
"4?
3
3 x((8$$C"'' (
h',,#,66
:= :C@@@%7"6":!5!8&* H
6
sZ 1D C+$A$C C+C C+C C+ D+C/ /D2C/ 3DD
D)__name__
__module__ __qualname__r r r" r7 r r r r * s[ 59 9 9 9 9 9
1
( r r r r c D dt | |z t j z S )Nz /var/imunify360/cleanup_storage/)
hash_provider)r
hashlibsha384r r s r r r X s+ - xw~1 1 1 r sinkc K d | D }|D ];}| t j |j |j d {V z restore_hits.._ s7 7 : @ @ @ @ @ @r r@ )process_messager MalwareDatabaseRestoreTask orig_filer )hitsrA db_hitsrH s r restore_hitsrO ^ s G
"" 2]S\
r ) __doc__r> loggingr defence360agent.contracts.configr r "defence360agent.contracts.messagesr "defence360agent.internals.the_sinkr defence360agent.utilsr imav.malwarelib.configr r imav.malwarelib.modelr
imav.malwarelib.scan.mdsr !imav.malwarelib.scan.mds.detachedr imav.malwarelib.utilsr
getLoggerr8 r0 r r r rO r; r r r\ sn * J J J J J J J J : : : : : : 6 6 6 6 6 6 1 1 1 1 1 1 - , , , , , - - - - - - C C C C C C + + + + + + 8 $ $+ + + + + + + +\s c
7
r