HOME

Mini Shell 1.0
DIR:/proc/self/root/usr/libexec/kcare/python/kcarectl/
Current File : //proc/self/root/usr/libexec/kcare/python/kcarectl/auth.pyo
�
,w�gc@s�ddlZddlZddlZddlmZddlmZddlmZddlmZddlmZddlm	Z	dd	lm
Z
dd
lmZm
Z
mZmZdZdZd
Zd�Zd�Zd�Ze
jd��Zd�Zd�Zed�Zd�Zd�Zed�Ze
jd��Zd�Z d�Z!d�Z"d�Z#d�Z$d�Z%dd �Z&d!�Z'dS("i����Ni(terrors(tconfig(t	constants(t
http_utils(t	log_utils(tplatform_utils(tutils(t	urlencodetjson_loads_nstrtURLErrort	HTTPErrors/etc/sysconfig/kcare/systemids&/etc/sysconfig/kcare/systemid.almacares/var/imunify360/license.jsoncCs�tjjt�sdSttd��d}xZ|D]R}|jd�\}}}|j�dkri|j�Stj	dj
t���q/WWdQXdS(Ntrt=t	server_idsUnable to parse {0}.(tostpathtexiststSYSTEMIDtNonetopent	partitiontstripRt
KcareErrortformat(tfdtlinetparamt_tvalue((s*/usr/libexec/kcare/python/kcarectl/auth.pyt	_systemids

cCsBtjjt�sdSttd��}|j�j�SWdQXdS(NR(RRRt
ALMA_SYSTEMIDRRtreadlineR(tf((s*/usr/libexec/kcare/python/kcarectl/auth.pyt_alma_systemid$sc	Cswtjjt�sdSi}tt��@}|j�}|rdyt|�}Wqdtk
r`qdXnWdQX|j	d�S(Ntid(
RRRtIM360_LICENSE_FILERRtreadRt	Exceptiontget(tdataR tcontent((s*/usr/libexec/kcare/python/kcarectl/auth.pyt_im360_systemid+s

cCst�pt�pt�S(sqGet server_id or None if not present.

    Lookup order: SYSTEMID then IM360_LICENSE_FILE then ALMA_SYSTEMID
    (RR)R!(((s*/usr/libexec/kcare/python/kcarectl/auth.pytget_serverid:scCstjt�dS(N(RtunlinkR(((s*/usr/libexec/kcare/python/kcarectl/auth.pyt_rm_serveridCscCstjtdj|��dS(Nsserver_id={0}
(Rtatomic_writeRR(R
((s*/usr/libexec/kcare/python/kcarectl/auth.pyt_set_server_idGscCsd}y�t�}|dkr8|s4tjd�ndStjdj|�}tj|�}t	j
|j��}t	j|�}|ddkr�t
�|s�tjd�q�n+|s�tj|�tjd|d�nWn,tk
r}|stj||�qnXdS(Ns1Error unregistering server: cannot find server ids&/unregister_server.plain?server_id={0}tsuccessttruesServer was unregisteredsError unregistering server: tmessage(RR*RtlogerrorRtREGISTRATION_URLRRturlopenRtnstrR$tdata_as_dictR,tloginfoR
tprint_cln_http_error(tsilentturlR
tresponseR(treste((s*/usr/libexec/kcare/python/kcarectl/auth.pyt
unregisterKs*	
cCsPdGHtj�}|dkr!dStj�tj�}ddl}|dkr_|jd�n|jj�tdd�}tdd�}tj|j	�|j
j	��tj|j	�|jj	��tj|j	�|jj	��xbtrKt
jd
�t|�\}}}|dkr�|r�t|�t|�|jd�q�q�WdS(NsHRegister auto-retry has been enabled, the system can be registered laterii����s	/dev/nullRsa+i<iii (RtforktsetsidtsystexittstdouttflushRtdup2tfilenotstdintstderrtTruettimetsleept
_try_registerR.t_set_auth_token(R:tpidRAtsitsotcodeR
t
auth_token((s*/usr/libexec/kcare/python/kcarectl/auth.pyt_register_retrycs,

	


cCs�yitj|�}|jjtjd�}tjtj	|j
���}t|d�|jd�|fSWnOtt
fk
r�}tj||�dStk
r�tjjd|�dSXdS(NRQR
s)Exception while trying to register URL %s(NNN(NNN(RR4theadersR&RtAUTH_TOKEN_HEADERRRR6R5R$tintR
R	RR8R%tkcarelogt	exception(R:R;RRR<R=((s*/usr/libexec/kcare/python/kcarectl/auth.pyRL}s$
cCs�ytt�Wn!tk
r1tjjd�nXtj�}td|fd|fg�}dj	t
j|�}t|�\}}}|dkr�t
|�t|�tjd�dS|dkr�tjd�n�|d	kr�tjd
�n�|dkrtjd�nj|d
kr'tjd�nN|dkrCtjd�n2|dkr_tjd�ntjdj	|��|r�t|�dS|p�dS(Ns9Exception while trying to unregister URL before register.thostnametkeys{0}/register_server.plain?{1}isServer RegisteredisAccount LockedisInvalid Keyis}You have reached maximum registered servers for this key. Please go to your CLN account, remove unused servers and try again.is[IP is not allowed. Please change allowed IP ranges for the key in KernelCare Key tab in CLNisEThis IP was already used for trial, you cannot use it for trial againisfThis IP was banned. Please contact support for more information at https://www.kernelcare.com/support/sUnknown Error {0}i����(R>RIR%RRWRXRtget_hostnameRRRR3RLR.RMR7R2RS(RZtretryRYtqueryR:RQR
RR((s*/usr/libexec/kcare/python/kcarectl/auth.pytregister�s>





cCstjtj�S(N(Rttry_to_readRtAUTH_TOKEN_DUMP_PATH(((s*/usr/libexec/kcare/python/kcarectl/auth.pyt_get_auth_token�scCs!|s
dStjtj|�dS(N(RR-RR`(RR((s*/usr/libexec/kcare/python/kcarectl/auth.pyRM�scOs�|jdd�}|jdt�r-t}n	tj}tj|�rXtj|||�Stj|t	�t
�d|�}tj|dd�tj�|||�S(Ntmethodt
check_licensetcounti(
tpopRRIt_check_auth_retryRtcheck_urlopen_retrytis_local_urlturlopen_basethttp_requesttget_http_auth_stringRaRR\(R:targstkwargsRbtchecktrequest((s*/usr/libexec/kcare/python/kcarectl/auth.pyturlopen_auth�s		!cCs>t�}|r:tjtjtjdj|d����SdS(Ns{0}:{1}t
kernelcare(R*RR5tbase64t	b64encodetbstrRR(R
((s*/usr/libexec/kcare/python/kcarectl/auth.pyRk�s	+cCsLt|t�r5|jdkr(t|�S|jdkSt|t�rHtSdS(Ni�i�i�(i�i�(t
isinstanceR
RQt_handle_forbiddenR	RI(R=tstate((s*/usr/libexec/kcare/python/kcarectl/auth.pyRf�s

cCs#d|krtStjrt�}|rEtjddj|�}n
tjd}y7tjtj	|dt
�j��}tj|�}Wn*t
k
r�}tj||dt
�dSX|s�|jd�r�tjjdj|��dS|ddkrt|d<tjd�tSt�ndS(
sIn case of 403 error we should check what's happen.
    Case #1. We are trying to register unlicensed machine and should try to register trial.
    Case #2. We have a valid license but access restrictions on server are not consistent yet
             and we had to try later.
    tlicenses/check.plains?server_id={0}tretry_on_500RCNRQsUnexpected CLN response: {0}t0t1s$Unable to access server. Retrying...(RzR{(RIRtCHECK_CLN_LICENSE_STATUSR*R3RRR5RR4tFalseR$R6R	RR8R&RWterrorR2t_register_trial(RwR
R:R(tinfotex((s*/usr/libexec/kcare/python/kcarectl/auth.pyRv�s*		
$

cCst�}|r�tjdj|�}y�tj|�}tj|j��}tj	|�}|sr|j
d�r�dj|�GHdSt|d�}|dkr�dGHdStdd�}|dkr�dGHn|SWqt
k
r�}tj||�dSXnt�SdS(	Ns/check.plain?server_id={0}RQsUnexpected CLN response: {0}iisKey-based valid license foundtkey_checkeds No valid key-based license found(R*RR3RRR4RR5R$R6R&RVt_get_license_info_by_ipR
RR8(R
R:R;R(R<RQtlicense_typeR=((s*/usr/libexec/kcare/python/kcarectl/auth.pytlicense_infos,	icCs�tjd}y|tj|�}tj|j��}tj|�}|dj�dkrht	|d�}|dkr�dj
|d�GHdS|dkr�|d}tj|d	�jd
�}dj
||�GHdS|dkr$|dkr$|d}tj|d	�jd
�}d
j
||�GHn|dkr�|dkr�d|kr]dj
|d�GHqedGHq�n |j
dd�}dj
|�GHWnLtk
r�}	tj|	|�n*tk
r�}
dj
|
|j��GHnXdS(Ns/check.plainR/R0RQisValid license found for IP {0}tipitexpire_dates%Y-%m-%ds?You have a trial license for the IP {0} that will expire on {1}is0Your trial license for the IP {0} expired on {1}isThe IP {0} hasn't been licenseds This server hasn't been licensedR1ts"Error retrieving license info: {0}s1Unexpected CLN response, cannot find {0} key:
{1}(RR3RR4RR5R$R6tlowerRVRtparse_response_datetstrftimeR&R
RR8tKeyErrorR(R�R:R;R(R<RQR�texpires_strR1R=RZ((s*/usr/libexec/kcare/python/kcarectl/auth.pyR� s<


cCsytjjtjd�}tjj|�r.dSytjtj	d�}t
jt
j|j
���}y�|dj�dkr�t
j|ddt�|ddkr�tj|d|d	��ntjd
j|d��dS|ddkrt
j|ddt�tjd��ntjd
��Wn"tk
rK}tj|��nXWn%tk
rt}tj|j��nXdS(Nstrial-requesteds/trial.plainR/R0R�t
ensure_dirtexpiredR�tcreateds3Requesting trial license for IP {0}. Please wait...tnasInvalid Licensei����(RRtjoinRtPATCH_CACHERRR4RR3RR6R5R$R�R-RIRtAlreadyTrialedExceptionRR7RRRtUnableToGetLicenseExceptionR�R
RQ(t
trial_markR;R<tkeR=((s*/usr/libexec/kcare/python/kcarectl/auth.pyRCs*((RrRRJR�RRRRRRRtpy23RRR	R
RRR#RR!R)tcachedR*R,R.R}R>RSRLR^RaRMRpRkRfRvR�R�R(((s*/usr/libexec/kcare/python/kcarectl/auth.pyt<module>s@"								(		
				)	#