h T d Z ddlZddlZddlmZ ddlmZ ddlmZ G d de Z G d d e Z
G d
d e Z G d d
e Z G d de Z
G d de Z G d de Z G d de Z G d de 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)ABC)
NamedTuple)Enumc e Zd ZdZdZdZdS ) HashStater N)__name__
__module__ __qualname__ACTIVE BLACKLISTED
SUPERSEDED P/opt/imunify360/venv/lib/python3.11/site-packages/imav/patchman/fs_scanner/db.pyr r s
FKJJJr r c . e Zd ZdZdZdZdZdZdZdZ dZ
d S )
DefinitionTyper r
N) r
r r MALWARE
VULNERABILITY APPLICATIONDRYRUN MALWARE_RULEMALWARE_RULE_DRYRUNVULNERABILITY_ECOMMERCEVULNERABILITY_PLUGINr r r r r # s; GMK
FL r r c b e Zd ZU eed< eed< eed< d
dej defdZ defd Z
d
Z d Z d S )Table
table_namefieldscreate_table_query conn buffer_sizec X || _ || _ g | _ | d S N)r) r* buffer create_table)selfr) r* s r __init__zTable.__init__3 s0 &#% r rowc t | t | j k sJ | j | t | j | j k r| dS dS )z$
Insert with buffer
N)lenr& r- appendr* flush)r/ r1 s r buffered_insertzTable.buffered_insert9 sl 3xx3t{++++++ 3 t{ t / / / JJLLLLL 0 /r c 6 d | j }| j 5 | j d| j d| dd dgt | j z d| j ddd n # 1 swxY w Y g | _ dS )z
Flush buffer
z, z INSERT INTO z (z
) VALUES (?)N)joinr& r) executemanyr% r3 r- )r/ r& s r r5 z Table.flushB s 4;''
Y I ! ! s r __enter__z DB.__enter__ s r c V | | d S r, )r5 close)r/ exc_typeexc_valexc_tbs r __exit__z DB.__exit__ s
r r( db_namer* c r t j | rt j | t j | | _ t | j | | _ t | j | | _
t | j | | _ | j | j
| j f| _
d S )N)r* )osrI existsremoverC connectr) rL rM rW rX r] r^ _tables)r/ ri r* s r r0 z DB.__init__ s
7>>' " " Ig OG,, 1 I;!
!
!
#7 I;#
#
#
- I;
! #
r c ( d | j D dS )z&
Flush tables buffers
c 6 g | ]}| S r )r5 ).0_s r
z DB.flush.. s )))q)))r N)ro r> s r r5 zDB.flush s *)DL))))))r c 8 | j d S r, )r) rd r> s r rd zDB.close s r Nr? )
r
r r rb rh r@ rE r0 r5 rd r r r r` r` sw
#
** * * r r` )__doc__rk rC abcr typingr enumr r r r$ rG rL rQ rW rZ r] r` r r r rz s *
T #7 #7 #7 #7 #7C #7 #7 #7L :
# # # # #: # # # 5
U $ $ $ $ $ $ $ $ $ $r