5&˷ . d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZ d dlmZ d dlmZmZ d d lmZ d d
lmZmZ d d lmZ d d l m Z d d
l m Z m Z e e! Z"dZ#dZ$ dZ% G d de& Z' G d de Z(dS ) N) dataclass) getLogger)Path)Callable)urljoin)Request)APIAPIError)
LicenseCLN)atomic_rewritesafe_cancel_task)DAY)g)DeadlockDetectingLock
DeadlockError
< c e Zd ZdZdS )IAIDTokenErrorz$Can't get iaid token for any reason.N)__name__
__module__ __qualname____doc__ S/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/internals/iaid.pyr r & s ....r r c e Zd ZdZ eej e d Z eej e d Z eej e d Z
eej e d Z e d Z
e
dz Ze
dz Ze
d z Ze
d
z Zg g g d Z e Z ej Z ed
G d d Zedddefd Z e dddefd Z e d Z e d Z ed Z!e d Z"ed)d Z#e d Z$e d Z%e d efd Z&e d Z'e d*d# Z(e d$ Z)e d% Z*e d+d& Z+e d' Z,e d+d( Z-dS ),IndependentAgentIDAPIz/api/auth/agent/{}registeractivateloginz
token-infoz/var/imunify360iaidz
iaid-passwordz
iaid-tokenziaid-activated)r r! r" T)frozenc J e Zd ZU g dZeed< eed< eed< eed< eed< dS ) IndependentAgentIDAPI.TokenInfo)validr# license_status server_id
need_renewr' r# r( r) r* N)r r r __slots__bool__annotations__strr r r TokenInfor& > sX
r r/ r )timeoutcoroc K t j |t j d|z t z z d {V | | d {V d S )N )asynciosleeprandom randrange_TIMEOUT_MULTIPLICATOR)r1 attemptr0 argss r _retry_on_errorz%IndependentAgentIDAPI._retry_on_errorM s} m f&qG|447MM M
dDkr c
f d | j | D | j |< t | j | dk rXt j }| j | | | j |g|R ||d d S t d| d S )Nc : g | ]}| |S r )done).0tasks r
z3IndependentAgentIDAPI._add_task..W s5
TYY[[
r r3 r9 r0 z Task %s already in retry queue) _taskslenr4 get_event_loopappend create_taskr; loggerinfo)clstyper1 r9 r0 r: loops r _add_taskz IndependentAgentIDAPI._add_taskU s
Z -
4 sz$ A % %)++D Jt # # 'C' # -4g
KK8$ ? ? ? ? ?r c @ | d| j d d S )Nr! r r9 )rM r! rJ s r add_initial_taskz&IndependentAgentIDAPI.add_initial_taskf s"
j#,
:::::r c K | j D ]N\ }}|D ]F}| s0t | d {V t d| GOd S )NzRetry task %s was canceled.)rC itemsr> r
rH rI )rJ rK tasksr@ s r shutdownz IndependentAgentIDAPI.shutdownj s :++-- E EKD%
E
Eyy{{ E*4000000000KK =tDDD
E E Er c 4 t j d j S )N_imunify)grpgetgrnamgr_gidr r r _gidzIndependentAgentIDAPI._gidr s |J''..r c j | j r| j S d S N) IAID_FILEexists read_textrP s r get_iaidz IndependentAgentIDAPI.get_iaidv s1 = ! ! -=**,, ,tr NPOSTc ddi}|| | t | |||r&t j | nd S )Nz Content-Typezapplication/json)methodheadersdata)updater jsondumpsencode)urlre rd kwargs_headerss r _requestz IndependentAgentIDAPI._request| sh "$67 OOG $ $ $ 06@F##**,,,D
r c L t d | j | j fD S )Nc 3 > K | ]}| V d S r] )r_ )r? iaid_files r z6IndependentAgentIDAPI.is_registered.. sB
r )allr^ IAID_PASSWORD_FILErP s r
is_registeredz#IndependentAgentIDAPI.is_registered s:
!mS-C D
r c K t rGt d{V t rt d | j d }|st d |S # t $ r}t d| |d}~ww xY w)zWEnsure that iaid token is up to date
Return iaid token or raise IAIDTokenError.NzIAID token is expiredascii)encodingzIAID_TOKEN_FILE is emptyz Can't get iaid token, reason: )r is_token_expiredr" r IAID_TOKEN_FILEr` strip Exception)rJ tokenes r get_tokenz IndependentAgentIDAPI.get_token s ! 1 1 3 3 >'--// / / / / / / /$5577
>$%<=== N'1171CCIIKKE
A$%?@@@L N N N !E!!E!EFFA M Ns $?B$ $
C.CCreturnc l K | d {V }d|i}| | j |d }| | d {V }| d }|t d| | j di |S # t $ r}t d|d| |d }~ww xY w) NzX-AuthGET)re rd
token_infozwrong response %rzincomplete token_info z: r )r rn
TOKEN_INFO
async_requestgetr
r/ TypeError)rJ
iaid_tokenre requestresultr} r~ s r _get_token_infoz%IndependentAgentIDAPI._get_token_info s ==??******
Z(,,s~wu,MM((11111111
<(( =.77 7 O 3=))5)) ) O O O(UUUAAFGGQ N Os B
B3B..B3c t j | j }|j }n# t $ r d}Y nw xY wt j |z
t k S )Ng )osstatrz st_mtimeFileNotFoundErrortimer )rJ r r s r ry z&IndependentAgentIDAPI.is_token_expired sb %73.//D }HH ! HHH y{{X%++s # 22Fr3 c
` K | j } | j 4 d {V | rF|r|r d d d d {V d S |,|| k r d d d d {V d S t }t j }|r||d< | j | j fi |} | | d {V }| j
d t t | j |d | j d| d t t | j |d | j d
| d {V n# t& $ r} t( d | | | j | j d k s | j d
k r/|t. k r$| d| j |||dz | n"t( d|j || Y d } ~ d d d d {V d S d } ~ ww xY w d d d d {V d S # 1 d {V swxY w Y d S # t8 $ r t( d| Y d S w xY w)Nr) T
missing_okr# backupuidgid permissionspasswordi )r r z0Something went wrong on register %r - attempt %s r r3 rO z-Failed to register (%s) after %s attempts: %rz >$) ,,..((((((((W NNJ
- =C// =C//!J..
& L ! 0 #aK$+
& K # , # FFFgC
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
)0V )GC
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
) C
)H LLN
s~
J I1J I12J ;I16F*7B3I1*
I4B
I>I1J II1 J 1
I;;J >I;?J %J-,J-c P K | j s | d{V dS t j }| d{V }|j | d k s |j | d k r7t
d| | d{V dS | j
}|j r|j |k s|j r | d{V dS dS )z!Check whether the agent activatedNstatusidzGot a corrupted token: %r)r r_ r! r r r r( r r) rH r
reactivater^ r` r' r# r* r" )rJ licr} r# s r ensure_is_activated_and_validz3IndependentAgentIDAPI.ensure_is_activated_and_valid
sJ &--// ,,.. F"$$))++++++++ 377 $
$
_
-
- LL4e < < <.."" " " " " " " " F}&&(({ ejD 0 0E4D 0))++ 1 0r c > | j j S r] )rt r r rP s r r z)IndependentAgentIDAPI._get_credentials_ts s %**,,55r c K | j r | t d< d S | s6t
d | d {V d S t j rL| t d< |
r| d {V d S t j }|s t
d d S | j
4 d {V | j r0| t d< d d d d {V d S | j }| j }| }| | j ||| }|t d< d} | | d {V | j | j d | d {V n# t0 $ r}t
d|| |j r|j d k rd}nr|j rI|j d
k s |j d k r3|t4 k r(| d | j |d
z |t: n"t
d|j || Y d }~nd }~ww xY wd d d d {V n# 1 d {V swxY w Y |r | d| d {V d S d S )Nr# z&need to register first before activatez9Can't continue iaid activation: no valid license is found)r# r licenseFTr z.Something went wrong on activate %r attempt %s r r r! r3 rB z-Failed to activate (%s) after %s attempts: %rr r ) r r_ ra r ru rH r r r is_freery r" r _activate_lockr^ r` rt r rn ACTIVATE_URLr touchrz r r
r r rM r! _ACTIVATE_MINIMUM_TIMEOUTr r ) rJ r9 r r# r credentials_tsr need_to_registerr~ s r r! z IndependentAgentIDAPI.activate! sc " ) ) + + AfI F "" NNC D D D,,.. F AfI##%%
"iikk!!!!!!! F"$$ NNK
F% 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " &--//
LLNN& 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " =**,,D-7799H 4 4 6 6Nll th # G AfI $ $
"''000000000'--///B #**d*;;;iikk!!!!!!!!C
D
= Q]c%9%9'+$$M #-- #1E1E *,,
MM" ! ' 9 " LLG ( 3
#2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 " 2 "f P,,T,OO O O O O O O O O O P PsD !7L+A0L 4I6L
K=B#K83L8K==L
L L c t K | j d | d {V d S )NTr )r r r! rP s r r z IndependentAgentIDAPI.reactivatek sF &&$&777llnn r c K | s t d d S | j }| j }| }| | j || } | | d {V }t t | j |d | j
d| d d S # t $ r}t d|| |t" k r]|j |j dk r"| d | j |d
z | nW|j d k r | d
| d {V n(t d|j || Y d }~d S Y d }~d S Y d }~d S d }~ww xY w)Nz#need to register first before login)r# r r} r r r z/Something wrong happened on login %r attempt %sr r" r3 rO r Tr z*Failed to login (%s) after %s attempts: %r)ru rH r r^ r` rt r rn LOGIN_URLr r r. rz r_ r[ r
r r r rM r" r r )rJ r9 r# r r r r r~ s r r" zIndependentAgentIDAPI.loginp sS "" LL> ? ? ? F}&&(()3355 0022,,s}4(,KK ,,W55555555F.
C'((w*1133HHJJ !
- NNA1g
##=(AMS,@,@MM GaK " ]c)) ,, " ' @$ s D
GB&G G)Nrb )FNr3 )r3 ).r r r API_PATHr r _BASE_URLformatr r r r r IAID_DIRr^ rt rz r rC r r r4 Lockr r r/ staticmethodr r; classmethodrM rQ rU r[ ra rn ru r r ry r r r r! r r" r r r r r * s #H73=(//**E*EFFL73=(//**E*EFFL
xw'?'?@@I (E(EFFJt%&&H6!I!O3 -O"%55 F
+*,,N!W\^^NYd FG H \ EF @ @ @8 @ @ @ [@ ; ; [; E E [E / / \/ [
\
[
N
N [
N Oi O O O [ O , , [, P P P [Pd [ $ 6 6 [6 GP GP GP [GPR [ ( ( ( [( ( (r r ))r4 rX rh r r6 r dataclassesr loggingr pathlibr typingr urllib.parser urllib.requestr defence360agent.api.serverr r
!defence360agent.contracts.licenser defence360agent.utilsr r
defence360agent.utils.commonr &defence360agent.internals.global_scoper 1defence360agent.internals.deadlock_detecting_lockr r r rH r r8 r RuntimeErrorr r r r r r s
! ! ! ! ! ! " " " " " " 4 4 4 4 4 4 4 4 8 8 8 8 8 8 B B B B B B B B , , , , , , 4 4 4 4 4 4
8
/ / / / /\ / / /o o o o oC o o o o or