­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ 3 ;j/@sddlZddlZddlZddlZddlmZmZmZmZm Z m Z m Z m Z m Z ddlmZmZmZd"ddZdd Zd d Zd d Zd#ddZe jddZddZddZddZddZddZddZd$ddZd d!Z dS)%N) config constantserrors http_utils ipv6_support log_utilsplatform_utilsserveridutils) HTTPErrorURLError urlencodeFcCsd}ytj}|dkr(|s$tjddStjdj|}tj|}t j |j }t j |}|ddkrtj tj|stjdn |stj|tjd|dWn2tk r}z|stj||WYdd}~XnXdS)Nz1Error unregistering server: cannot find server idz&/unregister_server.plain?server_id={0}successtruezServer was unregisteredzError unregistering server: message)r get_serveridrlogerrorrget_registration_urlformatrurlopenr nstrread data_as_dictZ rm_serverid clear_cacheloginfor print_cln_http_error)Zsilenturl server_idresponsecontentreser#*/usr/libexec/kcare/python/kcarectl/auth.py unregisters,      r%cCstjdtj}|dkrdStjtj}ddl}|dkrH|jd|jjt dd}t dd}tj |j |j j tj |j |jj tj |j |j j xLtjd t|\}}}|dkr|rtj|t|tj|jdqWdS) NzHRegister auto-retry has been enabled, the system can be registered laterrz /dev/nullrza+<ii )r print_wrapperosforksetsidsysexitstdoutflushopendup2filenostdinstderrtimesleep _try_registerr set_server_id_set_auth_tokenrr)rpidr-sisocoder auth_tokenr#r#r$_register_retry)s.        r@cCs&|dk r"tjd| r"td||S)Nz ^[\w.-]+$zInvalid value received: %s)rematch ValueError)valuer#r#r$_validate_urlsafe_encodingGs rEcCsyNtj|}|jjtjd}tjtj|j }t |dt |jdt |fSt t fk r~}ztj||dSd}~Xn$tk rtjjd|dSXdS)Nr>rz)Exception while trying to register URL %s)NNN)NNN)rrheadersgetrZAUTH_TOKEN_HEADERr rrrintrEr r rr Exceptionkcarelog exception)rrr?r!r"r#r#r$r8Ms   r8c Cs>y tdWn tk r,tjjdYnXtj}td|fd|fg}djt j |}t |\}}}|dkrt j |t|t jtjddS|dkrtjd nv|d krtjd nb|d krtjd nN|dkrtjdn:|dkrtjdn&|dkrtjdntjdj||r4t|dS|prr?r#r#r$register[s@         rUcCs tjtjS)N)r try_to_readrAUTH_TOKEN_DUMP_PATHr#r#r#r$_get_auth_tokensrXcCs|sdStjtj|dS)N)r atomic_writerrW)r?r#r#r$r:sr:cOsv|jdd}|jddrt}ntj}tj|r@tj|f||Stj|tt|d}t j |ddtj|f||S)Nmethod check_licenseT)rZ)count) pop_check_auth_retryrcheck_urlopen_retryZ is_local_url urlopen_base http_requestget_http_auth_stringrXr rS)rargskwargsrZcheckrequestr#r#r$ urlopen_auths   rhcCs.tj}|r*tjtjtjdj|dSdS)Nz{0}:{1}Z kernelcare)r rr rbase64 b64encodebstrr)rr#r#r$rcsrccCs(t|tr|jdkrt|Stj||S)N)rlrm) isinstancer r>_handle_forbiddenrr`)r"stater#r#r$r_sr_cCsd|kr dStjrtj}tjd}|r8|dj|7}y&tjt j |ddj }tj |}Wn0t k r}ztj||dddSd}~XnX| s|jd  rtjjd j|dS|d dkrd|d<tjd dStdS)aIn 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. licenseTz /check.plainz?server_id={0}F)Z retry_on_500)r/Nr>zUnexpected CLN response: {0}01z$Unable to access server. Retrying...)rrrs)rCHECK_CLN_LICENSE_STATUSr rrrrr rrrrrr rrrGrJerrorr_register_trial)rprrr infoexr#r#r$ros*   rocCstj}|rtjdj|}ytj|}tj|j }tj |}| sT|j d rhtj dj|dSt |d}|dkrtj ddStdd}|dkrtj d|SWqtk r}ztj||dSd}~XqXntSdS) Nz/check.plain?server_id={0}r>zUnexpected CLN response: {0}rrzKey-based valid license found) key_checkedz No valid key-based license found)r rrrrrrr rrrrGr)rH_get_license_info_by_ipr rr)rrrr r!r>Z license_typer"r#r#r$ license_infos,       r{c Cstjd}y6tj|}tj|j}tj|}|djdkr&t |d}|dkrptj dj |ddS|dkr|d}tj |d j d }tj d j ||d S|d kr|dkr|d}tj |d j d }tj d j |||dko|dkrBd|krtj dj |dn tj dn|jdd}tj dj |Wnhtk rt} ztj| |WYdd} ~ Xn:tk r} ztj dj | |jWYdd} ~ XnXdS)Nz /check.plainrrr>rzValid license found for IP {0}iprZ expire_datez%Y-%m-%dz?You have a trial license for the IP {0} that will expire on {1}r(z0Your trial license for the IP {0} expired on {1}rNzThe IP {0} hasn't been licensedz This server hasn't been licensedrz"Error retrieving license info: {0}z1Unexpected CLN response, cannot find {0} key: {1})rrrrr rrrlowerrHr)rZparse_response_datestrftimerGr rrKeyErrorstrip) ryrrr r!r>r|Z expires_strrr"rMr#r#r$rzs<       (rzcCsFtjjtjd}tjj|r dSytjtj d}t j t j |j }y|djdkrt j|dddtj|ddkrtj|d |d tjd j|d dS|dd krt j|dddtjd n tjdWn.tk r }ztj|WYdd}~XnXWn0tk r@}ztj|jWYdd}~XnXdS)Nztrial-requestedz /trial.plainrrr}T) ensure_dirZexpiredr|createdz3Requesting trial license for IP {0}. Please wait...ZnazInvalid LicenserrR)r*pathjoinr PATCH_CACHEexistsrrrrr rrrr~rYrrZAlreadyTrialedExceptionrrr KcareErrorZUnableToGetLicenseExceptionrr r>)Z trial_markrr!Zker"r#r#r$rvs,     rv)F)F)r)!rir*rAr6r}rrrrrrr r r py23r r rr%r@rEr8rUcachedrXr:rhrcr_ror{rzrvr#r#r#r$s&,  . ) #