t d dl Z d dlZd dlmZ d dlmZ d dlmZmZ d dl m
Z
d dl m Z d dl
mZmZmZ e j e Z e Zd Zd Zd
Zd Zd Zd
Zd Zd Z d Z d Z dS ) Nwraps)eula)CoreUserType)
LicenseCLN) MessageType)Genmessage_id_gen publisherc < t fd }|S )Nc K | i | d {V }t |t s
J d|z t j |d< |S )N Result should be a dictionary %slicense)
isinstancedictr license_infoargskwargsresultfs Y/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/rpc_tools/middleware.pywrapperz add_license..wrapper ss q$)&))))))))&$''
. 7
'
' 3 5 5y
r r r s ` r add_licenser s3
1XX X Nr c < t fd }|S )Nc K | i | d {V }t |t s
J d|z t j }|d | d | d d|d< |S )Nr status license_typeeligible_for_imunify_patch)r r! r" r )r r r r get)r r r r r s r r z!add_license_user..wrapper$ s q$)&))))))))&$''
. 7
'
)++ h' #KK 7 7*1++,+ +
y
r r r s ` r add_license_userr$ # s3
1XX X" Nr c < t fd }|S )Nc K | i | d {V }t |t s
J d|z d }t j rt j st j d {V sx t j t j t j d}n=# t $ r0}dd t | dd}Y d }~nd }~ww xY w||d< |S )Nr )messagetextupdatedzFailed to read EULAzFailed to read EULA: {} r )
r r r is_validis_freer is_acceptedr' r( r) OSErrorformatstr)r r r eula_dicter s r r zadd_eula..wrapper: s6 q$)&))))))))&$''
. 7
'
**<*>*>
)++++++++
#'<>> $ #'<>>! !II
#8 9 @ @Q H H#%! !IIIIII #v
s 69B0 0
C*:&C%%C*r r s ` r add_eular3 9 s3
1XX X0 Nr c < t fd }|S )Nc K | i | d {V }t |t s
J d|z t j |d< |S )Nr version)r r r VERSIONr s r r z add_version..wrapperW sl q$)&))))))))&$''
. 7
' !Ly
r r r s ` r add_versionr8 V s3
1XX X Nr c < t fd }|S )Nc 4 K | i | d {V \ }}||dS )N) max_countitems )r r countr< r s r r zmax_count..wrappere s? Q / / /////// u "U333r r r s ` r r; r; d s3
1XX4 4 4 4 X4 Nr c < t fd }|S )Nc 8 K | i | d {V \ }}}|||dS )N)r; countsr< r= )r r r; rA r< r s r r zcounts..wrappern sD )*D);F););#;#;#;#;#;#; 65 &&5IIIr r r s ` r rA rA m s8
1XXJ J J J XJ Nr c < t fd }|S )Nc K t j dt t j d 5 } | i | d {V }d |D |d< |cd d d S # 1 swxY w Y d S )NalwaysT)recordc L g | ]!}d |j j "S ) )joinr' r ).0ws r
z5collect_warnings..wrapper..| s( !J!J!Jq#((19>":":!J!J!Jr warnings)rL simplefilterDeprecationWarningcatch_warnings)r r warnsr r s r r z!collect_warnings..wrapperw s h(:;;;
$D
1
1
1 U 1d-f--------F!J!JE!J!J!JF: s A A#&A#r r s ` r collect_warningsrQ v s3
1XX X Nr c < t fd }|S )Nc Z K | i | d {V }t |t sd|i}|S )Nr< )r r r s r r z!default_to_items..wrapper sN q$)&))))))))&$'' ' v&F
r r r s ` r default_to_itemsrT s3
1XX X Nr c < t fd }|S )a
This middleware copies 'remote_addr' to 'client_addr'.
This is needed because send_command_invoke middleware may remove
remote_addr parameter from request.
Used for endpoints that need remote_addr in their logic.
:param f:
:return:
c l K | d d }|| d< | g|R i | d {V S )Nparams remote_addr client_addr)r# )requestr r rX r s r r z%preserve_remote_addr..wrapper s[ h'++M:: !,
Qw0000000000000r r r s ` r preserve_remote_addrr[ s5 1XX1 1 1 1 X1 Nr c < t fd }|S )Nc T K d }|r |d }n d|v r|d }|t | d }d|vr?d }t | dk r |d }n d|v r|d }|t j k rd|d< d|v rd|d< t j | d || d
d }t j | t j
|t d
| | d {V | d dd | g|R i | d {V S )Nr sinkrW user Tpasswordz***commandcalling_process)rb rW rc create)stagerX )
r lenr NON_ROOTr
CommandInvokepopr enrichr report_reporter_gen_createprocess_message)rZ r r r^ rW user_typemsgcoros r r z,send_command_invoke_message..wrapper s "7DD
v
&>D '(+,,FV## t99q== $QIIv%% &vI 111%)F6N V##%*z" + * ' ,=t D D C
!# & & & S"6h G G G G&&s++ + + + + + + + H ! !- 6 6 6T'3D333F333333333r r )rp r s ` r send_command_invoke_messagerq s4
4[[!4 !4 !4 !4 [!4F Nr ) loggingrL functoolsr defence360agent.contractsr defence360agent.contracts.configr r !defence360agent.contracts.licenser "defence360agent.contracts.messagesr 2defence360agent.internals.message_status_publisherr
r r getLogger__name__loggerrl r r$ r3 r8 r; rA rQ rT r[ rq r= r r r| si * * * * * * ; ; ; ; ; ; ; ; 8 8 8 8 8 8 : : : : : :
8 $ $suu , : *% % % % %r