$pTo d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z ddl
mZ ddlZddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
l m Z ddl m Z ddl!m"Z"m#Z#m$Z$ ddl%m&Z& e
e' Z(dZ)ej* de+de,fd Z-dedee+ fdZ. e/ fdee+ de/e/e+e+f df fdZ0d Z1d ed d Z2e'd k r e2 dS dS ) zbThis module import peewee_migrate and apply migrations, for Imunify-AV
it's entrypoint for service N)Iterable) getLogger)migrator)SqliteExtDatabase)app) configure)Core)Model)Router)systemd_notifier)db) tls_check)write_pid_fileIM360_RESIDENT_PID_PATHcleanup_pid_file)recreate_schema_modelsz/usr/bin/imunify-residentlog_msgreraisec # v K dV dS # t $ r$ t | | |r Y dS w xY w)z
Logs error in case of exception.
Depending on `reraise`:
- re-raise exception and don't include exception info in the log operation
- do not re-raise exception and include exception info in the log operation
N)exc_info) Exceptionloggererror)r r s L/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/migrate.py exc_handlerr ) se
W7{ 333 s
*88r
migrations_dirsc ~ t | |t }t t _ | dS )z4Apply migrations: restructure db, config files, etc.)r r N)r r r LOGGERrun)r
r routers r apply_migrationsr! : s;
' F HO
JJLLLLL attached_dbs.c t j t j t j g }|D ]1\ }}t j d||f | | 2 t d t j t j j
t j d 5 t dd 5 t! t | ddd n # 1 swxY w Y ddd n # 1 swxY w Y t d t j d 5 t d d
5 t# t | t d d
5 t! t | ddd n # 1 swxY w Y ddd n # 1 swxY w Y ddd n # 1 swxY w Y t j dS # t j w xY w) a>
Apply migrations and recreate attached databases.
The workflow:
1. Apply migrations
2. Regardless whether the migrations were applied - recreate attached databases
3. If the recreation of the attached databases was successful - apply migrations again
- this is done to verify that migrations will successfully apply in future for the recreated databases
- the recreation + the migrations in this step are within the same transaction,
so databases will only be recreated if the migrations can applied after the recreation.
z
ATTACH ? AS ?z Applying database migrations... EXCLUSIVEzError applying migrationsF)r Nz Recreating attached databases...z#Error recreating attached databasesTz=Error applying migrations after recreating attached databases)r reset db_instanceinitr
PATH execute_sqlappendr infor notify
AgentState MIGRATINGatomicr r! r close)r r# attached_schemasdb_path schema_names r prepare_databasesr5 H s2 " OUZ , - - ';1GHHH ,,,, 5666 0 ; EFFF
,
, ; ;k '/
/
/
; ;
[/ : : : ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;
6777
,
,
?
?k 14/
/
/
?
?
#;0@ A A A
?
?
!o>>>
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?
?" s )AG' :D C." D.C2 2D5C2 6D9 G' D G' D
1G' >G'F07F
F0F F0 F !F0$ G0F4 4G7F4 8G; G' G G' G G' 'G<c t d| t dt j t j t j t d t j d d S )Nz$Received signal %s in signal_handlerz0waiting %d seconds so that migrations can finishExitingr ) r warningr %SIGNAL_HANDLER_MIGRATION_TIMEOUT_SECStimesleepr, sysexit)sig_s r signal_handlerr@ ~ sn
NN93???
NN: 2 Jt9:::
KK HQKKKKKr" defence360agent) start_pkgr c t j t j t j fD ] }t j |t | dk rt t t j t j
| t j j
t j t" t$ j t$ j f }| | t/ j t. j j t d t/ j t. j j | dk rqt j d t d t j t@ t@ gtB j" dd z dS t j tB j# tB j# d d
$ | gtB j" dd z dS # tJ $ r! | dk rtM t Y dS Y dS w xY w) zoEntry point for Imunify-AV service. Apply migrations,
and then replace process with {start_pkg}.run module.zim360.run_resident)targetargszStarting main process...T)exist_okz Run imunify-resident service Nz-mz{})'signalSIGINTSIGTERMSIGHUPr@ r r osumaskr
FILE_UMASKrA internalsr reconfigure threadingThreadr5 r MIGRATIONS_DIRSMIGRATIONS_ATTACHED_DBSstartjoinr r- r. READYr, STARTING GO_FLAG_FILEtouchexecvGO_SERVICE_NAMEr< argv
executableformatr r )rB r r> migration_threads r r r s*
v~v}= + +
c>****"6 , , , 2 3 3 3
!!! !(44666$+$%s'BC
0 ; ABBB ./// 0 ; DEEE , , , # #T # 2 2 2 KK6 7 7 7 H # (122,
H t{{9'='=>!"" M
6 6 6 , , , 4 5 5 5 5 5 5 - , ,6s FH- AH- -$II__main__)3__doc__
contextlibrL r< rH rQ r: collections.abcr loggingr peewee_migrater playhouse.sqlite_extr defence360agent.internals.loggerrA defence360agent.applicationr $defence360agent.application.settingsr defence360agent.contracts.configr r
defence360agent.routerr defence360agent.subsysr defence360agent.model.instancer
r' defence360agent.modelr defence360agent.utilsr r r defence360agent.utils.check_dbr __name__r r\ contextmanagerstrboolr r! tupler5 r@ r r" r rx s
$ $ $ $ $ $ # # # # # # 2 2 2 2 2 2 ' ' ' ' + + + + + + : : : : : : 1 1 1 1 1 1 2 2 2 2 2 2 ) ) ) ) ) ) 3 3 3 3 3 3 < < < < < < + + + + + +
8 -
t
* Xc] 162 2 c]2 c3h,-2 2 2 2 l ') (6 (6 (6 (6 (6V zCEEEEE r"