@S09 F d dl Z d dlZd dlZd dlmZ d dlmZmZ d dlm Z d dl
m Z d dl m
Z
mZmZ d dlZd dlZd dlmZ d dlmZ d d lmZ d d
lmZmZm Z m Z d Z d Z ddZ ddee ddf fdZ!dde
ee fdZ" ed d Z#d Z$ ddZ%dS ) N)suppress) lru_cachewraps)chain)Path)OptionalTuple Generator) SimpleRpc)run_in_executor)ValidationError) AV_PID_PATHIM360_NON_RESIDENT_PID_PATHIM360_RESIDENT_PID_PATHantivirus_modec V t j rt nt } | rzt j }t t 5 t |
}||k ot j | cddd S # 1 swxY w Y dS )z/Check if non-resident agent instance is runningNF)
r enabledr r existsosgetpidr Exceptionint read_textpsutil
pid_exists)rpc_process_pid_path current_pidpids T/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/rpc_tools/utils.pyrpc_is_runningr s
&-N 3N ""$$ Aikk
i
A A*446677C+%@&*;C*@*@ A A A A A A A A A A A A A A A A 5s ;B B"%B"c t j rt S t j rMt j } t t j }|| k ot j
| S dS )z&Check if the agent instance is runningF) ConfigSOCKET_ACTIVATIONr r r r r r r r r )r r s r
is_runningr$ ' sl
%'' =ikk )35566k! resultr5 s r get_schema_pathsrH G si $
,T%[[^^"F >> ', 6D
MM K. . 6
Mr% c g }t | D ]$}| t | %t t | S N)rH appendr: dictr )r> r& base_paths r prepare_schemarO S sM
F%e,, / /
l9--.... v r% c < t fd }|S )Nc d K t t j fd d {V S )Nc i S rK )argsfkwargss r z.wrapper.._ s aa.@.@.@ r% )r asyncioget_event_loop)rT rV rU s ``r wrapperz*run_in_executor_decorator..wrapper\ sW $ " $ $&@&@&@&@&@&@
r% )r )rU rZ s ` r run_in_executor_decoratorr[ [ s3
1XX
X
Nr% c g }|D ]w}|d | d| }
} |
r,|r*| | | |
N| | | | xt | } t | } | dk r/| | k r)| | | | |rt | i S )aP
:param list affected: IPs that were changed during operation
:param list of tuples || list of str not_affected: IPs & it's listnames
that weren't changed during operation
:param list all_list: list of all IPs that take place in operation
:param str success_warning: msg if IP was changed
:param str failure_warning: msg if IPs wasn't changed and it's absent
in any other lists
:param str in_another_list_warning: msg if IPs wasn't changed , however
it present in another list
:return list of st warnings: msg to be printed
reclistnamerI )getrL formatlenr
)
affected not_affected
dest_listnameall_listsuccess_warningfailure_warningin_another_list_warningwarningsitemrecordr^ num_deleted total_nums
r generate_warningsrn e s * H K K ;](K(K K/ K OO 3 : :68 L L M M M M OOO 2 26= I I J J J Jh--KH
I1}}k11..{IFFGGG ( h'''
Ir% )Nr& rK )&r+ rX r
contextlibr functoolsr r itertoolsr pathlibr typingr r r
r. r defence360agent.contracts.configr r" $defence360agent.model.simplificationr "defence360agent.rpc_tools.validater
defence360agent.utilsr r r r r r$ r: r) rH rO r[ rn rS r% r rx s
& & & & & & & & - - - - - - - - - -
@ @ @ @ @ @ @ @ @ @ @ @ > > > > > > ) ) ) ) # tT4 HU4[ 1 1 " !& & & & & &r%