LaaJ
` d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z m
Z
ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ ej e Z e d Z G d
dej Z G d de Z G d de Z G d d Z de!de!de!de!fdZ"de!fdZ#dd d d d Z$de!fd Z% e e d!d" d#e$j& $ e e d%d& d' ee"dd( $ fZ' e e d)d* e d+ e% fd,e#$ e e d-d. d/ ee(d0 $ e e d1d2 d3e)$ e e d4d2 d5e)$ fe'z Z*d;d6Z+d.
Copyright © 2019 Cloud Linux Software Inc.
This software is also available under ImunifyAV commercial license,
see
N)suppress)partial)chain)Callable)SqliteDatabase TextField) LocalConfig)
im360_present)PleskSdk)find_revisium_db)SettingsExtractorc T e Zd ZdZ e Z e Z G d d ZdS )KeyValuez#ex-Revisium settings database modelc e Zd ZdZdS )
KeyValue.MetaFN)__name__
__module__ __qualname__ primary_key V/opt/imunify360/venv/lib/python3.11/site-packages/imav/migrations/004_plesk_configs.pyMetar 0 s r r N)r r r __doc__r keyvaluer r r r r r * sX --
)++C IKKE r r c e Zd ZdZd ZdS )Revisiumz'ex-Revisium database settings extractorc f t t j | j k j S N)r getr r selfs r _getz
Revisium._get7 s ||HLDH 455;;r Nr r r r r$ r r r r r 4 s) 11< < < < s %dh ===r Nr% r r r r' r' ; s) 44> > > > >r r' c X e Zd ZdZdeeedf z deeedf z defdZd Z dS )
ConfigMappingz
Transform a single or multiple dependent ex-Revisium settings parameters
to a single ImunifyAV/360 config value and map it to a single or multiple
config keys
source.target converterc t |t s|f}t |t s|f}|| _ || _ || _ dS )z
:param source: settings extractor(s) for ex-Revisium / Plesk
:param target: name(s) of target config parameter(s) for ImunifyAV/360
:param converter: a callable to convert source value(s) to a target one
N)
isinstancetupler- r. r/ )r# r- r. r/ s r __init__zConfigMapping.__init__I sO &%(( YF&%(( YF "r c d | j D } | j | }n:# t $ r-}t d| j || Y d}~dS d}~ww xY wt t
| j |ft | j z S )z@Get target value from source value(s) and assign it to target(s)c 6 g | ]}| S r )r! ).0r- s r
z)ConfigMapping.convert..` s @@@& @@@r z$Fail to convert %r value(s) (%r): %rNr ) r- r/ Exceptionloggerwarningr2 zipr. len)r#
source_values target_valuees r convertzConfigMapping.convert^ s @@DK@@@
)4>=9LL NN6
22222 S|oDK8H8H&HIIJJJs
A"AAN)
r r r r r
r2 strr r3 r@ r r r r, r, B s #!E*;S*@$AA# eCHo%#
# # # #*K K K K Kr r, minimummaximumr returnc > t | t || S )z.
Ensure that a value is within limits
)maxmin)rB rC r s r clamprH p s wE7++ , ,,r c d}t t 5 || z |z }ddd n # 1 swxY w Y t dd| S )zK
Calculate ImunifyAV/360 CPU intensity based on a value in a range
N )r ZeroDivisionErrorrH )r max_value intensitys r
intensity_cpurP w s I # $ $ 3 3 %2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 Ay ! !!s ,00nonedayweekmonth)neverdailyweeklymonthlyc N t t j pddz d S )z-A half of available CPUs/cores (at least one) rK )rF os cpu_countr r r get_max_possible_cpur] s" #!)1 - --r ra_auto_scan_periodrX z MALWARE_SCAN_SCHEDULE.interval)r/ ra_hour_auto_scanrJ zMALWARE_SCAN_SCHEDULE.hour ra_max_worker_countrK ra_max_possible_worker_count)zMALWARE_SCAN_INTENSITY.cpuz$MALWARE_SCAN_INTENSITY.user_scan_cpura_keep_backups_daysrL z(MALWARE_CLEANUP.keep_original_files_daysrZ
ra_trim_filesTz,MALWARE_CLEANUP.trim_file_instead_of_removalra_use_ignore_list_by_userz PERMISSIONS.user_ignore_listc t | i }t rt nt }t j d |D D ]?\ }}|}| d ^ }}|D ]}| |i }|||< @t
dt | |S )Nc 3 > K | ]}| V d S r )r@ )r6 ms r z.migrate_imav4plesk_settings.. s* ,K,KQQYY[[,K,K,K,K,K,Kr .z$Those settings=%s are to be migrated) dbinitr
scan_time_mapping_only all_mappingsr
from_iterablesplit
setdefaultr9 inforA ) databaseconfigmappingsr. r dpathparamr s r migrate_imav4plesk_settingsry s GGH F)6 J%%lH ,,K,K(,K,K,KKK
||C(( u & &C S"%%AA%
KK6F DDD Mr Fc : |rd S t }| t d d S t | x}r%t |d d S d S # t $ r&}t d| Y d }~d S d }~ww xY w)Nz/No legacy ImunifyAV database found. Skipping...F) normalizez2Failed to migrate ImunifyAV for Plesk settings: %r)r r9 rr ry r dict_to_configr8 r: )migratorrs fakekwargsrevisium_db_pathmigrated_configr? s r migrater s ')) EFFFP 9:J K K K? K MM ( ( E ( J J J J J K K P P PKQOOOOOOOOOPs 4A* *
B4BBc d S r r )r} rs r~ r s r rollbackr s Dr __main__r )F)/r loggingr[
contextlibr functoolsr itertoolsr typingr peeweer r defence360agent.contracts.configr imav.migration_utils.otherr
imav.migration_utils.plesk_sdkr imav.migration_utils.revisiumr imav.migration_utils.settingsr
getLoggerr r9 rk Modelr r r' r, intrH rP schedule_interval_mappingr] __getitem__rm rF boolrn ry r r printr r r r s * , , , , , , , , 8 8 8 8 8 8 4 4 4 4 4 4 3 3 3 3 3 3 : : : : : : ; ; ; ; ; ; 8 $ $^D rx < < < < < < < <> > > > > > > >+K +K +K +K +K +K +K +K\-3 - -S -S - - - -"s " " " "
.c . . . . M
#Y//(+7
M
!1%%$'%B''
M E' + + E02F2F2H2H I I
M
$a((2'#q//
M
ot$$6
M-t44& -6 7 < $
P
P
P
P z E
%
%&6&6&8&8
9
9::::: r