G𢏸 " d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z m
Z
d dl m Z d dl
mZ d dlmZ d d lmZ d d
lmZ e j e Z G d d e Zd
Zd ZddZ d Z d Z d Z d Z d Z!ddZ"de de#e$ ddfdZ%dS ) N)suppress)datetime)copy)connect
DatabaseError)SqliteExtDatabase)app)
simple_rpc)Model)simplificationc e Zd ZdS )OperationErrorN)__name__
__module__ __qualname__ S/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/utils/check_db.pyr r s Dr r z.Blank database will be created on agent start c V t j } t j rt d t
j | st d| dt t | rt | }|st d t | }t
d| z t j | |st d|dt t | | }|st d t | r+t j | t d t z t
d
|z t j | t
d t! j t% s+t j | t d t z d S # t& $ r3}t j | t d
|dt d }~ww xY wd S )NzCannot perform database check and backup while agent is running. Please, stop the imunify360 agent with `service imunify360 stop`zDB z is not exists. z{Cannot proceed without backup copy of the database.Please contact imunify360 support team at https://cloudlinux.zendesk.comz*Removing original corrupted database at %sz0Cannot dump database to sql. Old DB backuped at z. zRLoading dump to new database failed. Database will be recreated during migrations.zBRestored database is still corrupt. Removing restored database. %sz0Database restored successfully. Removing dump %sz*Performing migrations on restored databasez;Restored database does not contain all necessary tables. %sz(Migrations on restored database failed: )r PATHr
is_runningr ospathisfileWORKAROUND_MSGis_db_corrupted make_backup dump_to_sqlloggerinforemove
load_from_sqlr migrateall_tables_are_present Exception)basebackupdumprestoredes r check_and_repairr+ su :D >
O
W^^D
!
! 9 n)-~~ >
4 4 &&F
$5
t$$D KKDtK L L L IdOOO (
$n vv~~/
)t44 (:
#8,, Ih'''(02@A
FM $KK LMMM"*,,, 233 $ , !#1 2 ! IdOOO(. 11nn. O4 4 s -G) )
H&3.H!!H&c t j }d | | d }|r|d|z z S |S )a
>>> mark_with_timestamp('/var/imunify360/imunify360.db')
'/var/imunify360/imunify360.db_2017-09-26_03:33:44.705967'
>>> mark_with_timestamp('/var/imunify360/imunify360.db', extension='sql')
'/var/imunify360/imunify360.db_2017-09-26_03:34:01.098544.sql'
z{}_{}_z.%s)r nowformat isoformat)filename extensioninstantbasenames r mark_with_timestampr5 ` sM lnnG~~h(9(9#(>(>??H %)+++r c t d| z d}t | 5 } | d }t | }d|v r t d d}n2# t
$ r%}t d| Y d }~nd }~ww xY w|cd d d S # 1 swxY w Y d S )Nz Database %s integrity check...TzPRAGMA INTEGRITY_CHECK;okz#Database integrity check succeeded.FzDatabaseError detected: %s)r r r executenextr warning)db_path is_corrupted
connectioncursorresultr* s r r r o s
KK07:;;;L Z < ''(ABBF&\\Fv~~ ABBB $ < < < NN7 ; ; ; ; ; ; ; ; < s; B6AA65B66
B% B B6 B%%B66B:=B:c X t | d }t d|z t |d 5 }t | 5 }| D ]}| | d d d n # 1 swxY w Y d d d n # 1 swxY w Y n}# t t f$ ri}t d|z t t 5 t j | d d d n # 1 swxY w Y d }Y d }~nd }~ww xY w|S )Nsql)r2 z!Dumping imunify360 database to %swz(Error during dump: %s. Operation aborted)
r5 r r openr iterdumpwriter OSErrorerrorr r r! )r; dumpfiler( r= rowr* s r r r ~ s "7e<<???
(C
D''*:*: j!**,,
3
7 # ?!CDDD
g
Ih
Os B- B!-B
> B!
B B!B B! B- !B%%B- (B%)B- -D'>1D"/D D"D D"D D""D'c t j | r t d d S t d|d| d t
|d 5 }t | 5 } | }| | ns# t $ rf}t | t t 5 t j | d d d n # 1 swxY w Y d } Y d }~nd }~ww xY wd d d n # 1 swxY w Y d d d n # 1 swxY w Y | S )Nz^Database already exists. Loading dump to existing database may cause errors. Operation abortedz
Reading dump z into new database z...r)r r existsr r: r rC r read
executescript MemoryErrorrG r rF r! )r; rH r( r= rA r* s r r" r" s
w~~g
;
t
KKK5=XXwwwG
h
gg&6&6
* ))++C $ $S ) ) ) ) LLOOO'""
#
# '"""
#
#
#
#
#
#
#
#
#
#
#
#
#
#
#GGGGGG
Ns -D?=D(?)B)(D()
D3.D!D 6 DD
D D
DD(DD( D?(D, ,D?/D, 0D??EEc t d| z t | d } t | | t d|z nt# t $ rg}t d| t
t 5 t j | d d d n # 1 swxY w Y d }Y d }~nd }~ww xY w|S )NzMaking backup of the %s...r' z$Database copied successfully to: %s zMaking backup failed: %s)
r r r5 r r% rG r rF r r! )r; backup_filenamer* s r r r s
KK,w6777)'8<B; ?C Cc t d t j d t j D } t
d | D r t d dS t d dS )NzDVerifying that db schema is up-to-date and all tables are present...c 6 g | ]}t j | S r r
get_models.0modules r
z*all_tables_are_present.. s3
%f - -
r c 3 > K | ]}| V d S N) table_exists)rW models r z)all_tables_are_present.. s.
4
4E5
4
4
4
4
4
4r zAll tables are presentTz Some tables are missing in db.F)r r itertoolschainr MODULES_WITH_MODELSallrG )modelss r r$ r$ s
KKN _
1
F
4
4V
4
4
444 ,---t 5666ur returnc t j j t j t d g } t j D ]\\ }}t d| t j j
d||f | | ]t t j j | t d d S )Nz#Recreating schema for linked DBs...z
Attach db: %sz
ATTACH ? AS ?z Schema recreated successfully.)
r instancedbinitr r r r r MIGRATIONS_ATTACHED_DBS execute_sqlappendrecreate_schema_models)attached_schemasr; schemas r recreate_schemaro s ##EJ///
KK5666 6 ( ( OW--- ".. gv .
''''>257GHHH
KK011111r rg target_schemasc " fdt j d t j D D }t d| | | | | t d d S )Nc 0 g | ]}|j j v |S r )_metarn )rW r] rp s r rY z*recreate_schema_models.. s7 ; / / 0 / /r c 6 g | ]}t j | S r rT rV s r rY z*recreate_schema_models.. s3 )&11 r z%rz%Schema models recreated successfully.)r_ r` r ra r r bind
create_tables)rg rp models_to_creates ` r rl rl s _ ! 5
KK&''' GG %&&&
KK788888r r[ )rd N)&loggingr_ r
contextlibr r shutilr sqlite3r r playhouse.sqlite_extr defence360agent.applicationr defence360agentr
defence360agent.contracts.configr defence360agent.modelr getLoggerr r r% r r r+ r5 r r r" r r$ ro liststrrl r r r r s * * * * * * * * 2 2 2 2 2 2 + + + + + + & & & & & & 2 2 2 2 2 2 0 0 0 0 0 0
8 $ $ Y B@ @ @F 4 $
2
2
2
2 99+/99 9 9 9 9 9 9r