Ҹ q d Z ddlZddlZddlmZmZmZ ddlmZ ddl m
Z
m Z ddl m
Z
ddlmZ ej e Z G d d 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
N)MalwareScanResourceTypeMalwareScanTypeQueuedScanState)
DetachedState)QueuedScanBase ScanQueue) create_task_and_log_exceptions) MessageTypec ~ e Zd Z e ZddZ ddefdZd Zd Z ddZ
d Z d
Z d Z
d Zd
ZdefdZd ZdS )QueueSupervisorSyncNc |pt j | _ || _ | j j }|rD|j t j t j
fv rt j |_
d S d S d S N)asyncioget_event_looploopsinkqueue current_scan
detached_scanget_detached_process_stater RUNNINGFINISHEDr runningstate)selfr r r s _/opt/imunify360/venv/lib/python3.11/site-packages/imav/malwarelib/scan/queue_supervisor_sync.py__init__z QueueSupervisorSync.__init__) s 4G 2 4 4 z. 9 *EEGG % & "1!8L 9 9 F
resource_typec K t d|j d | | j j }|D ]:}| j || x}r| j | ;|rA|j |v r8|j |k r-|j
| j d|j
d {V | j j |f||d| d {V | d S )NzQueuing %s scan for %s, )pathr T)r kill scan_started)r
prioritize)loggerinfovaluejoinr r findremover" r r handle_aborted_processr startedputrecheck)r pathsr r% scan_args running_scanr" scans r r. zQueueSupervisorSync.put8 s^ $m&9499U;K;K
z. ( (DzD
NNNt
(
!!$'''
!U** *m;; ,CCYT 8L D
djn
'!
r c X | j r"| j | j d S d S r )r r+ r s r clearzQueueSupervisorSync.clear[ s@ j J j r c | j t j }|sd S | j | D)NT) scan_type)r r* r
BACKGROUNDr+ r r3 s r clear_backgroundz$QueueSupervisorSync.clear_background_ sG $:??_-G?HHD
J d # # # $r c @ |r| j | }n| j j x}r|gng }d |D }t dd | |D ] }| j | | d S )Nc g | ] }|j
S )r" ).0ss r
z.QueueSupervisorSync.remove..l s '''A'''r z$Removing scans from the queue for %sr! )r find_allr r& r' r) r+ r/ )r scan_idsscansr3 r0 s r r+ zQueueSupervisorSync.removef s HJ''11EE'+z'> >tGTFFRE''''' :DIIe%D *17 c$*oo 1155
'>'DEEEr c 6 | j | S r )r get_scans_from_paths)r r0 s r rS z(QueueSupervisorSync.get_scans_from_paths s z..u555r c v | j j }t j |_ t | j | j | d S r )r r r r r r r
_scan_coror: s r rH z%QueueSupervisorSync._create_scan_task s2 z&$,
&ty$/4HHHHHr r3 c K | d {V }|1| j t j di | d {V d S | d S )Nr> )startr process_messager
MalwareScanr/ )r r3 scan_datas r rU z QueueSupervisorSync._scan_coro s **,,&&&&&& )++'44)44
LLNNNNNr c @ K | j || d S r )r update)r rC rN s r update_scan_statusz&QueueSupervisorSync.update_scan_status s$
(F+++++r )NN)Fr )__name__
__module__ __qualname__r r r r r. r6 r; r+ r/ rK rN rS rH r rU r] r> r r r r & s IKKE
9
9
9
9& ! ! /! ! ! !F $ $ $ ) ) ) $ $ $F F F 6 6 6I I I
^ , , , , ,r r )__doc__r loggingimav.malwarelib.configr r r imav.malwarelib.scan.detachedr imav.malwarelib.scan.queuer r defence360agent.utilsr "defence360agent.contracts.messagesr
getLoggerr^ r& r r> r r ri s *
8 7 7 7 7 7 @ @ @ @ @ @ @ @ @ @ @ @ @ @ : : : : : : 8 $ $C, C, C, C, C, C, C, C, C, C,r