}g d Z ddlZddlZddlZddlZddlmZmZmZm Z m
Z
ddl m Z ddl
mZ ddlmZ ddlmZ dd lmZmZmZmZ ej e Z G d
d Z 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)DictIteratorListOptionalTuple)ScanFailedError)Core) MessageType )AiBolit)
DirectAiBolitPrepareFileList
ScanResult
event_hookc e Zd ZddZd Zd Zd Zd Zd Zd Z d
Z
d Z d e e
d
eeee e e
f fdZdS )MalwareScannerNFc |p t j | _ || _ || _ d | _ d | _ t j | _ || _ d | _
d | _ t j | _
d S N)ConfigTMPDIR_tmpdir_sink_hooksaibolit_taskasyncioget_event_loop_loop
_current_task_future_resultLock _is_running)selftmpdirsinkhookss Q/opt/imunify360/venv/lib/python3.11/site-packages/imav/malwarelib/scan/scanner.py__init__zMalwareScanner.__init__' sg .
+--
! ""<>> c t j | _ | j | j |g|R i | | _ | j S r )r Futurer r create_task_startr )r# pathargskwargss r' startzMalwareScanner.start4 sM %n . .Z++KDK,Nt,N,N,Nv,N,NOO
zr) c | j t j d | j | j | j S NzNeed to start action)r r
exceptionsInvalidStateErrorr run_until_completeresultr# s r' waitzMalwareScanner.wait9 sQ &$667MNN N
%%d&9:::"))+++r) c K | j t j d | j d {V | j S r3 )r r r4 r5 r7 r8 s r'
async_waitzMalwareScanner.async_wait@ sW &$667MNN N!!!!!!!!"))+++r) c d | r| j d S d S r )statusr cancelr8 s r' stopzMalwareScanner.stopG s6 ;;== J r) c 4 | j S r )r" lockedr8 s r' r= zMalwareScanner.statusK s &&(((r) c K | j 4 d {V | j }| j r t | j | } ||g|R i | d {V }| j | n[# t j $ r | j d Y n0t $ r$}| j
| Y d }~nd }~ww xY wd d d d {V d S # 1 d {V swxY w Y d S r ) r"
_execute_scanr r r r
set_resultr CancelledError Exception
set_exception)r# r. r/ r0 execute_scanr7 es r' r- zMalwareScanner._startN s # 7 7 7 7 7 7 7 7
7 # 1 ; H#9:dj#9#9,#G#GL +|D B4 B B B6 B BBBBBBB #..v6666 )
5
5
5#..t44444
5
5
5#11!44444444
5 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7sF C=A*C*)CC C B=8C=CC
C"%C"c K | j r t j }|j j |d< t |t rG|j |d< |j |d< |j |d< |j
|d< |j |d< |j |d< |j
|d< nt | |d< | j | d {V d S # t j $ r t$ $ r t& d Y d S w xY wd S )
N exception return_codecommandouterrmessager. scan_idz*Exception while sending ScanFailed message)r r
ScanFailed __class____name__
isinstancer rL rM rN rO rP r. rQ strprocess_messager rE rF loggerrK )r# rI msgs r' _send_scan_failed_messagez(MalwareScanner._send_scan_failed_message\ s4 : O
O!,..#$;#7K a 11 ,)* C
&%&YC N!"CJ!"CJ%&YC N"#&CK%&YC NN%(VVC Nj0055555555555)
O
O
O !MNNNNNN
O' O Os B0B= =3C43C4c | K t d t |t rt nt
} || j }t j |d< || j } ||g|R i | d {V }t |t sJ |
rd nt j }| | |
r| S | j
r | j
j r| j
j |_ | S )Nz Scan started
begin_time)end_time)rX inforU rV r
r r time _run_aibolitr is_detachedset_start_stopto_dict_initialr cmdr/ to_dict) r# r. r/ r0 files_providerprepare_file_listrH r7 r] s r' rC z MalwareScanner._execute_scans sB N### (c 2 2 GMM +N4< 8 8 #y{{| (():;; #|D:4:::6::::::::&*-----!--//@44TY[[ x 000 $))++ +|
/ 0
/ "l . >>## #r) filereturnc ~ K g }d}t | _ | j j } || _ ||fi | d{V }n# t j $ r t $ rH}| | d{V t d g }|j
}Y d}~n`d}~wt $ rP}| | d{V t d g }t | }Y d}~nd}~ww xY wt j | \ }} | | d| _ n # d| _ w xY w||fS )z
:param file: file with list of files, that should be scanned
:param scan_options: some options for the scan
:return: parse report iterator and error
Nz%Scan failed with ScanFailed exceptionz#Scan failed with ordinary exception)r r scanr r rE r rZ rX rK rP rF repr itertoolsteeappend)
r# rh scan_options iteratorserrorrk iteratorrI r7 _s
r' r` zMalwareScanner._run_aibolit s yy | &!%D
!%d!;!;l!;!;;;;;;;)
"
"
"
"44Q777777777 !HIII
44Q777777777 !FGGG Q
"
h//IFA V $ $ $!%D D % % % %% sD D/ ? D/ C8>BD/
C8(AC3.D/ 3C88/D/ / D8)NNF)rT
__module__ __qualname__r( r1 r9 r; r? r= r- rZ rC r rV r r r r r` r) r' r r & s * * * *
, , ,, , , ) ) ) 7 7 7 O O O.$ $ $0& SM& tHTN#Xc]2 3& & & & & & r) r ) __doc__r rm loggingr_ typingr r r r r imav.malwarelib.scanr defence360agent.contracts.configr r "defence360agent.contracts.messagesr
ai_bolit.scannerr scan_resultr
r r r getLoggerrT rX r rw r) r' r s * 8 8 8 8 8 8 8 8 8 8 8 8 8 8 0 0 0 0 0 0 ; ; ; ; ; ; : : : : : : % % % % % % O O O O O O O O O O O O 8 $ $K K K K K K K K K K r)