Uiє d dl Z d dlZd dlmZ d dlmZ d dlmZmZ d dl m
Z
d dl m Z e j
e Zd Zd Zd Zd
Zd Zd Zd
Zd Zd Zd ZdS ) Nwraps)eula)CoreUserType)
LicenseCLN) MessageTypec < 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..wrapper3 s6 q$)&))))))))&$''
. 7
'
**<*>*>
)++++++++
#'<>> $ #'<>>! !II
#8 9 @ @Q H H#%! !IIIIII #v
s 69B0 0
C*:&C%%C*r r s ` r add_eular0 2 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..wrapperP sl q$)&))))))))&$''
. 7
' !Ly
r r r s ` r add_versionr5 O s3
1XX X Nr c < t fd }|S )Nc 4 K | i | d {V \ }}||dS )N) max_countitems )r r countr9 r s r r zmax_count..wrapper^ s? Q / / /////// u "U333r r r s ` r r8 r8 ] s3
1XX4 4 4 4 X4 Nr c < t fd }|S )Nc 8 K | i | d {V \ }}}|||dS )N)r8 countsr9 r: )r r r8 r> r9 r s r r zcounts..wrapperg sD )*D);F););#;#;#;#;#;#; 65 &&5IIIr r r s ` r r> r> f 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..u s( !J!J!Jq#((19>":":!J!J!Jr warnings)rI simplefilterDeprecationWarningcatch_warnings)r r warnsr r s r r z!collect_warnings..wrapperp s h(:;;;
$D
1
1
1 U 1d-f--------F!J!JE!J!J!JF: s A A#&A#r r s ` r collect_warningsrN o s3
1XX X Nr c < t fd }|S )Nc Z K | i | d {V }t |t sd|i}|S )Nr9 )r r r s r r z!default_to_items..wrapper} sN q$)&))))))))&$'' ' v&F
r r r s ` r default_to_itemsrQ | 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 rU r s r r z%preserve_remote_addr..wrapper s[ h'++M:: !,
Qw0000000000000r r r s ` r preserve_remote_addrrX s5 1XX1 1 1 1 X1 Nr c < t fd }|S )Nc 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 }| | d {V | d d d | g|R i | d {V S )
Nr sinkrT user Tpasswordz***commandcalling_process)r_ rT r` rU )r lenr NON_ROOTr
CommandInvokepopprocess_message)rW r r r[ rT user_typemsgcoros r r z,send_command_invoke_message..wrapper sb "7DD
v
&>D '(+,,FV## t99q== $QIIv%% &vI 111%)F6N V##%*z" + * ' ,=t D D C &&s++ + + + + + + + H ! !- 6 6 6T'3D333F333333333r r )rh r s ` r send_command_invoke_messageri s4
4[[$4 $4 $4 $4 [$4L Nr )loggingrI functoolsr defence360agent.contractsr defence360agent.contracts.configr r !defence360agent.contracts.licenser "defence360agent.contracts.messagesr getLogger__name__loggerr r! r0 r5 r8 r> rN rQ rX ri r: r r rs s) * * * * * * ; ; ; ; ; ; ; ; 8 8 8 8 8 8 : : : : : : 8 $ $ , : *( ( ( ( (r