File: //proc/thread-self/root/usr/libexec/kcare/python/kcarectl/__pycache__/auth.cpython-38.pyc
U
n�h~+ � @ s� d dl Z d dlZd dlZd dlZddlmZmZmZmZm Z m
Z
m Z m Z m
Z
ddlmZmZmZ d"dd�Zdd � Zd
d � Zd d
� Zd#dd�Ze
jdd� �Zdd� Zdd� Zdd� Zdd� Z dd� Z d d � Z d$d d �Z d d!� Z dS )%� N� ) �config� constants�errors�
http_utils� ipv6_support� log_utils�platform_utils�serverid�utils)� HTTPError�URLError� urlencodeFc
C s� d }z�t �� }|d kr*| s$t�d� W d S t�� d�|� }t�|�}t �
|� � �}t � |�}|d dkr�t �
� | s�t�d� n | s�t�|� t�d|d � W n2 tk
r� } z| s�t�||� W 5 d }~X Y nX d S )Nz1Error unregistering server: cannot find server idz&/unregister_server.plain?server_id={0}�success�truezServer was unregisteredz Error unregistering server: �message)r
� get_serveridr �logerrorr �get_registration_url�formatr �urlopenr �nstr�read� data_as_dictZ rm_serverid�loginfor �print_cln_http_error)Zsilent�url� server_id�response�content�res�e� r" �*/usr/libexec/kcare/python/kcarectl/auth.py�
unregister s*
r$ c C s� t �d� t�� }|dkr d S t�� t�� }dd l}|dkrH|�d� |j�� t dd�}t dd�}t�
|� � |j � � � t�
|� � |j� � � t�
|� � |j
� � � t�d� t| �\}}}|dkr�|r�t�|� t|� |�d� q�d S )NzHRegister auto-retry has been enabled, the system can be registered laterr z /dev/null�rza+i )r �
print_wrapper�os�fork�setsid�sys�exit�stdout�flush�open�dup2�fileno�stdin�stderr�time�sleep�
_try_registerr
�
set_server_id�_set_auth_token)r �pidr* �si�so�coder �
auth_tokenr" r" r# �_register_retry& s*
r= c C s$ | d k r t �d| �s td| ��| S )Nz ^[\w.-]+$zInvalid value received: %s)�re�match�
ValueError)�valuer" r" r# �_validate_urlsafe_encoding@ s rB c
C s� zPt �| �}|j�tjd �}t�t�|� � ��}t
|d �t |�d��t |�fW S t t
fk
r� } zt�|| � W Y �dS d }~X Y n& tk
r� tj�d| � Y dS X d S )Nr; r )NNNz)Exception while trying to register URL %s)r r �headers�getr ZAUTH_TOKEN_HEADERr r r r �intrB r r
r r � Exception�kcarelog� exception)r r r< r r! r" r" r# r5 F s
" r5 c C s6 z t d� W n tk
r, tj�d� Y nX t�� }td|fd| fg�}d�t �
� |�}t |�\}}}|dkr�t �
|� t|� t�d� dS |dkr�t�d � nv|d
kr�t�d � nb|d kr�t�d
� nN|dkr�t�d� n:|dkr�t�d� n&|dk�r
t�d� nt�d�|�� |�r,t|� dS |�p4dS )NTz9Exception while trying to unregister URL before register.�hostname�keyz {0}/register_server.plain?{1}r zServer Registeredr zAccount Locked� z Invalid Key� z}You have reached maximum registered servers for this key. Please go to your CLN account, remove unused servers and try again.� z[IP is not allowed. Please change allowed IP ranges for the key in KernelCare Key tab in CLN� zEThis IP was already used for trial, you cannot use it for trial again� zfThis IP was banned. Please contact support for more information at https://www.kernelcare.com/support/zUnknown Error {0}���)r$ rF r rG rH r Z get_hostnamer r r r r5 r
r6 r7 r r r= )rJ �retryrI �queryr r; r r< r" r" r# �registerT s@
�
rS c C s t �tj�S �N)r � try_to_readr �AUTH_TOKEN_DUMP_PATHr" r" r" r# �_get_auth_token| s rW c C s | sd S t �tj| � d S rT )r � atomic_writer rV )r< r" r" r# r7 � s r7 c O sv |� dd �}|� dd�r t}ntj}t�| �r@tj| f|�|�S tj| t� t� |d�}t j
|dd�tj�|f|�|�S )N�method�
check_licenseT)rY � )�count) �pop�_check_auth_retryr Zcheck_urlopen_retryZ is_local_url� urlopen_base� http_request�get_http_auth_stringrW r rQ )r �args�kwargsrY �check�requestr" r" r# � urlopen_auth� s
rf c
C s. t �� } | r*t�t�t�d�| d����S d S )Nz{0}:{1}Z
kernelcare)r
r r r �base64� b64encode�bstrr )r r" r" r# ra � s ra c C s8 t | t�r&| jdkr t|�S | jdkS t | t�r4dS d S )N)i� i� i� T)�
isinstancer r; �_handle_forbiddenr
)r! �stater" r" r# r^ � s
r^ c
C s� d| kr dS t jr�t�� }t�� d }|r8|d�|�7 }z&t�t j
|dd�� � �}t� |�}W n8 t
k
r� } ztj||dd� W Y �dS d}~X Y nX |r�|�d �s�tj�d
�|�� dS |d d kr�d| d< t�d � dS t� dS )
a In 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.
�licenseT� /check.plainz?server_id={0}F)Z retry_on_500)r, Nr; � Unexpected CLN response: {0})�0�1z$Unable to access server. Retrying...)r �CHECK_CLN_LICENSE_STATUSr
r r r r r r r r r r r
r r rD rG �errorr �_register_trial)rl r r r �info�exr" r" r# rk � s*
rk c
C s� t �� } | r�t�� d�| � }z�t�|�}t�|� � �}t�
|�}|rP|� d�sft� d�|�� W dS t
|d �}|dkr�t� d� W dS tdd�}|dkr�t� d� |W S W q� tk
r� } zt�||� W Y �dS d }~X Y q�X nt� S d S ) Nz/check.plain?server_id={0}r; ro r r z Key-based valid license found)� key_checkedz No valid key-based license found)r
r r r r r r r r r r rD r& rE �_get_license_info_by_ipr
r r )r r r r r r; Z license_typer! r" r" r# � license_info� s,
ry c
C s� t �� d }�z<t�|�}t�|�� �}t�|�}|d �� dk�r,t |d �}|dkrrt�
d� |d �� W dS |dkr�|d }t� |d ��
d
�}t�
d � ||�� W d S |d kr�| dkr�|d }t� |d ��
d
�}t�
d
� ||�� |dk�rH| dk�rHd|k�r t�
d� |d �� n
t�
d� n |�dd�}t�
d� |�� W nh tk
�rz } zt�| |� W 5 d } ~ X Y n: tk
�r� }
zt�
d� |
|�� �� W 5 d }
~
X Y nX dS )Nrn r r r; r z Valid license found for IP {0}�ipr Z expire_datez%Y-%m-%dz?You have a trial license for the IP {0} that will expire on {1}rK z0Your trial license for the IP {0} expired on {1}rL z The IP {0} hasn't been licensedz This server hasn't been licensedr � z"Error retrieving license info: {0}z1Unexpected CLN response, cannot find {0} key:
{1})r r r r r r r r �lowerrE r& r Zparse_response_date�strftimerD r
r r �KeyError�strip) rw r r r r r; rz Z expires_strr r! rJ r" r" r# rx � s<
(rx c
C sB t j�tjd�} t j�| �r d S z�t�t� � d �}t
� t
� |�
� ��}z�|d �� dkr�t
j| ddd� |d dkr�t�|d |d
��t�d �|d �� W W d S |d d kr�t
j| ddd� t�d
��n
t�d��W n. tk
�r } zt�|��W 5 d }~X Y nX W n0 tk
�r< } zt�|j��W 5 d }~X Y nX d S )Nztrial-requestedz /trial.plainr r r{ T)�
ensure_dirZexpiredrz �createdz3Requesting trial license for IP {0}. Please wait...ZnazInvalid LicenserP )r' �path�joinr � PATCH_CACHE�existsr r r r r r r r r| rX r ZAlreadyTrialedExceptionr r r �
KcareErrorZUnableToGetLicenseExceptionr~ r r; )Z
trial_markr r Zker! r" r" r# rt s* rt )F)F)r )!rg r' r> r3 r{ r r r r r r r r
r �py23r r
r r$ r= rB r5 rS �cachedrW r7 rf ra r^ rk ry rx rt r" r" r" r# �<module> s( ,
(
)
#